lavc/audiotoolboxdec: move to new BSF API
This commit is contained in:
parent
95116bf35f
commit
e1c20485c1
@ -43,10 +43,13 @@ typedef struct ATDecodeContext {
|
|||||||
AudioStreamPacketDescription pkt_desc;
|
AudioStreamPacketDescription pkt_desc;
|
||||||
AVPacket in_pkt;
|
AVPacket in_pkt;
|
||||||
AVPacket new_in_pkt;
|
AVPacket new_in_pkt;
|
||||||
AVBitStreamFilterContext *bsf;
|
AVBSFContext *bsf;
|
||||||
char *decoded_data;
|
char *decoded_data;
|
||||||
int channel_map[64];
|
int channel_map[64];
|
||||||
|
|
||||||
|
uint8_t *extradata;
|
||||||
|
int extradata_size;
|
||||||
|
|
||||||
int64_t last_pts;
|
int64_t last_pts;
|
||||||
int eof;
|
int eof;
|
||||||
} ATDecodeContext;
|
} ATDecodeContext;
|
||||||
@ -224,22 +227,23 @@ static void put_descr(PutByteContext *pb, int tag, unsigned int size)
|
|||||||
|
|
||||||
static uint8_t* ffat_get_magic_cookie(AVCodecContext *avctx, UInt32 *cookie_size)
|
static uint8_t* ffat_get_magic_cookie(AVCodecContext *avctx, UInt32 *cookie_size)
|
||||||
{
|
{
|
||||||
|
ATDecodeContext *at = avctx->priv_data;
|
||||||
if (avctx->codec_id == AV_CODEC_ID_AAC) {
|
if (avctx->codec_id == AV_CODEC_ID_AAC) {
|
||||||
char *extradata;
|
char *extradata;
|
||||||
PutByteContext pb;
|
PutByteContext pb;
|
||||||
*cookie_size = 5 + 3 + 5+13 + 5+avctx->extradata_size;
|
*cookie_size = 5 + 3 + 5+13 + 5+at->extradata_size;
|
||||||
if (!(extradata = av_malloc(*cookie_size)))
|
if (!(extradata = av_malloc(*cookie_size)))
|
||||||
return NULL;
|
return NULL;
|
||||||
|
|
||||||
bytestream2_init_writer(&pb, extradata, *cookie_size);
|
bytestream2_init_writer(&pb, extradata, *cookie_size);
|
||||||
|
|
||||||
// ES descriptor
|
// ES descriptor
|
||||||
put_descr(&pb, 0x03, 3 + 5+13 + 5+avctx->extradata_size);
|
put_descr(&pb, 0x03, 3 + 5+13 + 5+at->extradata_size);
|
||||||
bytestream2_put_be16(&pb, 0);
|
bytestream2_put_be16(&pb, 0);
|
||||||
bytestream2_put_byte(&pb, 0x00); // flags (= no flags)
|
bytestream2_put_byte(&pb, 0x00); // flags (= no flags)
|
||||||
|
|
||||||
// DecoderConfig descriptor
|
// DecoderConfig descriptor
|
||||||
put_descr(&pb, 0x04, 13 + 5+avctx->extradata_size);
|
put_descr(&pb, 0x04, 13 + 5+at->extradata_size);
|
||||||
|
|
||||||
// Object type indication
|
// Object type indication
|
||||||
bytestream2_put_byte(&pb, 0x40);
|
bytestream2_put_byte(&pb, 0x40);
|
||||||
@ -252,18 +256,19 @@ static uint8_t* ffat_get_magic_cookie(AVCodecContext *avctx, UInt32 *cookie_size
|
|||||||
bytestream2_put_be32(&pb, 0); // avgbitrate
|
bytestream2_put_be32(&pb, 0); // avgbitrate
|
||||||
|
|
||||||
// DecoderSpecific info descriptor
|
// DecoderSpecific info descriptor
|
||||||
put_descr(&pb, 0x05, avctx->extradata_size);
|
put_descr(&pb, 0x05, at->extradata_size);
|
||||||
bytestream2_put_buffer(&pb, avctx->extradata, avctx->extradata_size);
|
bytestream2_put_buffer(&pb, at->extradata, at->extradata_size);
|
||||||
return extradata;
|
return extradata;
|
||||||
} else {
|
} else {
|
||||||
*cookie_size = avctx->extradata_size;
|
*cookie_size = at->extradata_size;
|
||||||
return avctx->extradata;
|
return at->extradata;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
static av_cold int ffat_usable_extradata(AVCodecContext *avctx)
|
static av_cold int ffat_usable_extradata(AVCodecContext *avctx)
|
||||||
{
|
{
|
||||||
return avctx->extradata_size &&
|
ATDecodeContext *at = avctx->priv_data;
|
||||||
|
return at->extradata_size &&
|
||||||
(avctx->codec_id == AV_CODEC_ID_ALAC ||
|
(avctx->codec_id == AV_CODEC_ID_ALAC ||
|
||||||
avctx->codec_id == AV_CODEC_ID_AAC);
|
avctx->codec_id == AV_CODEC_ID_AAC);
|
||||||
}
|
}
|
||||||
@ -284,7 +289,7 @@ static int ffat_set_extradata(AVCodecContext *avctx)
|
|||||||
if (status != 0)
|
if (status != 0)
|
||||||
av_log(avctx, AV_LOG_WARNING, "AudioToolbox cookie error: %i\n", (int)status);
|
av_log(avctx, AV_LOG_WARNING, "AudioToolbox cookie error: %i\n", (int)status);
|
||||||
|
|
||||||
if (cookie != avctx->extradata)
|
if (cookie != at->extradata)
|
||||||
av_free(cookie);
|
av_free(cookie);
|
||||||
}
|
}
|
||||||
return 0;
|
return 0;
|
||||||
@ -320,7 +325,7 @@ static av_cold int ffat_create_decoder(AVCodecContext *avctx, AVPacket *pkt)
|
|||||||
return AVERROR(ENOMEM);
|
return AVERROR(ENOMEM);
|
||||||
status = AudioFormatGetProperty(kAudioFormatProperty_FormatInfo,
|
status = AudioFormatGetProperty(kAudioFormatProperty_FormatInfo,
|
||||||
cookie_size, cookie, &format_size, &in_format);
|
cookie_size, cookie, &format_size, &in_format);
|
||||||
if (cookie != avctx->extradata)
|
if (cookie != at->extradata)
|
||||||
av_free(cookie);
|
av_free(cookie);
|
||||||
if (status != 0) {
|
if (status != 0) {
|
||||||
av_log(avctx, AV_LOG_ERROR, "AudioToolbox header-parse error: %i\n", (int)status);
|
av_log(avctx, AV_LOG_ERROR, "AudioToolbox header-parse error: %i\n", (int)status);
|
||||||
@ -391,6 +396,10 @@ static av_cold int ffat_create_decoder(AVCodecContext *avctx, AVPacket *pkt)
|
|||||||
|
|
||||||
static av_cold int ffat_init_decoder(AVCodecContext *avctx)
|
static av_cold int ffat_init_decoder(AVCodecContext *avctx)
|
||||||
{
|
{
|
||||||
|
ATDecodeContext *at = avctx->priv_data;
|
||||||
|
at->extradata = avctx->extradata;
|
||||||
|
at->extradata_size = avctx->extradata_size;
|
||||||
|
|
||||||
if ((avctx->channels && avctx->sample_rate) || ffat_usable_extradata(avctx))
|
if ((avctx->channels && avctx->sample_rate) || ffat_usable_extradata(avctx))
|
||||||
return ffat_create_decoder(avctx, NULL);
|
return ffat_create_decoder(avctx, NULL);
|
||||||
else
|
else
|
||||||
@ -416,8 +425,6 @@ static OSStatus ffat_decode_callback(AudioConverterRef converter, UInt32 *nb_pac
|
|||||||
|
|
||||||
av_packet_unref(&at->in_pkt);
|
av_packet_unref(&at->in_pkt);
|
||||||
av_packet_move_ref(&at->in_pkt, &at->new_in_pkt);
|
av_packet_move_ref(&at->in_pkt, &at->new_in_pkt);
|
||||||
at->new_in_pkt.data = 0;
|
|
||||||
at->new_in_pkt.size = 0;
|
|
||||||
|
|
||||||
if (!at->in_pkt.data) {
|
if (!at->in_pkt.data) {
|
||||||
*nb_packets = 0;
|
*nb_packets = 0;
|
||||||
@ -464,32 +471,48 @@ static int ffat_decode(AVCodecContext *avctx, void *data,
|
|||||||
ATDecodeContext *at = avctx->priv_data;
|
ATDecodeContext *at = avctx->priv_data;
|
||||||
AVFrame *frame = data;
|
AVFrame *frame = data;
|
||||||
int pkt_size = avpkt->size;
|
int pkt_size = avpkt->size;
|
||||||
AVPacket filtered_packet;
|
AVPacket filtered_packet = {0};
|
||||||
OSStatus ret;
|
OSStatus ret;
|
||||||
AudioBufferList out_buffers;
|
AudioBufferList out_buffers;
|
||||||
|
|
||||||
if (avctx->codec_id == AV_CODEC_ID_AAC && avpkt->size > 2 &&
|
if (avctx->codec_id == AV_CODEC_ID_AAC && avpkt->size > 2 &&
|
||||||
(AV_RB16(avpkt->data) & 0xfff0) == 0xfff0) {
|
(AV_RB16(avpkt->data) & 0xfff0) == 0xfff0) {
|
||||||
uint8_t *p_filtered = NULL;
|
AVPacket filter_pkt = {0};
|
||||||
int n_filtered = 0;
|
|
||||||
if (!at->bsf) {
|
if (!at->bsf) {
|
||||||
if(!(at->bsf = av_bitstream_filter_init("aac_adtstoasc")))
|
const AVBitStreamFilter *bsf = av_bsf_get_by_name("aac_adtstoasc");
|
||||||
return AVERROR(ENOMEM);
|
if(!bsf)
|
||||||
|
return AVERROR_BSF_NOT_FOUND;
|
||||||
|
if ((ret = av_bsf_alloc(bsf, &at->bsf)))
|
||||||
|
return ret;
|
||||||
|
if (((ret = avcodec_parameters_from_context(at->bsf->par_in, avctx)) < 0) ||
|
||||||
|
((ret = av_bsf_init(at->bsf)) < 0)) {
|
||||||
|
av_bsf_free(&at->bsf);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
ret = av_bitstream_filter_filter(at->bsf, avctx, NULL, &p_filtered, &n_filtered,
|
if ((ret = av_packet_ref(&filter_pkt, avpkt)) < 0)
|
||||||
avpkt->data, avpkt->size, 0);
|
return ret;
|
||||||
if (ret >= 0 && p_filtered != avpkt->data) {
|
|
||||||
filtered_packet = *avpkt;
|
if ((ret = av_bsf_send_packet(at->bsf, &filter_pkt)) < 0) {
|
||||||
avpkt = &filtered_packet;
|
av_packet_unref(&filter_pkt);
|
||||||
avpkt->data = p_filtered;
|
return ret;
|
||||||
avpkt->size = n_filtered;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if ((ret = av_bsf_receive_packet(at->bsf, &filtered_packet)) < 0)
|
||||||
|
return ret;
|
||||||
|
|
||||||
|
at->extradata = at->bsf->par_out->extradata;
|
||||||
|
at->extradata_size = at->bsf->par_out->extradata_size;
|
||||||
|
|
||||||
|
avpkt = &filtered_packet;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!at->converter) {
|
if (!at->converter) {
|
||||||
if ((ret = ffat_create_decoder(avctx, avpkt)) < 0)
|
if ((ret = ffat_create_decoder(avctx, avpkt)) < 0) {
|
||||||
|
av_packet_unref(&filtered_packet);
|
||||||
return ret;
|
return ret;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
out_buffers = (AudioBufferList){
|
out_buffers = (AudioBufferList){
|
||||||
@ -506,9 +529,11 @@ static int ffat_decode(AVCodecContext *avctx, void *data,
|
|||||||
av_packet_unref(&at->new_in_pkt);
|
av_packet_unref(&at->new_in_pkt);
|
||||||
|
|
||||||
if (avpkt->size) {
|
if (avpkt->size) {
|
||||||
if ((ret = av_packet_ref(&at->new_in_pkt, avpkt)) < 0)
|
if (filtered_packet.data) {
|
||||||
|
at->new_in_pkt = filtered_packet;
|
||||||
|
} else if ((ret = av_packet_ref(&at->new_in_pkt, avpkt)) < 0) {
|
||||||
return ret;
|
return ret;
|
||||||
at->new_in_pkt.data = avpkt->data;
|
}
|
||||||
} else {
|
} else {
|
||||||
at->eof = 1;
|
at->eof = 1;
|
||||||
}
|
}
|
||||||
@ -551,6 +576,7 @@ static av_cold int ffat_close_decoder(AVCodecContext *avctx)
|
|||||||
{
|
{
|
||||||
ATDecodeContext *at = avctx->priv_data;
|
ATDecodeContext *at = avctx->priv_data;
|
||||||
AudioConverterDispose(at->converter);
|
AudioConverterDispose(at->converter);
|
||||||
|
av_bsf_free(&at->bsf);
|
||||||
av_packet_unref(&at->new_in_pkt);
|
av_packet_unref(&at->new_in_pkt);
|
||||||
av_packet_unref(&at->in_pkt);
|
av_packet_unref(&at->in_pkt);
|
||||||
av_free(at->decoded_data);
|
av_free(at->decoded_data);
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user