checkasm/llviddsp : add test for other dsp func

add_median_pred
add_left_pred : add two func one with acc 0, and one with random acc
add_left_pred16

1 files changed, 108 insertions, 8 deletions

diff --git a/tests/checkasm/llviddsp.c b/tests/checkasm/llviddsp.c
index b1f3b83b43..9b35e2cadc 100644
--- a/tests/checkasm/llviddsp.c
+++ b/tests/checkasm/llviddsp.c
@@ -31,23 +31,29 @@
 #define randomize_buffers(buf, size)     \
     do {                                 \
         int j;                           \
+        uint8_t *tmp_buf = (uint8_t *)buf;\
         for (j = 0; j < size; j++)       \
-            buf[j] = rnd() & 0xFF;       \
+            tmp_buf[j] = rnd() & 0xFF;       \
     } while (0)
 
+#define init_buffer(a0, a1, type, width)\
+    type *a0 = av_mallocz_array(width, sizeof(type));\
+    type *a1 = av_mallocz_array(width, sizeof(type));\
+    if (!a0 || !a1)\
+        fail();\
+    randomize_buffers(a0, width * sizeof(type));\
+    memcpy(a1, a0, width*sizeof(type));\
+
 static void check_add_bytes(LLVidDSPContext c, int width)
 {
-    uint8_t *src0 = av_mallocz(width);
-    uint8_t *src1 = av_mallocz(width);
     uint8_t *dst0 = av_mallocz(width);
     uint8_t *dst1 = av_mallocz(width);
-    declare_func_emms(AV_CPU_FLAG_MMX, void, uint8_t *dst, uint8_t *src, ptrdiff_t w);
+    init_buffer(src0, src1, uint8_t, width);
 
-    if (!src0 || !src1 || !dst0 || !dst1)
+    if (!dst0 || !dst1)
         fail();
 
-    randomize_buffers(src0, width);
-    memcpy(src1, src0, width);
+    declare_func_emms(AV_CPU_FLAG_MMX, void, uint8_t *dst, uint8_t *src, ptrdiff_t w);
 
     if (check_func(c.add_bytes, "add_bytes")) {
         call_ref(dst0, src0, width);
@@ -63,14 +69,108 @@ static void check_add_bytes(LLVidDSPContext c, int width)
     av_free(dst1);
 }
 
+static void check_add_median_pred(LLVidDSPContext c, int width) {
+    int A0, A1, B0, B1;
+    uint8_t *dst0 = av_mallocz(width);
+    uint8_t *dst1 = av_mallocz(width);
+    init_buffer(src0, src1, uint8_t, width);
+    init_buffer(diff0, diff1, uint8_t, width);
+
+    A0 = rnd() & 0xFF;
+    B0 = rnd() & 0xFF;
+    A1 = A0;
+    B1 = B0;
+
+    declare_func_emms(AV_CPU_FLAG_MMX, void, uint8_t *dst, const uint8_t *src1,
+                      const uint8_t *diff, ptrdiff_t w,
+                      int *left, int *left_top);
+
+    if (check_func(c.add_median_pred, "add_median_pred")) {
+        call_ref(dst0, src0, diff0, width, &A0, &B0);
+        call_new(dst1, src1, diff1, width, &A1, &B1);
+        if (memcmp(dst0, dst1, width) || (A0 != A1) || (B0 != B1))
+            fail();
+        bench_new(dst1, src1, diff1, width, &A1, &B1);
+    }
+
+    av_free(src0);
+    av_free(src1);
+    av_free(diff0);
+    av_free(diff1);
+    av_free(dst0);
+    av_free(dst1);
+}
+
+static void check_add_left_pred(LLVidDSPContext c, int width, int acc, const char * report)
+{
+    uint8_t *dst0 = av_mallocz(width);
+    uint8_t *dst1 = av_mallocz(width);
+    init_buffer(src0, src1, uint8_t, width);
+
+    if (!dst0 || !dst1)
+        fail();
+
+    declare_func_emms(AV_CPU_FLAG_MMX, void, uint8_t *dst, uint8_t *src, ptrdiff_t w, int acc);
+
+    if (check_func(c.add_left_pred, "%s", report)) {
+        call_ref(dst0, src0, width, acc);
+        call_new(dst1, src1, width, acc);
+        if (memcmp(dst0, dst1, width))
+            fail();
+        bench_new(dst1, src1, width, acc);
+    }
+
+    av_free(src0);
+    av_free(src1);
+    av_free(dst0);
+    av_free(dst1);
+}
+
+static void check_add_left_pred_16(LLVidDSPContext c, unsigned mask, int width, unsigned acc, const char * report)
+{
+    uint16_t *dst0 = av_mallocz_array(width, sizeof(uint16_t));
+    uint16_t *dst1 = av_mallocz_array(width, sizeof(uint16_t));
+    init_buffer(src0, src1, uint16_t, width);
+
+    if (!dst0 || !dst1)
+        fail();
+
+    declare_func_emms(AV_CPU_FLAG_MMX, void, uint16_t *dst, uint16_t *src, unsigned mask, ptrdiff_t w, unsigned acc);
+
+    if (check_func(c.add_left_pred_int16, "%s", report)) {
+        call_ref(dst0, src0, mask, width, acc);
+        call_new(dst1, src1, mask, width, acc);
+        if (memcmp(dst0, dst1, width))
+            fail();
+        bench_new(dst1, src1, mask, width, acc);
+    }
+
+    av_free(src0);
+    av_free(src1);
+    av_free(dst0);
+    av_free(dst1);
+}
+
 void checkasm_check_llviddsp(void)
 {
     LLVidDSPContext c;
     int width = 16 * av_clip(rnd(), 16, 128);
+    int accRnd = rnd() & 0xFF;
 
     ff_llviddsp_init(&c);
 
     check_add_bytes(c, width);
-
     report("add_bytes");
+
+    check_add_median_pred(c, width);
+    report("add_median_pred");
+
+    check_add_left_pred(c, width, 0, "add_left_pred_zero");
+    report("add_left_pred_zero");
+
+    check_add_left_pred(c, width, accRnd, "add_left_pred_rnd_acc");
+    report("add_left_pred_rnd_acc");
+
+    check_add_left_pred_16(c, 255, width, accRnd, "add_left_pred_int16");
+    report("add_left_pred_int16");
 }