FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
vf_ssim.c
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2003-2013 Loren Merritt
3  * Copyright (c) 2015 Paul B Mahol
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21 
22 /* Computes the Structural Similarity Metric between two video streams.
23  * original algorithm:
24  * Z. Wang, A. C. Bovik, H. R. Sheikh and E. P. Simoncelli,
25  * "Image quality assessment: From error visibility to structural similarity,"
26  * IEEE Transactions on Image Processing, vol. 13, no. 4, pp. 600-612, Apr. 2004.
27  *
28  * To improve speed, this implementation uses the standard approximation of
29  * overlapped 8x8 block sums, rather than the original gaussian weights.
30  */
31 
32 /*
33  * @file
34  * Caculate the SSIM between two input videos.
35  */
36 
37 #include "libavutil/avstring.h"
38 #include "libavutil/opt.h"
39 #include "libavutil/pixdesc.h"
40 #include "avfilter.h"
41 #include "dualinput.h"
42 #include "drawutils.h"
43 #include "formats.h"
44 #include "internal.h"
45 #include "ssim.h"
46 #include "video.h"
47 
48 typedef struct SSIMContext {
49  const AVClass *class;
51  FILE *stats_file;
54  int max;
55  uint64_t nb_frames;
56  double ssim[4], ssim_total;
57  char comps[4];
58  float coefs[4];
60  int planewidth[4];
61  int planeheight[4];
62  int *temp;
63  int is_rgb;
65  uint8_t *main, int main_stride,
66  uint8_t *ref, int ref_stride,
67  int width, int height, void *temp,
68  int max);
70 } SSIMContext;
71 
72 #define OFFSET(x) offsetof(SSIMContext, x)
73 #define FLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM
74 
75 static const AVOption ssim_options[] = {
76  {"stats_file", "Set file where to store per-frame difference information", OFFSET(stats_file_str), AV_OPT_TYPE_STRING, {.str=NULL}, 0, 0, FLAGS },
77  {"f", "Set file where to store per-frame difference information", OFFSET(stats_file_str), AV_OPT_TYPE_STRING, {.str=NULL}, 0, 0, FLAGS },
78  { NULL }
79 };
80 
82 
83 static void set_meta(AVDictionary **metadata, const char *key, char comp, float d)
84 {
85  char value[128];
86  snprintf(value, sizeof(value), "%0.2f", d);
87  if (comp) {
88  char key2[128];
89  snprintf(key2, sizeof(key2), "%s%c", key, comp);
90  av_dict_set(metadata, key2, value, 0);
91  } else {
92  av_dict_set(metadata, key, value, 0);
93  }
94 }
95 
96 static void ssim_4x4xn_16bit(const uint8_t *main8, ptrdiff_t main_stride,
97  const uint8_t *ref8, ptrdiff_t ref_stride,
98  int64_t (*sums)[4], int width)
99 {
100  const uint16_t *main16 = (const uint16_t *)main8;
101  const uint16_t *ref16 = (const uint16_t *)ref8;
102  int x, y, z;
103 
104  main_stride >>= 1;
105  ref_stride >>= 1;
106 
107  for (z = 0; z < width; z++) {
108  uint64_t s1 = 0, s2 = 0, ss = 0, s12 = 0;
109 
110  for (y = 0; y < 4; y++) {
111  for (x = 0; x < 4; x++) {
112  int a = main16[x + y * main_stride];
113  int b = ref16[x + y * ref_stride];
114 
115  s1 += a;
116  s2 += b;
117  ss += a*a;
118  ss += b*b;
119  s12 += a*b;
120  }
121  }
122 
123  sums[z][0] = s1;
124  sums[z][1] = s2;
125  sums[z][2] = ss;
126  sums[z][3] = s12;
127  main16 += 4;
128  ref16 += 4;
129  }
130 }
131 
132 static void ssim_4x4xn_8bit(const uint8_t *main, ptrdiff_t main_stride,
133  const uint8_t *ref, ptrdiff_t ref_stride,
134  int (*sums)[4], int width)
135 {
136  int x, y, z;
137 
138  for (z = 0; z < width; z++) {
139  uint32_t s1 = 0, s2 = 0, ss = 0, s12 = 0;
140 
141  for (y = 0; y < 4; y++) {
142  for (x = 0; x < 4; x++) {
143  int a = main[x + y * main_stride];
144  int b = ref[x + y * ref_stride];
145 
146  s1 += a;
147  s2 += b;
148  ss += a*a;
149  ss += b*b;
150  s12 += a*b;
151  }
152  }
153 
154  sums[z][0] = s1;
155  sums[z][1] = s2;
156  sums[z][2] = ss;
157  sums[z][3] = s12;
158  main += 4;
159  ref += 4;
160  }
161 }
162 
163 static float ssim_end1x(int64_t s1, int64_t s2, int64_t ss, int64_t s12, int max)
164 {
165  int64_t ssim_c1 = (int64_t)(.01*.01*max*max*64 + .5);
166  int64_t ssim_c2 = (int64_t)(.03*.03*max*max*64*63 + .5);
167 
168  int64_t fs1 = s1;
169  int64_t fs2 = s2;
170  int64_t fss = ss;
171  int64_t fs12 = s12;
172  int64_t vars = fss * 64 - fs1 * fs1 - fs2 * fs2;
173  int64_t covar = fs12 * 64 - fs1 * fs2;
174 
175  return (float)(2 * fs1 * fs2 + ssim_c1) * (float)(2 * covar + ssim_c2)
176  / ((float)(fs1 * fs1 + fs2 * fs2 + ssim_c1) * (float)(vars + ssim_c2));
177 }
178 
179 static float ssim_end1(int s1, int s2, int ss, int s12)
180 {
181  static const int ssim_c1 = (int)(.01*.01*255*255*64 + .5);
182  static const int ssim_c2 = (int)(.03*.03*255*255*64*63 + .5);
183 
184  int fs1 = s1;
185  int fs2 = s2;
186  int fss = ss;
187  int fs12 = s12;
188  int vars = fss * 64 - fs1 * fs1 - fs2 * fs2;
189  int covar = fs12 * 64 - fs1 * fs2;
190 
191  return (float)(2 * fs1 * fs2 + ssim_c1) * (float)(2 * covar + ssim_c2)
192  / ((float)(fs1 * fs1 + fs2 * fs2 + ssim_c1) * (float)(vars + ssim_c2));
193 }
194 
195 static float ssim_endn_16bit(const int64_t (*sum0)[4], const int64_t (*sum1)[4], int width, int max)
196 {
197  float ssim = 0.0;
198  int i;
199 
200  for (i = 0; i < width; i++)
201  ssim += ssim_end1x(sum0[i][0] + sum0[i + 1][0] + sum1[i][0] + sum1[i + 1][0],
202  sum0[i][1] + sum0[i + 1][1] + sum1[i][1] + sum1[i + 1][1],
203  sum0[i][2] + sum0[i + 1][2] + sum1[i][2] + sum1[i + 1][2],
204  sum0[i][3] + sum0[i + 1][3] + sum1[i][3] + sum1[i + 1][3],
205  max);
206  return ssim;
207 }
208 
209 static float ssim_endn_8bit(const int (*sum0)[4], const int (*sum1)[4], int width)
210 {
211  float ssim = 0.0;
212  int i;
213 
214  for (i = 0; i < width; i++)
215  ssim += ssim_end1(sum0[i][0] + sum0[i + 1][0] + sum1[i][0] + sum1[i + 1][0],
216  sum0[i][1] + sum0[i + 1][1] + sum1[i][1] + sum1[i + 1][1],
217  sum0[i][2] + sum0[i + 1][2] + sum1[i][2] + sum1[i + 1][2],
218  sum0[i][3] + sum0[i + 1][3] + sum1[i][3] + sum1[i + 1][3]);
219  return ssim;
220 }
221 
222 #define SUM_LEN(w) (((w) >> 2) + 3)
223 
225  uint8_t *main, int main_stride,
226  uint8_t *ref, int ref_stride,
227  int width, int height, void *temp,
228  int max)
229 {
230  int z = 0, y;
231  float ssim = 0.0;
232  int64_t (*sum0)[4] = temp;
233  int64_t (*sum1)[4] = sum0 + SUM_LEN(width);
234 
235  width >>= 2;
236  height >>= 2;
237 
238  for (y = 1; y < height; y++) {
239  for (; z <= y; z++) {
240  FFSWAP(void*, sum0, sum1);
241  ssim_4x4xn_16bit(&main[4 * z * main_stride], main_stride,
242  &ref[4 * z * ref_stride], ref_stride,
243  sum0, width);
244  }
245 
246  ssim += ssim_endn_16bit((const int64_t (*)[4])sum0, (const int64_t (*)[4])sum1, width - 1, max);
247  }
248 
249  return ssim / ((height - 1) * (width - 1));
250 }
251 
252 static float ssim_plane(SSIMDSPContext *dsp,
253  uint8_t *main, int main_stride,
254  uint8_t *ref, int ref_stride,
255  int width, int height, void *temp,
256  int max)
257 {
258  int z = 0, y;
259  float ssim = 0.0;
260  int (*sum0)[4] = temp;
261  int (*sum1)[4] = sum0 + SUM_LEN(width);
262 
263  width >>= 2;
264  height >>= 2;
265 
266  for (y = 1; y < height; y++) {
267  for (; z <= y; z++) {
268  FFSWAP(void*, sum0, sum1);
269  dsp->ssim_4x4_line(&main[4 * z * main_stride], main_stride,
270  &ref[4 * z * ref_stride], ref_stride,
271  sum0, width);
272  }
273 
274  ssim += dsp->ssim_end_line((const int (*)[4])sum0, (const int (*)[4])sum1, width - 1);
275  }
276 
277  return ssim / ((height - 1) * (width - 1));
278 }
279 
280 static double ssim_db(double ssim, double weight)
281 {
282  return 10 * log10(weight / (weight - ssim));
283 }
284 
286  const AVFrame *ref)
287 {
288  AVDictionary **metadata = avpriv_frame_get_metadatap(main);
289  SSIMContext *s = ctx->priv;
290  float c[4], ssimv = 0.0;
291  int i;
292 
293  s->nb_frames++;
294 
295  for (i = 0; i < s->nb_components; i++) {
296  c[i] = s->ssim_plane(&s->dsp, main->data[i], main->linesize[i],
297  ref->data[i], ref->linesize[i],
298  s->planewidth[i], s->planeheight[i], s->temp,
299  s->max);
300  ssimv += s->coefs[i] * c[i];
301  s->ssim[i] += c[i];
302  }
303  for (i = 0; i < s->nb_components; i++) {
304  int cidx = s->is_rgb ? s->rgba_map[i] : i;
305  set_meta(metadata, "lavfi.ssim.", s->comps[i], c[cidx]);
306  }
307  s->ssim_total += ssimv;
308 
309  set_meta(metadata, "lavfi.ssim.All", 0, ssimv);
310  set_meta(metadata, "lavfi.ssim.dB", 0, ssim_db(ssimv, 1.0));
311 
312  if (s->stats_file) {
313  fprintf(s->stats_file, "n:%"PRId64" ", s->nb_frames);
314 
315  for (i = 0; i < s->nb_components; i++) {
316  int cidx = s->is_rgb ? s->rgba_map[i] : i;
317  fprintf(s->stats_file, "%c:%f ", s->comps[i], c[cidx]);
318  }
319 
320  fprintf(s->stats_file, "All:%f (%f)\n", ssimv, ssim_db(ssimv, 1.0));
321  }
322 
323  return main;
324 }
325 
327 {
328  SSIMContext *s = ctx->priv;
329 
330  if (s->stats_file_str) {
331  if (!strcmp(s->stats_file_str, "-")) {
332  s->stats_file = stdout;
333  } else {
334  s->stats_file = fopen(s->stats_file_str, "w");
335  if (!s->stats_file) {
336  int err = AVERROR(errno);
337  char buf[128];
338  av_strerror(err, buf, sizeof(buf));
339  av_log(ctx, AV_LOG_ERROR, "Could not open stats file %s: %s\n",
340  s->stats_file_str, buf);
341  return err;
342  }
343  }
344  }
345 
346  s->dinput.process = do_ssim;
347  s->dinput.shortest = 1;
348  s->dinput.repeatlast = 0;
349  return 0;
350 }
351 
353 {
354  static const enum AVPixelFormat pix_fmts[] = {
362 #define PF(suf) AV_PIX_FMT_YUV420##suf, AV_PIX_FMT_YUV422##suf, AV_PIX_FMT_YUV444##suf, AV_PIX_FMT_GBR##suf
363  PF(P9), PF(P10), PF(P12), PF(P14), PF(P16),
365  };
366 
367  AVFilterFormats *fmts_list = ff_make_format_list(pix_fmts);
368  if (!fmts_list)
369  return AVERROR(ENOMEM);
370  return ff_set_common_formats(ctx, fmts_list);
371 }
372 
373 static int config_input_ref(AVFilterLink *inlink)
374 {
376  AVFilterContext *ctx = inlink->dst;
377  SSIMContext *s = ctx->priv;
378  int sum = 0, i;
379 
380  s->nb_components = desc->nb_components;
381 
382  if (ctx->inputs[0]->w != ctx->inputs[1]->w ||
383  ctx->inputs[0]->h != ctx->inputs[1]->h) {
384  av_log(ctx, AV_LOG_ERROR, "Width and height of input videos must be same.\n");
385  return AVERROR(EINVAL);
386  }
387  if (ctx->inputs[0]->format != ctx->inputs[1]->format) {
388  av_log(ctx, AV_LOG_ERROR, "Inputs must be of same pixel format.\n");
389  return AVERROR(EINVAL);
390  }
391 
392  s->is_rgb = ff_fill_rgba_map(s->rgba_map, inlink->format) >= 0;
393  s->comps[0] = s->is_rgb ? 'R' : 'Y';
394  s->comps[1] = s->is_rgb ? 'G' : 'U';
395  s->comps[2] = s->is_rgb ? 'B' : 'V';
396  s->comps[3] = 'A';
397 
398  s->planeheight[1] = s->planeheight[2] = AV_CEIL_RSHIFT(inlink->h, desc->log2_chroma_h);
399  s->planeheight[0] = s->planeheight[3] = inlink->h;
400  s->planewidth[1] = s->planewidth[2] = AV_CEIL_RSHIFT(inlink->w, desc->log2_chroma_w);
401  s->planewidth[0] = s->planewidth[3] = inlink->w;
402  for (i = 0; i < s->nb_components; i++)
403  sum += s->planeheight[i] * s->planewidth[i];
404  for (i = 0; i < s->nb_components; i++)
405  s->coefs[i] = (double) s->planeheight[i] * s->planewidth[i] / sum;
406 
407  s->temp = av_mallocz_array(2 * SUM_LEN(inlink->w), (desc->comp[0].depth > 8) ? sizeof(int64_t[4]) : sizeof(int[4]));
408  if (!s->temp)
409  return AVERROR(ENOMEM);
410  s->max = (1 << desc->comp[0].depth) - 1;
411 
412  s->ssim_plane = desc->comp[0].depth > 8 ? ssim_plane_16bit : ssim_plane;
415  if (ARCH_X86)
416  ff_ssim_init_x86(&s->dsp);
417 
418  return 0;
419 }
420 
421 static int config_output(AVFilterLink *outlink)
422 {
423  AVFilterContext *ctx = outlink->src;
424  SSIMContext *s = ctx->priv;
425  AVFilterLink *mainlink = ctx->inputs[0];
426  int ret;
427 
428  outlink->w = mainlink->w;
429  outlink->h = mainlink->h;
430  outlink->time_base = mainlink->time_base;
431  outlink->sample_aspect_ratio = mainlink->sample_aspect_ratio;
432  outlink->frame_rate = mainlink->frame_rate;
433 
434  if ((ret = ff_dualinput_init(ctx, &s->dinput)) < 0)
435  return ret;
436 
437  return 0;
438 }
439 
440 static int filter_frame(AVFilterLink *inlink, AVFrame *buf)
441 {
442  SSIMContext *s = inlink->dst->priv;
443  return ff_dualinput_filter_frame(&s->dinput, inlink, buf);
444 }
445 
446 static int request_frame(AVFilterLink *outlink)
447 {
448  SSIMContext *s = outlink->src->priv;
449  return ff_dualinput_request_frame(&s->dinput, outlink);
450 }
451 
453 {
454  SSIMContext *s = ctx->priv;
455 
456  if (s->nb_frames > 0) {
457  char buf[256];
458  int i;
459  buf[0] = 0;
460  for (i = 0; i < s->nb_components; i++) {
461  int c = s->is_rgb ? s->rgba_map[i] : i;
462  av_strlcatf(buf, sizeof(buf), " %c:%f (%f)", s->comps[i], s->ssim[c] / s->nb_frames,
463  ssim_db(s->ssim[c], s->nb_frames));
464  }
465  av_log(ctx, AV_LOG_INFO, "SSIM%s All:%f (%f)\n", buf,
466  s->ssim_total / s->nb_frames, ssim_db(s->ssim_total, s->nb_frames));
467  }
468 
470 
471  if (s->stats_file && s->stats_file != stdout)
472  fclose(s->stats_file);
473 
474  av_freep(&s->temp);
475 }
476 
477 static const AVFilterPad ssim_inputs[] = {
478  {
479  .name = "main",
480  .type = AVMEDIA_TYPE_VIDEO,
481  .filter_frame = filter_frame,
482  },{
483  .name = "reference",
484  .type = AVMEDIA_TYPE_VIDEO,
485  .filter_frame = filter_frame,
486  .config_props = config_input_ref,
487  },
488  { NULL }
489 };
490 
491 static const AVFilterPad ssim_outputs[] = {
492  {
493  .name = "default",
494  .type = AVMEDIA_TYPE_VIDEO,
495  .config_props = config_output,
496  .request_frame = request_frame,
497  },
498  { NULL }
499 };
500 
502  .name = "ssim",
503  .description = NULL_IF_CONFIG_SMALL("Calculate the SSIM between two video streams."),
504  .init = init,
505  .uninit = uninit,
506  .query_formats = query_formats,
507  .priv_size = sizeof(SSIMContext),
508  .priv_class = &ssim_class,
509  .inputs = ssim_inputs,
510  .outputs = ssim_outputs,
511 };
#define NULL
Definition: coverity.c:32
const char * s
Definition: avisynth_c.h:768
float(* ssim_plane)(SSIMDSPContext *dsp, uint8_t *main, int main_stride, uint8_t *ref, int ref_stride, int width, int height, void *temp, int max)
Definition: vf_ssim.c:64
const AVPixFmtDescriptor * av_pix_fmt_desc_get(enum AVPixelFormat pix_fmt)
Definition: pixdesc.c:2333
This structure describes decoded (raw) audio or video data.
Definition: frame.h:187
FFDualInputContext dinput
Definition: vf_ssim.c:50
AVOption.
Definition: opt.h:246
static av_cold void uninit(AVFilterContext *ctx)
Definition: vf_ssim.c:452
planar YUV 4:4:4, 24bpp, (1 Cr & Cb sample per 1x1 Y samples)
Definition: pixfmt.h:67
Main libavfilter public API header.
else temp
Definition: vf_mcdeint.c:259
const char * desc
Definition: nvenc.c:60
int nb_components
Definition: vf_ssim.c:53
planar GBR 4:4:4 24bpp
Definition: pixfmt.h:180
const char * b
Definition: vf_curves.c:113
int max
Definition: vf_ssim.c:54
static void ssim_4x4xn_8bit(const uint8_t *main, ptrdiff_t main_stride, const uint8_t *ref, ptrdiff_t ref_stride, int(*sums)[4], int width)
Definition: vf_ssim.c:132
uint8_t log2_chroma_w
Amount to shift the luma width right to find the chroma width.
Definition: pixdesc.h:92
AVFilterFormats * ff_make_format_list(const int *fmts)
Create a list of supported formats.
Definition: formats.c:283
#define AV_PIX_FMT_GRAY10
Definition: pixfmt.h:333
const char * name
Pad name.
Definition: internal.h:60
#define AV_PIX_FMT_GRAY12
Definition: pixfmt.h:334
int ff_dualinput_filter_frame(FFDualInputContext *s, AVFilterLink *inlink, AVFrame *in)
Definition: dualinput.c:76
AVFilterLink ** inputs
array of pointers to input links
Definition: avfilter.h:331
static av_cold int init(AVFilterContext *ctx)
Definition: vf_ssim.c:326
AVComponentDescriptor comp[4]
Parameters that describe how pixels are packed.
Definition: pixdesc.h:117
uint8_t
#define av_cold
Definition: attributes.h:82
AVOptions.
static const AVFilterPad ssim_inputs[]
Definition: vf_ssim.c:477
int shortest
terminate stream when the second input terminates
Definition: dualinput.h:36
static void ssim_4x4xn_16bit(const uint8_t *main8, ptrdiff_t main_stride, const uint8_t *ref8, ptrdiff_t ref_stride, int64_t(*sums)[4], int width)
Definition: vf_ssim.c:96
#define height
planar YUV 4:4:0 full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV440P and setting color_range...
Definition: pixfmt.h:101
planar YUV 4:2:2, 16bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV422P and setting col...
Definition: pixfmt.h:75
static float ssim_end1(int s1, int s2, int ss, int s12)
Definition: vf_ssim.c:179
void ff_dualinput_uninit(FFDualInputContext *s)
Definition: dualinput.c:87
static AVFrame * do_ssim(AVFilterContext *ctx, AVFrame *main, const AVFrame *ref)
Definition: vf_ssim.c:285
#define av_log(a,...)
A filter pad used for either input or output.
Definition: internal.h:54
#define FLAGS
Definition: vf_ssim.c:73
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:176
int ff_set_common_formats(AVFilterContext *ctx, AVFilterFormats *formats)
A helper for query_formats() which sets all links to the same list of formats.
Definition: formats.c:568
uint8_t log2_chroma_h
Amount to shift the luma height right to find the chroma height.
Definition: pixdesc.h:101
#define s2
Definition: regdef.h:39
#define AVERROR(e)
Definition: error.h:43
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:179
uint8_t rgba_map[4]
Definition: vf_ssim.c:59
void ff_ssim_init_x86(SSIMDSPContext *dsp)
Definition: vf_ssim_init.c:33
void * priv
private data for use by the filter
Definition: avfilter.h:338
static void * av_mallocz_array(size_t nmemb, size_t size)
Definition: mem.h:229
planar YUV 4:2:2, 16bpp, (1 Cr & Cb sample per 2x1 Y samples)
Definition: pixfmt.h:66
int planewidth[4]
Definition: vf_ssim.c:60
uint8_t nb_components
The number of components each pixel has, (1-4)
Definition: pixdesc.h:83
#define OFFSET(x)
Definition: vf_ssim.c:72
#define AV_PIX_FMT_GRAY16
Definition: pixfmt.h:335
#define PF(suf)
#define SUM_LEN(w)
Definition: vf_ssim.c:222
planar YUV 4:2:0, 12bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV420P and setting col...
Definition: pixfmt.h:74
#define width
GLsizei GLboolean const GLfloat * value
Definition: opengl_enc.c:109
static int request_frame(AVFilterLink *outlink)
Definition: vf_ssim.c:446
AVFormatContext * ctx
Definition: movenc.c:48
AVFrame *(* process)(AVFilterContext *ctx, AVFrame *main, const AVFrame *second)
Definition: dualinput.h:35
static const AVFilterPad outputs[]
Definition: af_afftfilt.c:386
int ff_fill_rgba_map(uint8_t *rgba_map, enum AVPixelFormat pix_fmt)
Definition: drawutils.c:35
static void comp(unsigned char *dst, ptrdiff_t dst_stride, unsigned char *src, ptrdiff_t src_stride, int add)
Definition: eamad.c:83
static const AVOption ssim_options[]
Definition: vf_ssim.c:75
static const uint8_t vars[2][12]
Definition: camellia.c:179
#define AV_LOG_INFO
Standard information.
Definition: log.h:187
static const AVFilterPad inputs[]
Definition: af_afftfilt.c:376
misc drawing utilities
AVDictionary ** avpriv_frame_get_metadatap(AVFrame *frame)
Definition: frame.c:51
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
Definition: frame.h:218
#define ss
Descriptor that unambiguously describes how the bits of a pixel are stored in the up to 4 data planes...
Definition: pixdesc.h:81
FILE * stats_file
Definition: vf_ssim.c:51
static float ssim_plane(SSIMDSPContext *dsp, uint8_t *main, int main_stride, uint8_t *ref, int ref_stride, int width, int height, void *temp, int max)
Definition: vf_ssim.c:252
int repeatlast
repeat last second frame
Definition: dualinput.h:37
void * buf
Definition: avisynth_c.h:690
static int filter_frame(AVFilterLink *inlink, AVFrame *buf)
Definition: vf_ssim.c:440
int av_dict_set(AVDictionary **pm, const char *key, const char *value, int flags)
Set the given entry in *pm, overwriting an existing entry.
Definition: dict.c:70
planar YUV 4:1:0, 9bpp, (1 Cr & Cb sample per 4x4 Y samples)
Definition: pixfmt.h:68
static int config_input_ref(AVFilterLink *inlink)
Definition: vf_ssim.c:373
AVFILTER_DEFINE_CLASS(ssim)
Describe the class of an AVClass context structure.
Definition: log.h:67
Filter definition.
Definition: avfilter.h:144
int planeheight[4]
Definition: vf_ssim.c:61
float coefs[4]
Definition: vf_ssim.c:58
void(* ssim_4x4_line)(const uint8_t *buf, ptrdiff_t buf_stride, const uint8_t *ref, ptrdiff_t ref_stride, int(*sums)[4], int w)
Definition: ssim.h:28
size_t av_strlcatf(char *dst, size_t size, const char *fmt,...)
Definition: avstring.c:101
uint64_t nb_frames
Definition: vf_ssim.c:55
const char * name
Filter name.
Definition: avfilter.h:148
#define s1
Definition: regdef.h:38
#define snprintf
Definition: snprintf.h:34
int ff_dualinput_init(AVFilterContext *ctx, FFDualInputContext *s)
Definition: dualinput.c:43
static int weight(int i, int blen, int offset)
Definition: diracdec.c:1523
static enum AVPixelFormat pix_fmts[]
Definition: libkvazaar.c:262
SSIMDSPContext dsp
Definition: vf_ssim.c:69
char comps[4]
Definition: vf_ssim.c:57
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
Definition: frame.h:201
int av_strerror(int errnum, char *errbuf, size_t errbuf_size)
Put a description of the AVERROR code errnum in errbuf.
Definition: error.c:105
static double ssim_db(double ssim, double weight)
Definition: vf_ssim.c:280
int
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
Definition: pixfmt.h:62
Y , 8bpp.
Definition: pixfmt.h:70
static int ref[MAX_W *MAX_W]
Definition: jpeg2000dwt.c:107
static double c[64]
double ssim_total
Definition: vf_ssim.c:56
planar YUV 4:4:4, 24bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV444P and setting col...
Definition: pixfmt.h:76
static int config_output(AVFilterLink *outlink)
Definition: vf_ssim.c:421
planar YUV 4:1:1, 12bpp, (1 Cr & Cb sample per 4x1 Y samples)
Definition: pixfmt.h:69
AVFilter ff_vf_ssim
Definition: vf_ssim.c:501
double ssim[4]
Definition: vf_ssim.c:56
static float ssim_endn_16bit(const int64_t(*sum0)[4], const int64_t(*sum1)[4], int width, int max)
Definition: vf_ssim.c:195
static float ssim_end1x(int64_t s1, int64_t s2, int64_t ss, int64_t s12, int max)
Definition: vf_ssim.c:163
static const AVFilterPad ssim_outputs[]
Definition: vf_ssim.c:491
Double input streams helper for filters.
A list of supported formats for one end of a filter link.
Definition: formats.h:64
planar YUV 4:1:1, 12bpp, (1 Cr & Cb sample per 4x1 Y samples) full scale (JPEG), deprecated in favor ...
Definition: pixfmt.h:272
float(* ssim_end_line)(const int(*sum0)[4], const int(*sum1)[4], int w)
Definition: ssim.h:31
static int query_formats(AVFilterContext *ctx)
Definition: vf_ssim.c:352
An instance of a filter.
Definition: avfilter.h:323
int ff_dualinput_request_frame(FFDualInputContext *s, AVFilterLink *outlink)
Definition: dualinput.c:82
#define av_freep(p)
int * temp
Definition: vf_ssim.c:62
planar YUV 4:4:0 (1 Cr & Cb sample per 1x2 Y samples)
Definition: pixfmt.h:100
static float ssim_endn_8bit(const int(*sum0)[4], const int(*sum1)[4], int width)
Definition: vf_ssim.c:209
#define FFSWAP(type, a, b)
Definition: common.h:99
int main(int argc, char **argv)
Definition: main.c:22
static float ssim_plane_16bit(SSIMDSPContext *dsp, uint8_t *main, int main_stride, uint8_t *ref, int ref_stride, int width, int height, void *temp, int max)
Definition: vf_ssim.c:224
internal API functions
int depth
Number of bits in the component.
Definition: pixdesc.h:58
static void set_meta(AVDictionary **metadata, const char *key, char comp, float d)
Definition: vf_ssim.c:83
AVPixelFormat
Pixel format.
Definition: pixfmt.h:60
int is_rgb
Definition: vf_ssim.c:63
char * stats_file_str
Definition: vf_ssim.c:52
for(j=16;j >0;--j)
#define AV_CEIL_RSHIFT(a, b)
Definition: common.h:58