91 lines
2.4 KiB
C
91 lines
2.4 KiB
C
|
|
||
|
|
#include "libavutil/opt.h"
|
||
|
|
#include "libavutil/samplefmt.h"
|
||
|
|
#include "avcodec.h"
|
||
|
|
#include "ac3.h"
|
||
|
|
|
||
|
|
typedef struct CombineContext{
|
||
|
|
AVClass *av_class; ///< AVClass used for AVOption
|
||
|
|
AC3EncOptions options; ///< encoding options
|
||
|
|
void *ctx;
|
||
|
|
AVCodec *codec;
|
||
|
|
}CombineContext;
|
||
|
|
|
||
|
|
static AVClass ac3enc_class = { "AC-3 Encoder", av_default_item_name,
|
||
|
|
ff_ac3_options, LIBAVUTIL_VERSION_INT };
|
||
|
|
|
||
|
|
static av_cold AVCodec *get_codec(enum AVSampleFormat s){
|
||
|
|
#if CONFIG_AC3_FIXED_ENCODER
|
||
|
|
if(s==AV_SAMPLE_FMT_S16) return &ff_ac3_fixed_encoder;
|
||
|
|
#endif
|
||
|
|
#if CONFIG_AC3_FLOAT_ENCODER
|
||
|
|
if(s==AV_SAMPLE_FMT_FLT) return &ff_ac3_float_encoder;
|
||
|
|
#endif
|
||
|
|
return NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
static av_cold int encode_init(AVCodecContext *avctx)
|
||
|
|
{
|
||
|
|
CombineContext *c= avctx->priv_data;
|
||
|
|
int ret;
|
||
|
|
int offset= (uint8_t*)&c->options - (uint8_t*)c;
|
||
|
|
|
||
|
|
c->codec= get_codec(avctx->sample_fmt);
|
||
|
|
if(!c->codec){
|
||
|
|
av_log(avctx, AV_LOG_ERROR, "Unsupported sample format\n");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
c->ctx= av_mallocz(c->codec->priv_data_size);
|
||
|
|
memcpy((uint8_t*)c->ctx + offset, &c->options, (uint8_t*)&c->ctx - (uint8_t*)&c->options);
|
||
|
|
FFSWAP(void *,avctx->priv_data, c->ctx);
|
||
|
|
ret= c->codec->init(avctx);
|
||
|
|
FFSWAP(void *,avctx->priv_data, c->ctx);
|
||
|
|
return ret;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int encode_frame(AVCodecContext *avctx, unsigned char *frame,
|
||
|
|
int buf_size, void *data)
|
||
|
|
{
|
||
|
|
CombineContext *c= avctx->priv_data;
|
||
|
|
int ret;
|
||
|
|
|
||
|
|
FFSWAP(void *,avctx->priv_data, c->ctx);
|
||
|
|
ret= c->codec->encode(avctx, frame, buf_size, data);
|
||
|
|
FFSWAP(void *,avctx->priv_data, c->ctx);
|
||
|
|
return ret;
|
||
|
|
}
|
||
|
|
|
||
|
|
static av_cold int encode_close(AVCodecContext *avctx)
|
||
|
|
{
|
||
|
|
CombineContext *c= avctx->priv_data;
|
||
|
|
int ret;
|
||
|
|
|
||
|
|
FFSWAP(void *,avctx->priv_data, c->ctx);
|
||
|
|
ret= c->codec->close(avctx);
|
||
|
|
FFSWAP(void *,avctx->priv_data, c->ctx);
|
||
|
|
return ret;
|
||
|
|
}
|
||
|
|
|
||
|
|
AVCodec ff_ac3_encoder = {
|
||
|
|
"ac3",
|
||
|
|
AVMEDIA_TYPE_AUDIO,
|
||
|
|
CODEC_ID_AC3,
|
||
|
|
sizeof(CombineContext),
|
||
|
|
encode_init,
|
||
|
|
encode_frame,
|
||
|
|
encode_close,
|
||
|
|
NULL,
|
||
|
|
.sample_fmts = (const enum AVSampleFormat[]){
|
||
|
|
#if CONFIG_AC3_FLOAT_ENCODER
|
||
|
|
AV_SAMPLE_FMT_FLT,
|
||
|
|
#endif
|
||
|
|
#if CONFIG_AC3_FIXED_ENCODER
|
||
|
|
AV_SAMPLE_FMT_S16,
|
||
|
|
#endif
|
||
|
|
AV_SAMPLE_FMT_NONE},
|
||
|
|
.long_name = NULL_IF_CONFIG_SMALL("ATSC A/52A (AC-3)"),
|
||
|
|
.priv_class = &ac3enc_class,
|
||
|
|
.channel_layouts = ff_ac3_channel_layouts,
|
||
|
|
};
|