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"
32 /* HACK Duplicated from swscale_internal.h.
33 * Should be removed when a cleaner pixel format system exists. */
34 const char *sws_format_name(enum PixelFormat format
);
37 || (x)==PIX_FMT_GRAY16BE \
38 || (x)==PIX_FMT_GRAY16LE \
40 #define hasChroma(x) (!( \
42 || (x)==PIX_FMT_MONOBLACK \
43 || (x)==PIX_FMT_MONOWHITE \
45 #define isALPHA(x) ( \
47 || (x)==PIX_FMT_BGR32_1 \
48 || (x)==PIX_FMT_RGB32 \
49 || (x)==PIX_FMT_RGB32_1 \
50 || (x)==PIX_FMT_YUVA420P \
53 static uint64_t getSSD(uint8_t *src1
, uint8_t *src2
, int stride1
, int stride2
, int w
, int h
)
58 //printf("%d %d\n", w, h);
62 int d
= src1
[x
+ y
*stride1
] - src2
[x
+ y
*stride2
];
64 //printf("%d", abs(src1[x + y*stride1] - src2[x + y*stride2])/26 );
71 // test by ref -> src -> dst -> out & compare out against ref
73 static int doTest(uint8_t *ref
[4], int refStride
[4], int w
, int h
, int srcFormat
, int dstFormat
,
74 int srcW
, int srcH
, int dstW
, int dstH
, int flags
)
76 uint8_t *src
[4] = {0};
77 uint8_t *dst
[4] = {0};
78 uint8_t *out
[4] = {0};
79 int srcStride
[4], dstStride
[4];
81 uint64_t ssdY
, ssdU
=0, ssdV
=0, ssdA
=0;
82 struct SwsContext
*srcContext
= NULL
, *dstContext
= NULL
,
88 // avoid stride % bpp != 0
89 if (srcFormat
==PIX_FMT_RGB24
|| srcFormat
==PIX_FMT_BGR24
)
91 else if (srcFormat
==PIX_FMT_RGB48BE
|| srcFormat
==PIX_FMT_RGB48LE
)
96 if (dstFormat
==PIX_FMT_RGB24
|| dstFormat
==PIX_FMT_BGR24
)
98 else if (dstFormat
==PIX_FMT_RGB48BE
|| dstFormat
==PIX_FMT_RGB48LE
)
101 dstStride
[i
]= dstW
*4;
103 src
[i
]= malloc(srcStride
[i
]*srcH
);
104 dst
[i
]= malloc(dstStride
[i
]*dstH
);
105 out
[i
]= malloc(refStride
[i
]*h
);
106 if (!src
[i
] || !dst
[i
] || !out
[i
]) {
114 srcContext
= sws_getContext(w
, h
, PIX_FMT_YUVA420P
, srcW
, srcH
, srcFormat
, flags
, NULL
, NULL
, NULL
);
116 fprintf(stderr
, "Failed to get %s ---> %s\n",
117 sws_format_name(PIX_FMT_YUVA420P
),
118 sws_format_name(srcFormat
));
123 dstContext
= sws_getContext(srcW
, srcH
, srcFormat
, dstW
, dstH
, dstFormat
, flags
, NULL
, NULL
, NULL
);
125 fprintf(stderr
, "Failed to get %s ---> %s\n",
126 sws_format_name(srcFormat
),
127 sws_format_name(dstFormat
));
132 outContext
= sws_getContext(dstW
, dstH
, dstFormat
, w
, h
, PIX_FMT_YUVA420P
, flags
, NULL
, NULL
, NULL
);
134 fprintf(stderr
, "Failed to get %s ---> %s\n",
135 sws_format_name(dstFormat
),
136 sws_format_name(PIX_FMT_YUVA420P
));
141 // printf("test %X %X %X -> %X %X %X\n", (int)ref[0], (int)ref[1], (int)ref[2],
142 // (int)src[0], (int)src[1], (int)src[2]);
144 sws_scale(srcContext
, ref
, refStride
, 0, h
, src
, srcStride
);
145 sws_scale(dstContext
, src
, srcStride
, 0, srcH
, dst
, dstStride
);
146 sws_scale(outContext
, dst
, dstStride
, 0, dstH
, out
, refStride
);
148 ssdY
= getSSD(ref
[0], out
[0], refStride
[0], refStride
[0], w
, h
);
149 if (hasChroma(srcFormat
) && hasChroma(dstFormat
)) {
150 //FIXME check that output is really gray
151 ssdU
= getSSD(ref
[1], out
[1], refStride
[1], refStride
[1], (w
+1)>>1, (h
+1)>>1);
152 ssdV
= getSSD(ref
[2], out
[2], refStride
[2], refStride
[2], (w
+1)>>1, (h
+1)>>1);
154 if (isALPHA(srcFormat
) && isALPHA(dstFormat
))
155 ssdA
= getSSD(ref
[3], out
[3], refStride
[3], refStride
[3], w
, h
);
162 printf(" %s %dx%d -> %s %4dx%4d flags=%2d SSD=%5"PRId64
",%5"PRId64
",%5"PRId64
",%5"PRId64
"\n",
163 sws_format_name(srcFormat
), srcW
, srcH
,
164 sws_format_name(dstFormat
), dstW
, dstH
,
165 flags
, ssdY
, ssdU
, ssdV
, ssdA
);
170 sws_freeContext(srcContext
);
171 sws_freeContext(dstContext
);
172 sws_freeContext(outContext
);
174 for (i
=0; i
<4; i
++) {
183 static void selfTest(uint8_t *ref
[4], int refStride
[4], int w
, int h
)
185 const int flags
[] = { SWS_FAST_BILINEAR
,
186 SWS_BILINEAR
, SWS_BICUBIC
,
187 SWS_X
, SWS_POINT
, SWS_AREA
, 0 };
190 const int dstW
[] = { srcW
- srcW
/3, srcW
, srcW
+ srcW
/3, 0 };
191 const int dstH
[] = { srcH
- srcH
/3, srcH
, srcH
+ srcH
/3, 0 };
192 enum PixelFormat srcFormat
, dstFormat
;
194 for (srcFormat
= 0; srcFormat
< PIX_FMT_NB
; srcFormat
++) {
195 for (dstFormat
= 0; dstFormat
< PIX_FMT_NB
; dstFormat
++) {
200 sws_format_name(srcFormat
),
201 sws_format_name(dstFormat
));
204 for (i
= 0; dstW
[i
] && !res
; i
++)
205 for (j
= 0; dstH
[j
] && !res
; j
++)
206 for (k
= 0; flags
[k
] && !res
; k
++)
207 res
= doTest(ref
, refStride
, w
, h
, srcFormat
, dstFormat
,
208 srcW
, srcH
, dstW
[i
], dstH
[j
], flags
[k
]);
216 int main(int argc
, char **argv
)
218 uint8_t *rgb_data
= malloc (W
*H
*4);
219 uint8_t *rgb_src
[3]= {rgb_data
, NULL
, NULL
};
220 int rgb_stride
[3]={4*W
, 0, 0};
221 uint8_t *data
= malloc (4*W
*H
);
222 uint8_t *src
[4]= {data
, data
+W
*H
, data
+W
*H
*2, data
+W
*H
*3};
223 int stride
[4]={W
, W
, W
, W
};
225 struct SwsContext
*sws
;
228 if (!rgb_data
|| !data
)
231 sws
= sws_getContext(W
/12, H
/12, PIX_FMT_RGB32
, W
, H
, PIX_FMT_YUVA420P
, SWS_BILINEAR
, NULL
, NULL
, NULL
);
233 av_lfg_init(&rand
, 1);
235 for (y
=0; y
<H
; y
++) {
236 for (x
=0; x
<W
*4; x
++) {
237 rgb_data
[ x
+ y
*4*W
]= av_lfg_get(&rand
);
240 sws_scale(sws
, rgb_src
, rgb_stride
, 0, H
, src
, stride
);
241 sws_freeContext(sws
);
244 selfTest(src
, stride
, W
, H
);