FFmpeg
Data Structures | Macros | Functions | Variables
vf_psnr.c File Reference
#include "libavutil/avstring.h"
#include "libavutil/file_open.h"
#include "libavutil/opt.h"
#include "libavutil/pixdesc.h"
#include "avfilter.h"
#include "drawutils.h"
#include "framesync.h"
#include "internal.h"
#include "psnr.h"

Go to the source code of this file.

Data Structures

struct  PSNRContext
 
struct  ThreadData
 Used for passing data between threads. More...
 

Macros

#define OFFSET(x)   offsetof(PSNRContext, x)
 
#define FLAGS   AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM
 
#define PF_NOALPHA(suf)   AV_PIX_FMT_YUV420##suf, AV_PIX_FMT_YUV422##suf, AV_PIX_FMT_YUV444##suf
 
#define PF_ALPHA(suf)   AV_PIX_FMT_YUVA420##suf, AV_PIX_FMT_YUVA422##suf, AV_PIX_FMT_YUVA444##suf
 
#define PF(suf)   PF_NOALPHA(suf), PF_ALPHA(suf)
 

Functions

 FRAMESYNC_DEFINE_CLASS (psnr, PSNRContext, fs)
 
static unsigned pow_2 (unsigned base)
 
static double get_psnr (double mse, uint64_t nb_frames, int max)
 
static uint64_t sse_line_8bit (const uint8_t *main_line, const uint8_t *ref_line, int outw)
 
static uint64_t sse_line_16bit (const uint8_t *_main_line, const uint8_t *_ref_line, int outw)
 
static int compute_images_mse (AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
 
static void set_meta (AVDictionary **metadata, const char *key, char comp, float d)
 
static int do_psnr (FFFrameSync *fs)
 
static av_cold int init (AVFilterContext *ctx)
 
static int config_input_ref (AVFilterLink *inlink)
 
static int config_output (AVFilterLink *outlink)
 
static int activate (AVFilterContext *ctx)
 
static av_cold void uninit (AVFilterContext *ctx)
 

Variables

static const AVOption psnr_options []
 
static enum AVPixelFormat pix_fmts []
 
static const AVFilterPad psnr_inputs []
 
static const AVFilterPad psnr_outputs []
 
const AVFilter ff_vf_psnr
 

Detailed Description

Calculate the PSNR between two input videos.

Definition in file vf_psnr.c.

Macro Definition Documentation

◆ OFFSET

#define OFFSET (   x)    offsetof(PSNRContext, x)

Definition at line 61 of file vf_psnr.c.

◆ FLAGS

#define FLAGS   AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM

Definition at line 62 of file vf_psnr.c.

◆ PF_NOALPHA

#define PF_NOALPHA (   suf)    AV_PIX_FMT_YUV420##suf, AV_PIX_FMT_YUV422##suf, AV_PIX_FMT_YUV444##suf

◆ PF_ALPHA

#define PF_ALPHA (   suf)    AV_PIX_FMT_YUVA420##suf, AV_PIX_FMT_YUVA422##suf, AV_PIX_FMT_YUVA444##suf

◆ PF

#define PF (   suf)    PF_NOALPHA(suf), PF_ALPHA(suf)

Function Documentation

◆ FRAMESYNC_DEFINE_CLASS()

FRAMESYNC_DEFINE_CLASS ( psnr  ,
PSNRContext  ,
fs   
)

◆ pow_2()

static unsigned pow_2 ( unsigned  base)
inlinestatic

Definition at line 74 of file vf_psnr.c.

Referenced by get_psnr(), sse_line_16bit(), and sse_line_8bit().

◆ get_psnr()

static double get_psnr ( double  mse,
uint64_t  nb_frames,
int  max 
)
inlinestatic

Definition at line 79 of file vf_psnr.c.

Referenced by do_psnr(), and uninit().

◆ sse_line_8bit()

static uint64_t sse_line_8bit ( const uint8_t *  main_line,
const uint8_t *  ref_line,
int  outw 
)
static

Definition at line 84 of file vf_psnr.c.

Referenced by config_input_ref().

◆ sse_line_16bit()

static uint64_t sse_line_16bit ( const uint8_t *  _main_line,
const uint8_t *  _ref_line,
int  outw 
)
static

Definition at line 95 of file vf_psnr.c.

Referenced by config_input_ref().

◆ compute_images_mse()

static int compute_images_mse ( AVFilterContext *  ctx,
void *  arg,
int  jobnr,
int  nb_jobs 
)
static

Definition at line 121 of file vf_psnr.c.

Referenced by do_psnr().

◆ set_meta()

static void set_meta ( AVDictionary **  metadata,
const char *  key,
char  comp,
float  d 
)
static

Definition at line 148 of file vf_psnr.c.

Referenced by do_psnr().

◆ do_psnr()

static int do_psnr ( FFFrameSync *  fs)
static

Definition at line 161 of file vf_psnr.c.

Referenced by init().

◆ init()

static av_cold int init ( AVFilterContext *  ctx)
static

Definition at line 266 of file vf_psnr.c.

◆ config_input_ref()

static int config_input_ref ( AVFilterLink *  inlink)
static

Definition at line 313 of file vf_psnr.c.

◆ config_output()

static int config_output ( AVFilterLink *  outlink)
static

Definition at line 373 of file vf_psnr.c.

◆ activate()

static int activate ( AVFilterContext *  ctx)
static

Definition at line 402 of file vf_psnr.c.

◆ uninit()

static av_cold void uninit ( AVFilterContext *  ctx)
static

Definition at line 408 of file vf_psnr.c.

Variable Documentation

◆ psnr_options

const AVOption psnr_options[]
static
Initial value:
= {
{"stats_file", "Set file where to store per-frame difference information", OFFSET(stats_file_str), AV_OPT_TYPE_STRING, {.str=NULL}, 0, 0, FLAGS },
{"f", "Set file where to store per-frame difference information", OFFSET(stats_file_str), AV_OPT_TYPE_STRING, {.str=NULL}, 0, 0, FLAGS },
{"stats_version", "Set the format version for the stats file.", OFFSET(stats_version), AV_OPT_TYPE_INT, {.i64=1}, 1, 2, FLAGS },
{"output_max", "Add raw stats (max values) to the output log.", OFFSET(stats_add_max), AV_OPT_TYPE_BOOL, {.i64=0}, 0, 1, FLAGS},
{ NULL }
}

Definition at line 64 of file vf_psnr.c.

◆ pix_fmts

enum AVPixelFormat pix_fmts[]
static
Initial value:
= {
AV_PIX_FMT_GRAY8, AV_PIX_FMT_GRAY9, AV_PIX_FMT_GRAY10, AV_PIX_FMT_GRAY12, AV_PIX_FMT_GRAY14, AV_PIX_FMT_GRAY16,
#define PF_NOALPHA(suf)
#define PF_ALPHA(suf)
#define PF(suf)
PF(P), PF(P9), PF(P10), PF_NOALPHA(P12), PF_NOALPHA(P14), PF(P16),
AV_PIX_FMT_YUV440P, AV_PIX_FMT_YUV411P, AV_PIX_FMT_YUV410P,
AV_PIX_FMT_YUVJ411P, AV_PIX_FMT_YUVJ420P, AV_PIX_FMT_YUVJ422P,
AV_PIX_FMT_YUVJ440P, AV_PIX_FMT_YUVJ444P,
AV_PIX_FMT_GBRP, AV_PIX_FMT_GBRP9, AV_PIX_FMT_GBRP10,
AV_PIX_FMT_GBRP12, AV_PIX_FMT_GBRP14, AV_PIX_FMT_GBRP16,
AV_PIX_FMT_GBRAP, AV_PIX_FMT_GBRAP10, AV_PIX_FMT_GBRAP12, AV_PIX_FMT_GBRAP16,
AV_PIX_FMT_NONE
}

Definition at line 298 of file vf_psnr.c.

◆ psnr_inputs

const AVFilterPad psnr_inputs[]
static
Initial value:
= {
{
.name = "main",
.type = AVMEDIA_TYPE_VIDEO,
},{
.name = "reference",
.type = AVMEDIA_TYPE_VIDEO,
.config_props = config_input_ref,
},
}

Definition at line 438 of file vf_psnr.c.

◆ psnr_outputs

const AVFilterPad psnr_outputs[]
static
Initial value:
= {
{
.name = "default",
.type = AVMEDIA_TYPE_VIDEO,
.config_props = config_output,
},
}

Definition at line 449 of file vf_psnr.c.

◆ ff_vf_psnr

const AVFilter ff_vf_psnr
Initial value:
= {
.name = "psnr",
.description = NULL_IF_CONFIG_SMALL("Calculate the PSNR between two video streams."),
.preinit = psnr_framesync_preinit,
.init = init,
.uninit = uninit,
.activate = activate,
.priv_size = sizeof(PSNRContext),
.priv_class = &psnr_class,
FILTER_INPUTS(psnr_inputs),
FILTER_OUTPUTS(psnr_outputs),
FILTER_PIXFMTS_ARRAY(pix_fmts),
.flags = AVFILTER_FLAG_SUPPORT_TIMELINE_INTERNAL |
AVFILTER_FLAG_SLICE_THREADS |
AVFILTER_FLAG_METADATA_ONLY,
}

Definition at line 457 of file vf_psnr.c.

AV_PIX_FMT_GBRAP16
#define AV_PIX_FMT_GBRAP16
Definition: pixfmt.h:481
config_input_ref
static int config_input_ref(AVFilterLink *inlink)
Definition: vf_psnr.c:313
PSNRContext
Definition: vf_psnr.c:38
FILTER_PIXFMTS_ARRAY
#define FILTER_PIXFMTS_ARRAY(array)
Definition: internal.h:174
AV_PIX_FMT_YUV440P
@ AV_PIX_FMT_YUV440P
planar YUV 4:4:0 (1 Cr & Cb sample per 1x2 Y samples)
Definition: pixfmt.h:99
init
static av_cold int init(AVFilterContext *ctx)
Definition: vf_psnr.c:266
AV_PIX_FMT_GRAY9
#define AV_PIX_FMT_GRAY9
Definition: pixfmt.h:439
psnr_inputs
static const AVFilterPad psnr_inputs[]
Definition: vf_psnr.c:438
AV_PIX_FMT_GBRP14
#define AV_PIX_FMT_GBRP14
Definition: pixfmt.h:477
AV_PIX_FMT_GBRAP
@ AV_PIX_FMT_GBRAP
planar GBRA 4:4:4:4 32bpp
Definition: pixfmt.h:205
AV_PIX_FMT_GBRP10
#define AV_PIX_FMT_GBRP10
Definition: pixfmt.h:475
AV_PIX_FMT_GRAY16
#define AV_PIX_FMT_GRAY16
Definition: pixfmt.h:443
AV_PIX_FMT_YUVJ411P
@ AV_PIX_FMT_YUVJ411P
planar YUV 4:1:1, 12bpp, (1 Cr & Cb sample per 4x1 Y samples) full scale (JPEG), deprecated in favor ...
Definition: pixfmt.h:276
AV_PIX_FMT_YUVJ422P
@ AV_PIX_FMT_YUVJ422P
planar YUV 4:2:2, 16bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV422P and setting col...
Definition: pixfmt.h:79
AV_PIX_FMT_GBRAP10
#define AV_PIX_FMT_GBRAP10
Definition: pixfmt.h:479
AV_PIX_FMT_GBRAP12
#define AV_PIX_FMT_GBRAP12
Definition: pixfmt.h:480
AV_PIX_FMT_GRAY14
#define AV_PIX_FMT_GRAY14
Definition: pixfmt.h:442
FILTER_INPUTS
#define FILTER_INPUTS(array)
Definition: internal.h:194
AV_PIX_FMT_YUVJ444P
@ AV_PIX_FMT_YUVJ444P
planar YUV 4:4:4, 24bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV444P and setting col...
Definition: pixfmt.h:80
AV_PIX_FMT_GRAY10
#define AV_PIX_FMT_GRAY10
Definition: pixfmt.h:440
AV_PIX_FMT_GBRP16
#define AV_PIX_FMT_GBRP16
Definition: pixfmt.h:478
NULL
#define NULL
Definition: coverity.c:32
AV_PIX_FMT_YUVJ420P
@ AV_PIX_FMT_YUVJ420P
planar YUV 4:2:0, 12bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV420P and setting col...
Definition: pixfmt.h:78
AV_PIX_FMT_GRAY8
@ AV_PIX_FMT_GRAY8
Y , 8bpp.
Definition: pixfmt.h:74
AV_PIX_FMT_GBRP9
#define AV_PIX_FMT_GBRP9
Definition: pixfmt.h:474
PF_NOALPHA
#define PF_NOALPHA(suf)
psnr_outputs
static const AVFilterPad psnr_outputs[]
Definition: vf_psnr.c:449
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:115
P
#define P
uninit
static av_cold void uninit(AVFilterContext *ctx)
Definition: vf_psnr.c:408
pix_fmts
static enum AVPixelFormat pix_fmts[]
Definition: vf_psnr.c:298
PF
#define PF(suf)
AV_PIX_FMT_GBRP12
#define AV_PIX_FMT_GBRP12
Definition: pixfmt.h:476
config_output
static int config_output(AVFilterLink *outlink)
Definition: vf_psnr.c:373
AV_PIX_FMT_YUVJ440P
@ AV_PIX_FMT_YUVJ440P
planar YUV 4:4:0 full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV440P and setting color_range
Definition: pixfmt.h:100
AV_PIX_FMT_NONE
@ AV_PIX_FMT_NONE
Definition: pixfmt.h:65
AV_OPT_TYPE_INT
@ AV_OPT_TYPE_INT
Definition: opt.h:225
AVFILTER_FLAG_METADATA_ONLY
#define AVFILTER_FLAG_METADATA_ONLY
The filter is a "metadata" filter - it does not modify the frame data in any way.
Definition: avfilter.h:133
activate
static int activate(AVFilterContext *ctx)
Definition: vf_psnr.c:402
AV_PIX_FMT_GBRP
@ AV_PIX_FMT_GBRP
planar GBR 4:4:4 24bpp
Definition: pixfmt.h:158
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:117
AVMEDIA_TYPE_VIDEO
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201
AV_OPT_TYPE_BOOL
@ AV_OPT_TYPE_BOOL
Definition: opt.h:244
FILTER_OUTPUTS
#define FILTER_OUTPUTS(array)
Definition: internal.h:195
FLAGS
#define FLAGS
Definition: vf_psnr.c:62
AV_PIX_FMT_YUV411P
@ AV_PIX_FMT_YUV411P
planar YUV 4:1:1, 12bpp, (1 Cr & Cb sample per 4x1 Y samples)
Definition: pixfmt.h:73
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:150
AV_PIX_FMT_YUV410P
@ AV_PIX_FMT_YUV410P
planar YUV 4:1:0, 9bpp, (1 Cr & Cb sample per 4x4 Y samples)
Definition: pixfmt.h:72
AV_OPT_TYPE_STRING
@ AV_OPT_TYPE_STRING
Definition: opt.h:229
AV_PIX_FMT_GRAY12
#define AV_PIX_FMT_GRAY12
Definition: pixfmt.h:441
OFFSET
#define OFFSET(x)
Definition: vf_psnr.c:61
Generated on Tue Feb 28 2023 21:34:33 for FFmpeg by   doxygen 1.8.17