Statistics
| Branch: | Revision:

ffmpeg / libavcodec / bitstream_filter.c @ 514e0831

History | View | Annotate | Download (3.8 KB)

1

    
2
#include "avcodec.h"
3

    
4
AVBitStreamFilter *first_bitstream_filter= NULL;
5

    
6
void av_register_bitstream_filter(AVBitStreamFilter *bsf){
7
    bsf->next = first_bitstream_filter;
8
    first_bitstream_filter= bsf;
9
}
10

    
11
AVBitStreamFilterContext *av_bitstream_filter_init(const char *name){
12
    AVBitStreamFilter *bsf= first_bitstream_filter;
13

    
14
    while(bsf){
15
        if(!strcmp(name, bsf->name)){
16
            AVBitStreamFilterContext *bsfc= av_mallocz(sizeof(AVBitStreamFilterContext));
17
            bsfc->filter= bsf;
18
            bsfc->priv_data= av_mallocz(bsf->priv_data_size);
19
            return bsfc;
20
        }
21
        bsf= bsf->next;
22
    }
23
    return NULL;
24
}
25

    
26
void av_bitstream_filter_close(AVBitStreamFilterContext *bsfc){
27
    av_freep(&bsfc->priv_data);
28
    av_parser_close(bsfc->parser);
29
    av_free(bsfc);
30
}
31

    
32
int av_bitstream_filter_filter(AVBitStreamFilterContext *bsfc,
33
                               AVCodecContext *avctx, const char *args,
34
                     uint8_t **poutbuf, int *poutbuf_size,
35
                     const uint8_t *buf, int buf_size, int keyframe){
36
    *poutbuf= (uint8_t *) buf;
37
    *poutbuf_size= buf_size;
38
    return bsfc->filter->filter(bsfc, avctx, args, poutbuf, poutbuf_size, buf, buf_size, keyframe);
39
}
40

    
41
static int dump_extradata(AVBitStreamFilterContext *bsfc, AVCodecContext *avctx, const char *args,
42
                     uint8_t **poutbuf, int *poutbuf_size,
43
                     const uint8_t *buf, int buf_size, int keyframe){
44
    int cmd= args ? *args : 0;
45
    /* cast to avoid warning about discarding qualifiers */
46
    if(avctx->extradata){
47
        if(  (keyframe && (avctx->flags2 & CODEC_FLAG2_LOCAL_HEADER) && cmd=='a')
48
           ||(keyframe && (cmd=='k' || !cmd))
49
           ||(cmd=='e')
50
            /*||(? && (s->flags & PARSER_FLAG_DUMP_EXTRADATA_AT_BEGIN)*/){
51
            int size= buf_size + avctx->extradata_size;
52
            *poutbuf_size= size;
53
            *poutbuf= av_malloc(size + FF_INPUT_BUFFER_PADDING_SIZE);
54

    
55
            memcpy(*poutbuf, avctx->extradata, avctx->extradata_size);
56
            memcpy((*poutbuf) + avctx->extradata_size, buf, buf_size + FF_INPUT_BUFFER_PADDING_SIZE);
57
            return 1;
58
        }
59
    }
60
    return 0;
61
}
62

    
63
static int remove_extradata(AVBitStreamFilterContext *bsfc, AVCodecContext *avctx, const char *args,
64
                     uint8_t **poutbuf, int *poutbuf_size,
65
                     const uint8_t *buf, int buf_size, int keyframe){
66
    int cmd= args ? *args : 0;
67
    AVCodecParserContext *s;
68

    
69
    if(!bsfc->parser){
70
        bsfc->parser= av_parser_init(avctx->codec_id);
71
    }
72
    s= bsfc->parser;
73

    
74
    if(s && s->parser->split){
75
        if(  (((avctx->flags & CODEC_FLAG_GLOBAL_HEADER) || (avctx->flags2 & CODEC_FLAG2_LOCAL_HEADER)) && cmd=='a')
76
           ||(!keyframe && cmd=='k')
77
           ||(cmd=='e' || !cmd)
78
          ){
79
            int i= s->parser->split(avctx, buf, buf_size);
80
            buf += i;
81
            buf_size -= i;
82
        }
83
    }
84
    *poutbuf= (uint8_t *) buf;
85
    *poutbuf_size= buf_size;
86

    
87
    return 0;
88
}
89

    
90
static int noise(AVBitStreamFilterContext *bsfc, AVCodecContext *avctx, const char *args,
91
                     uint8_t **poutbuf, int *poutbuf_size,
92
                     const uint8_t *buf, int buf_size, int keyframe){
93
    int amount= args ? atoi(args) : 10000;
94
    unsigned int *state= bsfc->priv_data;
95
    int i;
96

    
97
    *poutbuf= av_malloc(buf_size + FF_INPUT_BUFFER_PADDING_SIZE);
98

    
99
    memcpy(*poutbuf, buf, buf_size + FF_INPUT_BUFFER_PADDING_SIZE);
100
    for(i=0; i<buf_size; i++){
101
        (*state) += (*poutbuf)[i] + 1;
102
        if(*state % amount == 0)
103
            (*poutbuf)[i] = *state;
104
    }
105
    return 1;
106
}
107

    
108
AVBitStreamFilter dump_extradata_bsf={
109
    "dump_extra",
110
    0,
111
    dump_extradata,
112
};
113

    
114
AVBitStreamFilter remove_extradata_bsf={
115
    "remove_extra",
116
    0,
117
    remove_extradata,
118
};
119

    
120
AVBitStreamFilter noise_bsf={
121
    "noise",
122
    sizeof(int),
123
    noise,
124
};