Mercurial > libavcodec.hg
annotate imgresample.c @ 6612:80cadb0595fb libavcodec
simplify
author | michael |
---|---|
date | Wed, 16 Apr 2008 15:03:07 +0000 |
parents | 0c3cc1d7a0b3 |
children | 7829c158c1c4 |
rev | line source |
---|---|
0 | 1 /* |
2967 | 2 * High quality image resampling with polyphase filters |
429 | 3 * Copyright (c) 2001 Fabrice Bellard. |
0 | 4 * |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3589
diff
changeset
|
5 * This file is part of FFmpeg. |
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3589
diff
changeset
|
6 * |
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3589
diff
changeset
|
7 * FFmpeg is free software; you can redistribute it and/or |
429 | 8 * modify it under the terms of the GNU Lesser General Public |
9 * License as published by the Free Software Foundation; either | |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3589
diff
changeset
|
10 * version 2.1 of the License, or (at your option) any later version. |
0 | 11 * |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3589
diff
changeset
|
12 * FFmpeg is distributed in the hope that it will be useful, |
0 | 13 * but WITHOUT ANY WARRANTY; without even the implied warranty of |
429 | 14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
15 * Lesser General Public License for more details. | |
0 | 16 * |
429 | 17 * You should have received a copy of the GNU Lesser General Public |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3589
diff
changeset
|
18 * License along with FFmpeg; if not, write to the Free Software |
3036
0b546eab515d
Update licensing information: The FSF changed postal address.
diego
parents:
2979
diff
changeset
|
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
0 | 20 */ |
2967 | 21 |
1106 | 22 /** |
23 * @file imgresample.c | |
24 * High quality image resampling with polyphase filters . | |
25 */ | |
2967 | 26 |
396
fce0a2520551
removed useless header includes - use av memory functions
glantau
parents:
18
diff
changeset
|
27 #include "avcodec.h" |
3249 | 28 #include "swscale.h" |
0 | 29 #include "dsputil.h" |
30 | |
5750
09f99af1db40
Sanitize altivec code so it can be built with runtime check properly
lu_zero
parents:
5187
diff
changeset
|
31 #ifdef HAVE_ALTIVEC |
09f99af1db40
Sanitize altivec code so it can be built with runtime check properly
lu_zero
parents:
5187
diff
changeset
|
32 #include "ppc/imgresample_altivec.h" |
09f99af1db40
Sanitize altivec code so it can be built with runtime check properly
lu_zero
parents:
5187
diff
changeset
|
33 #endif |
09f99af1db40
Sanitize altivec code so it can be built with runtime check properly
lu_zero
parents:
5187
diff
changeset
|
34 |
0 | 35 #define NB_COMPONENTS 3 |
36 | |
37 #define PHASE_BITS 4 | |
38 #define NB_PHASES (1 << PHASE_BITS) | |
39 #define NB_TAPS 4 | |
40 #define FCENTER 1 /* index of the center of the filter */ | |
630
b4ee42142ad1
croping patch by (talus25 at speakeasy dot net) with fixes from atmos & me
michaelni
parents:
429
diff
changeset
|
41 //#define TEST 1 /* Test it */ |
0 | 42 |
43 #define POS_FRAC_BITS 16 | |
44 #define POS_FRAC (1 << POS_FRAC_BITS) | |
45 /* 6 bits precision is needed for MMX */ | |
46 #define FILTER_BITS 8 | |
47 | |
48 #define LINE_BUF_HEIGHT (NB_TAPS * 4) | |
49 | |
4065
93163e2a2398
Do not use a fake libavcodec/swscale.h, but always use the real one
lucabe
parents:
4036
diff
changeset
|
50 struct SwsContext { |
6464
0c3cc1d7a0b3
Make av_class a pointer to a const AVClass. Addresses one warning in
takis
parents:
6434
diff
changeset
|
51 const AVClass *av_class; |
4065
93163e2a2398
Do not use a fake libavcodec/swscale.h, but always use the real one
lucabe
parents:
4036
diff
changeset
|
52 struct ImgReSampleContext *resampling_ctx; |
93163e2a2398
Do not use a fake libavcodec/swscale.h, but always use the real one
lucabe
parents:
4036
diff
changeset
|
53 enum PixelFormat src_pix_fmt, dst_pix_fmt; |
93163e2a2398
Do not use a fake libavcodec/swscale.h, but always use the real one
lucabe
parents:
4036
diff
changeset
|
54 }; |
93163e2a2398
Do not use a fake libavcodec/swscale.h, but always use the real one
lucabe
parents:
4036
diff
changeset
|
55 |
0 | 56 struct ImgReSampleContext { |
1928
0c23a5564489
padding support in ffmpeg patch by (Todd Kirby <doubleshot at pacbell dot net>)
michael
parents:
1488
diff
changeset
|
57 int iwidth, iheight, owidth, oheight; |
0c23a5564489
padding support in ffmpeg patch by (Todd Kirby <doubleshot at pacbell dot net>)
michael
parents:
1488
diff
changeset
|
58 int topBand, bottomBand, leftBand, rightBand; |
0c23a5564489
padding support in ffmpeg patch by (Todd Kirby <doubleshot at pacbell dot net>)
michael
parents:
1488
diff
changeset
|
59 int padtop, padbottom, padleft, padright; |
0c23a5564489
padding support in ffmpeg patch by (Todd Kirby <doubleshot at pacbell dot net>)
michael
parents:
1488
diff
changeset
|
60 int pad_owidth, pad_oheight; |
0 | 61 int h_incr, v_incr; |
3089 | 62 DECLARE_ALIGNED_8(int16_t, h_filters[NB_PHASES][NB_TAPS]); /* horizontal filters */ |
63 DECLARE_ALIGNED_8(int16_t, v_filters[NB_PHASES][NB_TAPS]); /* vertical filters */ | |
1064 | 64 uint8_t *line_buf; |
0 | 65 }; |
66 | |
2082
3dc9bbe1b152
polyphase kaiser windowed sinc and blackman nuttall windowed sinc audio resample filters
michael
parents:
2064
diff
changeset
|
67 void av_build_filter(int16_t *filter, double factor, int tap_count, int phase_count, int scale, int type); |
3dc9bbe1b152
polyphase kaiser windowed sinc and blackman nuttall windowed sinc audio resample filters
michael
parents:
2064
diff
changeset
|
68 |
0 | 69 static inline int get_phase(int pos) |
70 { | |
71 return ((pos) >> (POS_FRAC_BITS - PHASE_BITS)) & ((1 << PHASE_BITS) - 1); | |
72 } | |
73 | |
74 /* This function must be optimized */ | |
1488
766a2f4edbea
avcodec const correctness patch by (Drew Hess <dhess at ilm dot com>)
michaelni
parents:
1106
diff
changeset
|
75 static void h_resample_fast(uint8_t *dst, int dst_width, const uint8_t *src, |
2979 | 76 int src_width, int src_start, int src_incr, |
77 int16_t *filters) | |
0 | 78 { |
79 int src_pos, phase, sum, i; | |
1488
766a2f4edbea
avcodec const correctness patch by (Drew Hess <dhess at ilm dot com>)
michaelni
parents:
1106
diff
changeset
|
80 const uint8_t *s; |
1064 | 81 int16_t *filter; |
0 | 82 |
83 src_pos = src_start; | |
84 for(i=0;i<dst_width;i++) { | |
85 #ifdef TEST | |
86 /* test */ | |
87 if ((src_pos >> POS_FRAC_BITS) < 0 || | |
88 (src_pos >> POS_FRAC_BITS) > (src_width - NB_TAPS)) | |
653 | 89 av_abort(); |
0 | 90 #endif |
91 s = src + (src_pos >> POS_FRAC_BITS); | |
92 phase = get_phase(src_pos); | |
93 filter = filters + phase * NB_TAPS; | |
94 #if NB_TAPS == 4 | |
95 sum = s[0] * filter[0] + | |
96 s[1] * filter[1] + | |
97 s[2] * filter[2] + | |
98 s[3] * filter[3]; | |
99 #else | |
100 { | |
101 int j; | |
102 sum = 0; | |
103 for(j=0;j<NB_TAPS;j++) | |
104 sum += s[j] * filter[j]; | |
105 } | |
106 #endif | |
107 sum = sum >> FILTER_BITS; | |
108 if (sum < 0) | |
109 sum = 0; | |
110 else if (sum > 255) | |
111 sum = 255; | |
112 dst[0] = sum; | |
113 src_pos += src_incr; | |
114 dst++; | |
115 } | |
116 } | |
117 | |
118 /* This function must be optimized */ | |
1488
766a2f4edbea
avcodec const correctness patch by (Drew Hess <dhess at ilm dot com>)
michaelni
parents:
1106
diff
changeset
|
119 static void v_resample(uint8_t *dst, int dst_width, const uint8_t *src, |
2979 | 120 int wrap, int16_t *filter) |
0 | 121 { |
122 int sum, i; | |
1488
766a2f4edbea
avcodec const correctness patch by (Drew Hess <dhess at ilm dot com>)
michaelni
parents:
1106
diff
changeset
|
123 const uint8_t *s; |
0 | 124 |
125 s = src; | |
126 for(i=0;i<dst_width;i++) { | |
127 #if NB_TAPS == 4 | |
128 sum = s[0 * wrap] * filter[0] + | |
129 s[1 * wrap] * filter[1] + | |
130 s[2 * wrap] * filter[2] + | |
131 s[3 * wrap] * filter[3]; | |
132 #else | |
133 { | |
134 int j; | |
1064 | 135 uint8_t *s1 = s; |
0 | 136 |
137 sum = 0; | |
138 for(j=0;j<NB_TAPS;j++) { | |
139 sum += s1[0] * filter[j]; | |
140 s1 += wrap; | |
141 } | |
142 } | |
143 #endif | |
144 sum = sum >> FILTER_BITS; | |
145 if (sum < 0) | |
146 sum = 0; | |
147 else if (sum > 255) | |
148 sum = 255; | |
149 dst[0] = sum; | |
150 dst++; | |
151 s++; | |
152 } | |
153 } | |
154 | |
2 | 155 #ifdef HAVE_MMX |
0 | 156 |
157 #include "i386/mmx.h" | |
158 | |
159 #define FILTER4(reg) \ | |
160 {\ | |
161 s = src + (src_pos >> POS_FRAC_BITS);\ | |
162 phase = get_phase(src_pos);\ | |
163 filter = filters + phase * NB_TAPS;\ | |
164 movq_m2r(*s, reg);\ | |
165 punpcklbw_r2r(mm7, reg);\ | |
166 movq_m2r(*filter, mm6);\ | |
167 pmaddwd_r2r(reg, mm6);\ | |
168 movq_r2r(mm6, reg);\ | |
169 psrlq_i2r(32, reg);\ | |
170 paddd_r2r(mm6, reg);\ | |
171 psrad_i2r(FILTER_BITS, reg);\ | |
172 src_pos += src_incr;\ | |
173 } | |
174 | |
4122
daae66c03857
Replace most of the %lld and %llx by their (cleaner) PRI*64 counterparts.
diego
parents:
4105
diff
changeset
|
175 #define DUMP(reg) movq_r2m(reg, tmp); printf(#reg "=%016"PRIx64"\n", tmp.uq); |
0 | 176 |
177 /* XXX: do four pixels at a time */ | |
1488
766a2f4edbea
avcodec const correctness patch by (Drew Hess <dhess at ilm dot com>)
michaelni
parents:
1106
diff
changeset
|
178 static void h_resample_fast4_mmx(uint8_t *dst, int dst_width, |
2979 | 179 const uint8_t *src, int src_width, |
1064 | 180 int src_start, int src_incr, int16_t *filters) |
0 | 181 { |
182 int src_pos, phase; | |
1488
766a2f4edbea
avcodec const correctness patch by (Drew Hess <dhess at ilm dot com>)
michaelni
parents:
1106
diff
changeset
|
183 const uint8_t *s; |
1064 | 184 int16_t *filter; |
0 | 185 mmx_t tmp; |
2967 | 186 |
0 | 187 src_pos = src_start; |
188 pxor_r2r(mm7, mm7); | |
189 | |
190 while (dst_width >= 4) { | |
191 | |
192 FILTER4(mm0); | |
193 FILTER4(mm1); | |
194 FILTER4(mm2); | |
195 FILTER4(mm3); | |
196 | |
197 packuswb_r2r(mm7, mm0); | |
198 packuswb_r2r(mm7, mm1); | |
199 packuswb_r2r(mm7, mm3); | |
200 packuswb_r2r(mm7, mm2); | |
201 movq_r2m(mm0, tmp); | |
202 dst[0] = tmp.ub[0]; | |
203 movq_r2m(mm1, tmp); | |
204 dst[1] = tmp.ub[0]; | |
205 movq_r2m(mm2, tmp); | |
206 dst[2] = tmp.ub[0]; | |
207 movq_r2m(mm3, tmp); | |
208 dst[3] = tmp.ub[0]; | |
209 dst += 4; | |
210 dst_width -= 4; | |
211 } | |
212 while (dst_width > 0) { | |
213 FILTER4(mm0); | |
214 packuswb_r2r(mm7, mm0); | |
215 movq_r2m(mm0, tmp); | |
216 dst[0] = tmp.ub[0]; | |
217 dst++; | |
218 dst_width--; | |
219 } | |
220 emms(); | |
221 } | |
222 | |
1488
766a2f4edbea
avcodec const correctness patch by (Drew Hess <dhess at ilm dot com>)
michaelni
parents:
1106
diff
changeset
|
223 static void v_resample4_mmx(uint8_t *dst, int dst_width, const uint8_t *src, |
2979 | 224 int wrap, int16_t *filter) |
0 | 225 { |
226 int sum, i, v; | |
1488
766a2f4edbea
avcodec const correctness patch by (Drew Hess <dhess at ilm dot com>)
michaelni
parents:
1106
diff
changeset
|
227 const uint8_t *s; |
0 | 228 mmx_t tmp; |
229 mmx_t coefs[4]; | |
2967 | 230 |
0 | 231 for(i=0;i<4;i++) { |
232 v = filter[i]; | |
233 coefs[i].uw[0] = v; | |
234 coefs[i].uw[1] = v; | |
235 coefs[i].uw[2] = v; | |
236 coefs[i].uw[3] = v; | |
237 } | |
2967 | 238 |
0 | 239 pxor_r2r(mm7, mm7); |
240 s = src; | |
241 while (dst_width >= 4) { | |
242 movq_m2r(s[0 * wrap], mm0); | |
243 punpcklbw_r2r(mm7, mm0); | |
244 movq_m2r(s[1 * wrap], mm1); | |
245 punpcklbw_r2r(mm7, mm1); | |
246 movq_m2r(s[2 * wrap], mm2); | |
247 punpcklbw_r2r(mm7, mm2); | |
248 movq_m2r(s[3 * wrap], mm3); | |
249 punpcklbw_r2r(mm7, mm3); | |
250 | |
251 pmullw_m2r(coefs[0], mm0); | |
252 pmullw_m2r(coefs[1], mm1); | |
253 pmullw_m2r(coefs[2], mm2); | |
254 pmullw_m2r(coefs[3], mm3); | |
255 | |
256 paddw_r2r(mm1, mm0); | |
257 paddw_r2r(mm3, mm2); | |
258 paddw_r2r(mm2, mm0); | |
259 psraw_i2r(FILTER_BITS, mm0); | |
2967 | 260 |
0 | 261 packuswb_r2r(mm7, mm0); |
262 movq_r2m(mm0, tmp); | |
263 | |
1064 | 264 *(uint32_t *)dst = tmp.ud[0]; |
0 | 265 dst += 4; |
266 s += 4; | |
267 dst_width -= 4; | |
268 } | |
269 while (dst_width > 0) { | |
270 sum = s[0 * wrap] * filter[0] + | |
271 s[1 * wrap] * filter[1] + | |
272 s[2 * wrap] * filter[2] + | |
273 s[3 * wrap] * filter[3]; | |
274 sum = sum >> FILTER_BITS; | |
275 if (sum < 0) | |
276 sum = 0; | |
277 else if (sum > 255) | |
278 sum = 255; | |
279 dst[0] = sum; | |
280 dst++; | |
281 s++; | |
282 dst_width--; | |
283 } | |
284 emms(); | |
285 } | |
5054 | 286 #endif /* HAVE_MMX */ |
0 | 287 |
5963 | 288 /* slow version to handle limit cases. Does not need optimization */ |
1488
766a2f4edbea
avcodec const correctness patch by (Drew Hess <dhess at ilm dot com>)
michaelni
parents:
1106
diff
changeset
|
289 static void h_resample_slow(uint8_t *dst, int dst_width, |
2979 | 290 const uint8_t *src, int src_width, |
1064 | 291 int src_start, int src_incr, int16_t *filters) |
0 | 292 { |
293 int src_pos, phase, sum, j, v, i; | |
1488
766a2f4edbea
avcodec const correctness patch by (Drew Hess <dhess at ilm dot com>)
michaelni
parents:
1106
diff
changeset
|
294 const uint8_t *s, *src_end; |
1064 | 295 int16_t *filter; |
0 | 296 |
297 src_end = src + src_width; | |
298 src_pos = src_start; | |
299 for(i=0;i<dst_width;i++) { | |
300 s = src + (src_pos >> POS_FRAC_BITS); | |
301 phase = get_phase(src_pos); | |
302 filter = filters + phase * NB_TAPS; | |
303 sum = 0; | |
304 for(j=0;j<NB_TAPS;j++) { | |
305 if (s < src) | |
306 v = src[0]; | |
307 else if (s >= src_end) | |
308 v = src_end[-1]; | |
309 else | |
310 v = s[0]; | |
311 sum += v * filter[j]; | |
312 s++; | |
313 } | |
314 sum = sum >> FILTER_BITS; | |
315 if (sum < 0) | |
316 sum = 0; | |
317 else if (sum > 255) | |
318 sum = 255; | |
319 dst[0] = sum; | |
320 src_pos += src_incr; | |
321 dst++; | |
322 } | |
323 } | |
324 | |
1488
766a2f4edbea
avcodec const correctness patch by (Drew Hess <dhess at ilm dot com>)
michaelni
parents:
1106
diff
changeset
|
325 static void h_resample(uint8_t *dst, int dst_width, const uint8_t *src, |
2979 | 326 int src_width, int src_start, int src_incr, |
327 int16_t *filters) | |
0 | 328 { |
329 int n, src_end; | |
330 | |
331 if (src_start < 0) { | |
332 n = (0 - src_start + src_incr - 1) / src_incr; | |
333 h_resample_slow(dst, n, src, src_width, src_start, src_incr, filters); | |
334 dst += n; | |
335 dst_width -= n; | |
336 src_start += n * src_incr; | |
337 } | |
338 src_end = src_start + dst_width * src_incr; | |
339 if (src_end > ((src_width - NB_TAPS) << POS_FRAC_BITS)) { | |
2967 | 340 n = (((src_width - NB_TAPS + 1) << POS_FRAC_BITS) - 1 - src_start) / |
0 | 341 src_incr; |
342 } else { | |
343 n = dst_width; | |
344 } | |
2 | 345 #ifdef HAVE_MMX |
4197 | 346 if ((mm_flags & MM_MMX) && NB_TAPS == 4) |
2967 | 347 h_resample_fast4_mmx(dst, n, |
0 | 348 src, src_width, src_start, src_incr, filters); |
349 else | |
350 #endif | |
2967 | 351 h_resample_fast(dst, n, |
0 | 352 src, src_width, src_start, src_incr, filters); |
353 if (n < dst_width) { | |
354 dst += n; | |
355 dst_width -= n; | |
356 src_start += n * src_incr; | |
2967 | 357 h_resample_slow(dst, dst_width, |
0 | 358 src, src_width, src_start, src_incr, filters); |
359 } | |
360 } | |
361 | |
2967 | 362 static void component_resample(ImgReSampleContext *s, |
1064 | 363 uint8_t *output, int owrap, int owidth, int oheight, |
364 uint8_t *input, int iwrap, int iwidth, int iheight) | |
0 | 365 { |
366 int src_y, src_y1, last_src_y, ring_y, phase_y, y1, y; | |
1064 | 367 uint8_t *new_line, *src_line; |
0 | 368 |
369 last_src_y = - FCENTER - 1; | |
370 /* position of the bottom of the filter in the source image */ | |
2967 | 371 src_y = (last_src_y + NB_TAPS) * POS_FRAC; |
0 | 372 ring_y = NB_TAPS; /* position in ring buffer */ |
373 for(y=0;y<oheight;y++) { | |
374 /* apply horizontal filter on new lines from input if needed */ | |
375 src_y1 = src_y >> POS_FRAC_BITS; | |
376 while (last_src_y < src_y1) { | |
377 if (++ring_y >= LINE_BUF_HEIGHT + NB_TAPS) | |
378 ring_y = NB_TAPS; | |
379 last_src_y++; | |
630
b4ee42142ad1
croping patch by (talus25 at speakeasy dot net) with fixes from atmos & me
michaelni
parents:
429
diff
changeset
|
380 /* handle limit conditions : replicate line (slightly |
b4ee42142ad1
croping patch by (talus25 at speakeasy dot net) with fixes from atmos & me
michaelni
parents:
429
diff
changeset
|
381 inefficient because we filter multiple times) */ |
0 | 382 y1 = last_src_y; |
383 if (y1 < 0) { | |
384 y1 = 0; | |
385 } else if (y1 >= iheight) { | |
386 y1 = iheight - 1; | |
387 } | |
388 src_line = input + y1 * iwrap; | |
389 new_line = s->line_buf + ring_y * owidth; | |
390 /* apply filter and handle limit cases correctly */ | |
2967 | 391 h_resample(new_line, owidth, |
392 src_line, iwidth, - FCENTER * POS_FRAC, s->h_incr, | |
0 | 393 &s->h_filters[0][0]); |
6055 | 394 /* handle ring buffer wrapping */ |
0 | 395 if (ring_y >= LINE_BUF_HEIGHT) { |
396 memcpy(s->line_buf + (ring_y - LINE_BUF_HEIGHT) * owidth, | |
397 new_line, owidth); | |
398 } | |
399 } | |
400 /* apply vertical filter */ | |
401 phase_y = get_phase(src_y); | |
2 | 402 #ifdef HAVE_MMX |
0 | 403 /* desactivated MMX because loss of precision */ |
4197 | 404 if ((mm_flags & MM_MMX) && NB_TAPS == 4 && 0) |
2967 | 405 v_resample4_mmx(output, owidth, |
406 s->line_buf + (ring_y - NB_TAPS + 1) * owidth, owidth, | |
0 | 407 &s->v_filters[phase_y][0]); |
408 else | |
409 #endif | |
894
a408778eff87
altivec accelerated v-resample patch by (Brian Foley <bfoley at compsoc dot nuigalway dot ie>)
michaelni
parents:
653
diff
changeset
|
410 #ifdef HAVE_ALTIVEC |
4197 | 411 if ((mm_flags & MM_ALTIVEC) && NB_TAPS == 4 && FILTER_BITS <= 6) |
894
a408778eff87
altivec accelerated v-resample patch by (Brian Foley <bfoley at compsoc dot nuigalway dot ie>)
michaelni
parents:
653
diff
changeset
|
412 v_resample16_altivec(output, owidth, |
a408778eff87
altivec accelerated v-resample patch by (Brian Foley <bfoley at compsoc dot nuigalway dot ie>)
michaelni
parents:
653
diff
changeset
|
413 s->line_buf + (ring_y - NB_TAPS + 1) * owidth, owidth, |
a408778eff87
altivec accelerated v-resample patch by (Brian Foley <bfoley at compsoc dot nuigalway dot ie>)
michaelni
parents:
653
diff
changeset
|
414 &s->v_filters[phase_y][0]); |
a408778eff87
altivec accelerated v-resample patch by (Brian Foley <bfoley at compsoc dot nuigalway dot ie>)
michaelni
parents:
653
diff
changeset
|
415 else |
a408778eff87
altivec accelerated v-resample patch by (Brian Foley <bfoley at compsoc dot nuigalway dot ie>)
michaelni
parents:
653
diff
changeset
|
416 #endif |
2967 | 417 v_resample(output, owidth, |
418 s->line_buf + (ring_y - NB_TAPS + 1) * owidth, owidth, | |
0 | 419 &s->v_filters[phase_y][0]); |
2967 | 420 |
0 | 421 src_y += s->v_incr; |
2967 | 422 |
0 | 423 output += owrap; |
424 } | |
425 } | |
426 | |
427 ImgReSampleContext *img_resample_init(int owidth, int oheight, | |
428 int iwidth, int iheight) | |
429 { | |
2967 | 430 return img_resample_full_init(owidth, oheight, iwidth, iheight, |
1928
0c23a5564489
padding support in ffmpeg patch by (Todd Kirby <doubleshot at pacbell dot net>)
michael
parents:
1488
diff
changeset
|
431 0, 0, 0, 0, 0, 0, 0, 0); |
630
b4ee42142ad1
croping patch by (talus25 at speakeasy dot net) with fixes from atmos & me
michaelni
parents:
429
diff
changeset
|
432 } |
b4ee42142ad1
croping patch by (talus25 at speakeasy dot net) with fixes from atmos & me
michaelni
parents:
429
diff
changeset
|
433 |
b4ee42142ad1
croping patch by (talus25 at speakeasy dot net) with fixes from atmos & me
michaelni
parents:
429
diff
changeset
|
434 ImgReSampleContext *img_resample_full_init(int owidth, int oheight, |
b4ee42142ad1
croping patch by (talus25 at speakeasy dot net) with fixes from atmos & me
michaelni
parents:
429
diff
changeset
|
435 int iwidth, int iheight, |
b4ee42142ad1
croping patch by (talus25 at speakeasy dot net) with fixes from atmos & me
michaelni
parents:
429
diff
changeset
|
436 int topBand, int bottomBand, |
1928
0c23a5564489
padding support in ffmpeg patch by (Todd Kirby <doubleshot at pacbell dot net>)
michael
parents:
1488
diff
changeset
|
437 int leftBand, int rightBand, |
0c23a5564489
padding support in ffmpeg patch by (Todd Kirby <doubleshot at pacbell dot net>)
michael
parents:
1488
diff
changeset
|
438 int padtop, int padbottom, |
0c23a5564489
padding support in ffmpeg patch by (Todd Kirby <doubleshot at pacbell dot net>)
michael
parents:
1488
diff
changeset
|
439 int padleft, int padright) |
630
b4ee42142ad1
croping patch by (talus25 at speakeasy dot net) with fixes from atmos & me
michaelni
parents:
429
diff
changeset
|
440 { |
0 | 441 ImgReSampleContext *s; |
442 | |
2904 | 443 if (!owidth || !oheight || !iwidth || !iheight) |
2979 | 444 return NULL; |
2904 | 445 |
0 | 446 s = av_mallocz(sizeof(ImgReSampleContext)); |
447 if (!s) | |
448 return NULL; | |
2422 | 449 if((unsigned)owidth >= UINT_MAX / (LINE_BUF_HEIGHT + NB_TAPS)) |
450 return NULL; | |
0 | 451 s->line_buf = av_mallocz(owidth * (LINE_BUF_HEIGHT + NB_TAPS)); |
2967 | 452 if (!s->line_buf) |
0 | 453 goto fail; |
2967 | 454 |
0 | 455 s->owidth = owidth; |
456 s->oheight = oheight; | |
457 s->iwidth = iwidth; | |
458 s->iheight = iheight; | |
2967 | 459 |
630
b4ee42142ad1
croping patch by (talus25 at speakeasy dot net) with fixes from atmos & me
michaelni
parents:
429
diff
changeset
|
460 s->topBand = topBand; |
b4ee42142ad1
croping patch by (talus25 at speakeasy dot net) with fixes from atmos & me
michaelni
parents:
429
diff
changeset
|
461 s->bottomBand = bottomBand; |
b4ee42142ad1
croping patch by (talus25 at speakeasy dot net) with fixes from atmos & me
michaelni
parents:
429
diff
changeset
|
462 s->leftBand = leftBand; |
b4ee42142ad1
croping patch by (talus25 at speakeasy dot net) with fixes from atmos & me
michaelni
parents:
429
diff
changeset
|
463 s->rightBand = rightBand; |
2967 | 464 |
1928
0c23a5564489
padding support in ffmpeg patch by (Todd Kirby <doubleshot at pacbell dot net>)
michael
parents:
1488
diff
changeset
|
465 s->padtop = padtop; |
0c23a5564489
padding support in ffmpeg patch by (Todd Kirby <doubleshot at pacbell dot net>)
michael
parents:
1488
diff
changeset
|
466 s->padbottom = padbottom; |
0c23a5564489
padding support in ffmpeg patch by (Todd Kirby <doubleshot at pacbell dot net>)
michael
parents:
1488
diff
changeset
|
467 s->padleft = padleft; |
0c23a5564489
padding support in ffmpeg patch by (Todd Kirby <doubleshot at pacbell dot net>)
michael
parents:
1488
diff
changeset
|
468 s->padright = padright; |
0c23a5564489
padding support in ffmpeg patch by (Todd Kirby <doubleshot at pacbell dot net>)
michael
parents:
1488
diff
changeset
|
469 |
0c23a5564489
padding support in ffmpeg patch by (Todd Kirby <doubleshot at pacbell dot net>)
michael
parents:
1488
diff
changeset
|
470 s->pad_owidth = owidth - (padleft + padright); |
0c23a5564489
padding support in ffmpeg patch by (Todd Kirby <doubleshot at pacbell dot net>)
michael
parents:
1488
diff
changeset
|
471 s->pad_oheight = oheight - (padtop + padbottom); |
0c23a5564489
padding support in ffmpeg patch by (Todd Kirby <doubleshot at pacbell dot net>)
michael
parents:
1488
diff
changeset
|
472 |
0c23a5564489
padding support in ffmpeg patch by (Todd Kirby <doubleshot at pacbell dot net>)
michael
parents:
1488
diff
changeset
|
473 s->h_incr = ((iwidth - leftBand - rightBand) * POS_FRAC) / s->pad_owidth; |
2967 | 474 s->v_incr = ((iheight - topBand - bottomBand) * POS_FRAC) / s->pad_oheight; |
1928
0c23a5564489
padding support in ffmpeg patch by (Todd Kirby <doubleshot at pacbell dot net>)
michael
parents:
1488
diff
changeset
|
475 |
2967 | 476 av_build_filter(&s->h_filters[0][0], (float) s->pad_owidth / |
2082
3dc9bbe1b152
polyphase kaiser windowed sinc and blackman nuttall windowed sinc audio resample filters
michael
parents:
2064
diff
changeset
|
477 (float) (iwidth - leftBand - rightBand), NB_TAPS, NB_PHASES, 1<<FILTER_BITS, 0); |
2967 | 478 av_build_filter(&s->v_filters[0][0], (float) s->pad_oheight / |
2082
3dc9bbe1b152
polyphase kaiser windowed sinc and blackman nuttall windowed sinc audio resample filters
michael
parents:
2064
diff
changeset
|
479 (float) (iheight - topBand - bottomBand), NB_TAPS, NB_PHASES, 1<<FILTER_BITS, 0); |
0 | 480 |
481 return s; | |
1928
0c23a5564489
padding support in ffmpeg patch by (Todd Kirby <doubleshot at pacbell dot net>)
michael
parents:
1488
diff
changeset
|
482 fail: |
396
fce0a2520551
removed useless header includes - use av memory functions
glantau
parents:
18
diff
changeset
|
483 av_free(s); |
0 | 484 return NULL; |
485 } | |
486 | |
2967 | 487 void img_resample(ImgReSampleContext *s, |
1488
766a2f4edbea
avcodec const correctness patch by (Drew Hess <dhess at ilm dot com>)
michaelni
parents:
1106
diff
changeset
|
488 AVPicture *output, const AVPicture *input) |
0 | 489 { |
490 int i, shift; | |
1928
0c23a5564489
padding support in ffmpeg patch by (Todd Kirby <doubleshot at pacbell dot net>)
michael
parents:
1488
diff
changeset
|
491 uint8_t* optr; |
0 | 492 |
1928
0c23a5564489
padding support in ffmpeg patch by (Todd Kirby <doubleshot at pacbell dot net>)
michael
parents:
1488
diff
changeset
|
493 for (i=0;i<3;i++) { |
0 | 494 shift = (i == 0) ? 0 : 1; |
1928
0c23a5564489
padding support in ffmpeg patch by (Todd Kirby <doubleshot at pacbell dot net>)
michael
parents:
1488
diff
changeset
|
495 |
2967 | 496 optr = output->data[i] + (((output->linesize[i] * |
1928
0c23a5564489
padding support in ffmpeg patch by (Todd Kirby <doubleshot at pacbell dot net>)
michael
parents:
1488
diff
changeset
|
497 s->padtop) + s->padleft) >> shift); |
0c23a5564489
padding support in ffmpeg patch by (Todd Kirby <doubleshot at pacbell dot net>)
michael
parents:
1488
diff
changeset
|
498 |
2967 | 499 component_resample(s, optr, output->linesize[i], |
1928
0c23a5564489
padding support in ffmpeg patch by (Todd Kirby <doubleshot at pacbell dot net>)
michael
parents:
1488
diff
changeset
|
500 s->pad_owidth >> shift, s->pad_oheight >> shift, |
2967 | 501 input->data[i] + (input->linesize[i] * |
1928
0c23a5564489
padding support in ffmpeg patch by (Todd Kirby <doubleshot at pacbell dot net>)
michael
parents:
1488
diff
changeset
|
502 (s->topBand >> shift)) + (s->leftBand >> shift), |
2967 | 503 input->linesize[i], ((s->iwidth - s->leftBand - |
1928
0c23a5564489
padding support in ffmpeg patch by (Todd Kirby <doubleshot at pacbell dot net>)
michael
parents:
1488
diff
changeset
|
504 s->rightBand) >> shift), |
630
b4ee42142ad1
croping patch by (talus25 at speakeasy dot net) with fixes from atmos & me
michaelni
parents:
429
diff
changeset
|
505 (s->iheight - s->topBand - s->bottomBand) >> shift); |
0 | 506 } |
507 } | |
508 | |
509 void img_resample_close(ImgReSampleContext *s) | |
510 { | |
396
fce0a2520551
removed useless header includes - use av memory functions
glantau
parents:
18
diff
changeset
|
511 av_free(s->line_buf); |
fce0a2520551
removed useless header includes - use av memory functions
glantau
parents:
18
diff
changeset
|
512 av_free(s); |
0 | 513 } |
514 | |
6433
d42d5a90f7de
Provide non null class name to avoid crash with ffmpeg -h.
michael
parents:
6055
diff
changeset
|
515 static const AVClass context_class = { "imgresample", NULL, NULL }; |
d42d5a90f7de
Provide non null class name to avoid crash with ffmpeg -h.
michael
parents:
6055
diff
changeset
|
516 |
3249 | 517 struct SwsContext *sws_getContext(int srcW, int srcH, int srcFormat, |
518 int dstW, int dstH, int dstFormat, | |
519 int flags, SwsFilter *srcFilter, | |
520 SwsFilter *dstFilter, double *param) | |
521 { | |
522 struct SwsContext *ctx; | |
523 | |
524 ctx = av_malloc(sizeof(struct SwsContext)); | |
6433
d42d5a90f7de
Provide non null class name to avoid crash with ffmpeg -h.
michael
parents:
6055
diff
changeset
|
525 if (!ctx) { |
3249 | 526 av_log(NULL, AV_LOG_ERROR, "Cannot allocate a resampling context!\n"); |
527 | |
528 return NULL; | |
529 } | |
6433
d42d5a90f7de
Provide non null class name to avoid crash with ffmpeg -h.
michael
parents:
6055
diff
changeset
|
530 ctx->av_class = &context_class; |
3249 | 531 |
532 if ((srcH != dstH) || (srcW != dstW)) { | |
533 if ((srcFormat != PIX_FMT_YUV420P) || (dstFormat != PIX_FMT_YUV420P)) { | |
534 av_log(NULL, AV_LOG_INFO, "PIX_FMT_YUV420P will be used as an intermediate format for rescaling\n"); | |
535 } | |
536 ctx->resampling_ctx = img_resample_init(dstW, dstH, srcW, srcH); | |
537 } else { | |
538 ctx->resampling_ctx = av_malloc(sizeof(ImgReSampleContext)); | |
539 ctx->resampling_ctx->iheight = srcH; | |
540 ctx->resampling_ctx->iwidth = srcW; | |
541 ctx->resampling_ctx->oheight = dstH; | |
542 ctx->resampling_ctx->owidth = dstW; | |
543 } | |
544 ctx->src_pix_fmt = srcFormat; | |
545 ctx->dst_pix_fmt = dstFormat; | |
546 | |
547 return ctx; | |
548 } | |
549 | |
550 void sws_freeContext(struct SwsContext *ctx) | |
551 { | |
4357
1654075b205c
fix segfault with http://sam.zoy.org/zzuf/lol-ffplay.ogm and
takis
parents:
4197
diff
changeset
|
552 if (!ctx) |
1654075b205c
fix segfault with http://sam.zoy.org/zzuf/lol-ffplay.ogm and
takis
parents:
4197
diff
changeset
|
553 return; |
3249 | 554 if ((ctx->resampling_ctx->iwidth != ctx->resampling_ctx->owidth) || |
555 (ctx->resampling_ctx->iheight != ctx->resampling_ctx->oheight)) { | |
556 img_resample_close(ctx->resampling_ctx); | |
557 } else { | |
558 av_free(ctx->resampling_ctx); | |
559 } | |
560 av_free(ctx); | |
561 } | |
562 | |
4036
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
563 |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
564 /** |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
565 * Checks if context is valid or reallocs a new one instead. |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
566 * If context is NULL, just calls sws_getContext() to get a new one. |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
567 * Otherwise, checks if the parameters are the same already saved in context. |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
568 * If that is the case, returns the current context. |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
569 * Otherwise, frees context and gets a new one. |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
570 * |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
571 * Be warned that srcFilter, dstFilter are not checked, they are |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
572 * asumed to remain valid. |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
573 */ |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
574 struct SwsContext *sws_getCachedContext(struct SwsContext *ctx, |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
575 int srcW, int srcH, int srcFormat, |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
576 int dstW, int dstH, int dstFormat, int flags, |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
577 SwsFilter *srcFilter, SwsFilter *dstFilter, double *param) |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
578 { |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
579 if (ctx != NULL) { |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
580 if ((ctx->resampling_ctx->iwidth != srcW) || |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
581 (ctx->resampling_ctx->iheight != srcH) || |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
582 (ctx->src_pix_fmt != srcFormat) || |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
583 (ctx->resampling_ctx->owidth != dstW) || |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
584 (ctx->resampling_ctx->oheight != dstH) || |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
585 (ctx->dst_pix_fmt != dstFormat)) |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
586 { |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
587 sws_freeContext(ctx); |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
588 ctx = NULL; |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
589 } |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
590 } |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
591 if (ctx == NULL) { |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
592 return sws_getContext(srcW, srcH, srcFormat, |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
593 dstW, dstH, dstFormat, flags, |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
594 srcFilter, dstFilter, param); |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
595 } |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
596 return ctx; |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
597 } |
207c22206d53
Implement sws_getCachedContext() in swscale emulation
lucabe
parents:
3947
diff
changeset
|
598 |
3249 | 599 int sws_scale(struct SwsContext *ctx, uint8_t* src[], int srcStride[], |
600 int srcSliceY, int srcSliceH, uint8_t* dst[], int dstStride[]) | |
601 { | |
602 AVPicture src_pict, dst_pict; | |
603 int i, res = 0; | |
604 AVPicture picture_format_temp; | |
605 AVPicture picture_resample_temp, *formatted_picture, *resampled_picture; | |
606 uint8_t *buf1 = NULL, *buf2 = NULL; | |
607 enum PixelFormat current_pix_fmt; | |
608 | |
4105 | 609 for (i = 0; i < 4; i++) { |
3249 | 610 src_pict.data[i] = src[i]; |
611 src_pict.linesize[i] = srcStride[i]; | |
612 dst_pict.data[i] = dst[i]; | |
613 dst_pict.linesize[i] = dstStride[i]; | |
614 } | |
615 if ((ctx->resampling_ctx->iwidth != ctx->resampling_ctx->owidth) || | |
616 (ctx->resampling_ctx->iheight != ctx->resampling_ctx->oheight)) { | |
617 /* We have to rescale the picture, but only YUV420P rescaling is supported... */ | |
618 | |
619 if (ctx->src_pix_fmt != PIX_FMT_YUV420P) { | |
620 int size; | |
621 | |
622 /* create temporary picture for rescaling input*/ | |
623 size = avpicture_get_size(PIX_FMT_YUV420P, ctx->resampling_ctx->iwidth, ctx->resampling_ctx->iheight); | |
624 buf1 = av_malloc(size); | |
625 if (!buf1) { | |
626 res = -1; | |
627 goto the_end; | |
628 } | |
629 formatted_picture = &picture_format_temp; | |
630 avpicture_fill((AVPicture*)formatted_picture, buf1, | |
631 PIX_FMT_YUV420P, ctx->resampling_ctx->iwidth, ctx->resampling_ctx->iheight); | |
632 | |
633 if (img_convert((AVPicture*)formatted_picture, PIX_FMT_YUV420P, | |
634 &src_pict, ctx->src_pix_fmt, | |
635 ctx->resampling_ctx->iwidth, ctx->resampling_ctx->iheight) < 0) { | |
636 | |
637 av_log(NULL, AV_LOG_ERROR, "pixel format conversion not handled\n"); | |
638 res = -1; | |
639 goto the_end; | |
640 } | |
641 } else { | |
642 formatted_picture = &src_pict; | |
643 } | |
644 | |
645 if (ctx->dst_pix_fmt != PIX_FMT_YUV420P) { | |
646 int size; | |
647 | |
648 /* create temporary picture for rescaling output*/ | |
649 size = avpicture_get_size(PIX_FMT_YUV420P, ctx->resampling_ctx->owidth, ctx->resampling_ctx->oheight); | |
650 buf2 = av_malloc(size); | |
651 if (!buf2) { | |
652 res = -1; | |
653 goto the_end; | |
654 } | |
655 resampled_picture = &picture_resample_temp; | |
656 avpicture_fill((AVPicture*)resampled_picture, buf2, | |
657 PIX_FMT_YUV420P, ctx->resampling_ctx->owidth, ctx->resampling_ctx->oheight); | |
658 | |
659 } else { | |
660 resampled_picture = &dst_pict; | |
661 } | |
662 | |
663 /* ...and finally rescale!!! */ | |
664 img_resample(ctx->resampling_ctx, resampled_picture, formatted_picture); | |
665 current_pix_fmt = PIX_FMT_YUV420P; | |
666 } else { | |
667 resampled_picture = &src_pict; | |
668 current_pix_fmt = ctx->src_pix_fmt; | |
669 } | |
670 | |
671 if (current_pix_fmt != ctx->dst_pix_fmt) { | |
672 if (img_convert(&dst_pict, ctx->dst_pix_fmt, | |
673 resampled_picture, current_pix_fmt, | |
674 ctx->resampling_ctx->owidth, ctx->resampling_ctx->oheight) < 0) { | |
675 | |
676 av_log(NULL, AV_LOG_ERROR, "pixel format conversion not handled\n"); | |
677 | |
678 res = -1; | |
679 goto the_end; | |
680 } | |
3516 | 681 } else if (resampled_picture != &dst_pict) { |
4624
6a900f539e2c
Add the prefix "av_" to img_crop(), img_copy() and img_pad(), and rename "img"
takis
parents:
4357
diff
changeset
|
682 av_picture_copy(&dst_pict, resampled_picture, current_pix_fmt, |
3516 | 683 ctx->resampling_ctx->owidth, ctx->resampling_ctx->oheight); |
3249 | 684 } |
685 | |
686 the_end: | |
687 av_free(buf1); | |
688 av_free(buf2); | |
689 return res; | |
690 } | |
691 | |
692 | |
0 | 693 #ifdef TEST |
2400
17ec73c65748
imgresample test cleanup patch by (Panagiotis Issaris <takis )( lumumba d0t luc d0t ac.be>)
michael
parents:
2082
diff
changeset
|
694 #include <stdio.h> |
5055 | 695 #undef exit |
630
b4ee42142ad1
croping patch by (talus25 at speakeasy dot net) with fixes from atmos & me
michaelni
parents:
429
diff
changeset
|
696 |
0 | 697 /* input */ |
698 #define XSIZE 256 | |
699 #define YSIZE 256 | |
1064 | 700 uint8_t img[XSIZE * YSIZE]; |
0 | 701 |
702 /* output */ | |
703 #define XSIZE1 512 | |
704 #define YSIZE1 512 | |
1064 | 705 uint8_t img1[XSIZE1 * YSIZE1]; |
706 uint8_t img2[XSIZE1 * YSIZE1]; | |
0 | 707 |
1064 | 708 void save_pgm(const char *filename, uint8_t *img, int xsize, int ysize) |
0 | 709 { |
2846
40765c51a7a9
Compilation fixes part 1 patch by (Arvind R. and Burkhard Plaum, plaum, ipf uni-stuttgart de)
michael
parents:
2423
diff
changeset
|
710 #undef fprintf |
0 | 711 FILE *f; |
712 f=fopen(filename,"w"); | |
713 fprintf(f,"P5\n%d %d\n%d\n", xsize, ysize, 255); | |
714 fwrite(img,1, xsize * ysize,f); | |
715 fclose(f); | |
2846
40765c51a7a9
Compilation fixes part 1 patch by (Arvind R. and Burkhard Plaum, plaum, ipf uni-stuttgart de)
michael
parents:
2423
diff
changeset
|
716 #define fprintf please_use_av_log |
0 | 717 } |
718 | |
1064 | 719 static void dump_filter(int16_t *filter) |
0 | 720 { |
721 int i, ph; | |
722 | |
723 for(ph=0;ph<NB_PHASES;ph++) { | |
2400
17ec73c65748
imgresample test cleanup patch by (Panagiotis Issaris <takis )( lumumba d0t luc d0t ac.be>)
michael
parents:
2082
diff
changeset
|
724 av_log(NULL, AV_LOG_INFO, "%2d: ", ph); |
0 | 725 for(i=0;i<NB_TAPS;i++) { |
2400
17ec73c65748
imgresample test cleanup patch by (Panagiotis Issaris <takis )( lumumba d0t luc d0t ac.be>)
michael
parents:
2082
diff
changeset
|
726 av_log(NULL, AV_LOG_INFO, " %5.2f", filter[ph * NB_TAPS + i] / 256.0); |
0 | 727 } |
2400
17ec73c65748
imgresample test cleanup patch by (Panagiotis Issaris <takis )( lumumba d0t luc d0t ac.be>)
michael
parents:
2082
diff
changeset
|
728 av_log(NULL, AV_LOG_INFO, "\n"); |
0 | 729 } |
730 } | |
731 | |
2 | 732 #ifdef HAVE_MMX |
644 | 733 int mm_flags; |
0 | 734 #endif |
735 | |
736 int main(int argc, char **argv) | |
737 { | |
738 int x, y, v, i, xsize, ysize; | |
739 ImgReSampleContext *s; | |
740 float fact, factors[] = { 1/2.0, 3.0/4.0, 1.0, 4.0/3.0, 16.0/9.0, 2.0 }; | |
741 char buf[256]; | |
742 | |
743 /* build test image */ | |
744 for(y=0;y<YSIZE;y++) { | |
745 for(x=0;x<XSIZE;x++) { | |
746 if (x < XSIZE/2 && y < YSIZE/2) { | |
747 if (x < XSIZE/4 && y < YSIZE/4) { | |
748 if ((x % 10) <= 6 && | |
749 (y % 10) <= 6) | |
750 v = 0xff; | |
751 else | |
752 v = 0x00; | |
753 } else if (x < XSIZE/4) { | |
2967 | 754 if (x & 1) |
0 | 755 v = 0xff; |
2967 | 756 else |
0 | 757 v = 0; |
758 } else if (y < XSIZE/4) { | |
2967 | 759 if (y & 1) |
0 | 760 v = 0xff; |
2967 | 761 else |
0 | 762 v = 0; |
763 } else { | |
764 if (y < YSIZE*3/8) { | |
2967 | 765 if ((y+x) & 1) |
0 | 766 v = 0xff; |
2967 | 767 else |
0 | 768 v = 0; |
769 } else { | |
770 if (((x+3) % 4) <= 1 && | |
771 ((y+3) % 4) <= 1) | |
772 v = 0xff; | |
773 else | |
774 v = 0x00; | |
775 } | |
776 } | |
777 } else if (x < XSIZE/2) { | |
778 v = ((x - (XSIZE/2)) * 255) / (XSIZE/2); | |
779 } else if (y < XSIZE/2) { | |
780 v = ((y - (XSIZE/2)) * 255) / (XSIZE/2); | |
781 } else { | |
782 v = ((x + y - XSIZE) * 255) / XSIZE; | |
783 } | |
630
b4ee42142ad1
croping patch by (talus25 at speakeasy dot net) with fixes from atmos & me
michaelni
parents:
429
diff
changeset
|
784 img[(YSIZE - y) * XSIZE + (XSIZE - x)] = v; |
0 | 785 } |
786 } | |
787 save_pgm("/tmp/in.pgm", img, XSIZE, YSIZE); | |
788 for(i=0;i<sizeof(factors)/sizeof(float);i++) { | |
789 fact = factors[i]; | |
790 xsize = (int)(XSIZE * fact); | |
630
b4ee42142ad1
croping patch by (talus25 at speakeasy dot net) with fixes from atmos & me
michaelni
parents:
429
diff
changeset
|
791 ysize = (int)((YSIZE - 100) * fact); |
2400
17ec73c65748
imgresample test cleanup patch by (Panagiotis Issaris <takis )( lumumba d0t luc d0t ac.be>)
michael
parents:
2082
diff
changeset
|
792 s = img_resample_full_init(xsize, ysize, XSIZE, YSIZE, 50 ,50, 0, 0, 0, 0, 0, 0); |
17ec73c65748
imgresample test cleanup patch by (Panagiotis Issaris <takis )( lumumba d0t luc d0t ac.be>)
michael
parents:
2082
diff
changeset
|
793 av_log(NULL, AV_LOG_INFO, "Factor=%0.2f\n", fact); |
0 | 794 dump_filter(&s->h_filters[0][0]); |
795 component_resample(s, img1, xsize, xsize, ysize, | |
630
b4ee42142ad1
croping patch by (talus25 at speakeasy dot net) with fixes from atmos & me
michaelni
parents:
429
diff
changeset
|
796 img + 50 * XSIZE, XSIZE, XSIZE, YSIZE - 100); |
0 | 797 img_resample_close(s); |
798 | |
2423 | 799 snprintf(buf, sizeof(buf), "/tmp/out%d.pgm", i); |
0 | 800 save_pgm(buf, img1, xsize, ysize); |
801 } | |
802 | |
803 /* mmx test */ | |
2 | 804 #ifdef HAVE_MMX |
2400
17ec73c65748
imgresample test cleanup patch by (Panagiotis Issaris <takis )( lumumba d0t luc d0t ac.be>)
michael
parents:
2082
diff
changeset
|
805 av_log(NULL, AV_LOG_INFO, "MMX test\n"); |
0 | 806 fact = 0.72; |
807 xsize = (int)(XSIZE * fact); | |
808 ysize = (int)(YSIZE * fact); | |
809 mm_flags = MM_MMX; | |
810 s = img_resample_init(xsize, ysize, XSIZE, YSIZE); | |
811 component_resample(s, img1, xsize, xsize, ysize, | |
812 img, XSIZE, XSIZE, YSIZE); | |
813 | |
814 mm_flags = 0; | |
815 s = img_resample_init(xsize, ysize, XSIZE, YSIZE); | |
816 component_resample(s, img2, xsize, xsize, ysize, | |
817 img, XSIZE, XSIZE, YSIZE); | |
818 if (memcmp(img1, img2, xsize * ysize) != 0) { | |
2400
17ec73c65748
imgresample test cleanup patch by (Panagiotis Issaris <takis )( lumumba d0t luc d0t ac.be>)
michael
parents:
2082
diff
changeset
|
819 av_log(NULL, AV_LOG_ERROR, "mmx error\n"); |
0 | 820 exit(1); |
821 } | |
2400
17ec73c65748
imgresample test cleanup patch by (Panagiotis Issaris <takis )( lumumba d0t luc d0t ac.be>)
michael
parents:
2082
diff
changeset
|
822 av_log(NULL, AV_LOG_INFO, "MMX OK\n"); |
5054 | 823 #endif /* HAVE_MMX */ |
0 | 824 return 0; |
825 } | |
826 | |
5054 | 827 #endif /* TEST */ |