Function ff_set_cmp

libavcodec/me_cmp.c:443–497

}

av_cold int ff_set_cmp(const MECmpContext *c, me_cmp_func *cmp, int type, int mpvenc)
{
#define ENTRY(CMP_FLAG, ARRAY, MPVENC_ONLY)          \
    [FF_CMP_ ## CMP_FLAG] = {                        \
        .offset    = offsetof(MECmpContext, ARRAY),  \
        .mpv_only  = MPVENC_ONLY,                    \
        .available = 1,                              \
    }
    static const struct {
        char available;
        char mpv_only;
        uint16_t offset;
    } cmp_func_list[] = {
        ENTRY(SAD,        sad,            0),
        ENTRY(SSE,        sse,            0),
        ENTRY(SATD,       hadamard8_diff, 0),
        ENTRY(DCT,        dct_sad,        1),
        ENTRY(PSNR,       quant_psnr,     1),
        ENTRY(BIT,        bit,            1),
        ENTRY(RD,         rd,             1),
        ENTRY(VSAD,       vsad,           0),
        ENTRY(VSSE,       vsse,           0),
        ENTRY(NSSE,       nsse,           0),
#if CONFIG_SNOW_DECODER || CONFIG_SNOW_ENCODER
        ENTRY(W53,        w53,            0),
        ENTRY(W97,        w97,            0),
#endif
        ENTRY(DCTMAX,     dct_max,        1),
#if CONFIG_GPL
        ENTRY(DCT264,     dct264_sad,     1),
#endif
        ENTRY(MEDIAN_SAD, median_sad,     0),
    };
    const me_cmp_func *me_cmp_func_array;

    type &= 0xFF;

    if (type == FF_CMP_ZERO) {
        for (int i = 0; i < 6; i++)
            cmp[i] = zero_cmp;
        return 0;
    }
    if (type >= FF_ARRAY_ELEMS(cmp_func_list) ||
        !cmp_func_list[type].available ||
        !mpvenc && cmp_func_list[type].mpv_only) {
        av_log(NULL, AV_LOG_ERROR,
               "invalid cmp function selection\n");
        return AVERROR(EINVAL);
    }
    me_cmp_func_array = (const me_cmp_func*)(((const char*)c) + cmp_func_list[type].offset);
    for (int i = 0; i < 6; i++)
        cmp[i] = me_cmp_func_array[i];

    return 0;
}

#define BUTTERFLY2(o1, o2, i1, i2)              \
    o1 = (i1) + (i2);                           \