FFmpeg
af_crystalizer.c
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2016 The FFmpeg Project
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 
22 #include "libavutil/opt.h"
23 #include "avfilter.h"
24 #include "audio.h"
25 #include "filters.h"
26 
27 typedef struct CrystalizerContext {
28  const AVClass *class;
29  float mult;
30  int clip;
32  int (*filter[2][2])(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs);
34 
35 #define OFFSET(x) offsetof(CrystalizerContext, x)
36 #define A AV_OPT_FLAG_AUDIO_PARAM|AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_RUNTIME_PARAM
37 
38 static const AVOption crystalizer_options[] = {
39  { "i", "set intensity", OFFSET(mult), AV_OPT_TYPE_FLOAT, {.dbl=2.0},-10, 10, A },
40  { "c", "enable clipping", OFFSET(clip), AV_OPT_TYPE_BOOL, {.i64=1}, 0, 1, A },
41  { NULL }
42 };
43 
44 AVFILTER_DEFINE_CLASS(crystalizer);
45 
46 typedef struct ThreadData {
47  void **d;
48  void **p;
49  const void **s;
50  int nb_samples;
51  int channels;
52  float mult;
53 } ThreadData;
54 
55 #define filters(fmt, type, inverse, clp, inverset, clip, one, clip_fn, packed) \
56 static int filter_## inverse ##_## fmt ##_## clp(AVFilterContext *ctx, \
57  void *arg, int jobnr,\
58  int nb_jobs) \
59 { \
60  ThreadData *td = arg; \
61  void **d = td->d; \
62  void **p = td->p; \
63  const void **s = td->s; \
64  const int nb_samples = td->nb_samples; \
65  const int channels = td->channels; \
66  const type mult = td->mult; \
67  const type scale = one / (-mult + one); \
68  const int start = (channels * jobnr) / nb_jobs; \
69  const int end = (channels * (jobnr+1)) / nb_jobs; \
70  \
71  if (packed) { \
72  type *prv = p[0]; \
73  for (int c = start; c < end; c++) { \
74  const type *src = s[0]; \
75  type *dst = d[0]; \
76  \
77  for (int n = 0; n < nb_samples; n++) { \
78  type current = src[c]; \
79  \
80  if (inverset) { \
81  dst[c] = (current - prv[c] * mult) * scale; \
82  prv[c] = dst[c]; \
83  } else { \
84  dst[c] = current + (current - prv[c]) * mult; \
85  prv[c] = current; \
86  } \
87  if (clip) { \
88  dst[c] = clip_fn(dst[c], -one, one); \
89  } \
90  \
91  dst += channels; \
92  src += channels; \
93  } \
94  } \
95  } else { \
96  for (int c = start; c < end; c++) { \
97  const type *src = s[c]; \
98  type *dst = d[c]; \
99  type *prv = p[c]; \
100  \
101  for (int n = 0; n < nb_samples; n++) { \
102  type current = src[n]; \
103  \
104  if (inverset) { \
105  dst[n] = (current - prv[0] * mult) * scale; \
106  prv[0] = dst[n]; \
107  } else { \
108  dst[n] = current + (current - prv[0]) * mult; \
109  prv[0] = current; \
110  } \
111  if (clip) { \
112  dst[n] = clip_fn(dst[n], -one, one); \
113  } \
114  } \
115  } \
116  } \
117  \
118  return 0; \
119 }
120 
121 filters(flt, float, inverse, noclip, 1, 0, 1.f, av_clipf, 1)
122 filters(flt, float, inverse, clip, 1, 1, 1.f, av_clipf, 1)
123 filters(flt, float, noinverse, noclip, 0, 0, 1.f, av_clipf, 1)
124 filters(flt, float, noinverse, clip, 0, 1, 1.f, av_clipf, 1)
125 
126 filters(fltp, float, inverse, noclip, 1, 0, 1.f, av_clipf, 0)
127 filters(fltp, float, inverse, clip, 1, 1, 1.f, av_clipf, 0)
128 filters(fltp, float, noinverse, noclip, 0, 0, 1.f, av_clipf, 0)
129 filters(fltp, float, noinverse, clip, 0, 1, 1.f, av_clipf, 0)
130 
131 filters(dbl, double, inverse, noclip, 1, 0, 1.0, av_clipd, 1)
132 filters(dbl, double, inverse, clip, 1, 1, 1.0, av_clipd, 1)
133 filters(dbl, double, noinverse, noclip, 0, 0, 1.0, av_clipd, 1)
134 filters(dbl, double, noinverse, clip, 0, 1, 1.0, av_clipd, 1)
135 
136 filters(dblp, double, inverse, noclip, 1, 0, 1.0, av_clipd, 0)
137 filters(dblp, double, inverse, clip, 1, 1, 1.0, av_clipd, 0)
138 filters(dblp, double, noinverse, noclip, 0, 0, 1.0, av_clipd, 0)
139 filters(dblp, double, noinverse, clip, 0, 1, 1.0, av_clipd, 0)
140 
142 {
143  AVFilterContext *ctx = inlink->dst;
144  CrystalizerContext *s = ctx->priv;
145 
146  switch (inlink->format) {
147  case AV_SAMPLE_FMT_FLT:
148  s->filter[0][0] = filter_inverse_flt_noclip;
149  s->filter[1][0] = filter_noinverse_flt_noclip;
150  s->filter[0][1] = filter_inverse_flt_clip;
151  s->filter[1][1] = filter_noinverse_flt_clip;
152  break;
153  case AV_SAMPLE_FMT_FLTP:
154  s->filter[0][0] = filter_inverse_fltp_noclip;
155  s->filter[1][0] = filter_noinverse_fltp_noclip;
156  s->filter[0][1] = filter_inverse_fltp_clip;
157  s->filter[1][1] = filter_noinverse_fltp_clip;
158  break;
159  case AV_SAMPLE_FMT_DBL:
160  s->filter[0][0] = filter_inverse_dbl_noclip;
161  s->filter[1][0] = filter_noinverse_dbl_noclip;
162  s->filter[0][1] = filter_inverse_dbl_clip;
163  s->filter[1][1] = filter_noinverse_dbl_clip;
164  break;
165  case AV_SAMPLE_FMT_DBLP:
166  s->filter[0][0] = filter_inverse_dblp_noclip;
167  s->filter[1][0] = filter_noinverse_dblp_noclip;
168  s->filter[0][1] = filter_inverse_dblp_clip;
169  s->filter[1][1] = filter_noinverse_dblp_clip;
170  break;
171  default:
172  return AVERROR_BUG;
173  }
174 
175  return 0;
176 }
177 
179 {
180  AVFilterContext *ctx = inlink->dst;
181  AVFilterLink *outlink = ctx->outputs[0];
182  CrystalizerContext *s = ctx->priv;
183  AVFrame *out;
184  ThreadData td;
185 
186  if (!s->prev) {
187  s->prev = ff_get_audio_buffer(inlink, 1);
188  if (!s->prev) {
189  av_frame_free(&in);
190  return AVERROR(ENOMEM);
191  }
192  }
193 
194  if (av_frame_is_writable(in)) {
195  out = in;
196  } else {
197  out = ff_get_audio_buffer(outlink, in->nb_samples);
198  if (!out) {
199  av_frame_free(&in);
200  return AVERROR(ENOMEM);
201  }
203  }
204 
205  td.d = (void **)out->extended_data;
206  td.s = (const void **)in->extended_data;
207  td.p = (void **)s->prev->extended_data;
208  td.nb_samples = in->nb_samples;
209  td.channels = in->ch_layout.nb_channels;
210  td.mult = ctx->is_disabled ? 0.f : s->mult;
211  ff_filter_execute(ctx, s->filter[td.mult >= 0.f][s->clip], &td, NULL,
212  FFMIN(inlink->ch_layout.nb_channels, ff_filter_get_nb_threads(ctx)));
213 
214  if (out != in)
215  av_frame_free(&in);
216 
217  return ff_filter_frame(outlink, out);
218 }
219 
221 {
222  CrystalizerContext *s = ctx->priv;
223 
224  av_frame_free(&s->prev);
225 }
226 
227 static const AVFilterPad inputs[] = {
228  {
229  .name = "default",
230  .type = AVMEDIA_TYPE_AUDIO,
231  .filter_frame = filter_frame,
232  .config_props = config_input,
233  },
234 };
235 
237  .name = "crystalizer",
238  .description = NULL_IF_CONFIG_SMALL("Simple audio noise sharpening filter."),
239  .priv_size = sizeof(CrystalizerContext),
240  .priv_class = &crystalizer_class,
241  .uninit = uninit,
246  .process_command = ff_filter_process_command,
249 };
ff_get_audio_buffer
AVFrame * ff_get_audio_buffer(AVFilterLink *link, int nb_samples)
Request an audio samples buffer with a specific set of permissions.
Definition: audio.c:98
AV_SAMPLE_FMT_FLTP
@ AV_SAMPLE_FMT_FLTP
float, planar
Definition: samplefmt.h:66
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
out
FILE * out
Definition: movenc.c:55
inverse
inverse
Definition: af_crystalizer.c:122
CrystalizerContext::filter
int(* filter[2][2])(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
Definition: af_crystalizer.c:32
ff_filter_frame
int ff_filter_frame(AVFilterLink *link, AVFrame *frame)
Send a frame of data to the next filter.
Definition: avfilter.c:1061
config_input
static int config_input(AVFilterLink *inlink)
Definition: af_crystalizer.c:141
inlink
The exact code depends on how similar the blocks are and how related they are to the and needs to apply these operations to the correct inlink or outlink if there are several Macros are available to factor that when no extra processing is inlink
Definition: filter_design.txt:212
av_frame_free
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
Definition: frame.c:162
FILTER_INPUTS
#define FILTER_INPUTS(array)
Definition: filters.h:262
AVFrame
This structure describes decoded (raw) audio or video data.
Definition: frame.h:389
CrystalizerContext::clip
int clip
Definition: af_crystalizer.c:30
AVOption
AVOption.
Definition: opt.h:429
uninit
static av_cold void uninit(AVFilterContext *ctx)
Definition: af_crystalizer.c:220
AVFilter::name
const char * name
Filter name.
Definition: avfilter.h:205
AVChannelLayout::nb_channels
int nb_channels
Number of channels in this layout.
Definition: channel_layout.h:327
ThreadData::channels
int channels
Definition: af_asoftclip.c:402
AVFILTER_DEFINE_CLASS
AVFILTER_DEFINE_CLASS(crystalizer)
OFFSET
#define OFFSET(x)
Definition: af_crystalizer.c:35
AVFrame::ch_layout
AVChannelLayout ch_layout
Channel layout of the audio data.
Definition: frame.h:790
AVFilterPad
A filter pad used for either input or output.
Definition: filters.h:38
mult
static int16_t mult(Float11 *f1, Float11 *f2)
Definition: g726.c:60
av_cold
#define av_cold
Definition: attributes.h:90
FILTER_SAMPLEFMTS
#define FILTER_SAMPLEFMTS(...)
Definition: filters.h:250
clip
clip
Definition: af_crystalizer.c:122
s
#define s(width, name)
Definition: cbs_vp9.c:198
AVMEDIA_TYPE_AUDIO
@ AVMEDIA_TYPE_AUDIO
Definition: avutil.h:202
CrystalizerContext::mult
float mult
Definition: af_crystalizer.c:29
filters
#define filters(fmt, type, inverse, clp, inverset, clip, one, clip_fn, packed)
Definition: af_crystalizer.c:55
ThreadData::nb_samples
int nb_samples
Definition: af_asoftclip.c:401
filters.h
ctx
AVFormatContext * ctx
Definition: movenc.c:49
FILTER_OUTPUTS
#define FILTER_OUTPUTS(array)
Definition: filters.h:263
arg
const char * arg
Definition: jacosubdec.c:67
AVClass
Describe the class of an AVClass context structure.
Definition: log.h:75
NULL
#define NULL
Definition: coverity.c:32
ThreadData::mult
float mult
Definition: af_crystalizer.c:52
av_frame_copy_props
int av_frame_copy_props(AVFrame *dst, const AVFrame *src)
Copy only "metadata" fields from src to dst.
Definition: frame.c:725
ff_audio_default_filterpad
const AVFilterPad ff_audio_default_filterpad[1]
An AVFilterPad array whose only entry has name "default" and is of type AVMEDIA_TYPE_AUDIO.
Definition: audio.c:34
av_clipf
av_clipf
Definition: af_crystalizer.c:122
filter_frame
static int filter_frame(AVFilterLink *inlink, AVFrame *in)
Definition: af_crystalizer.c:178
f
f
Definition: af_crystalizer.c:122
NULL_IF_CONFIG_SMALL
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
Definition: internal.h:94
CrystalizerContext::prev
AVFrame * prev
Definition: af_crystalizer.c:31
av_frame_is_writable
int av_frame_is_writable(AVFrame *frame)
Check if the frame data is writable.
Definition: frame.c:661
ff_filter_process_command
int ff_filter_process_command(AVFilterContext *ctx, const char *cmd, const char *arg, char *res, int res_len, int flags)
Generic processing of user supplied commands that are set in the same way as the filter options.
Definition: avfilter.c:900
AV_OPT_TYPE_FLOAT
@ AV_OPT_TYPE_FLOAT
Underlying C type is float.
Definition: opt.h:271
ThreadData::d
void ** d
Definition: af_crystalizer.c:47
AVFrame::nb_samples
int nb_samples
number of audio samples (per channel) described by this frame
Definition: frame.h:469
inputs
static const AVFilterPad inputs[]
Definition: af_crystalizer.c:227
crystalizer_options
static const AVOption crystalizer_options[]
Definition: af_crystalizer.c:38
AVFrame::extended_data
uint8_t ** extended_data
pointers to the data planes/channels.
Definition: frame.h:450
ff_filter_get_nb_threads
int ff_filter_get_nb_threads(AVFilterContext *ctx)
Get number of threads for current filter instance.
Definition: avfilter.c:840
ThreadData
Used for passing data between threads.
Definition: dsddec.c:71
FFMIN
#define FFMIN(a, b)
Definition: macros.h:49
CrystalizerContext
Definition: af_crystalizer.c:27
AVFilterPad::name
const char * name
Pad name.
Definition: filters.h:44
AVFilter
Filter definition.
Definition: avfilter.h:201
A
#define A
Definition: af_crystalizer.c:36
channel_layout.h
ff_filter_execute
int ff_filter_execute(AVFilterContext *ctx, avfilter_action_func *func, void *arg, int *ret, int nb_jobs)
Definition: avfilter.c:1666
avfilter.h
AV_SAMPLE_FMT_DBLP
@ AV_SAMPLE_FMT_DBLP
double, planar
Definition: samplefmt.h:67
AVFilterContext
An instance of a filter.
Definition: avfilter.h:457
ThreadData::s
const void ** s
Definition: af_crystalizer.c:49
AVFILTER_FLAG_SLICE_THREADS
#define AVFILTER_FLAG_SLICE_THREADS
The filter supports multithreading by splitting frames into multiple parts and processing them concur...
Definition: avfilter.h:152
audio.h
ThreadData::p
void ** p
Definition: af_crystalizer.c:48
AV_OPT_TYPE_BOOL
@ AV_OPT_TYPE_BOOL
Underlying C type is int.
Definition: opt.h:327
AVFILTER_FLAG_SUPPORT_TIMELINE_INTERNAL
#define AVFILTER_FLAG_SUPPORT_TIMELINE_INTERNAL
Same as AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC, except that the filter will have its filter_frame() c...
Definition: avfilter.h:190
AVERROR_BUG
#define AVERROR_BUG
Internal bug, also see AVERROR_BUG2.
Definition: error.h:52
ff_af_crystalizer
const AVFilter ff_af_crystalizer
Definition: af_crystalizer.c:236
AV_SAMPLE_FMT_DBL
@ AV_SAMPLE_FMT_DBL
double
Definition: samplefmt.h:61
AV_SAMPLE_FMT_FLT
@ AV_SAMPLE_FMT_FLT
float
Definition: samplefmt.h:60
noinverse
noinverse
Definition: af_crystalizer.c:124
av_clipd
av_clipd
Definition: af_crystalizer.c:132