FFmpeg
noise_bsf.c
Go to the documentation of this file.
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 <stdlib.h>
22 
23 #include "bsf.h"
24 #include "bsf_internal.h"
25 
26 #include "libavutil/avstring.h"
27 #include "libavutil/log.h"
28 #include "libavutil/opt.h"
29 #include "libavutil/eval.h"
30 
31 static const char *const var_names[] = {
32  "n", /// packet index, starting from zero
33  "tb", /// timebase
34  "pts", /// packet presentation timestamp
35  "dts", /// packet decoding timestamp
36  "nopts", /// AV_NOPTS_VALUE
37  "startpts", /// first seen non-AV_NOPTS_VALUE packet timestamp
38  "startdts", /// first seen non-AV_NOPTS_VALUE packet timestamp
39  "duration", "d", /// packet duration
40  "pos", /// original position of packet in its source
41  "size", /// packet size
42  "key" , /// packet keyframe flag
43  "state", /// random-ish state
44  NULL
45 };
46 
47 enum var_name {
61 };
62 
63 typedef struct NoiseContext {
64  const AVClass *class;
65 
66  char *amount_str;
67  char *drop_str;
69 
72 
74 
75  unsigned int state;
76  unsigned int pkt_idx;
77 } NoiseContext;
78 
80 {
82  int ret;
83 
84  if (!s->amount_str) {
85  s->amount_str = (!s->drop_str && !s->dropamount) ? av_strdup("-1") : av_strdup("0");
86  if (!s->amount_str)
87  return AVERROR(ENOMEM);
88  }
89 
90  ret = av_expr_parse(&s->amount_pexpr, s->amount_str,
91  var_names, NULL, NULL, NULL, NULL, 0, ctx);
92  if (ret < 0) {
93  av_log(ctx, AV_LOG_ERROR, "Error in parsing expr for amount: %s\n", s->amount_str);
94  return ret;
95  }
96 
97  if (s->drop_str && s->dropamount) {
98  av_log(ctx, AV_LOG_WARNING, "Both drop '%s' and dropamount=%d set. Ignoring dropamount.\n",
99  s->drop_str, s->dropamount);
100  s->dropamount = 0;
101  }
102 
103  if (s->drop_str) {
104  ret = av_expr_parse(&s->drop_pexpr, s->drop_str,
105  var_names, NULL, NULL, NULL, NULL, 0, ctx);
106  if (ret < 0) {
107  av_log(ctx, AV_LOG_ERROR, "Error in parsing expr for drop: %s\n", s->drop_str);
108  return ret;
109  }
110  }
111 
112  s->var_values[VAR_TB] = ctx->time_base_out.den ? av_q2d(ctx->time_base_out) : 0;
113  s->var_values[VAR_NOPTS] = AV_NOPTS_VALUE;
114  s->var_values[VAR_STARTPTS] = AV_NOPTS_VALUE;
115  s->var_values[VAR_STARTDTS] = AV_NOPTS_VALUE;
116  s->var_values[VAR_STATE] = 0;
117 
118  return 0;
119 }
120 
122 {
124  int i, ret, amount, drop = 0;
125  double res;
126 
128  if (ret < 0)
129  return ret;
130 
131  s->var_values[VAR_N] = s->pkt_idx++;
132  s->var_values[VAR_PTS] = pkt->pts;
133  s->var_values[VAR_DTS] = pkt->dts;
134  s->var_values[VAR_DURATION] =
135  s->var_values[VAR_D] = pkt->duration;
136  s->var_values[VAR_SIZE] = pkt->size;
137  s->var_values[VAR_KEY] = !!(pkt->flags & AV_PKT_FLAG_KEY);
138  s->var_values[VAR_POS] = pkt->pos;
139 
140  if (s->var_values[VAR_STARTPTS] == AV_NOPTS_VALUE)
141  s->var_values[VAR_STARTPTS] = pkt->pts;
142 
143  if (s->var_values[VAR_STARTDTS] == AV_NOPTS_VALUE)
144  s->var_values[VAR_STARTDTS] = pkt->dts;
145 
146  res = av_expr_eval(s->amount_pexpr, s->var_values, NULL);
147 
148  if (isnan(res))
149  amount = 0;
150  else if (res < 0)
151  amount = (s->state % 10001 + 1);
152  else
153  amount = (int)res;
154 
155  if (s->drop_str) {
156  res = av_expr_eval(s->drop_pexpr, s->var_values, NULL);
157 
158  if (isnan(res))
159  drop = 0;
160  else if (res < 0)
161  drop = !(s->state % FFABS((int)res));
162  else
163  drop = !!res;
164  }
165 
166  if(s->dropamount) {
167  drop = !(s->state % s->dropamount);
168  }
169 
170  av_log(ctx, AV_LOG_VERBOSE, "Stream #%d packet %d pts %"PRId64" - amount %d drop %d\n",
171  pkt->stream_index, (unsigned int)s->var_values[VAR_N], pkt->pts, amount, drop);
172 
173  if (drop) {
174  s->var_values[VAR_STATE] = ++s->state;
176  return AVERROR(EAGAIN);
177  }
178 
179  if (amount) {
181  if (ret < 0) {
183  return ret;
184  }
185  }
186 
187  for (i = 0; i < pkt->size; i++) {
188  s->state += pkt->data[i] + 1;
189  if (amount && s->state % amount == 0)
190  pkt->data[i] = s->state;
191  }
192 
193  s->var_values[VAR_STATE] = s->state;
194 
195  return 0;
196 }
197 
198 static void noise_close(AVBSFContext *bsf)
199 {
200  NoiseContext *s = bsf->priv_data;
201 
202  av_expr_free(s->amount_pexpr);
203  av_expr_free(s->drop_pexpr);
204  s->amount_pexpr = s->drop_pexpr = NULL;
205 }
206 
207 #define OFFSET(x) offsetof(NoiseContext, x)
208 #define FLAGS (AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_AUDIO_PARAM|AV_OPT_FLAG_BSF_PARAM)
209 static const AVOption options[] = {
210  { "amount", NULL, OFFSET(amount_str), AV_OPT_TYPE_STRING, { .str = NULL }, 0, 0, FLAGS },
211  { "drop", NULL, OFFSET(drop_str), AV_OPT_TYPE_STRING, { .str = NULL }, 0, 0, FLAGS },
212  { "dropamount", NULL, OFFSET(dropamount), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, INT_MAX, FLAGS },
213  { NULL },
214 };
215 
216 static const AVClass noise_class = {
217  .class_name = "noise",
218  .item_name = av_default_item_name,
219  .option = options,
220  .version = LIBAVUTIL_VERSION_INT,
221 };
222 
224  .name = "noise",
225  .priv_data_size = sizeof(NoiseContext),
226  .priv_class = &noise_class,
227  .init = noise_init,
228  .close = noise_close,
229  .filter = noise,
230 };
av_packet_unref
void av_packet_unref(AVPacket *pkt)
Wipe the packet.
Definition: avpacket.c:424
AV_LOG_WARNING
#define AV_LOG_WARNING
Something somehow does not look correct.
Definition: log.h:186
AVERROR
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a all references to both lists are replaced with a reference to the intersection And when a single format is eventually chosen for a link amongst the remaining all references to the list are updated That means that if a filter requires that its input and output have the same format amongst a supported all it has to do is use a reference to the same list of formats query_formats can leave some formats unset and return AVERROR(EAGAIN) to cause the negotiation mechanism toagain later. That can be used by filters with complex requirements to use the format negotiated on one link to set the formats supported on another. Frame references ownership and permissions
opt.h
bsf_internal.h
VAR_D
@ VAR_D
Definition: noise_bsf.c:55
var_names
static const char *const var_names[]
Definition: noise_bsf.c:31
AVBitStreamFilter::name
const char * name
Definition: bsf.h:91
AVPacket::data
uint8_t * data
Definition: packet.h:373
AVOption
AVOption.
Definition: opt.h:247
AV_LOG_VERBOSE
#define AV_LOG_VERBOSE
Detailed information.
Definition: log.h:196
AVPacket::duration
int64_t duration
Duration of this packet in AVStream->time_base units, 0 if unknown.
Definition: packet.h:391
filter
filter_frame For filters that do not use the this method is called when a frame is pushed to the filter s input It can be called at any time except in a reentrant way If the input frame is enough to produce then the filter should push the output frames on the output link immediately As an exception to the previous rule if the input frame is enough to produce several output frames then the filter needs output only at least one per link The additional frames can be left buffered in the filter
Definition: filter_design.txt:228
VAR_SIZE
@ VAR_SIZE
Definition: noise_bsf.c:57
AV_PKT_FLAG_KEY
#define AV_PKT_FLAG_KEY
The packet contains a keyframe.
Definition: packet.h:428
AVBSFContext
The bitstream filter state.
Definition: bsf.h:47
NoiseContext::drop_pexpr
AVExpr * drop_pexpr
Definition: noise_bsf.c:71
av_expr_parse
int av_expr_parse(AVExpr **expr, const char *s, const char *const *const_names, const char *const *func1_names, double(*const *funcs1)(void *, double), const char *const *func2_names, double(*const *funcs2)(void *, double, double), int log_offset, void *log_ctx)
Parse an expression.
Definition: eval.c:685
init
static int init
Definition: av_tx.c:47
bsf.h
FLAGS
#define FLAGS
Definition: noise_bsf.c:208
VAR_NOPTS
@ VAR_NOPTS
Definition: noise_bsf.c:52
NoiseContext::state
unsigned int state
Definition: noise_bsf.c:75
NoiseContext
Definition: noise_bsf.c:63
av_expr_free
void av_expr_free(AVExpr *e)
Free a parsed expression previously created with av_expr_parse().
Definition: eval.c:336
pkt
AVPacket * pkt
Definition: movenc.c:59
AV_LOG_ERROR
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:180
s
#define s(width, name)
Definition: cbs_vp9.c:257
noise_init
static int noise_init(AVBSFContext *ctx)
Definition: noise_bsf.c:79
av_q2d
static double av_q2d(AVRational a)
Convert an AVRational to a double.
Definition: rational.h:104
var_name
var_name
Definition: noise_bsf.c:47
ctx
AVFormatContext * ctx
Definition: movenc.c:48
av_expr_eval
double av_expr_eval(AVExpr *e, const double *const_values, void *opaque)
Evaluate a previously parsed expression.
Definition: eval.c:766
AVExpr
Definition: eval.c:157
VAR_VARS_NB
@ VAR_VARS_NB
Definition: noise_bsf.c:60
VAR_KEY
@ VAR_KEY
Definition: noise_bsf.c:58
FFABS
#define FFABS(a)
Absolute value, Note, INT_MIN / INT64_MIN result in undefined behavior as they are not representable ...
Definition: common.h:65
LIBAVUTIL_VERSION_INT
#define LIBAVUTIL_VERSION_INT
Definition: version.h:85
AVClass
Describe the class of an AVClass context structure.
Definition: log.h:66
ff_noise_bsf
const AVBitStreamFilter ff_noise_bsf
Definition: noise_bsf.c:223
NULL
#define NULL
Definition: coverity.c:32
isnan
#define isnan(x)
Definition: libm.h:340
NoiseContext::dropamount
int dropamount
Definition: noise_bsf.c:68
av_default_item_name
const char * av_default_item_name(void *ptr)
Return the context name.
Definition: log.c:235
VAR_POS
@ VAR_POS
Definition: noise_bsf.c:56
VAR_PTS
@ VAR_PTS
Definition: noise_bsf.c:50
eval.h
VAR_DURATION
@ VAR_DURATION
Definition: noise_bsf.c:55
AVPacket::size
int size
Definition: packet.h:374
VAR_N
@ VAR_N
Definition: noise_bsf.c:48
AV_NOPTS_VALUE
#define AV_NOPTS_VALUE
Undefined timestamp value.
Definition: avutil.h:248
AVPacket::dts
int64_t dts
Decompression timestamp in AVStream->time_base units; the time at which the packet is decompressed.
Definition: packet.h:372
VAR_STATE
@ VAR_STATE
Definition: noise_bsf.c:59
AVPacket::flags
int flags
A combination of AV_PKT_FLAG values.
Definition: packet.h:379
OFFSET
#define OFFSET(x)
Definition: noise_bsf.c:207
log.h
i
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:271
AVPacket::pts
int64_t pts
Presentation timestamp in AVStream->time_base units; the time at which the decompressed packet will b...
Definition: packet.h:366
VAR_STARTDTS
@ VAR_STARTDTS
Definition: noise_bsf.c:54
NoiseContext::amount_pexpr
AVExpr * amount_pexpr
Definition: noise_bsf.c:70
AVBSFContext::priv_data
void * priv_data
Opaque filter-specific private data.
Definition: bsf.h:62
ret
ret
Definition: filter_design.txt:187
AVClass::class_name
const char * class_name
The name of the class; usually it is the same name as the context structure type to which the AVClass...
Definition: log.h:71
noise
static int noise(AVBSFContext *ctx, AVPacket *pkt)
Definition: noise_bsf.c:121
AVBitStreamFilter
Definition: bsf.h:90
AV_OPT_TYPE_INT
@ AV_OPT_TYPE_INT
Definition: opt.h:224
av_packet_make_writable
int av_packet_make_writable(AVPacket *pkt)
Create a writable reference for the data described by a given packet, avoiding data copy if possible.
Definition: avpacket.c:506
NoiseContext::amount_str
char * amount_str
Definition: noise_bsf.c:66
AVPacket::stream_index
int stream_index
Definition: packet.h:375
NoiseContext::var_values
double var_values[VAR_VARS_NB]
Definition: noise_bsf.c:73
av_strdup
char * av_strdup(const char *s)
Duplicate a string.
Definition: mem.c:279
VAR_TB
@ VAR_TB
Definition: noise_bsf.c:49
options
static const AVOption options[]
Definition: noise_bsf.c:209
AVPacket
This structure stores compressed data.
Definition: packet.h:350
VAR_DTS
@ VAR_DTS
Definition: noise_bsf.c:51
AVPacket::pos
int64_t pos
byte position in stream, -1 if unknown
Definition: packet.h:393
NoiseContext::drop_str
char * drop_str
Definition: noise_bsf.c:67
noise_close
static void noise_close(AVBSFContext *bsf)
Definition: noise_bsf.c:198
av_log
#define av_log(a,...)
Definition: tableprint_vlc.h:28
ff_bsf_get_packet_ref
int ff_bsf_get_packet_ref(AVBSFContext *ctx, AVPacket *pkt)
Called by bitstream filters to get packet for filtering.
Definition: bsf.c:252
VAR_STARTPTS
@ VAR_STARTPTS
Definition: noise_bsf.c:53
avstring.h
AV_OPT_TYPE_STRING
@ AV_OPT_TYPE_STRING
Definition: opt.h:228
NoiseContext::pkt_idx
unsigned int pkt_idx
Definition: noise_bsf.c:76
int
int
Definition: ffmpeg_filter.c:153
AVFormatContext::priv_data
void * priv_data
Format private data.
Definition: avformat.h:1228
noise_class
static const AVClass noise_class
Definition: noise_bsf.c:216