mirror of https://github.com/FFmpeg/FFmpeg.git
Ramiro Polla
5 months ago
parent
f9074427db
commit
834964ce1a
5 changed files with 153 additions and 0 deletions
@ -0,0 +1,147 @@ |
||||
/*
|
||||
* This file is part of FFmpeg. |
||||
* |
||||
* FFmpeg is free software; you can redistribute it and/or |
||||
* modify it under the terms of the GNU Lesser General Public |
||||
* License as published by the Free Software Foundation; either |
||||
* version 2.1 of the License, or (at your option) any later version. |
||||
* |
||||
* FFmpeg is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
||||
* Lesser General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU Lesser General Public |
||||
* License along with FFmpeg; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
||||
*/ |
||||
|
||||
#include "libavutil/intreadwrite.h" |
||||
#include "libavutil/mem.h" |
||||
#include "libavutil/mem_internal.h" |
||||
|
||||
#include "libavcodec/mpegvideoencdsp.h" |
||||
|
||||
#include "checkasm.h" |
||||
|
||||
#define randomize_buffers(buf, size) \ |
||||
do { \
|
||||
for (int j = 0; j < size; j += 4) \
|
||||
AV_WN32(buf + j, rnd()); \
|
||||
} while (0) |
||||
|
||||
static void check_pix_sum(MpegvideoEncDSPContext *c) |
||||
{ |
||||
LOCAL_ALIGNED_16(uint8_t, src, [16 * 16]); |
||||
|
||||
declare_func(int, const uint8_t *pix, int line_size); |
||||
|
||||
randomize_buffers(src, 16 * 16); |
||||
|
||||
for (int n = 0; n < 2; n++) { |
||||
const char *negstride_str = n ? "_negstride" : ""; |
||||
if (check_func(c->pix_sum, "pix_sum%s", negstride_str)) { |
||||
int sum0, sum1; |
||||
const uint8_t *pix = src + (n ? (15 * 16) : 0); |
||||
int line_size = 16 * (n ? -1 : 1); |
||||
sum0 = call_ref(pix, line_size); |
||||
sum1 = call_new(pix, line_size); |
||||
if (sum0 != sum1) |
||||
fail(); |
||||
bench_new(pix, line_size); |
||||
} |
||||
} |
||||
} |
||||
|
||||
static void check_pix_norm1(MpegvideoEncDSPContext *c) |
||||
{ |
||||
LOCAL_ALIGNED_16(uint8_t, src, [16 * 16]); |
||||
|
||||
declare_func(int, const uint8_t *pix, int line_size); |
||||
|
||||
randomize_buffers(src, 16 * 16); |
||||
|
||||
for (int n = 0; n < 2; n++) { |
||||
const char *negstride_str = n ? "_negstride" : ""; |
||||
if (check_func(c->pix_norm1, "pix_norm1%s", negstride_str)) { |
||||
int sum0, sum1; |
||||
const uint8_t *pix = src + (n ? (15 * 16) : 0); |
||||
int line_size = 16 * (n ? -1 : 1); |
||||
sum0 = call_ref(pix, line_size); |
||||
sum1 = call_new(pix, line_size); |
||||
if (sum0 != sum1) |
||||
fail(); |
||||
bench_new(pix, line_size); |
||||
} |
||||
} |
||||
} |
||||
|
||||
#define NUM_LINES 4 |
||||
#define MAX_LINE_SIZE 1920 |
||||
#define EDGE_WIDTH 16 |
||||
#define LINESIZE (EDGE_WIDTH + MAX_LINE_SIZE + EDGE_WIDTH) |
||||
#define BUFSIZE ((EDGE_WIDTH + NUM_LINES + EDGE_WIDTH) * LINESIZE) |
||||
|
||||
static void check_draw_edges(MpegvideoEncDSPContext *c) |
||||
{ |
||||
static const int input_sizes[] = {8, 128, 1080, MAX_LINE_SIZE, -MAX_LINE_SIZE}; |
||||
LOCAL_ALIGNED_16(uint8_t, buf0, [BUFSIZE]); |
||||
LOCAL_ALIGNED_16(uint8_t, buf1, [BUFSIZE]); |
||||
|
||||
declare_func_emms(AV_CPU_FLAG_MMX, void, uint8_t *buf, int wrap, int width, int height, |
||||
int w, int h, int sides); |
||||
|
||||
for (int isi = 0; isi < FF_ARRAY_ELEMS(input_sizes); isi++) { |
||||
int input_size = input_sizes[isi]; |
||||
int negstride = input_size < 0; |
||||
const char *negstride_str = negstride ? "_negstride" : ""; |
||||
int width = FFABS(input_size); |
||||
int linesize = EDGE_WIDTH + width + EDGE_WIDTH; |
||||
/* calculate height based on specified width to use the entire buffer. */ |
||||
int height = (BUFSIZE / linesize) - (2 * EDGE_WIDTH); |
||||
uint8_t *dst0 = buf0 + EDGE_WIDTH * linesize + EDGE_WIDTH; |
||||
uint8_t *dst1 = buf1 + EDGE_WIDTH * linesize + EDGE_WIDTH; |
||||
|
||||
if (negstride) { |
||||
dst0 += (height - 1) * linesize; |
||||
dst1 += (height - 1) * linesize; |
||||
linesize *= -1; |
||||
} |
||||
|
||||
for (int shift = 0; shift < 3; shift++) { |
||||
int edge = EDGE_WIDTH >> shift; |
||||
if (check_func(c->draw_edges, "draw_edges_%d_%d_%d%s", width, height, edge, negstride_str)) { |
||||
randomize_buffers(buf0, BUFSIZE); |
||||
memcpy(buf1, buf0, BUFSIZE); |
||||
call_ref(dst0, linesize, width, height, edge, edge, EDGE_BOTTOM | EDGE_TOP); |
||||
call_new(dst1, linesize, width, height, edge, edge, EDGE_BOTTOM | EDGE_TOP); |
||||
if (memcmp(buf0, buf1, BUFSIZE)) |
||||
fail(); |
||||
bench_new(dst1, linesize, width, height, edge, edge, EDGE_BOTTOM | EDGE_TOP); |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
#undef NUM_LINES |
||||
#undef MAX_LINE_SIZE |
||||
#undef EDGE_WIDTH |
||||
#undef LINESIZE |
||||
#undef BUFSIZE |
||||
|
||||
void checkasm_check_mpegvideoencdsp(void) |
||||
{ |
||||
AVCodecContext avctx = { |
||||
.bits_per_raw_sample = 8, |
||||
}; |
||||
MpegvideoEncDSPContext c = { 0 }; |
||||
|
||||
ff_mpegvideoencdsp_init(&c, &avctx); |
||||
|
||||
check_pix_sum(&c); |
||||
report("pix_sum"); |
||||
check_pix_norm1(&c); |
||||
report("pix_norm1"); |
||||
check_draw_edges(&c); |
||||
report("draw_edges"); |
||||
} |
||||
Loading…
Reference in new issue