tests/checkasm/motion.c

   1 /*
   2  *
   3  * This file is part of FFmpeg.
   4  *
   5  * FFmpeg is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation; either version 2 of the License, or
   8  * (at your option) any later version.
   9  *
  10  * FFmpeg is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License along
  16  * with FFmpeg; if not, write to the Free Software Foundation, Inc.,
  17  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18  */
  19
  20 #include <string.h>
  21
  22 #include "libavutil/common.h"
  23 #include "libavutil/intreadwrite.h"
  24 #include "libavutil/mem_internal.h"
  25
  26 #include "libavcodec/me_cmp.h"
  27
  28 #include "checkasm.h"
  29
  30 static void fill_random(uint8_t *tab, int size)
  31 {
  32     int i;
  33     for (i = 0; i < size; i++) {
  34         tab[i] = rnd() % 256;
  35     }
  36 }
  37
  38 static void test_motion(const char *name, me_cmp_func test_func)
  39 {
  40     /* test configurarion */
  41 #define ITERATIONS 16
  42 #define WIDTH 64
  43 #define HEIGHT 64
  44
  45     /* motion estimation can look up to 17 bytes ahead */
  46     static const int look_ahead = 17;
  47
  48     int i, x, y, h, d1, d2;
  49     uint8_t *ptr;
  50
  51     LOCAL_ALIGNED_16(uint8_t, img1, [WIDTH * HEIGHT]);
  52     LOCAL_ALIGNED_16(uint8_t, img2, [WIDTH * HEIGHT]);
  53
  54     declare_func_emms(AV_CPU_FLAG_MMX, int, MPVEncContext *c,
  55                       const uint8_t *blk1 /* align width (8 or 16) */,
  56                       const uint8_t *blk2 /* align 1 */, ptrdiff_t stride,
  57                       int h);
  58
  59     if (test_func == NULL) {
  60         return;
  61     }
  62
  63     /* test correctness */
  64     fill_random(img1, WIDTH * HEIGHT);
  65     fill_random(img2, WIDTH * HEIGHT);
  66
  67     if (check_func(test_func, "%s", name)) {
  68         for (i = 0; i < ITERATIONS; i++) {
  69             x = rnd() % (WIDTH - look_ahead);
  70             y = rnd() % (HEIGHT - look_ahead);
  71             // Pick a random h between 4 and 16; pick an even value.
  72             h = 4 + ((rnd() % (16 + 1 - 4)) & ~1);
  73
  74             ptr = img2 + y * WIDTH + x;
  75             d2 = call_ref(NULL, img1, ptr, WIDTH, h);
  76             d1 = call_new(NULL, img1, ptr, WIDTH, h);
  77
  78             if (d1 != d2) {
  79                 fail();
  80                 printf("func: %s, x=%d y=%d h=%d, error: asm=%d c=%d\n", name, x, y, h, d1, d2);
  81                 break;
  82             }
  83         }
  84         // Test with a fixed offset, for benchmark stability
  85         ptr = img2 + 3 * WIDTH + 3;
  86         bench_new(NULL, img1, ptr, WIDTH, 8);
  87     }
  88 }
  89
  90 #define ME_CMP_1D_ARRAYS(XX)                                                   \
  91     XX(sad)                                                                    \
  92     XX(sse)                                                                    \
  93     XX(hadamard8_diff)                                                         \
  94     XX(vsad)                                                                   \
  95     XX(vsse)                                                                   \
  96     XX(nsse)                                                                   \
  97     XX(median_sad)
  98
  99 // tests for functions not yet implemented
 100 #if 0
 101     XX(dct_sad)                                                                \
 102     XX(quant_psnr)                                                             \
 103     XX(bit)                                                                    \
 104     XX(rd)                                                                     \
 105     XX(w53)                                                                    \
 106     XX(w97)                                                                    \
 107     XX(dct_max)                                                                \
 108     XX(dct264_sad)                                                             \
 109
 110 #endif
 111
 112 static void check_motion(void)
 113 {
 114     char buf[64];
 115     /* Setup AVCodecContext in a way that does not pull in all of libavcodec */
 116     AVCodecContext av_ctx = { .codec_id = AV_CODEC_ID_NONE, .flags = AV_CODEC_FLAG_BITEXACT };
 117     MECmpContext me_ctx;
 118
 119     ff_me_cmp_init(&me_ctx, &av_ctx);
 120
 121     for (int i = 0; i < FF_ARRAY_ELEMS(me_ctx.pix_abs); i++) {
 122         for (int j = 0; j < FF_ARRAY_ELEMS(me_ctx.pix_abs[0]); j++) {
 123             snprintf(buf, sizeof(buf), "pix_abs_%d_%d", i, j);
 124             test_motion(buf, me_ctx.pix_abs[i][j]);
 125         }
 126     }
 127
 128 #define XX(me_cmp_array)                                                        \
 129     for (int i = 0; i < FF_ARRAY_ELEMS(me_ctx.me_cmp_array); i++) {             \
 130         snprintf(buf, sizeof(buf), #me_cmp_array "_%d", i);                     \
 131         test_motion(buf, me_ctx.me_cmp_array[i]);                               \
 132     }
 133     ME_CMP_1D_ARRAYS(XX)
 134 #undef XX
 135 }
 136
 137 void checkasm_check_motion(void)
 138 {
 139     check_motion();
 140     report("motion");
 141 }