git.sesse.net Git - ffmpeg/blob - tests/checkasm/vf_blend.c

   1 /*
   2  * Copyright (c) 2016 Tiancheng "Timothy" Gu
   3  *
   4  * This file is part of FFmpeg.
   5  *
   6  * FFmpeg is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * FFmpeg is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License along
  17  * with FFmpeg; if not, write to the Free Software Foundation, Inc.,
  18  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  19  */
  20
  21 #include <string.h>
  22 #include "checkasm.h"
  23 #include "libavfilter/blend.h"
  24 #include "libavutil/common.h"
  25 #include "libavutil/internal.h"
  26 #include "libavutil/intreadwrite.h"
  27
  28 #define WIDTH 256
  29 #define HEIGHT 256
  30 #define BUF_UNITS 3
  31 #define SIZE_PER_UNIT (WIDTH * HEIGHT)
  32 #define BUF_SIZE (BUF_UNITS * SIZE_PER_UNIT)
  33
  34 #define randomize_buffers()                   \
  35     do {                                      \
  36         int i, j;                             \
  37         for (i = 0; i < HEIGHT; i++) {        \
  38             for (j = 0; j < WIDTH; j++) {     \
  39                 top1[i * WIDTH + j] =         \
  40                 top2[i * WIDTH + j] = i;      \
  41                 bot1[i * WIDTH + j] =         \
  42                 bot2[i * WIDTH + j] = j;      \
  43             }                                 \
  44         }                                     \
  45         for (i = 0; i < SIZE_PER_UNIT; i += 4) { \
  46             uint32_t r = rnd();               \
  47             AV_WN32A(dst1 + i, r);            \
  48             AV_WN32A(dst2 + i, r);            \
  49         }                                     \
  50         for (; i < BUF_SIZE; i += 4) {        \
  51             uint32_t r = rnd();               \
  52             AV_WN32A(top1 + i, r);            \
  53             AV_WN32A(top2 + i, r);            \
  54             r = rnd();                        \
  55             AV_WN32A(bot1 + i, r);            \
  56             AV_WN32A(bot2 + i, r);            \
  57             r = rnd();                        \
  58             AV_WN32A(dst1 + i, r);            \
  59             AV_WN32A(dst2 + i, r);            \
  60         }                                     \
  61     } while (0)
  62
  63 #define check_blend_func(depth)                                                            \
  64     do {                                                                                   \
  65         int i, w;                                                                          \
  66         declare_func(void, const uint8_t *top, ptrdiff_t top_linesize,                     \
  67                      const uint8_t *bottom, ptrdiff_t bottom_linesize,                     \
  68                      uint8_t *dst, ptrdiff_t dst_linesize,                                 \
  69                      ptrdiff_t width, ptrdiff_t height,                                    \
  70                      struct FilterParams *param, double *values);                          \
  71         w = WIDTH / depth;                                                                 \
  72                                                                                            \
  73         for (i = 0; i < BUF_UNITS - 1; i++) {                                              \
  74             int src_offset = i * SIZE_PER_UNIT + (BUF_UNITS - 1 - i) * depth; /* Test various alignments */  \
  75             int dst_offset = i * SIZE_PER_UNIT; /* dst must be aligned */                  \
  76             randomize_buffers();                                                           \
  77             call_ref(top1 + src_offset, w, bot1 + src_offset, w,                           \
  78                      dst1 + dst_offset, w, w, HEIGHT, &param, NULL);                       \
  79             call_new(top2 + src_offset, w, bot2 + src_offset, w,                           \
  80                      dst2 + dst_offset, w, w, HEIGHT, &param, NULL);                       \
  81             if (memcmp(top1, top2, BUF_SIZE) || memcmp(bot1, bot2, BUF_SIZE) || memcmp(dst1, dst2, BUF_SIZE)) \
  82                 fail();                                                                    \
  83         }                                                                                  \
  84         bench_new(top2, w / 4, bot2, w / 4, dst2, w / 4,                                   \
  85                   w / 4, HEIGHT / 4, &param, NULL);                                        \
  86     } while (0)
  87
  88 void checkasm_check_blend(void)
  89 {
  90     uint8_t *top1 = av_malloc(BUF_SIZE);
  91     uint8_t *top2 = av_malloc(BUF_SIZE);
  92     uint8_t *bot1 = av_malloc(BUF_SIZE);
  93     uint8_t *bot2 = av_malloc(BUF_SIZE);
  94     uint8_t *dst1 = av_malloc(BUF_SIZE);
  95     uint8_t *dst2 = av_malloc(BUF_SIZE);
  96     FilterParams param = {
  97         .opacity = 1.0,
  98     };
  99
 100 #define check_and_report(name, val, depth)        \
 101     param.mode = val;                             \
 102     ff_blend_init(&param, depth * 8);             \
 103     if (check_func(param.blend, #name))           \
 104         check_blend_func(depth);
 105
 106     check_and_report(addition, BLEND_ADDITION, 1)
 107     check_and_report(grainmerge, BLEND_GRAINMERGE, 1)
 108     check_and_report(and, BLEND_AND, 1)
 109     check_and_report(average, BLEND_AVERAGE, 1)
 110     check_and_report(darken, BLEND_DARKEN, 1)
 111     check_and_report(grainextract, BLEND_GRAINEXTRACT, 1)
 112     check_and_report(hardmix, BLEND_HARDMIX, 1)
 113     check_and_report(lighten, BLEND_LIGHTEN, 1)
 114     check_and_report(multiply, BLEND_MULTIPLY, 1)
 115     check_and_report(or, BLEND_OR, 1)
 116     check_and_report(phoenix, BLEND_PHOENIX, 1)
 117     check_and_report(screen, BLEND_SCREEN, 1)
 118     check_and_report(subtract, BLEND_SUBTRACT, 1)
 119     check_and_report(xor, BLEND_XOR, 1)
 120     check_and_report(difference, BLEND_DIFFERENCE, 1)
 121     check_and_report(extremity, BLEND_EXTREMITY, 1)
 122     check_and_report(negation, BLEND_NEGATION, 1)
 123
 124     report("8bit");
 125
 126     check_and_report(addition_16, BLEND_ADDITION, 2)
 127     check_and_report(grainmerge_16, BLEND_GRAINMERGE, 2)
 128     check_and_report(and_16, BLEND_AND, 2)
 129     check_and_report(average_16, BLEND_AVERAGE, 2)
 130     check_and_report(darken_16, BLEND_DARKEN, 2)
 131     check_and_report(grainextract_16, BLEND_GRAINEXTRACT, 2)
 132     check_and_report(difference_16, BLEND_DIFFERENCE, 2)
 133     check_and_report(extremity_16, BLEND_EXTREMITY, 2)
 134     check_and_report(negation_16, BLEND_NEGATION, 2)
 135     check_and_report(lighten_16, BLEND_LIGHTEN, 2)
 136     check_and_report(or_16, BLEND_OR, 2)
 137     check_and_report(phoenix_16, BLEND_PHOENIX, 2)
 138     check_and_report(subtract_16, BLEND_SUBTRACT, 2)
 139     check_and_report(xor_16, BLEND_SUBTRACT, 2)
 140
 141     report("16bit");
 142
 143     av_freep(&top1);
 144     av_freep(&top2);
 145     av_freep(&bot1);
 146     av_freep(&bot2);
 147     av_freep(&dst1);
 148     av_freep(&dst2);
 149 }