128 lines
4.2 KiB
C
128 lines
4.2 KiB
C
|
/*
|
||
|
|
* Copyright (c) 2015 Muhammad Faiz <mfcc64@gmail.com>
|
||
|
|
*
|
||
|
|
* This file is part of FFmpeg.
|
||
|
|
*
|
||
|
|
* FFmpeg is free software; you can redistribute it and/or
|
||
|
|
* modify it under the terms of the GNU Lesser General Public
|
||
|
|
* License as published by the Free Software Foundation; either
|
||
|
|
* version 2.1 of the License, or (at your option) any later version.
|
||
|
|
*
|
||
|
|
* FFmpeg is distributed in the hope that it will be useful,
|
||
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||
|
|
* Lesser General Public License for more details.
|
||
|
|
*
|
||
|
|
* You should have received a copy of the GNU Lesser General Public
|
||
|
|
* License along with FFmpeg; if not, write to the Free Software
|
||
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
||
|
|
*/
|
||
|
|
|
||
|
|
#ifndef AVFILTER_SHOWCQT_H
|
||
|
|
#define AVFILTER_SHOWCQT_H
|
||
|
|
|
||
|
|
#include "libavcodec/avfft.h"
|
||
|
|
#include "avfilter.h"
|
||
|
|
#include "internal.h"
|
||
|
|
|
||
|
|
typedef struct Coeffs {
|
||
|
|
FFTSample *val;
|
||
|
|
int start, len;
|
||
|
|
} Coeffs;
|
||
|
|
|
||
|
|
typedef struct RGBFloat {
|
||
|
|
float r, g, b;
|
||
|
|
} RGBFloat;
|
||
|
|
|
||
|
|
typedef struct YUVFloat {
|
||
|
|
float y, u, v;
|
||
|
|
} YUVFloat;
|
||
|
|
|
||
|
|
typedef union {
|
||
|
|
RGBFloat rgb;
|
||
|
|
YUVFloat yuv;
|
||
|
|
} ColorFloat;
|
||
|
|
|
||
|
|
typedef struct ShowCQTContext {
|
||
|
|
const AVClass *class;
|
||
|
|
AVFilterContext *ctx;
|
||
|
|
AVFrame *axis_frame;
|
||
|
|
AVFrame *sono_frame;
|
||
|
|
enum AVPixelFormat format;
|
||
|
|
int sono_idx;
|
||
|
|
int sono_count;
|
||
|
|
int step;
|
||
|
|
AVRational step_frac;
|
||
|
|
int remaining_frac;
|
||
|
|
int remaining_fill;
|
||
|
|
int remaining_fill_max;
|
||
|
|
int64_t next_pts;
|
||
|
|
double *freq;
|
||
|
|
FFTContext *fft_ctx;
|
||
|
|
Coeffs *coeffs;
|
||
|
|
FFTComplex *fft_data;
|
||
|
|
FFTComplex *fft_result;
|
||
|
|
FFTComplex *cqt_result;
|
||
|
|
float *attack_data;
|
||
|
|
int fft_bits;
|
||
|
|
int fft_len;
|
||
|
|
int cqt_len;
|
||
|
|
int cqt_align;
|
||
|
|
ColorFloat *c_buf;
|
||
|
|
float *h_buf;
|
||
|
|
float *rcp_h_buf;
|
||
|
|
float *sono_v_buf;
|
||
|
|
float *bar_v_buf;
|
||
|
|
float cmatrix[3][3];
|
||
|
|
float cscheme_v[6];
|
||
|
|
/* callback */
|
||
|
|
void (*cqt_calc)(FFTComplex *dst, const FFTComplex *src, const Coeffs *coeffs,
|
||
|
|
int len, int fft_len);
|
||
|
|
void (*permute_coeffs)(float *v, int len);
|
||
|
|
void (*draw_bar)(AVFrame *out, const float *h, const float *rcp_h,
|
||
|
|
const ColorFloat *c, int bar_h, float bar_t);
|
||
|
|
void (*draw_axis)(AVFrame *out, AVFrame *axis, const ColorFloat *c, int off);
|
||
|
|
void (*draw_sono)(AVFrame *out, AVFrame *sono, int off, int idx);
|
||
|
|
void (*update_sono)(AVFrame *sono, const ColorFloat *c, int idx);
|
||
|
|
/* performance debugging */
|
||
|
|
int64_t fft_time;
|
||
|
|
int64_t cqt_time;
|
||
|
|
int64_t process_cqt_time;
|
||
|
|
int64_t update_sono_time;
|
||
|
|
int64_t alloc_time;
|
||
|
|
int64_t bar_time;
|
||
|
|
int64_t axis_time;
|
||
|
|
int64_t sono_time;
|
||
|
|
/* option */
|
||
|
|
int width, height;
|
||
|
|
AVRational rate;
|
||
|
|
int bar_h;
|
||
|
|
int axis_h;
|
||
|
|
int sono_h;
|
||
|
|
int fullhd; /* deprecated */
|
||
|
|
char *sono_v;
|
||
|
|
char *bar_v;
|
||
|
|
float sono_g;
|
||
|
|
float bar_g;
|
||
|
|
float bar_t;
|
||
|
|
double timeclamp;
|
||
|
|
double attack;
|
||
|
|
double basefreq;
|
||
|
|
double endfreq;
|
||
|
|
float coeffclamp; /* deprecated - ignored */
|
||
|
|
char *tlength;
|
||
|
|
int count;
|
||
|
|
int fcount;
|
||
|
|
char *fontfile;
|
||
|
|
char *font;
|
||
|
|
char *fontcolor;
|
||
|
|
char *axisfile;
|
||
|
|
int axis;
|
||
|
|
int csp;
|
||
|
|
char *cscheme;
|
||
|
|
} ShowCQTContext;
|
||
|
|
|
||
|
|
void ff_showcqt_init_x86(ShowCQTContext *s);
|
||
|
|
|
||
|
|
#endif
|