avdct.c (5733B)
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 "libavutil/mem.h" 22 #include "avcodec.h" 23 #include "idctdsp.h" 24 #include "fdctdsp.h" 25 #include "pixblockdsp.h" 26 #include "avdct.h" 27 28 #define OFFSET(x) offsetof(AVDCT,x) 29 #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 30 //these names are too long to be readable 31 #define V AV_OPT_FLAG_VIDEO_PARAM 32 #define A AV_OPT_FLAG_AUDIO_PARAM 33 #define E AV_OPT_FLAG_ENCODING_PARAM 34 #define D AV_OPT_FLAG_DECODING_PARAM 35 36 static const AVOption avdct_options[] = { 37 {"dct", "DCT algorithm", OFFSET(dct_algo), AV_OPT_TYPE_INT, {.i64 = DEFAULT }, 0, INT_MAX, V|E, .unit = "dct"}, 38 {"auto", "autoselect a good one", 0, AV_OPT_TYPE_CONST, {.i64 = FF_DCT_AUTO }, INT_MIN, INT_MAX, V|E, .unit = "dct"}, 39 {"fastint", "fast integer (experimental / for debugging)", 0, AV_OPT_TYPE_CONST, {.i64 = FF_DCT_FASTINT }, INT_MIN, INT_MAX, V|E, .unit = "dct"}, 40 {"int", "accurate integer", 0, AV_OPT_TYPE_CONST, {.i64 = FF_DCT_INT }, INT_MIN, INT_MAX, V|E, .unit = "dct"}, 41 {"mmx", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_DCT_MMX }, INT_MIN, INT_MAX, V|E, .unit = "dct"}, 42 {"altivec", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_DCT_ALTIVEC }, INT_MIN, INT_MAX, V|E, .unit = "dct"}, 43 {"faan", "floating point AAN DCT (experimental / for debugging)", 0, AV_OPT_TYPE_CONST, {.i64 = FF_DCT_FAAN }, INT_MIN, INT_MAX, V|E, .unit = "dct"}, 44 45 {"idct", "select IDCT implementation", OFFSET(idct_algo), AV_OPT_TYPE_INT, {.i64 = DEFAULT }, 0, INT_MAX, V|E|D, .unit = "idct"}, 46 {"auto", "autoselect a good one", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_AUTO }, INT_MIN, INT_MAX, V|E|D, .unit = "idct"}, 47 {"int", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_INT }, INT_MIN, INT_MAX, V|E|D, .unit = "idct"}, 48 {"simple", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_SIMPLE }, INT_MIN, INT_MAX, V|E|D, .unit = "idct"}, 49 {"simplemmx", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_SIMPLEMMX }, INT_MIN, INT_MAX, V|E|D, .unit = "idct"}, 50 {"arm", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_ARM }, INT_MIN, INT_MAX, V|E|D, .unit = "idct"}, 51 {"altivec", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_ALTIVEC }, INT_MIN, INT_MAX, V|E|D, .unit = "idct"}, 52 {"simplearm", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_SIMPLEARM }, INT_MIN, INT_MAX, V|E|D, .unit = "idct"}, 53 {"simplearmv5te", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_SIMPLEARMV5TE }, INT_MIN, INT_MAX, V|E|D, .unit = "idct"}, 54 {"simplearmv6", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_SIMPLEARMV6 }, INT_MIN, INT_MAX, V|E|D, .unit = "idct"}, 55 {"simpleneon", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_SIMPLENEON }, INT_MIN, INT_MAX, V|E|D, .unit = "idct"}, 56 {"xvid", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_XVID }, INT_MIN, INT_MAX, V|E|D, .unit = "idct"}, 57 {"xvidmmx", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_XVID }, INT_MIN, INT_MAX, V|E|D, .unit = "idct"}, 58 {"faani", "floating point AAN IDCT (experimental / for debugging)", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_FAAN }, INT_MIN, INT_MAX, V|D|E, .unit = "idct"}, 59 {"simpleauto", "experimental / for debugging", 0, AV_OPT_TYPE_CONST, {.i64 = FF_IDCT_SIMPLEAUTO }, INT_MIN, INT_MAX, V|E|D, .unit = "idct"}, 60 61 {"bits_per_sample", "", OFFSET(bits_per_sample), AV_OPT_TYPE_INT, {.i64 = 8 }, 0, 14, 0,}, 62 {NULL}, 63 }; 64 65 static const AVClass avdct_class = { 66 .class_name = "AVDCT", 67 .option = avdct_options, 68 .version = LIBAVUTIL_VERSION_INT, 69 }; 70 71 const AVClass *avcodec_dct_get_class(void) 72 { 73 return &avdct_class; 74 } 75 76 AVDCT *avcodec_dct_alloc(void) 77 { 78 AVDCT *dsp = av_mallocz(sizeof(AVDCT)); 79 80 if (!dsp) 81 return NULL; 82 83 dsp->av_class = &avdct_class; 84 av_opt_set_defaults(dsp); 85 86 return dsp; 87 } 88 89 int avcodec_dct_init(AVDCT *dsp) 90 { 91 AVCodecContext *avctx = avcodec_alloc_context3(NULL); 92 93 if (!avctx) 94 return AVERROR(ENOMEM); 95 96 avctx->idct_algo = dsp->idct_algo; 97 avctx->dct_algo = dsp->dct_algo; 98 avctx->bits_per_raw_sample = dsp->bits_per_sample; 99 100 #define COPY(src, name) memcpy(&dsp->name, &src.name, sizeof(dsp->name)) 101 102 #if CONFIG_IDCTDSP 103 { 104 IDCTDSPContext idsp = {0}; 105 ff_idctdsp_init(&idsp, avctx); 106 COPY(idsp, idct); 107 COPY(idsp, idct_permutation); 108 } 109 #endif 110 111 #if CONFIG_FDCTDSP 112 { 113 FDCTDSPContext fdsp; 114 ff_fdctdsp_init(&fdsp, avctx); 115 COPY(fdsp, fdct); 116 } 117 #endif 118 119 #if CONFIG_PIXBLOCKDSP 120 { 121 PixblockDSPContext pdsp; 122 ff_pixblockdsp_init(&pdsp, avctx); 123 COPY(pdsp, get_pixels); 124 COPY(pdsp, get_pixels_unaligned); 125 } 126 #endif 127 128 avcodec_free_context(&avctx); 129 130 return 0; 131 }