62 #define OFFSET(x) offsetof(PSNRContext, x)
63 #define FLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM
68 {
"stats_version",
"Set the format version for the stats file.",
OFFSET(stats_version),
AV_OPT_TYPE_INT, {.i64=1}, 1, 2,
FLAGS },
80 static inline double get_psnr(
double mse, uint64_t nb_frames,
int max)
82 return 10.0 * log10(
pow_2(
max) / (mse / nb_frames));
90 for (j = 0; j < outw; j++)
91 m2 +=
pow_2(main_line[j] - ref_line[j]);
100 const uint16_t *main_line = (
const uint16_t *) _main_line;
101 const uint16_t *ref_line = (
const uint16_t *) _ref_line;
103 for (j = 0; j < outw; j++)
104 m2 +=
pow_2(main_line[j] - ref_line[j]);
123 int jobnr,
int nb_jobs)
126 uint64_t *score =
td->score[jobnr];
128 for (
int c = 0;
c <
td->nb_components;
c++) {
129 const int outw =
td->planewidth[
c];
130 const int outh =
td->planeheight[
c];
131 const int slice_start = (outh * jobnr) / nb_jobs;
132 const int slice_end = (outh * (jobnr+1)) / nb_jobs;
133 const int ref_linesize =
td->ref_linesize[
c];
134 const int main_linesize =
td->main_linesize[
c];
135 const uint8_t *main_line =
td->main_data[
c] + main_linesize * slice_start;
136 const uint8_t *ref_line =
td->ref_data[
c] + ref_linesize * slice_start;
139 m +=
td->dsp->sse_line(main_line, ref_line, outw);
140 ref_line += ref_linesize;
141 main_line += main_linesize;
167 double comp_mse[4], mse = 0.;
168 uint64_t comp_sum[4] = { 0 };
176 if (
ctx->is_disabled || !
ref)
178 metadata = &
master->metadata;
180 td.nb_components =
s->nb_components;
183 for (
int c = 0;
c <
s->nb_components;
c++) {
187 td.ref_linesize[
c] =
ref->linesize[
c];
188 td.planewidth[
c] =
s->planewidth[
c];
189 td.planeheight[
c] =
s->planeheight[
c];
194 for (
int j = 0; j <
s->nb_threads; j++) {
195 for (
int c = 0;
c <
s->nb_components;
c++)
196 comp_sum[
c] +=
s->score[j][
c];
199 for (
int c = 0;
c <
s->nb_components;
c++)
200 comp_mse[
c] = comp_sum[
c] / ((
double)
s->planewidth[
c] *
s->planeheight[
c]);
202 for (
int c = 0;
c <
s->nb_components;
c++)
203 mse += comp_mse[
c] *
s->planeweight[
c];
205 s->min_mse =
FFMIN(
s->min_mse, mse);
206 s->max_mse =
FFMAX(
s->max_mse, mse);
210 for (
int j = 0; j <
s->nb_components; j++)
211 s->mse_comp[j] += comp_mse[j];
214 for (
int j = 0; j <
s->nb_components; j++) {
215 int c =
s->is_rgb ?
s->rgba_map[j] : j;
216 set_meta(metadata,
"lavfi.psnr.mse.",
s->comps[j], comp_mse[
c]);
219 set_meta(metadata,
"lavfi.psnr.mse_avg", 0, mse);
223 if (
s->stats_version == 2 && !
s->stats_header_written) {
224 fprintf(
s->stats_file,
"psnr_log_version:2 fields:n");
225 fprintf(
s->stats_file,
",mse_avg");
226 for (
int j = 0; j <
s->nb_components; j++) {
227 fprintf(
s->stats_file,
",mse_%c",
s->comps[j]);
229 fprintf(
s->stats_file,
",psnr_avg");
230 for (
int j = 0; j <
s->nb_components; j++) {
231 fprintf(
s->stats_file,
",psnr_%c",
s->comps[j]);
233 if (
s->stats_add_max) {
234 fprintf(
s->stats_file,
",max_avg");
235 for (
int j = 0; j <
s->nb_components; j++) {
236 fprintf(
s->stats_file,
",max_%c",
s->comps[j]);
239 fprintf(
s->stats_file,
"\n");
240 s->stats_header_written = 1;
242 fprintf(
s->stats_file,
"n:%"PRId64
" mse_avg:%0.2f ",
s->nb_frames, mse);
243 for (
int j = 0; j <
s->nb_components; j++) {
244 int c =
s->is_rgb ?
s->rgba_map[j] : j;
245 fprintf(
s->stats_file,
"mse_%c:%0.2f ",
s->comps[j], comp_mse[
c]);
247 fprintf(
s->stats_file,
"psnr_avg:%0.2f ",
get_psnr(mse, 1,
s->average_max));
248 for (
int j = 0; j <
s->nb_components; j++) {
249 int c =
s->is_rgb ?
s->rgba_map[j] : j;
250 fprintf(
s->stats_file,
"psnr_%c:%0.2f ",
s->comps[j],
253 if (
s->stats_version == 2 &&
s->stats_add_max) {
254 fprintf(
s->stats_file,
"max_avg:%d ",
s->average_max);
255 for (
int j = 0; j <
s->nb_components; j++) {
256 int c =
s->is_rgb ?
s->rgba_map[j] : j;
257 fprintf(
s->stats_file,
"max_%c:%d ",
s->comps[j],
s->max[
c]);
260 fprintf(
s->stats_file,
"\n");
273 if (
s->stats_file_str) {
274 if (
s->stats_version < 2 &&
s->stats_add_max) {
276 "stats_add_max was specified but stats_version < 2.\n" );
279 if (!strcmp(
s->stats_file_str,
"-")) {
280 s->stats_file = stdout;
282 s->stats_file = fopen(
s->stats_file_str,
"w");
283 if (!
s->stats_file) {
288 s->stats_file_str, buf);
302 #define PF_NOALPHA(suf) AV_PIX_FMT_YUV420##suf, AV_PIX_FMT_YUV422##suf, AV_PIX_FMT_YUV444##suf
303 #define PF_ALPHA(suf) AV_PIX_FMT_YUVA420##suf, AV_PIX_FMT_YUVA422##suf, AV_PIX_FMT_YUVA444##suf
304 #define PF(suf) PF_NOALPHA(suf), PF_ALPHA(suf)
331 s->nb_components =
desc->nb_components;
332 if (
ctx->inputs[0]->w !=
ctx->inputs[1]->w ||
333 ctx->inputs[0]->h !=
ctx->inputs[1]->h) {
337 if (
ctx->inputs[0]->format !=
ctx->inputs[1]->format) {
342 s->max[0] = (1 <<
desc->comp[0].depth) - 1;
343 s->max[1] = (1 <<
desc->comp[1].depth) - 1;
344 s->max[2] = (1 <<
desc->comp[2].depth) - 1;
345 s->max[3] = (1 <<
desc->comp[3].depth) - 1;
348 s->comps[0] =
s->is_rgb ?
'r' :
'y' ;
349 s->comps[1] =
s->is_rgb ?
'g' :
'u' ;
350 s->comps[2] =
s->is_rgb ?
'b' :
'v' ;
354 s->planeheight[0] =
s->planeheight[3] = inlink->
h;
356 s->planewidth[0] =
s->planewidth[3] = inlink->
w;
358 for (j = 0; j <
s->nb_components; j++)
359 sum +=
s->planeheight[j] *
s->planewidth[j];
361 for (j = 0; j <
s->nb_components; j++) {
362 s->planeweight[j] = (double)
s->planeheight[j] *
s->planewidth[j] / sum;
363 average_max +=
s->max[j] *
s->planeweight[j];
365 s->average_max =
lrint(average_max);
375 for (
int t = 0; t <
s->nb_threads &&
s->score; t++) {
376 s->score[t] =
av_calloc(
s->nb_components,
sizeof(*
s->score[0]));
394 outlink->
w = mainlink->
w;
395 outlink->
h = mainlink->
h;
406 av_log(
ctx,
AV_LOG_WARNING,
"not matching timebases found between first input: %d/%d and second input %d/%d, results may be incorrect!\n",
408 ctx->inputs[1]->time_base.num,
ctx->inputs[1]->time_base.den);
423 if (
s->nb_frames > 0) {
428 for (j = 0; j <
s->nb_components; j++) {
429 int c =
s->is_rgb ?
s->rgba_map[j] : j;
441 for (
int t = 0; t <
s->nb_threads &&
s->score; t++)
445 if (
s->stats_file &&
s->stats_file != stdout)
446 fclose(
s->stats_file);
473 .preinit = psnr_framesync_preinit,
479 .priv_class = &psnr_class,
static const AVFilterPad inputs[]
static const AVFilterPad outputs[]
int ff_filter_frame(AVFilterLink *link, AVFrame *frame)
Send a frame of data to the next filter.
int ff_filter_get_nb_threads(AVFilterContext *ctx)
Get number of threads for current filter instance.
Main libavfilter public API header.
size_t av_strlcatf(char *dst, size_t size, const char *fmt,...)
#define flags(name, subs,...)
#define fs(width, name, subs,...)
#define AV_CEIL_RSHIFT(a, b)
int ff_fill_rgba_map(uint8_t *rgba_map, enum AVPixelFormat pix_fmt)
static void comp(unsigned char *dst, ptrdiff_t dst_stride, unsigned char *src, ptrdiff_t src_stride, int add)
static double psnr(double d)
int ff_framesync_configure(FFFrameSync *fs)
Configure a frame sync structure.
int ff_framesync_dualinput_get(FFFrameSync *fs, AVFrame **f0, AVFrame **f1)
int ff_framesync_activate(FFFrameSync *fs)
Examine the frames in the filter's input and try to produce output.
int ff_framesync_init_dualinput(FFFrameSync *fs, AVFilterContext *parent)
Initialize a frame sync structure for dualinput.
void ff_framesync_uninit(FFFrameSync *fs)
Free all memory currently allocated.
#define AVFILTER_FLAG_SLICE_THREADS
The filter supports multithreading by splitting frames into multiple parts and processing them concur...
#define AVFILTER_FLAG_SUPPORT_TIMELINE_INTERNAL
Same as AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC, except that the filter will have its filter_frame() c...
int av_dict_set(AVDictionary **pm, const char *key, const char *value, int flags)
Set the given entry in *pm, overwriting an existing entry.
int av_strerror(int errnum, char *errbuf, size_t errbuf_size)
Put a description of the AVERROR code errnum in errbuf.
#define AV_LOG_WARNING
Something somehow does not look correct.
#define AV_LOG_INFO
Standard information.
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
static int av_cmp_q(AVRational a, AVRational b)
Compare two rationals.
void * av_calloc(size_t nmemb, size_t size)
Non-inlined equivalent of av_mallocz_array().
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 int slice_end(AVCodecContext *avctx, AVFrame *pict)
Handle slice ends.
const AVPixFmtDescriptor * av_pix_fmt_desc_get(enum AVPixelFormat pix_fmt)
#define AV_PIX_FMT_GBRAP12
#define AV_PIX_FMT_GBRAP16
#define AV_PIX_FMT_GBRP10
#define AV_PIX_FMT_GRAY12
#define AV_PIX_FMT_GBRP12
AVPixelFormat
Pixel format.
@ AV_PIX_FMT_YUV440P
planar YUV 4:4:0 (1 Cr & Cb sample per 1x2 Y samples)
@ AV_PIX_FMT_GRAY8
Y , 8bpp.
@ AV_PIX_FMT_YUVJ440P
planar YUV 4:4:0 full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV440P and setting color_range
@ AV_PIX_FMT_YUV410P
planar YUV 4:1:0, 9bpp, (1 Cr & Cb sample per 4x4 Y samples)
@ AV_PIX_FMT_YUV411P
planar YUV 4:1:1, 12bpp, (1 Cr & Cb sample per 4x1 Y samples)
@ AV_PIX_FMT_YUVJ411P
planar YUV 4:1:1, 12bpp, (1 Cr & Cb sample per 4x1 Y samples) full scale (JPEG), deprecated in favor ...
@ AV_PIX_FMT_GBRAP
planar GBRA 4:4:4:4 32bpp
@ AV_PIX_FMT_YUVJ422P
planar YUV 4:2:2, 16bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV422P and setting col...
@ AV_PIX_FMT_GBRP
planar GBR 4:4:4 24bpp
@ AV_PIX_FMT_YUVJ444P
planar YUV 4:4:4, 24bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV444P and setting col...
@ AV_PIX_FMT_YUVJ420P
planar YUV 4:2:0, 12bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV420P and setting col...
#define AV_PIX_FMT_GRAY10
#define AV_PIX_FMT_GRAY14
#define AV_PIX_FMT_GRAY16
#define AV_PIX_FMT_GBRAP10
#define AV_PIX_FMT_GBRP16
#define AV_PIX_FMT_GBRP14
void ff_psnr_init_x86(PSNRDSPContext *dsp, int bpp)
Describe the class of an AVClass context structure.
A link between two filters.
int w
agreed upon image width
int h
agreed upon image height
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.
Descriptor that unambiguously describes how the bits of a pixel are stored in the up to 4 data planes...
Used for passing data between threads.
const uint8_t * ref_data[4]
const uint8_t * main_data[4]
static int ref[MAX_W *MAX_W]
static const AVOption psnr_options[]
static unsigned pow_2(unsigned base)
static int compute_images_mse(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
FRAMESYNC_DEFINE_CLASS(psnr, PSNRContext, fs)
static int query_formats(AVFilterContext *ctx)
static uint64_t sse_line_8bit(const uint8_t *main_line, const uint8_t *ref_line, int outw)
static const AVFilterPad psnr_outputs[]
static int config_input_ref(AVFilterLink *inlink)
static void set_meta(AVDictionary **metadata, const char *key, char comp, float d)
static double get_psnr(double mse, uint64_t nb_frames, int max)
static const AVFilterPad psnr_inputs[]
static int activate(AVFilterContext *ctx)
static av_cold int init(AVFilterContext *ctx)
static av_cold void uninit(AVFilterContext *ctx)
static int config_output(AVFilterLink *outlink)
static uint64_t sse_line_16bit(const uint8_t *_main_line, const uint8_t *_ref_line, int outw)
static int do_psnr(FFFrameSync *fs)