# HG changeset patch # User michael # Date 1152198286 0 # Node ID b7826511f7b6b84fe12756391626d6fc07ee92cc # Parent 54814e15aa3d70aa6c7bd0045bb7acf6344b2b4f AVBitStreamFilter (some thingy which can modify the bitstream like add or remove global headers or change the headers or ...) diff -r 54814e15aa3d -r b7826511f7b6 Makefile --- a/Makefile Thu Jul 06 13:53:07 2006 +0000 +++ b/Makefile Thu Jul 06 15:04:46 2006 +0000 @@ -17,7 +17,9 @@ fft.o mdct.o raw.o golomb.o cabac.o\ dpcm.o adx.o faandct.o parser.o g726.o \ vp3dsp.o h264idct.o rangecoder.o pnm.o h263.o msmpeg4.o h263dec.o \ - opt.o + opt.o \ + bitstream_filter.o \ + HEADERS = avcodec.h diff -r 54814e15aa3d -r b7826511f7b6 allcodecs.c --- a/allcodecs.c Thu Jul 06 13:53:07 2006 +0000 +++ b/allcodecs.c Thu Jul 06 15:04:46 2006 +0000 @@ -661,5 +661,8 @@ av_register_codec_parser(&dvbsub_parser); #endif av_register_codec_parser(&aac_parser); + + av_register_bitstream_filter(&dump_extradata_bsf); + av_register_bitstream_filter(&remove_extradata_bsf); } diff -r 54814e15aa3d -r b7826511f7b6 avcodec.h --- a/avcodec.h Thu Jul 06 13:53:07 2006 +0000 +++ b/avcodec.h Thu Jul 06 15:04:46 2006 +0000 @@ -2571,6 +2571,37 @@ extern AVCodecParser dvbsub_parser; extern AVCodecParser aac_parser; + +typedef struct AVBitStreamFilterContext { + struct AVBitStreamFilter *filter; + AVCodecParserContext *parser; + struct AVBitStreamFilterContext *next; +} AVBitStreamFilterContext; + + +typedef struct AVBitStreamFilter { + const char *name; + int (*filter)(AVBitStreamFilterContext *bsfc, + AVCodecContext *avctx, const char *args, + uint8_t **poutbuf, int *poutbuf_size, + const uint8_t *buf, int buf_size, int keyframe); + struct AVBitStreamFilter *next; +} AVBitStreamFilter; + +extern AVBitStreamFilter *av_first_bitstream_filter; + +void av_register_bitstream_filter(AVBitStreamFilter *bsf); +AVBitStreamFilterContext *av_bitstream_filter_init(const char *name); +int av_bitstream_filter_filter(AVBitStreamFilterContext *bsfc, + AVCodecContext *avctx, const char *args, + uint8_t **poutbuf, int *poutbuf_size, + const uint8_t *buf, int buf_size, int keyframe); +void av_bitstream_filter_close(AVBitStreamFilterContext *bsf); + +extern AVBitStreamFilter dump_extradata_bsf; +extern AVBitStreamFilter remove_extradata_bsf; + + /* memory */ void *av_malloc(unsigned int size); void *av_mallocz(unsigned int size); diff -r 54814e15aa3d -r b7826511f7b6 bitstream_filter.c --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/bitstream_filter.c Thu Jul 06 15:04:46 2006 +0000 @@ -0,0 +1,96 @@ + +#include "avcodec.h" + +AVBitStreamFilter *first_bitstream_filter= NULL; + +void av_register_bitstream_filter(AVBitStreamFilter *bsf){ + bsf->next = first_bitstream_filter; + first_bitstream_filter= bsf; +} + +AVBitStreamFilterContext *av_bitstream_filter_init(const char *name){ + AVBitStreamFilter *bsf= first_bitstream_filter; + + while(bsf){ + if(!strcmp(name, bsf->name)){ + AVBitStreamFilterContext *bsfc= av_mallocz(sizeof(AVBitStreamFilterContext)); + bsfc->filter= bsf; + return bsfc; + } + bsf= bsf->next; + } + return NULL; +} + +void av_bitstream_filter_close(AVBitStreamFilterContext *bsfc){ + av_parser_close(bsfc->parser); + av_free(bsfc); +} + +int av_bitstream_filter_filter(AVBitStreamFilterContext *bsfc, + AVCodecContext *avctx, const char *args, + uint8_t **poutbuf, int *poutbuf_size, + const uint8_t *buf, int buf_size, int keyframe){ + return bsfc->filter->filter(bsfc, avctx, args, poutbuf, poutbuf_size, buf, buf_size, keyframe); +} + +static int dump_extradata(AVBitStreamFilterContext *bsfc, AVCodecContext *avctx, const char *args, + uint8_t **poutbuf, int *poutbuf_size, + const uint8_t *buf, int buf_size, int keyframe){ + int cmd= args ? *args : 0; + /* cast to avoid warning about discarding qualifiers */ + *poutbuf= (uint8_t *) buf; + *poutbuf_size= buf_size; + if(avctx->extradata){ + if( (keyframe && (avctx->flags2 & CODEC_FLAG2_LOCAL_HEADER) && cmd=='a') + ||(keyframe && (cmd=='k' || !cmd)) + ||(cmd=='e') + /*||(? && (s->flags & PARSER_FLAG_DUMP_EXTRADATA_AT_BEGIN)*/){ + int size= buf_size + avctx->extradata_size; + *poutbuf_size= size; + *poutbuf= av_malloc(size + FF_INPUT_BUFFER_PADDING_SIZE); + + memcpy(*poutbuf, avctx->extradata, avctx->extradata_size); + memcpy((*poutbuf) + avctx->extradata_size, buf, buf_size + FF_INPUT_BUFFER_PADDING_SIZE); + return 1; + } + } + return 0; +} + +static int remove_extradata(AVBitStreamFilterContext *bsfc, AVCodecContext *avctx, const char *args, + uint8_t **poutbuf, int *poutbuf_size, + const uint8_t *buf, int buf_size, int keyframe){ + int cmd= args ? *args : 0; + AVCodecParserContext *s; + + if(!bsfc->parser){ + bsfc->parser= av_parser_init(avctx->codec_id); + } + s= bsfc->parser; + + if(s && s->parser->split){ + if( (((avctx->flags & CODEC_FLAG_GLOBAL_HEADER) || (avctx->flags2 & CODEC_FLAG2_LOCAL_HEADER)) && cmd=='a') + ||(!keyframe && cmd=='k') + ||(cmd=='e' || !cmd) + ){ + int i= s->parser->split(avctx, buf, buf_size); + buf += i; + buf_size -= i; + } + } + *poutbuf= (uint8_t *) buf; + *poutbuf_size= buf_size; + + return 0; +} + +AVBitStreamFilter dump_extradata_bsf={ + "dump_extra", + dump_extradata, +}; + +AVBitStreamFilter remove_extradata_bsf={ + "remove_extra", + remove_extradata, +}; diff -r 54814e15aa3d -r b7826511f7b6 parser.c --- a/parser.c Thu Jul 06 13:53:07 2006 +0000 +++ b/parser.c Thu Jul 06 15:04:46 2006 +0000 @@ -145,6 +145,7 @@ /** * * @return 0 if the output buffer is a subset of the input, 1 if it is allocated and must be freed + * @deprecated use AVBitstreamFilter */ int av_parser_change(AVCodecParserContext *s, AVCodecContext *avctx,