3 * This file is part of FFmpeg.
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.
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.
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.
22 #include "libavutil/common.h"
23 #include "libavutil/intreadwrite.h"
24 #include "libavutil/mem_internal.h"
26 #include "libavcodec/me_cmp.h"
30 static void fill_random(uint8_t *tab, int size)
33 for (i = 0; i < size; i++) {
38 static void test_motion(const char *name, me_cmp_func test_func)
40 /* test configurarion */
45 /* motion estimation can look up to 17 bytes ahead */
46 static const int look_ahead = 17;
51 LOCAL_ALIGNED_16(uint8_t, img1, [WIDTH * HEIGHT]);
52 LOCAL_ALIGNED_16(uint8_t, img2, [WIDTH * HEIGHT]);
54 declare_func_emms(AV_CPU_FLAG_MMX, int, struct MpegEncContext *c,
55 uint8_t *blk1 /* align width (8 or 16) */,
56 uint8_t *blk2 /* align 1 */, ptrdiff_t stride,
59 if (test_func == NULL) {
63 /* test correctness */
64 fill_random(img1, WIDTH * HEIGHT);
65 fill_random(img2, WIDTH * HEIGHT);
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);
72 ptr = img2 + y * WIDTH + x;
73 d2 = call_ref(NULL, img1, ptr, WIDTH, 8);
74 d1 = call_new(NULL, img1, ptr, WIDTH, 8);
78 printf("func: %s, x=%d y=%d, error: asm=%d c=%d\n", name, x, y, d1, d2);
82 // benchmark with the final value of ptr
83 bench_new(NULL, img1, ptr, WIDTH, 8);
87 #define ME_CMP_1D_ARRAYS(XX) \
102 // tests for functions not yet implemented
115 static void check_motion(void)
118 AVCodecContext *av_ctx;
121 memset(&me_ctx, 0, sizeof(me_ctx));
123 /* allocate AVCodecContext */
124 av_ctx = avcodec_alloc_context3(NULL);
125 av_ctx->flags |= AV_CODEC_FLAG_BITEXACT;
127 ff_me_cmp_init(&me_ctx, av_ctx);
129 for (int i = 0; i < FF_ARRAY_ELEMS(me_ctx.pix_abs); i++) {
130 for (int j = 0; j < FF_ARRAY_ELEMS(me_ctx.pix_abs[0]); j++) {
131 snprintf(buf, sizeof(buf), "pix_abs_%d_%d", i, j);
132 test_motion(buf, me_ctx.pix_abs[i][j]);
136 #define XX(me_cmp_array) \
137 for (int i = 0; i < FF_ARRAY_ELEMS(me_ctx.me_cmp_array); i++) { \
138 snprintf(buf, sizeof(buf), #me_cmp_array "_%d", i); \
139 test_motion(buf, me_ctx.me_cmp_array[i]); \
144 avcodec_free_context(&av_ctx);
147 void checkasm_check_motion(void)