mirror of https://git.ffmpeg.org/ffmpeg.git
checkasm/h264: add loop filter tests
This commit is contained in:
Janne Grunau
parent
bb515e3a73
commit
d7f4f5c4a1
|
|
@ -28,6 +28,7 @@
|
||||||
#include "libavutil/intreadwrite.h"
|
#include "libavutil/intreadwrite.h"
|
||||||
|
|
||||||
static const uint32_t pixel_mask[3] = { 0xffffffff, 0x01ff01ff, 0x03ff03ff };
|
static const uint32_t pixel_mask[3] = { 0xffffffff, 0x01ff01ff, 0x03ff03ff };
|
||||||
|
static const uint32_t pixel_mask_lf[3] = { 0xff0fff0f, 0x01ff000f, 0x03ff000f };
|
||||||
|
|
||||||
#define SIZEOF_PIXEL ((bit_depth + 7) / 8)
|
#define SIZEOF_PIXEL ((bit_depth + 7) / 8)
|
||||||
#define SIZEOF_COEF (2 * ((bit_depth + 7) / 8))
|
#define SIZEOF_COEF (2 * ((bit_depth + 7) / 8))
|
||||||
|
|
@ -312,9 +313,132 @@ static void check_idct_multiple(void)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
static void check_loop_filter(void)
|
||||||
|
{
|
||||||
|
LOCAL_ALIGNED_16(uint8_t, dst, [32 * 16 * 2]);
|
||||||
|
LOCAL_ALIGNED_16(uint8_t, dst0, [32 * 16 * 2]);
|
||||||
|
LOCAL_ALIGNED_16(uint8_t, dst1, [32 * 16 * 2]);
|
||||||
|
H264DSPContext h;
|
||||||
|
int bit_depth;
|
||||||
|
int alphas[36], betas[36];
|
||||||
|
int8_t tc0[36][4];
|
||||||
|
|
||||||
|
declare_func_emms(AV_CPU_FLAG_MMX, void, uint8_t *pix, int stride,
|
||||||
|
int alpha, int beta, int8_t *tc0);
|
||||||
|
|
||||||
|
for (bit_depth = 8; bit_depth <= 10; bit_depth++) {
|
||||||
|
int i, j, a, c;
|
||||||
|
uint32_t mask = pixel_mask_lf[bit_depth - 8];
|
||||||
|
ff_h264dsp_init(&h, bit_depth, 1);
|
||||||
|
for (i = 35, a = 255, c = 250; i >= 0; i--) {
|
||||||
|
alphas[i] = a << (bit_depth - 8);
|
||||||
|
betas[i] = (i + 1) / 2 << (bit_depth - 8);
|
||||||
|
tc0[i][0] = tc0[i][3] = (c + 6) / 10;
|
||||||
|
tc0[i][1] = (c + 7) / 15;
|
||||||
|
tc0[i][2] = (c + 9) / 20;
|
||||||
|
a = a*9/10;
|
||||||
|
c = c*9/10;
|
||||||
|
}
|
||||||
|
|
||||||
|
#define CHECK_LOOP_FILTER(name, align, ...) \
|
||||||
|
do { \
|
||||||
|
if (check_func(h.name, #name "_%dbpp", bit_depth)) { \
|
||||||
|
for (j = 0; j < 36; j++) { \
|
||||||
|
intptr_t off = 8 * 32 + (j & 15) * 4 * !align; \
|
||||||
|
for (i = 0; i < 1024; i+=4) { \
|
||||||
|
AV_WN32A(dst + i, rnd() & mask); \
|
||||||
|
} \
|
||||||
|
memcpy(dst0, dst, 32 * 16 * 2); \
|
||||||
|
memcpy(dst1, dst, 32 * 16 * 2); \
|
||||||
|
\
|
||||||
|
call_ref(dst0 + off, 32, alphas[j], betas[j], tc0[j]); \
|
||||||
|
call_new(dst1 + off, 32, alphas[j], betas[j], tc0[j]); \
|
||||||
|
if (memcmp(dst0, dst1, 32 * 16 * SIZEOF_PIXEL)) { \
|
||||||
|
fprintf(stderr, #name ": j:%d, alpha:%d beta:%d " \
|
||||||
|
"tc0:{%d,%d,%d,%d}\n", j, alphas[j], betas[j], \
|
||||||
|
tc0[j][0], tc0[j][1], tc0[j][2], tc0[j][3]); \
|
||||||
|
fail(); \
|
||||||
|
} \
|
||||||
|
bench_new(dst1, 32, alphas[j], betas[j], tc0[j]); \
|
||||||
|
} \
|
||||||
|
} \
|
||||||
|
} while (0)
|
||||||
|
|
||||||
|
CHECK_LOOP_FILTER(h264_v_loop_filter_luma, 1);
|
||||||
|
CHECK_LOOP_FILTER(h264_h_loop_filter_luma, 0);
|
||||||
|
CHECK_LOOP_FILTER(h264_h_loop_filter_luma_mbaff, 0);
|
||||||
|
CHECK_LOOP_FILTER(h264_v_loop_filter_chroma, 1);
|
||||||
|
CHECK_LOOP_FILTER(h264_h_loop_filter_chroma, 0);
|
||||||
|
CHECK_LOOP_FILTER(h264_h_loop_filter_chroma_mbaff, 0);
|
||||||
|
#undef CHECK_LOOP_FILTER
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static void check_loop_filter_intra(void)
|
||||||
|
{
|
||||||
|
LOCAL_ALIGNED_16(uint8_t, dst, [32 * 16 * 2]);
|
||||||
|
LOCAL_ALIGNED_16(uint8_t, dst0, [32 * 16 * 2]);
|
||||||
|
LOCAL_ALIGNED_16(uint8_t, dst1, [32 * 16 * 2]);
|
||||||
|
H264DSPContext h;
|
||||||
|
int bit_depth;
|
||||||
|
int alphas[36], betas[36];
|
||||||
|
|
||||||
|
declare_func_emms(AV_CPU_FLAG_MMX, void, uint8_t *pix, int stride,
|
||||||
|
int alpha, int beta);
|
||||||
|
|
||||||
|
for (bit_depth = 8; bit_depth <= 10; bit_depth++) {
|
||||||
|
int i, j, a;
|
||||||
|
uint32_t mask = pixel_mask_lf[bit_depth - 8];
|
||||||
|
ff_h264dsp_init(&h, bit_depth, 1);
|
||||||
|
for (i = 35, a = 255; i >= 0; i--) {
|
||||||
|
alphas[i] = a << (bit_depth - 8);
|
||||||
|
betas[i] = (i + 1) / 2 << (bit_depth - 8);
|
||||||
|
a = a*9/10;
|
||||||
|
}
|
||||||
|
|
||||||
|
#define CHECK_LOOP_FILTER(name, align) \
|
||||||
|
do { \
|
||||||
|
if (check_func(h.name, #name "_%dbpp", bit_depth)) { \
|
||||||
|
for (j = 0; j < 36; j++) { \
|
||||||
|
intptr_t off = 8 * 32 + (j & 15) * 4 * !align; \
|
||||||
|
for (i = 0; i < 1024; i+=4) { \
|
||||||
|
AV_WN32A(dst + i, rnd() & mask); \
|
||||||
|
} \
|
||||||
|
memcpy(dst0, dst, 32 * 16 * 2); \
|
||||||
|
memcpy(dst1, dst, 32 * 16 * 2); \
|
||||||
|
\
|
||||||
|
call_ref(dst0 + off, 32, alphas[j], betas[j]); \
|
||||||
|
call_new(dst1 + off, 32, alphas[j], betas[j]); \
|
||||||
|
if (memcmp(dst0, dst1, 32 * 16 * SIZEOF_PIXEL)) { \
|
||||||
|
fprintf(stderr, #name ": j:%d, alpha:%d beta:%d\n", \
|
||||||
|
j, alphas[j], betas[j]); \
|
||||||
|
fail(); \
|
||||||
|
} \
|
||||||
|
bench_new(dst1, 32, alphas[j], betas[j]); \
|
||||||
|
} \
|
||||||
|
} \
|
||||||
|
} while (0)
|
||||||
|
|
||||||
|
CHECK_LOOP_FILTER(h264_v_loop_filter_luma_intra, 1);
|
||||||
|
CHECK_LOOP_FILTER(h264_h_loop_filter_luma_intra, 0);
|
||||||
|
CHECK_LOOP_FILTER(h264_h_loop_filter_luma_mbaff_intra, 0);
|
||||||
|
CHECK_LOOP_FILTER(h264_v_loop_filter_chroma_intra, 1);
|
||||||
|
CHECK_LOOP_FILTER(h264_h_loop_filter_chroma_intra, 0);
|
||||||
|
CHECK_LOOP_FILTER(h264_h_loop_filter_chroma_mbaff_intra, 0);
|
||||||
|
#undef CHECK_LOOP_FILTER
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
void checkasm_check_h264dsp(void)
|
void checkasm_check_h264dsp(void)
|
||||||
{
|
{
|
||||||
check_idct();
|
check_idct();
|
||||||
check_idct_multiple();
|
check_idct_multiple();
|
||||||
report("idct");
|
report("idct");
|
||||||
|
|
||||||
|
check_loop_filter();
|
||||||
|
report("loop_filter");
|
||||||
|
|
||||||
|
check_loop_filter_intra();
|
||||||
|
report("loop_filter_intra");
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue