2 * Copyright (C) 2003 Michael Niedermayer <michaelni@gmx.at>
4 * This file is part of FFmpeg.
6 * FFmpeg is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * FFmpeg is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with FFmpeg; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
27 #undef HAVE_AV_CONFIG_H
28 #include "libavutil/avutil.h"
29 #include "libavutil/lfg.h"
31 #include "swscale_internal.h"
39 static uint64_t getSSD(uint8_t *src1
, uint8_t *src2
, int stride1
, int stride2
, int w
, int h
){
43 //printf("%d %d\n", w, h);
47 int d
= src1
[x
+ y
*stride1
] - src2
[x
+ y
*stride2
];
49 //printf("%d", abs(src1[x + y*stride1] - src2[x + y*stride2])/26 );
56 // test by ref -> src -> dst -> out & compare out against ref
58 static int doTest(uint8_t *ref
[4], int refStride
[4], int w
, int h
, int srcFormat
, int dstFormat
,
59 int srcW
, int srcH
, int dstW
, int dstH
, int flags
){
60 uint8_t *src
[4] = {0};
61 uint8_t *dst
[4] = {0};
62 uint8_t *out
[4] = {0};
63 int srcStride
[4], dstStride
[4];
65 uint64_t ssdY
, ssdU
, ssdV
, ssdA
=0;
66 struct SwsContext
*srcContext
= NULL
, *dstContext
= NULL
,
72 // avoid stride % bpp != 0
73 if (srcFormat
==PIX_FMT_RGB24
|| srcFormat
==PIX_FMT_BGR24
)
75 else if (srcFormat
==PIX_FMT_RGB48BE
|| srcFormat
==PIX_FMT_RGB48LE
)
80 if (dstFormat
==PIX_FMT_RGB24
|| dstFormat
==PIX_FMT_BGR24
)
82 else if (dstFormat
==PIX_FMT_RGB48BE
|| dstFormat
==PIX_FMT_RGB48LE
)
87 src
[i
]= (uint8_t*) malloc(srcStride
[i
]*srcH
);
88 dst
[i
]= (uint8_t*) malloc(dstStride
[i
]*dstH
);
89 out
[i
]= (uint8_t*) malloc(refStride
[i
]*h
);
90 if (!src
[i
] || !dst
[i
] || !out
[i
]) {
98 srcContext
= sws_getContext(w
, h
, PIX_FMT_YUVA420P
, srcW
, srcH
, srcFormat
, flags
, NULL
, NULL
, NULL
);
100 fprintf(stderr
, "Failed to get %s ---> %s\n",
101 sws_format_name(PIX_FMT_YUVA420P
),
102 sws_format_name(srcFormat
));
107 dstContext
= sws_getContext(srcW
, srcH
, srcFormat
, dstW
, dstH
, dstFormat
, flags
, NULL
, NULL
, NULL
);
109 fprintf(stderr
, "Failed to get %s ---> %s\n",
110 sws_format_name(srcFormat
),
111 sws_format_name(dstFormat
));
116 outContext
= sws_getContext(dstW
, dstH
, dstFormat
, w
, h
, PIX_FMT_YUVA420P
, flags
, NULL
, NULL
, NULL
);
118 fprintf(stderr
, "Failed to get %s ---> %s\n",
119 sws_format_name(dstFormat
),
120 sws_format_name(PIX_FMT_YUVA420P
));
125 // printf("test %X %X %X -> %X %X %X\n", (int)ref[0], (int)ref[1], (int)ref[2],
126 // (int)src[0], (int)src[1], (int)src[2]);
128 sws_scale(srcContext
, ref
, refStride
, 0, h
, src
, srcStride
);
129 sws_scale(dstContext
, src
, srcStride
, 0, srcH
, dst
, dstStride
);
130 sws_scale(outContext
, dst
, dstStride
, 0, dstH
, out
, refStride
);
132 ssdY
= getSSD(ref
[0], out
[0], refStride
[0], refStride
[0], w
, h
);
133 ssdU
= getSSD(ref
[1], out
[1], refStride
[1], refStride
[1], (w
+1)>>1, (h
+1)>>1);
134 ssdV
= getSSD(ref
[2], out
[2], refStride
[2], refStride
[2], (w
+1)>>1, (h
+1)>>1);
135 if (isALPHA(srcFormat
) && isALPHA(dstFormat
))
136 ssdA
= getSSD(ref
[3], out
[3], refStride
[3], refStride
[3], w
, h
);
138 if (srcFormat
== PIX_FMT_GRAY8
|| dstFormat
==PIX_FMT_GRAY8
) ssdU
=ssdV
=0; //FIXME check that output is really gray
145 printf(" %s %dx%d -> %s %4dx%4d flags=%2d SSD=%5"PRId64
",%5"PRId64
",%5"PRId64
",%5"PRId64
"\n",
146 sws_format_name(srcFormat
), srcW
, srcH
,
147 sws_format_name(dstFormat
), dstW
, dstH
,
148 flags
, ssdY
, ssdU
, ssdV
, ssdA
);
153 sws_freeContext(srcContext
);
154 sws_freeContext(dstContext
);
155 sws_freeContext(outContext
);
166 static void selfTest(uint8_t *src
[4], int stride
[4], int w
, int h
){
167 enum PixelFormat srcFormat
, dstFormat
;
168 int srcW
, srcH
, dstW
, dstH
;
171 for (srcFormat
= 0; srcFormat
< PIX_FMT_NB
; srcFormat
++) {
172 for (dstFormat
= 0; dstFormat
< PIX_FMT_NB
; dstFormat
++) {
174 sws_format_name(srcFormat
),
175 sws_format_name(dstFormat
));
180 for (dstW
=w
- w
/3; dstW
<= 4*w
/3; dstW
+= w
/3){
181 for (dstH
=h
- h
/3; dstH
<= 4*h
/3; dstH
+= h
/3){
182 for (flags
=1; flags
<33; flags
*=2) {
185 res
= doTest(src
, stride
, w
, h
, srcFormat
, dstFormat
,
186 srcW
, srcH
, dstW
, dstH
, flags
);
202 int main(int argc
, char **argv
){
203 uint8_t *rgb_data
= malloc (W
*H
*4);
204 uint8_t *rgb_src
[3]= {rgb_data
, NULL
, NULL
};
205 int rgb_stride
[3]={4*W
, 0, 0};
206 uint8_t *data
= malloc (4*W
*H
);
207 uint8_t *src
[4]= {data
, data
+W
*H
, data
+W
*H
*2, data
+W
*H
*3};
208 int stride
[4]={W
, W
, W
, W
};
210 struct SwsContext
*sws
;
213 sws
= sws_getContext(W
/12, H
/12, PIX_FMT_RGB32
, W
, H
, PIX_FMT_YUVA420P
, 2, NULL
, NULL
, NULL
);
215 av_lfg_init(&rand
, 1);
218 for (x
=0; x
<W
*4; x
++){
219 rgb_data
[ x
+ y
*4*W
]= av_lfg_get(&rand
);
222 sws_scale(sws
, rgb_src
, rgb_stride
, 0, H
, src
, stride
);
224 selfTest(src
, stride
, W
, H
);