virtualenv/FFmpeg
1
0
Fork 0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2025-01-13 21:28:01 +02:00
Code Issues Releases Activity

swscale-test: take input from reference file

Browse Source 
If the CRC from the src->dst conversion matches a reference, it is not
necessary to perform a dst->yuva420p conversion and check the SSD.

Originally committed as revision 32213 to svn://svn.mplayerhq.hu/mplayer/trunk/libswscale
This commit is contained in:
Ramiro Polla 2010-09-12 18:16:33 +00:00
parent 6bbae4c8b3
commit 4bf4478598
1 changed files with 118 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

137
libswscale/swscale-test.c
View File

@ -71,11 +71,20 @@ static uint64_t getSSD(uint8_t *src1, uint8_t *src2, int stride1, int stride2, i
return ssd; return ssd;
} }
struct Results {
uint64_t ssdY;
uint64_t ssdU;
uint64_t ssdV;
uint64_t ssdA;
uint32_t crc;
};
// test by ref -> src -> dst -> out & compare out against ref // test by ref -> src -> dst -> out & compare out against ref
// ref & out are YV12 // ref & out are YV12
static int doTest(uint8_t *ref[4], int refStride[4], int w, int h, static int doTest(uint8_t *ref[4], int refStride[4], int w, int h,
enum PixelFormat srcFormat, enum PixelFormat dstFormat, enum PixelFormat srcFormat, enum PixelFormat dstFormat,
int srcW, int srcH, int dstW, int dstH, int flags) int srcW, int srcH, int dstW, int dstH, int flags,
struct Results *r)
{ {
static enum PixelFormat cur_srcFormat; static enum PixelFormat cur_srcFormat;
static int cur_srcW, cur_srcH; static int cur_srcW, cur_srcH;
@ -137,9 +146,7 @@ static int doTest(uint8_t *ref[4], int refStride[4], int w, int h,
* out of bounds. */ * out of bounds. */
if (dstStride[i]) if (dstStride[i])
dst[i]= av_mallocz(dstStride[i]*dstH+16); dst[i]= av_mallocz(dstStride[i]*dstH+16);
if (refStride[i]) if (dstStride[i] && !dst[i]) {
out[i]= av_mallocz(refStride[i]*h);
if ((dstStride[i] && !dst[i]) || (refStride[i] && !out[i])) {
perror("Malloc"); perror("Malloc");
res = -1; res = -1;
@ -154,15 +161,6 @@ static int doTest(uint8_t *ref[4], int refStride[4], int w, int h,
av_pix_fmt_descriptors[dstFormat].name); av_pix_fmt_descriptors[dstFormat].name);
res = -1; res = -1;
goto end;
}
outContext= sws_getContext(dstW, dstH, dstFormat, w, h, PIX_FMT_YUVA420P, flags, NULL, NULL, NULL);
if (!outContext) {
fprintf(stderr, "Failed to get %s ---> %s\n",
av_pix_fmt_descriptors[dstFormat].name,
av_pix_fmt_descriptors[PIX_FMT_YUVA420P].name);
res = -1;
goto end; goto end;
} }
// printf("test %X %X %X -> %X %X %X\n", (int)ref[0], (int)ref[1], (int)ref[2], // printf("test %X %X %X -> %X %X %X\n", (int)ref[0], (int)ref[1], (int)ref[2],
@ -175,12 +173,38 @@ static int doTest(uint8_t *ref[4], int refStride[4], int w, int h,
fflush(stdout); fflush(stdout);
sws_scale(dstContext, src, srcStride, 0, srcH, dst, dstStride); sws_scale(dstContext, src, srcStride, 0, srcH, dst, dstStride);
sws_scale(outContext, dst, dstStride, 0, dstH, out, refStride);
for (i = 0; i < 4 && dstStride[i]; i++) { for (i = 0; i < 4 && dstStride[i]; i++) {
crc = av_crc(av_crc_get_table(AV_CRC_32_IEEE), crc, dst[i], dstStride[i] * dstH); crc = av_crc(av_crc_get_table(AV_CRC_32_IEEE), crc, dst[i], dstStride[i] * dstH);
} }
if (r && crc == r->crc) {
ssdY = r->ssdY;
ssdU = r->ssdU;
ssdV = r->ssdV;
ssdA = r->ssdA;
} else {
for (i=0; i<4; i++) {
if (refStride[i])
out[i]= av_mallocz(refStride[i]*h);
if (refStride[i] && !out[i]) {
perror("Malloc");
res = -1;
goto end;
}
}
outContext= sws_getContext(dstW, dstH, dstFormat, w, h, PIX_FMT_YUVA420P, flags, NULL, NULL, NULL);
if (!outContext) {
fprintf(stderr, "Failed to get %s ---> %s\n",
av_pix_fmt_descriptors[dstFormat].name,
av_pix_fmt_descriptors[PIX_FMT_YUVA420P].name);
res = -1;
goto end;
}
sws_scale(outContext, dst, dstStride, 0, dstH, out, refStride);
ssdY= getSSD(ref[0], out[0], refStride[0], refStride[0], w, h); ssdY= getSSD(ref[0], out[0], refStride[0], refStride[0], w, h);
if (hasChroma(srcFormat) && hasChroma(dstFormat)) { if (hasChroma(srcFormat) && hasChroma(dstFormat)) {
//FIXME check that output is really gray //FIXME check that output is really gray
@ -195,19 +219,24 @@ static int doTest(uint8_t *ref[4], int refStride[4], int w, int h,
ssdV/= w*h/4; ssdV/= w*h/4;
ssdA/= w*h; ssdA/= w*h;
sws_freeContext(outContext);
for (i=0; i<4; i++) {
if (refStride[i])
av_free(out[i]);
}
}
printf(" CRC=%08x SSD=%5"PRId64",%5"PRId64",%5"PRId64",%5"PRId64"\n", printf(" CRC=%08x SSD=%5"PRId64",%5"PRId64",%5"PRId64",%5"PRId64"\n",
crc, ssdY, ssdU, ssdV, ssdA); crc, ssdY, ssdU, ssdV, ssdA);
end: end:
sws_freeContext(dstContext); sws_freeContext(dstContext);
sws_freeContext(outContext);
for (i=0; i<4; i++) { for (i=0; i<4; i++) {
if (dstStride[i]) if (dstStride[i])
av_free(dst[i]); av_free(dst[i]);
if (refStride[i])
av_free(out[i]);
} }
return res; return res;
@ -245,12 +274,58 @@ static void selfTest(uint8_t *ref[4], int refStride[4], int w, int h)
for (j = 0; dstH[j] && !res; j++) for (j = 0; dstH[j] && !res; j++)
res = doTest(ref, refStride, w, h, res = doTest(ref, refStride, w, h,
srcFormat, dstFormat, srcFormat, dstFormat,
srcW, srcH, dstW[i], dstH[j], flags[k]); srcW, srcH, dstW[i], dstH[j], flags[k],
NULL);
} }
} }
} }
} }
static int fileTest(uint8_t *ref[4], int refStride[4], int w, int h, FILE *fp)
{
char buf[256];
while (fgets(buf, sizeof(buf), fp)) {
struct Results r;
enum PixelFormat srcFormat;
char srcStr[12];
int srcW, srcH;
enum PixelFormat dstFormat;
char dstStr[12];
int dstW, dstH;
int flags;
int ret;
ret = sscanf(buf, " %12s %dx%d -> %12s %dx%d flags=%d CRC=%x"
" SSD=%"PRId64", %"PRId64", %"PRId64", %"PRId64"\n",
srcStr, &srcW, &srcH, dstStr, &dstW, &dstH,
&flags, &r.crc, &r.ssdY, &r.ssdU, &r.ssdV, &r.ssdA);
if (ret != 12) {
srcStr[0] = dstStr[0] = 0;
ret = sscanf(buf, "%12s -> %12s\n", srcStr, dstStr);
}
srcFormat = av_get_pix_fmt(srcStr);
dstFormat = av_get_pix_fmt(dstStr);
if (srcFormat == PIX_FMT_NONE || dstFormat == PIX_FMT_NONE) {
fprintf(stderr, "malformed input file\n");
return -1;
}
if (ret != 12) {
printf("%s", buf);
continue;
}
doTest(ref, refStride, w, h,
srcFormat, dstFormat,
srcW, srcH, dstW, dstH, flags,
&r);
}
return 0;
}
#define W 96 #define W 96
#define H 96 #define H 96
@ -265,6 +340,8 @@ int main(int argc, char **argv)
int x, y; int x, y;
struct SwsContext *sws; struct SwsContext *sws;
AVLFG rand; AVLFG rand;
int res = -1;
int i;
if (!rgb_data || !data) if (!rgb_data || !data)
return -1; return -1;
@ -282,8 +359,30 @@ int main(int argc, char **argv)
sws_freeContext(sws); sws_freeContext(sws);
av_free(rgb_data); av_free(rgb_data);
for (i = 1; i < argc; i += 2) {
if (argv[i][0] != '-' || i+1 == argc)
goto bad_option;
if (!strcmp(argv[i], "-ref")) {
FILE *fp = fopen(argv[i+1], "r");
if (!fp) {
fprintf(stderr, "could not open '%s'\n", argv[i+1]);
goto error;
}
res = fileTest(src, stride, W, H, fp);
fclose(fp);
goto end;
} else {
bad_option:
fprintf(stderr, "bad option or argument missing (%s)\n", argv[i]);
goto error;
}
}
selfTest(src, stride, W, H); selfTest(src, stride, W, H);
end:
res = 0;
error:
av_free(data); av_free(data);
return 0; return res;
} }