ipred.c 9.77 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48
/*
 * Copyright © 2018, VideoLAN and dav1d authors
 * Copyright © 2018, Two Orioles, LLC
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * 1. Redistributions of source code must retain the above copyright notice, this
 *    list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright notice,
 *    this list of conditions and the following disclaimer in the documentation
 *    and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include "tests/checkasm/checkasm.h"
#include "src/ipred.h"
#include "src/levels.h"

static const char *const intra_pred_mode_names[N_IMPL_INTRA_PRED_MODES] = {
    [DC_PRED]       = "dc",
    [DC_128_PRED]   = "dc_128",
    [TOP_DC_PRED]   = "dc_top",
    [LEFT_DC_PRED]  = "dc_left",
    [HOR_PRED]      = "h",
    [VERT_PRED]     = "v",
    [PAETH_PRED]    = "paeth",
    [SMOOTH_PRED]   = "smooth",
    [SMOOTH_V_PRED] = "smooth_v",
    [SMOOTH_H_PRED] = "smooth_h",
    [Z1_PRED]       = "z1",
    [Z2_PRED]       = "z2",
    [Z3_PRED]       = "z3",
    [FILTER_PRED]   = "filter"
};

49 50
static const char *const cfl_ac_names[3] = { "420", "422", "444" };

51 52 53 54 55 56 57
static const char *const cfl_pred_mode_names[DC_128_PRED + 1] = {
    [DC_PRED]       = "cfl",
    [DC_128_PRED]   = "cfl_128",
    [TOP_DC_PRED]   = "cfl_top",
    [LEFT_DC_PRED]  = "cfl_left",
};

58 59 60 61 62 63 64 65 66 67 68 69 70 71 72
static const uint8_t z_angles[27] = {
     3,  6,  9,
    14, 17, 20, 23, 26, 29, 32,
    36, 39, 42, 45, 48, 51, 54,
    58, 61, 64, 67, 70, 73, 76,
    81, 84, 87
};

static void check_intra_pred(Dav1dIntraPredDSPContext *const c) {
    ALIGN_STK_32(pixel, c_dst, 64 * 64,);
    ALIGN_STK_32(pixel, a_dst, 64 * 64,);
    ALIGN_STK_32(pixel, topleft_buf, 257,);
    pixel *const topleft = topleft_buf + 128;

    declare_func(void, pixel *dst, ptrdiff_t stride, const pixel *topleft,
73 74
                 int width, int height, int angle, int max_width, int max_height
                 HIGHBD_DECL_SUFFIX);
75 76 77 78 79 80 81 82 83 84 85 86 87

    for (int mode = 0; mode < N_IMPL_INTRA_PRED_MODES; mode++)
        for (int w = 4; w <= (mode == FILTER_PRED ? 32 : 64); w <<= 1)
            if (check_func(c->intra_pred[mode], "intra_pred_%s_w%d_%dbpc",
                intra_pred_mode_names[mode], w, BITDEPTH))
            {
                for (int h = imax(w / 4, 4); h <= imin(w * 4,
                    (mode == FILTER_PRED ? 32 : 64)); h <<= 1)
                {
                    const ptrdiff_t stride = w * sizeof(pixel);

                    int a = 0;
                    if (mode >= Z1_PRED && mode <= Z3_PRED) /* angle */
Henrik Gramner's avatar
Henrik Gramner committed
88 89
                        a = (90 * (mode - Z1_PRED) + z_angles[rand() % 27]) |
                            (rand() & 0x600);
90
                    else if (mode == FILTER_PRED) /* filter_idx */
91
                        a = (rand() % 5) | (rand() & ~511);
92

93 94 95 96 97 98
#if BITDEPTH == 16
                    const int bitdepth_max = rand() & 1 ? 0x3ff : 0xfff;
#else
                    const int bitdepth_max = 0xff;
#endif

99
                    for (int i = -h * 2; i <= w * 2; i++)
100
                        topleft[i] = rand() & bitdepth_max;
101

102
                    const int maxw = 1 + (rand() % 128), maxh = 1 + (rand() % 128);
103 104 105 106
                    call_ref(c_dst, stride, topleft, w, h, a, maxw, maxh
                             HIGHBD_TAIL_SUFFIX);
                    call_new(a_dst, stride, topleft, w, h, a, maxw, maxh
                             HIGHBD_TAIL_SUFFIX);
107 108 109
                    if (memcmp(c_dst, a_dst, w * h * sizeof(*c_dst)))
                        fail();

110 111
                    bench_new(a_dst, stride, topleft, w, h, a, 128, 128
                              HIGHBD_TAIL_SUFFIX);
112 113 114 115 116
                }
            }
    report("intra_pred");
}

117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133
static void check_cfl_ac(Dav1dIntraPredDSPContext *const c) {
    ALIGN_STK_32(int16_t, c_dst, 32 * 32,);
    ALIGN_STK_32(int16_t, a_dst, 32 * 32,);
    ALIGN_STK_32(pixel, luma, 32 * 32,);

    declare_func(void, int16_t *ac, const pixel *y, ptrdiff_t stride,
                 int w_pad, int h_pad, int cw, int ch);

    for (int layout = 1; layout <= DAV1D_PIXEL_LAYOUT_I444; layout++) {
        const int ss_ver = layout == DAV1D_PIXEL_LAYOUT_I420;
        const int ss_hor = layout != DAV1D_PIXEL_LAYOUT_I444;
        for (int w = 4; w <= (32 >> ss_hor); w <<= 1)
            if (check_func(c->cfl_ac[layout - 1], "cfl_ac_%s_w%d_%dbpc",
                cfl_ac_names[layout - 1], w, BITDEPTH))
            {
                for (int h = imax(w / 4, 4); h <= imin(w * 4, (32 >> ss_ver)); h <<= 1) {
                    const ptrdiff_t stride = 32 * sizeof(pixel);
134 135
                    for (int w_pad = (w >> 2) - 1; w_pad >= 0; w_pad--) {
                        for (int h_pad = (h >> 2) - 1; h_pad >= 0; h_pad--) {
136 137 138 139 140
#if BITDEPTH == 16
                            const int bitdepth_max = rand() & 1 ? 0x3ff : 0xfff;
#else
                            const int bitdepth_max = 0xff;
#endif
141 142
                            for (int y = 0; y < (h << ss_ver); y++)
                                for (int x = 0; x < (w << ss_hor); x++)
143
                                    luma[y * 32 + x] = rand() & bitdepth_max;
144 145 146 147 148 149 150

                            call_ref(c_dst, luma, stride, w_pad, h_pad, w, h);
                            call_new(a_dst, luma, stride, w_pad, h_pad, w, h);
                            if (memcmp(c_dst, a_dst, w * h * sizeof(*c_dst)))
                                fail();
                        }
                    }
151 152 153 154 155 156 157 158

                    bench_new(a_dst, luma, stride, 0, 0, w, h);
                }
            }
    }
    report("cfl_ac");
}

159 160 161 162 163 164 165 166
static void check_cfl_pred(Dav1dIntraPredDSPContext *const c) {
    ALIGN_STK_32(pixel, c_dst, 32 * 32,);
    ALIGN_STK_32(pixel, a_dst, 32 * 32,);
    ALIGN_STK_32(int16_t, ac, 32 * 32,);
    ALIGN_STK_32(pixel, topleft_buf, 257,);
    pixel *const topleft = topleft_buf + 128;

    declare_func(void, pixel *dst, ptrdiff_t stride, const pixel *topleft,
167 168
                 int width, int height, const int16_t *ac, int alpha
                 HIGHBD_DECL_SUFFIX);
169 170 171 172 173 174 175 176

    for (int mode = 0; mode <= DC_128_PRED; mode += 1 + 2 * !mode)
        for (int w = 4; w <= 32; w <<= 1)
            if (check_func(c->cfl_pred[mode], "cfl_pred_%s_w%d_%dbpc",
                cfl_pred_mode_names[mode], w, BITDEPTH))
            {
                for (int h = imax(w / 4, 4); h <= imin(w * 4, 32); h <<= 1)
                {
177 178 179 180 181 182
#if BITDEPTH == 16
                    const int bitdepth_max = rand() & 1 ? 0x3ff : 0xfff;
#else
                    const int bitdepth_max = 0xff;
#endif

183 184 185 186 187
                    const ptrdiff_t stride = w * sizeof(pixel);

                    int alpha = ((rand() & 15) + 1) * (1 - (rand() & 2));

                    for (int i = -h * 2; i <= w * 2; i++)
188
                        topleft[i] = rand() & bitdepth_max;
189 190 191

                    int luma_avg = w * h >> 1;
                    for (int i = 0; i < w * h; i++)
192
                        luma_avg += ac[i] = rand() & (bitdepth_max << 3);
193 194 195 196
                    luma_avg /= w * h;
                    for (int i = 0; i < w * h; i++)
                        ac[i] -= luma_avg;

197 198 199 200
                    call_ref(c_dst, stride, topleft, w, h, ac, alpha
                             HIGHBD_TAIL_SUFFIX);
                    call_new(a_dst, stride, topleft, w, h, ac, alpha
                             HIGHBD_TAIL_SUFFIX);
201 202 203
                    if (memcmp(c_dst, a_dst, w * h * sizeof(*c_dst)))
                        fail();

204 205
                    bench_new(a_dst, stride, topleft, w, h, ac, alpha
                              HIGHBD_TAIL_SUFFIX);
206 207 208 209 210
                }
            }
    report("cfl_pred");
}

211 212 213 214 215 216 217 218 219 220 221 222 223
static void check_pal_pred(Dav1dIntraPredDSPContext *const c) {
    ALIGN_STK_32(pixel, c_dst, 64 * 64,);
    ALIGN_STK_32(pixel, a_dst, 64 * 64,);
    ALIGN_STK_32(uint8_t, idx, 64 * 64,);
    ALIGN_STK_16(uint16_t, pal, 8,);

    declare_func(void, pixel *dst, ptrdiff_t stride, const uint16_t *pal,
                 const uint8_t *idx, int w, int h);

    for (int w = 4; w <= 64; w <<= 1)
        if (check_func(c->pal_pred, "pal_pred_w%d_%dbpc", w, BITDEPTH))
            for (int h = imax(w / 4, 4); h <= imin(w * 4, 64); h <<= 1)
            {
224 225 226 227 228
#if BITDEPTH == 16
                const int bitdepth_max = rand() & 1 ? 0x3ff : 0xfff;
#else
                const int bitdepth_max = 0xff;
#endif
229 230 231
                const ptrdiff_t stride = w * sizeof(pixel);

                for (int i = 0; i < 8; i++)
232
                    pal[i] = rand() & bitdepth_max;
233 234 235 236 237 238 239 240 241 242 243 244 245 246

                for (int i = 0; i < w * h; i++)
                    idx[i] = rand() & 7;

                call_ref(c_dst, stride, pal, idx, w, h);
                call_new(a_dst, stride, pal, idx, w, h);
                if (memcmp(c_dst, a_dst, w * h * sizeof(*c_dst)))
                    fail();

                bench_new(a_dst, stride, pal, idx, w, h);
            }
    report("pal_pred");
}

247 248 249 250 251
void bitfn(checkasm_check_ipred)(void) {
    Dav1dIntraPredDSPContext c;
    bitfn(dav1d_intra_pred_dsp_init)(&c);

    check_intra_pred(&c);
252
    check_cfl_ac(&c);
253
    check_cfl_pred(&c);
254
    check_pal_pred(&c);
255
}