Merge remote branch 'qatar/master'
[ffmpeg.git] / libavcodec / ac3enc_combined.c
1
2 #include "libavutil/opt.h"
3 #include "libavutil/samplefmt.h"
4 #include "avcodec.h"
5 #include "ac3.h"
6
7 typedef struct CombineContext{
8     AVClass *av_class;                      ///< AVClass used for AVOption
9     AC3EncOptions options;                  ///< encoding options
10     void *ctx;
11     AVCodec *codec;
12 }CombineContext;
13
14 static AVClass ac3enc_class = { "AC-3 Encoder", av_default_item_name,
15                                 ff_ac3_options, LIBAVUTIL_VERSION_INT };
16
17 static av_cold AVCodec *get_codec(enum AVSampleFormat s){
18 #if CONFIG_AC3_FIXED_ENCODER
19     if(s==AV_SAMPLE_FMT_S16) return &ff_ac3_fixed_encoder;
20 #endif
21 #if CONFIG_AC3_FLOAT_ENCODER
22     if(s==AV_SAMPLE_FMT_FLT) return &ff_ac3_float_encoder;
23 #endif
24     return NULL;
25 }
26
27
28 static av_cold int encode_init(AVCodecContext *avctx)
29 {
30     CombineContext *c= avctx->priv_data;
31     int ret;
32     int offset= (uint8_t*)&c->options - (uint8_t*)c;
33
34     c->codec= get_codec(avctx->sample_fmt);
35     if(!c->codec){
36         av_log(avctx, AV_LOG_ERROR, "Unsupported sample format\n");
37         return -1;
38     }
39     c->ctx= av_mallocz(c->codec->priv_data_size);
40     memcpy((uint8_t*)c->ctx + offset, &c->options, (uint8_t*)&c->ctx - (uint8_t*)&c->options);
41     FFSWAP(void *,avctx->priv_data, c->ctx);
42     ret= c->codec->init(avctx);
43     FFSWAP(void *,avctx->priv_data, c->ctx);
44     return ret;
45 }
46
47 static int encode_frame(AVCodecContext *avctx, unsigned char *frame,
48                         int buf_size, void *data)
49 {
50     CombineContext *c= avctx->priv_data;
51     int ret;
52
53     FFSWAP(void *,avctx->priv_data, c->ctx);
54     ret= c->codec->encode(avctx, frame, buf_size, data);
55     FFSWAP(void *,avctx->priv_data, c->ctx);
56     return ret;
57 }
58
59 static av_cold int encode_close(AVCodecContext *avctx)
60 {
61     CombineContext *c= avctx->priv_data;
62     int ret;
63
64     FFSWAP(void *,avctx->priv_data, c->ctx);
65     ret= c->codec->close(avctx);
66     FFSWAP(void *,avctx->priv_data, c->ctx);
67     return ret;
68 }
69
70 AVCodec ff_ac3_encoder = {
71     "ac3",
72     AVMEDIA_TYPE_AUDIO,
73     CODEC_ID_AC3,
74     sizeof(CombineContext),
75     encode_init,
76     encode_frame,
77     encode_close,
78     NULL,
79     .sample_fmts = (const enum AVSampleFormat[]){
80 #if CONFIG_AC3_FLOAT_ENCODER
81         AV_SAMPLE_FMT_FLT,
82 #endif
83 #if CONFIG_AC3_FIXED_ENCODER
84         AV_SAMPLE_FMT_S16,
85 #endif
86         AV_SAMPLE_FMT_NONE},
87     .long_name = NULL_IF_CONFIG_SMALL("ATSC A/52A (AC-3)"),
88     .priv_class = &ac3enc_class,
89     .channel_layouts = ff_ac3_channel_layouts,
90 };