1 /*
2 * Copyright (c) 2014 Michael Niedermayer <michaelni@gmx.at>
3 *
4 * This file is part of FFmpeg.
5 *
6 * FFmpeg is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * FFmpeg is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with FFmpeg; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21 #include "avcodec.h"
22 #include "idctdsp.h"
23 #include "fdctdsp.h"
24 #include "pixblockdsp.h"
25 #include "avdct.h"
26
27 #define OFFSET(x) offsetof(AVDCT,x)
28 #define DEFAULT 0 //should be NAN but it does not work as it is not a constant in glibc as required by ANSI/ISO C
29 //these names are too long to be readable
30 #define V AV_OPT_FLAG_VIDEO_PARAM
31 #define A AV_OPT_FLAG_AUDIO_PARAM
32 #define E AV_OPT_FLAG_ENCODING_PARAM
33 #define D AV_OPT_FLAG_DECODING_PARAM
34
35 static const AVOption avdct_options[] = {
36 {"dct", "DCT algorithm", OFFSET(dct_algo), AV_OPT_TYPE_INT, {.i64 = DEFAULT }, 0, INT_MAX, V|E, "dct"},
37 {"auto", "autoselect a good one", 0, AV_OPT_TYPE_CONST, {.i64 = FF_DCT_AUTO }, INT_MIN, INT_MAX, V|E, "dct"},
38 {"fastint", "fast integer (experimental / for debugging)", 0, AV_OPT_TYPE_CONST, {.i64 = FF_DCT_FASTINT }, INT_MIN, INT_MAX, V|E, "dct"},
39 {"int", "accurate integer", 0, AV_OPT_TYPE_CONST, {.i64 = FF_DCT_INT }, INT_MIN, INT_MAX, V|E, "dct"},
40 {"mmx", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_DCT_MMX }, INT_MIN, INT_MAX, V|E, "dct"},
41 {"altivec", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_DCT_ALTIVEC }, INT_MIN, INT_MAX, V|E, "dct"},
42 {"faan", "floating point AAN DCT (experimental / for debugging)", 0, AV_OPT_TYPE_CONST, {.i64 = FF_DCT_FAAN }, INT_MIN, INT_MAX, V|E, "dct"},
43
44 {"idct", "select IDCT implementation", OFFSET(idct_algo), AV_OPT_TYPE_INT, {.i64 = DEFAULT }, 0, INT_MAX, V|E|D, "idct"},
45 {"auto", "autoselect a good one", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_AUTO }, INT_MIN, INT_MAX, V|E|D, "idct"},
46 {"int", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_INT }, INT_MIN, INT_MAX, V|E|D, "idct"},
47 {"simple", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_SIMPLE }, INT_MIN, INT_MAX, V|E|D, "idct"},
48 {"simplemmx", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_SIMPLEMMX }, INT_MIN, INT_MAX, V|E|D, "idct"},
49 {"arm", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_ARM }, INT_MIN, INT_MAX, V|E|D, "idct"},
50 {"altivec", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_ALTIVEC }, INT_MIN, INT_MAX, V|E|D, "idct"},
51 {"simplearm", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_SIMPLEARM }, INT_MIN, INT_MAX, V|E|D, "idct"},
52 {"simplearmv5te", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_SIMPLEARMV5TE }, INT_MIN, INT_MAX, V|E|D, "idct"},
53 {"simplearmv6", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_SIMPLEARMV6 }, INT_MIN, INT_MAX, V|E|D, "idct"},
54 {"simpleneon", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_SIMPLENEON }, INT_MIN, INT_MAX, V|E|D, "idct"},
55 {"xvid", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_XVID }, INT_MIN, INT_MAX, V|E|D, "idct"},
56 {"xvidmmx", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_XVID }, INT_MIN, INT_MAX, V|E|D, "idct"},
57 {"faani", "floating point AAN IDCT (experimental / for debugging)", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_FAAN }, INT_MIN, INT_MAX, V|D|E, "idct"},
58 {"simpleauto", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_SIMPLEAUTO }, INT_MIN, INT_MAX, V|E|D, "idct"},
59
60 {"bits_per_sample", "", OFFSET(bits_per_sample), AV_OPT_TYPE_INT, {.i64 = 8 }, 0, 14, 0,},
61 {NULL},
62 };
63
64 static const AVClass avdct_class = {
65 .class_name = "AVDCT",
66 .option = avdct_options,
67 .version = LIBAVUTIL_VERSION_INT,
68 };
69
avcodec_dct_get_class(void)70 const AVClass *avcodec_dct_get_class(void)
71 {
72 return &avdct_class;
73 }
74
avcodec_dct_alloc(void)75 AVDCT *avcodec_dct_alloc(void)
76 {
77 AVDCT *dsp = av_mallocz(sizeof(AVDCT));
78
79 if (!dsp)
80 return NULL;
81
82 dsp->av_class = &avdct_class;
83 av_opt_set_defaults(dsp);
84
85 return dsp;
86 }
87
avcodec_dct_init(AVDCT * dsp)88 int avcodec_dct_init(AVDCT *dsp)
89 {
90 AVCodecContext *avctx = avcodec_alloc_context3(NULL);
91
92 if (!avctx)
93 return AVERROR(ENOMEM);
94
95 avctx->idct_algo = dsp->idct_algo;
96 avctx->dct_algo = dsp->dct_algo;
97 avctx->bits_per_raw_sample = dsp->bits_per_sample;
98
99 #define COPY(src, name) memcpy(&dsp->name, &src.name, sizeof(dsp->name))
100
101 #if CONFIG_IDCTDSP
102 {
103 IDCTDSPContext idsp = {0};
104 ff_idctdsp_init(&idsp, avctx);
105 COPY(idsp, idct);
106 COPY(idsp, idct_permutation);
107 }
108 #endif
109
110 #if CONFIG_FDCTDSP
111 {
112 FDCTDSPContext fdsp;
113 ff_fdctdsp_init(&fdsp, avctx);
114 COPY(fdsp, fdct);
115 }
116 #endif
117
118 #if CONFIG_PIXBLOCKDSP
119 {
120 PixblockDSPContext pdsp;
121 ff_pixblockdsp_init(&pdsp, avctx);
122 COPY(pdsp, get_pixels);
123 COPY(pdsp, get_pixels_unaligned);
124 }
125 #endif
126
127 avcodec_free_context(&avctx);
128
129 return 0;
130 }
131