path: root/libavcodec/remove_extradata_bsf.c (plain)
blob: a54bbdbacfcee867c79219b0daa47e0661abc23f

1	/*
2	* copyright (c) 2006 Michael Niedermayer <michaelni@gmx.at>
3	*
4	* This file is part of FFmpeg.
5	*
6	* FFmpeg is free software; you can redistribute it and/or
7	* modify it under the terms of the GNU Lesser General Public
8	* License as published by the Free Software Foundation; either
9	* version 2.1 of the License, or (at your option) any later version.
10	*
11	* FFmpeg is distributed in the hope that it will be useful,
12	* but WITHOUT ANY WARRANTY; without even the implied warranty of
13	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14	* Lesser General Public License for more details.
15	*
16	* You should have received a copy of the GNU Lesser General Public
17	* License along with FFmpeg; if not, write to the Free Software
18	* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19	*/
20
21	#include "libavutil/log.h"
22	#include "libavutil/opt.h"
23
24	#include "avcodec.h"
25	#include "bsf.h"
26
27	enum RemoveFreq {
28	REMOVE_FREQ_KEYFRAME,
29	REMOVE_FREQ_ALL,
30	REMOVE_FREQ_NONKEYFRAME,
31	};
32
33	typedef struct RemoveExtradataContext {
34	const AVClass *class;
35	int freq;
36
37	AVCodecParserContext *parser;
38	AVCodecContext *avctx;
39	} RemoveExtradataContext;
40
41	static int remove_extradata(AVBSFContext *ctx, AVPacket *out)
42	{
43	RemoveExtradataContext *s = ctx->priv_data;
44
45	AVPacket *in;
46	int ret;
47
48	ret = ff_bsf_get_packet(ctx, &in);
49	if (ret < 0)
50	return ret;
51
52	if (s->parser && s->parser->parser->split) {
53	if (s->freq == REMOVE_FREQ_ALL ||
54	(s->freq == REMOVE_FREQ_NONKEYFRAME && !(in->flags & AV_PKT_FLAG_KEY)) ||
55	(s->freq == REMOVE_FREQ_KEYFRAME && in->flags & AV_PKT_FLAG_KEY)) {
56	int i = s->parser->parser->split(s->avctx, in->data, in->size);
57	in->data += i;
58	in->size -= i;
59	}
60	}
61
62	av_packet_move_ref(out, in);
63	av_packet_free(&in);
64
65	return 0;
66	}
67
68	static int remove_extradata_init(AVBSFContext *ctx)
69	{
70	RemoveExtradataContext *s = ctx->priv_data;
71	int ret;
72
73	s->parser = av_parser_init(ctx->par_in->codec_id);
74
75	if (s->parser) {
76	s->avctx = avcodec_alloc_context3(NULL);
77	if (!s->avctx)
78	return AVERROR(ENOMEM);
79
80	ret = avcodec_parameters_to_context(s->avctx, ctx->par_in);
81	if (ret < 0)
82	return ret;
83	}
84
85	return 0;
86	}
87
88	static void remove_extradata_close(AVBSFContext *ctx)
89	{
90	RemoveExtradataContext *s = ctx->priv_data;
91
92	avcodec_free_context(&s->avctx);
93	av_parser_close(s->parser);
94	}
95
96	#define OFFSET(x) offsetof(RemoveExtradataContext, x)
97	static const AVOption options[] = {
98	{ "freq", NULL, OFFSET(freq), AV_OPT_TYPE_INT, { .i64 = REMOVE_FREQ_KEYFRAME }, REMOVE_FREQ_KEYFRAME, REMOVE_FREQ_NONKEYFRAME, 0, "freq" },
99	{ "k", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = REMOVE_FREQ_NONKEYFRAME }, .unit = "freq" },
100	{ "keyframe", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = REMOVE_FREQ_KEYFRAME }, .unit = "freq" },
101	{ "e", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = REMOVE_FREQ_ALL }, .unit = "freq" },
102	{ "all", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = REMOVE_FREQ_ALL }, .unit = "freq" },
103	{ NULL },
104	};
105
106	static const AVClass remove_extradata_class = {
107	.class_name = "remove_extradata",
108	.item_name = av_default_item_name,
109	.option = options,
110	.version = LIBAVUTIL_VERSION_INT,
111	};
112
113	const AVBitStreamFilter ff_remove_extradata_bsf = {
114	.name = "remove_extra",
115	.priv_data_size = sizeof(RemoveExtradataContext),
116	.priv_class = &remove_extradata_class,
117	.init = remove_extradata_init,
118	.close = remove_extradata_close,
119	.filter = remove_extradata,
120	};
121