72 #define OFFSET(x) offsetof(ShowFreqsContext, x)
73 #define FLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM
115 {
"colors",
"set channels colors",
OFFSET(colors),
AV_OPT_TYPE_STRING, {.str =
"red|green|blue|yellow|orange|lime|pink|magenta|brown" }, 0, 0,
FLAGS },
178 s->nb_freq = 1 << (
s->fft_bits - 1);
179 s->win_size =
s->nb_freq << 1;
185 "The window size might be too high.\n");
192 for (
i = 0;
i <
s->nb_channels;
i++) {
200 s->fft_data =
av_calloc(
s->nb_channels,
sizeof(*
s->fft_data));
203 s->avg_data =
av_calloc(
s->nb_channels,
sizeof(*
s->avg_data));
206 for (
i = 0;
i <
s->nb_channels;
i++) {
207 s->fft_data[
i] =
av_calloc(
s->win_size,
sizeof(**
s->fft_data));
208 s->avg_data[
i] =
av_calloc(
s->nb_freq,
sizeof(**
s->avg_data));
209 if (!
s->fft_data[
i] || !
s->avg_data[
i])
215 sizeof(*
s->window_func_lut));
216 if (!
s->window_func_lut)
219 if (
s->overlap == 1.)
220 s->overlap = overlap;
221 s->hop_size = (1. -
s->overlap) *
s->win_size;
222 if (
s->hop_size < 1) {
227 for (
s->scale = 0,
i = 0; i < s->win_size;
i++) {
228 s->scale +=
s->window_func_lut[
i] *
s->window_func_lut[
i];
247 if ((
color & 0xffffff) != 0)
257 return (
s->w/(
float)
s->nb_freq)*
f;
259 return s->w-pow(
s->w, (
s->nb_freq-
f-1)/(
s->nb_freq-1.));
261 return pow(
s->w,
f/(
s->nb_freq-1.));
271 return s->w/(float)
s->nb_freq;
273 return pow(
s->w, (
s->nb_freq-
f-1)/(
s->nb_freq-1.))-
274 pow(
s->w, (
s->nb_freq-
f-2)/(
s->nb_freq-1.));
276 return pow(
s->w, (
f+1)/(
s->nb_freq-1.))-
277 pow(
s->w,
f /(
s->nb_freq-1.));
284 double a,
int f,
uint8_t fg[4],
int *prev_y,
288 const float min =
s->minamp;
289 const float avg =
s->avg_data[ch][
f];
292 int end = outlink->
h;
312 y =
a * outlink->
h - 1;
315 end = (outlink->
h /
s->nb_channels) * (ch + 1);
316 y = (outlink->
h /
s->nb_channels) * ch +
a * (outlink->
h /
s->nb_channels) - 1;
332 y =
s->avg_data[ch][
f];
342 for (x = sx + 1; x < sx + bsize && x <
w; x++)
344 for (
i = y;
i <= *prev_y;
i++)
347 for (
i = *prev_y;
i <= y;
i++)
349 for (x = sx + 1; x < sx + bsize && x <
w; x++)
355 for (x = sx; x < sx + bsize && x <
w; x++)
356 for (
i = y;
i < end;
i++)
360 for (x = sx; x < sx + bsize && x <
w; x++)
371 const int win_size =
s->win_size;
380 for (n = 0; n < outlink->
h; n++)
381 memset(
out->data[0] +
out->linesize[0] * n, 0, outlink->
w * 4);
384 for (ch = 0; ch <
s->nb_channels; ch++) {
385 const float *p = (
float *)
in->extended_data[ch];
387 for (n = 0; n <
in->nb_samples; n++) {
388 s->fft_data[ch][n].re = p[n] *
s->window_func_lut[n];
389 s->fft_data[ch][n].im = 0;
391 for (; n < win_size; n++) {
392 s->fft_data[ch][n].re = 0;
393 s->fft_data[ch][n].im = 0;
398 for (ch = 0; ch <
s->nb_channels; ch++) {
403 #define RE(x, ch) s->fft_data[ch][x].re
404 #define IM(x, ch) s->fft_data[ch][x].im
405 #define M(a, b) (sqrt((a) * (a) + (b) * (b)))
406 #define P(a, b) (atan2((b), (a)))
414 for (ch = 0; ch <
s->nb_channels; ch++) {
415 uint8_t fg[4] = { 0xff, 0xff, 0xff, 0xff };
423 switch (
s->data_mode) {
428 for (
f = 1;
f <
s->nb_freq;
f++) {
438 for (
f = 1;
f <
s->nb_freq;
f++) {
447 for (
f = 1;
f <
s->nb_freq;
f++) {
477 s->pts +=
s->hop_size;
530 for (
i = 0;
i <
s->nb_channels;
i++) {
569 .priv_class = &showfreqs_class,
static enum AVSampleFormat sample_fmts[]
static const AVFilterPad inputs[]
static const AVFilterPad outputs[]
AVFrame * ff_get_audio_buffer(AVFilterLink *link, int nb_samples)
Request an audio samples buffer with a specific set of permissions.
uint8_t pi<< 24) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_U8, uint8_t,(*(const uint8_t *) pi - 0x80) *(1.0f/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_U8, uint8_t,(*(const uint8_t *) pi - 0x80) *(1.0/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S16, int16_t,(*(const int16_t *) pi >> 8)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S16, int16_t, *(const int16_t *) pi *(1.0f/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S16, int16_t, *(const int16_t *) pi *(1.0/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S32, int32_t,(*(const int32_t *) pi >> 24)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S32, int32_t, *(const int32_t *) pi *(1.0f/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S32, int32_t, *(const int32_t *) pi *(1.0/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_FLT, float, av_clip_uint8(lrintf(*(const float *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_FLT, float, av_clip_int16(lrintf(*(const float *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_FLT, float, av_clipl_int32(llrintf(*(const float *) pi *(1U<< 31)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_DBL, double, av_clip_uint8(lrint(*(const double *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_DBL, double, av_clip_int16(lrint(*(const double *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_DBL, double, av_clipl_int32(llrint(*(const double *) pi *(1U<< 31)))) #define SET_CONV_FUNC_GROUP(ofmt, ifmt) static void set_generic_function(AudioConvert *ac) { } void ff_audio_convert_free(AudioConvert **ac) { if(! *ac) return;ff_dither_free(&(*ac) ->dc);av_freep(ac);} AudioConvert *ff_audio_convert_alloc(AVAudioResampleContext *avr, enum AVSampleFormat out_fmt, enum AVSampleFormat in_fmt, int channels, int sample_rate, int apply_map) { AudioConvert *ac;int in_planar, out_planar;ac=av_mallocz(sizeof(*ac));if(!ac) return NULL;ac->avr=avr;ac->out_fmt=out_fmt;ac->in_fmt=in_fmt;ac->channels=channels;ac->apply_map=apply_map;if(avr->dither_method !=AV_RESAMPLE_DITHER_NONE &&av_get_packed_sample_fmt(out_fmt)==AV_SAMPLE_FMT_S16 &&av_get_bytes_per_sample(in_fmt) > 2) { ac->dc=ff_dither_alloc(avr, out_fmt, in_fmt, channels, sample_rate, apply_map);if(!ac->dc) { av_free(ac);return NULL;} return ac;} in_planar=ff_sample_fmt_is_planar(in_fmt, channels);out_planar=ff_sample_fmt_is_planar(out_fmt, channels);if(in_planar==out_planar) { ac->func_type=CONV_FUNC_TYPE_FLAT;ac->planes=in_planar ? ac->channels :1;} else if(in_planar) ac->func_type=CONV_FUNC_TYPE_INTERLEAVE;else ac->func_type=CONV_FUNC_TYPE_DEINTERLEAVE;set_generic_function(ac);if(ARCH_AARCH64) ff_audio_convert_init_aarch64(ac);if(ARCH_ARM) ff_audio_convert_init_arm(ac);if(ARCH_X86) ff_audio_convert_init_x86(ac);return ac;} int ff_audio_convert(AudioConvert *ac, AudioData *out, AudioData *in) { int use_generic=1;int len=in->nb_samples;int p;if(ac->dc) { av_log(ac->avr, AV_LOG_TRACE, "%d samples - audio_convert: %s to %s (dithered)\n", len, av_get_sample_fmt_name(ac->in_fmt), av_get_sample_fmt_name(ac->out_fmt));return ff_convert_dither(ac-> in
simple assert() macros that are a bit more flexible than ISO C assert().
#define av_assert0(cond)
assert() equivalent, that is always enabled.
static float get_bsize(ShowFreqsContext *s, int f)
static int get_sx(ShowFreqsContext *s, int f)
static const AVFilterPad showfreqs_outputs[]
static const AVFilterPad showfreqs_inputs[]
static int query_formats(AVFilterContext *ctx)
AVFILTER_DEFINE_CLASS(showfreqs)
static int filter_frame(AVFilterLink *inlink)
static void draw_dot(AVFrame *out, int x, int y, uint8_t fg[4])
static int activate(AVFilterContext *ctx)
static av_cold int init(AVFilterContext *ctx)
static av_cold void uninit(AVFilterContext *ctx)
static const AVOption showfreqs_options[]
static int config_output(AVFilterLink *outlink)
AVFilter ff_avf_showfreqs
static int plot_freqs(AVFilterLink *inlink, AVFrame *in)
static void plot_freq(ShowFreqsContext *s, int ch, double a, int f, uint8_t fg[4], int *prev_y, AVFrame *out, AVFilterLink *outlink)
int ff_filter_frame(AVFilterLink *link, AVFrame *frame)
Send a frame of data to the next filter.
int ff_inlink_consume_samples(AVFilterLink *link, unsigned min, unsigned max, AVFrame **rframe)
Take samples from the link's FIFO and update the link's stats.
Main libavfilter public API header.
audio channel layout utility functions
mode
Use these values in ebur128_init (or'ed).
#define FF_FILTER_FORWARD_WANTED(outlink, inlink)
Forward the frame_wanted_out flag from an output link to an input link.
#define FF_FILTER_FORWARD_STATUS(inlink, outlink)
Acknowledge the status on an input link and forward it to an output link.
#define FFERROR_NOT_READY
Filters implementation helper functions.
#define FF_FILTER_FORWARD_STATUS_BACK(outlink, inlink)
Forward the status on an output link to an input link.
@ AV_OPT_TYPE_IMAGE_SIZE
offset must point to two consecutive integers
void av_fft_permute(FFTContext *s, FFTComplex *z)
Do the permutation needed BEFORE calling ff_fft_calc().
void av_fft_calc(FFTContext *s, FFTComplex *z)
Do a complex FFT with the parameters defined in av_fft_init().
FFTContext * av_fft_init(int nbits, int inverse)
Set up a complex FFT.
av_cold void av_fft_end(FFTContext *s)
int av_audio_fifo_peek(AVAudioFifo *af, void **data, int nb_samples)
Peek data from an AVAudioFifo.
int av_audio_fifo_write(AVAudioFifo *af, void **data, int nb_samples)
Write data to an AVAudioFifo.
void av_audio_fifo_free(AVAudioFifo *af)
Free an AVAudioFifo.
AVAudioFifo * av_audio_fifo_alloc(enum AVSampleFormat sample_fmt, int channels, int nb_samples)
Allocate an AVAudioFifo.
int av_audio_fifo_size(AVAudioFifo *af)
Get the current number of samples in the AVAudioFifo available for reading.
int av_audio_fifo_drain(AVAudioFifo *af, int nb_samples)
Drain data from an AVAudioFifo.
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
static AVRational av_make_q(int num, int den)
Create an AVRational.
char * av_strdup(const char *s)
Duplicate a string.
void * av_calloc(size_t nmemb, size_t size)
Non-inlined equivalent of av_mallocz_array().
AVSampleFormat
Audio sample formats.
@ AV_SAMPLE_FMT_FLTP
float, planar
char * av_strtok(char *s, const char *delim, char **saveptr)
Split the string into several tokens which can be accessed by successive calls to av_strtok().
#define AV_NOPTS_VALUE
Undefined timestamp value.
common internal API header
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
static enum AVPixelFormat pix_fmts[]
static const uint32_t color[16+AV_CLASS_CATEGORY_NB]
enum MovChannelLayoutTag * layouts
int av_parse_color(uint8_t *rgba_color, const char *color_string, int slen, void *log_ctx)
Put the RGBA values that correspond to color_string in rgba_color.
AVPixelFormat
Pixel format.
@ AV_PIX_FMT_RGBA
packed RGBA 8:8:8:8, 32bpp, RGBARGBA...
Context for an Audio FIFO Buffer.
Describe the class of an AVClass context structure.
A list of supported channel layouts.
A link between two filters.
AVFilterFormatsConfig incfg
Lists of supported formats / etc.
int w
agreed upon image width
int h
agreed upon image height
int channels
Number of channels.
AVFilterFormatsConfig outcfg
Lists of supported formats / etc.
AVFilterContext * src
source filter
int64_t frame_count_in
Number of past frames sent through the link.
int sample_rate
samples per second
AVRational sample_aspect_ratio
agreed upon sample aspect ratio
AVRational frame_rate
Frame rate of the stream on the link, or 1/0 if unknown or variable; if left to 0/0,...
AVFilterContext * dst
dest filter
int format
agreed upon media format
A filter pad used for either input or output.
const char * name
Pad name.
const char * name
Filter name.
This structure describes decoded (raw) audio or video data.
int64_t pts
Presentation timestamp in time_base units (time when frame should be shown to user).
uint8_t ** extended_data
pointers to the data planes/channels.
Rational number (pair of numerator and denominator).
#define av_realloc_f(p, o, n)
AVFrame * ff_get_video_buffer(AVFilterLink *link, int w, int h)
Request a picture buffer with a specific set of permissions.
static void generate_window_func(float *lut, int N, int win_func, float *overlap)