56 #define OFFSET(x) offsetof(ShowSpatialContext, x)
57 #define FLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM
96 for (
i = 0;
i < 2;
i++)
98 for (
i = 0;
i < 2;
i++)
100 for (
i = 0;
i < 2;
i++)
136 const float *window_func_lut =
s->window_func_lut;
138 const int ch = jobnr;
142 s->fft_data[ch][n].re = p[n] * window_func_lut[n];
143 s->fft_data[ch][n].im = 0;
165 s->win_size =
s->buf_size;
169 if (fft_bits !=
s->fft_bits) {
170 s->fft_bits = fft_bits;
175 for (
i = 0;
i < 2;
i++) {
179 for (
i = 0;
i < 2;
i++) {
183 "The window size might be too high.\n");
188 for (
i = 0;
i < 2;
i++) {
189 s->fft_data[
i] =
av_calloc(
s->buf_size,
sizeof(**
s->fft_data));
197 sizeof(*
s->window_func_lut));
198 if (!
s->window_func_lut)
202 s->overlap = overlap;
204 s->hop_size = (1.f -
s->overlap) *
s->win_size;
205 if (
s->hop_size < 1) {
220 #define RE(y, ch) s->fft_data[ch][y].re
221 #define IM(y, ch) s->fft_data[ch][y].im
228 dst[linesize] =
value;
229 dst[-linesize] =
value;
240 int z =
s->win_size / 2;
247 for (
int i = 0;
i < outlink->
h;
i++) {
248 memset(outpicref->
data[0] +
i * outpicref->
linesize[0], 0, outlink->
w);
249 memset(outpicref->
data[1] +
i * outpicref->
linesize[1], 0, outlink->
w);
250 memset(outpicref->
data[2] +
i * outpicref->
linesize[2], 0, outlink->
w);
253 for (
int j = 0; j < z; j++) {
254 const int idx = z - 1 - j;
255 float l = hypotf(
RE(idx, 0),
IM(idx, 0));
256 float r = hypotf(
RE(idx, 1),
IM(idx, 1));
260 float diffp = ((rp - lp) / (2.f *
M_PI) + 1.f) * 0.5f;
261 float diff = (sum < 0.000001f ? 0.f : (
r - l) / sum) * 0.5f + 0.5f;
310 fin->
pts =
s->pts +
s->consumed;
311 s->consumed +=
s->hop_size;
362 .
name =
"showspatial",
370 .priv_class = &showspatial_class,
static enum AVSampleFormat sample_fmts[]
static const AVFilterPad inputs[]
static const AVFilterPad outputs[]
static int activate(AVFilterContext *ctx)
AVFrame * ff_get_audio_buffer(AVFilterLink *link, int nb_samples)
Request an audio samples buffer with a specific set of permissions.
simple assert() macros that are a bit more flexible than ISO C assert().
#define av_assert0(cond)
assert() equivalent, that is always enabled.
static const AVOption showspatial_options[]
static const AVFilterPad showspatial_inputs[]
static int draw_spatial(AVFilterLink *inlink, AVFrame *insamples)
AVFILTER_DEFINE_CLASS(showspatial)
static int query_formats(AVFilterContext *ctx)
AVFilter ff_avf_showspatial
static void draw_dot(uint8_t *dst, int linesize, int value)
static const AVFilterPad showspatial_outputs[]
static av_cold void uninit(AVFilterContext *ctx)
static int config_output(AVFilterLink *outlink)
static int run_channel_fft(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
static int spatial_activate(AVFilterContext *ctx)
int ff_filter_frame(AVFilterLink *link, AVFrame *frame)
Send a frame of data to the next filter.
void ff_filter_set_ready(AVFilterContext *filter, unsigned priority)
Mark a filter ready and schedule it for activation.
int ff_inlink_consume_frame(AVFilterLink *link, AVFrame **rframe)
Take a frame from the link's FIFO and update the link's stats.
void ff_inlink_request_frame(AVFilterLink *link)
Mark that a frame is wanted on the link.
Main libavfilter public API header.
#define flags(name, subs,...)
audio channel layout utility functions
#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.
static int ff_outlink_frame_wanted(AVFilterLink *link)
Test if a frame is wanted on an output link.
@ AV_OPT_TYPE_IMAGE_SIZE
offset must point to two consecutive integers
#define AV_CH_LAYOUT_STEREO
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)
#define AVFILTER_FLAG_SLICE_THREADS
The filter supports multithreading by splitting frames into multiple parts and processing them concur...
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 av_always_inline AVRational av_inv_q(AVRational q)
Invert a rational.
int64_t av_rescale_q(int64_t a, AVRational bq, AVRational cq)
Rescale a 64-bit integer by 2 rational numbers.
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
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 av_always_inline float cbrtf(float x)
AVPixelFormat
Pixel format.
@ AV_PIX_FMT_GBRP
planar GBR 4:4:4 24bpp
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
AVRational time_base
Define the time base used by the PTS of the frames/samples which will pass through this link.
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.
AVFormatInternal * internal
An opaque field for libavformat internal usage.
This structure describes decoded (raw) audio or video data.
int nb_samples
number of audio samples (per channel) described by this frame
int64_t pts
Presentation timestamp in time_base units (time when frame should be shown to user).
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
AVRational sample_aspect_ratio
Sample aspect ratio for the video frame, 0/1 if unknown/unspecified.
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
uint8_t ** extended_data
pointers to the data planes/channels.
Rational number (pair of numerator and denominator).
FFTContext * ifft[2]
Inverse Fast Fourier Transform context.
float * window_func_lut
Window function LUT.
FFTComplex * fft_data[2]
bins holder for each (displayed) channels
FFTContext * fft[2]
Fast Fourier Transform context.
int fft_bits
number of bits (FFT window size = 1<<fft_bits)
#define av_realloc_f(p, o, n)
static double cr(void *priv, double x, double y)
static double cb(void *priv, double x, double y)
static av_always_inline int diff(const uint32_t a, const uint32_t b)
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)