fa4424810485e8894fb04637319fbf1bca522af6
[libav.git] / libavcodec / imgresample.c
1 /*
2 * High quality image resampling with polyphase filters
3 * Copyright (c) 2001 Fabrice Bellard.
4 *
5 * This file is part of FFmpeg.
6 *
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22 /**
23 * @file imgresample.c
24 * High quality image resampling with polyphase filters .
25 */
26
27 #include "avcodec.h"
28 #include "dsputil.h"
29 #include "libswscale/swscale.h"
30
31 #ifdef HAVE_ALTIVEC
32 #include "ppc/imgresample_altivec.h"
33 #endif
34
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 */
41 //#define TEST 1 /* Test it */
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
50 struct SwsContext {
51 const AVClass *av_class;
52 struct ImgReSampleContext *resampling_ctx;
53 enum PixelFormat src_pix_fmt, dst_pix_fmt;
54 };
55
56 typedef struct ImgReSampleContext {
57 int iwidth, iheight, owidth, oheight;
58 int topBand, bottomBand, leftBand, rightBand;
59 int padtop, padbottom, padleft, padright;
60 int pad_owidth, pad_oheight;
61 int h_incr, v_incr;
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 */
64 uint8_t *line_buf;
65 } ImgReSampleContext;
66
67 void av_build_filter(int16_t *filter, double factor, int tap_count, int phase_count, int scale, int type);
68
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 */
75 static void h_resample_fast(uint8_t *dst, int dst_width, const uint8_t *src,
76 int src_width, int src_start, int src_incr,
77 int16_t *filters)
78 {
79 int src_pos, phase, sum, i;
80 const uint8_t *s;
81 int16_t *filter;
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))
89 av_abort();
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 */
119 static void v_resample(uint8_t *dst, int dst_width, const uint8_t *src,
120 int wrap, int16_t *filter)
121 {
122 int sum, i;
123 const uint8_t *s;
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;
135 uint8_t *s1 = s;
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
155 #ifdef HAVE_MMX
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
175 #define DUMP(reg) movq_r2m(reg, tmp); printf(#reg "=%016"PRIx64"\n", tmp.uq);
176
177 /* XXX: do four pixels at a time */
178 static void h_resample_fast4_mmx(uint8_t *dst, int dst_width,
179 const uint8_t *src, int src_width,
180 int src_start, int src_incr, int16_t *filters)
181 {
182 int src_pos, phase;
183 const uint8_t *s;
184 int16_t *filter;
185 mmx_t tmp;
186
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
223 static void v_resample4_mmx(uint8_t *dst, int dst_width, const uint8_t *src,
224 int wrap, int16_t *filter)
225 {
226 int sum, i, v;
227 const uint8_t *s;
228 mmx_t tmp;
229 mmx_t coefs[4];
230
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 }
238
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);
260
261 packuswb_r2r(mm7, mm0);
262 movq_r2m(mm0, tmp);
263
264 *(uint32_t *)dst = tmp.ud[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 }
286 #endif /* HAVE_MMX */
287
288 /* slow version to handle limit cases. Does not need optimization */
289 static void h_resample_slow(uint8_t *dst, int dst_width,
290 const uint8_t *src, int src_width,
291 int src_start, int src_incr, int16_t *filters)
292 {
293 int src_pos, phase, sum, j, v, i;
294 const uint8_t *s, *src_end;
295 int16_t *filter;
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
325 static void h_resample(uint8_t *dst, int dst_width, const uint8_t *src,
326 int src_width, int src_start, int src_incr,
327 int16_t *filters)
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)) {
340 n = (((src_width - NB_TAPS + 1) << POS_FRAC_BITS) - 1 - src_start) /
341 src_incr;
342 } else {
343 n = dst_width;
344 }
345 #ifdef HAVE_MMX
346 if ((mm_flags & MM_MMX) && NB_TAPS == 4)
347 h_resample_fast4_mmx(dst, n,
348 src, src_width, src_start, src_incr, filters);
349 else
350 #endif
351 h_resample_fast(dst, n,
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;
357 h_resample_slow(dst, dst_width,
358 src, src_width, src_start, src_incr, filters);
359 }
360 }
361
362 static void component_resample(ImgReSampleContext *s,
363 uint8_t *output, int owrap, int owidth, int oheight,
364 uint8_t *input, int iwrap, int iwidth, int iheight)
365 {
366 int src_y, src_y1, last_src_y, ring_y, phase_y, y1, y;
367 uint8_t *new_line, *src_line;
368
369 last_src_y = - FCENTER - 1;
370 /* position of the bottom of the filter in the source image */
371 src_y = (last_src_y + NB_TAPS) * POS_FRAC;
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++;
380 /* handle limit conditions : replicate line (slightly
381 inefficient because we filter multiple times) */
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 */
391 h_resample(new_line, owidth,
392 src_line, iwidth, - FCENTER * POS_FRAC, s->h_incr,
393 &s->h_filters[0][0]);
394 /* handle ring buffer wrapping */
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);
402 #ifdef HAVE_MMX
403 /* desactivated MMX because loss of precision */
404 if ((mm_flags & MM_MMX) && NB_TAPS == 4 && 0)
405 v_resample4_mmx(output, owidth,
406 s->line_buf + (ring_y - NB_TAPS + 1) * owidth, owidth,
407 &s->v_filters[phase_y][0]);
408 else
409 #endif
410 #ifdef HAVE_ALTIVEC
411 if ((mm_flags & MM_ALTIVEC) && NB_TAPS == 4 && FILTER_BITS <= 6)
412 v_resample16_altivec(output, owidth,
413 s->line_buf + (ring_y - NB_TAPS + 1) * owidth,
414 owidth, &s->v_filters[phase_y][0]);
415 else
416 #endif
417 v_resample(output, owidth,
418 s->line_buf + (ring_y - NB_TAPS + 1) * owidth, owidth,
419 &s->v_filters[phase_y][0]);
420
421 src_y += s->v_incr;
422
423 output += owrap;
424 }
425 }
426
427 ImgReSampleContext *img_resample_full_init(int owidth, int oheight,
428 int iwidth, int iheight,
429 int topBand, int bottomBand,
430 int leftBand, int rightBand,
431 int padtop, int padbottom,
432 int padleft, int padright)
433 {
434 ImgReSampleContext *s;
435
436 if (!owidth || !oheight || !iwidth || !iheight)
437 return NULL;
438
439 s = av_mallocz(sizeof(ImgReSampleContext));
440 if (!s)
441 return NULL;
442 if((unsigned)owidth >= UINT_MAX / (LINE_BUF_HEIGHT + NB_TAPS))
443 goto fail;
444 s->line_buf = av_mallocz(owidth * (LINE_BUF_HEIGHT + NB_TAPS));
445 if (!s->line_buf)
446 goto fail;
447
448 s->owidth = owidth;
449 s->oheight = oheight;
450 s->iwidth = iwidth;
451 s->iheight = iheight;
452
453 s->topBand = topBand;
454 s->bottomBand = bottomBand;
455 s->leftBand = leftBand;
456 s->rightBand = rightBand;
457
458 s->padtop = padtop;
459 s->padbottom = padbottom;
460 s->padleft = padleft;
461 s->padright = padright;
462
463 s->pad_owidth = owidth - (padleft + padright);
464 s->pad_oheight = oheight - (padtop + padbottom);
465
466 s->h_incr = ((iwidth - leftBand - rightBand) * POS_FRAC) / s->pad_owidth;
467 s->v_incr = ((iheight - topBand - bottomBand) * POS_FRAC) / s->pad_oheight;
468
469 av_build_filter(&s->h_filters[0][0], (float) s->pad_owidth /
470 (float) (iwidth - leftBand - rightBand), NB_TAPS, NB_PHASES, 1<<FILTER_BITS, 0);
471 av_build_filter(&s->v_filters[0][0], (float) s->pad_oheight /
472 (float) (iheight - topBand - bottomBand), NB_TAPS, NB_PHASES, 1<<FILTER_BITS, 0);
473
474 return s;
475 fail:
476 av_free(s);
477 return NULL;
478 }
479
480 ImgReSampleContext *img_resample_init(int owidth, int oheight,
481 int iwidth, int iheight)
482 {
483 return img_resample_full_init(owidth, oheight, iwidth, iheight,
484 0, 0, 0, 0, 0, 0, 0, 0);
485 }
486
487 void img_resample(ImgReSampleContext *s,
488 AVPicture *output, const AVPicture *input)
489 {
490 int i, shift;
491 uint8_t* optr;
492
493 for (i=0;i<3;i++) {
494 shift = (i == 0) ? 0 : 1;
495
496 optr = output->data[i] + (((output->linesize[i] *
497 s->padtop) + s->padleft) >> shift);
498
499 component_resample(s, optr, output->linesize[i],
500 s->pad_owidth >> shift, s->pad_oheight >> shift,
501 input->data[i] + (input->linesize[i] *
502 (s->topBand >> shift)) + (s->leftBand >> shift),
503 input->linesize[i], ((s->iwidth - s->leftBand -
504 s->rightBand) >> shift),
505 (s->iheight - s->topBand - s->bottomBand) >> shift);
506 }
507 }
508
509 void img_resample_close(ImgReSampleContext *s)
510 {
511 av_free(s->line_buf);
512 av_free(s);
513 }
514
515 static const char *context_to_name(void* ptr)
516 {
517 return "imgconvert";
518 }
519
520 static const AVClass context_class = { "imgresample", context_to_name, NULL };
521
522 struct SwsContext *sws_getContext(int srcW, int srcH, int srcFormat,
523 int dstW, int dstH, int dstFormat,
524 int flags, SwsFilter *srcFilter,
525 SwsFilter *dstFilter, double *param)
526 {
527 struct SwsContext *ctx;
528
529 ctx = av_malloc(sizeof(struct SwsContext));
530 if (!ctx) {
531 av_log(NULL, AV_LOG_ERROR, "Cannot allocate a resampling context!\n");
532
533 return NULL;
534 }
535 ctx->av_class = &context_class;
536
537 if ((srcH != dstH) || (srcW != dstW)) {
538 if ((srcFormat != PIX_FMT_YUV420P) || (dstFormat != PIX_FMT_YUV420P)) {
539 av_log(ctx, AV_LOG_INFO, "PIX_FMT_YUV420P will be used as an intermediate format for rescaling\n");
540 }
541 ctx->resampling_ctx = img_resample_init(dstW, dstH, srcW, srcH);
542 } else {
543 ctx->resampling_ctx = av_malloc(sizeof(ImgReSampleContext));
544 ctx->resampling_ctx->iheight = srcH;
545 ctx->resampling_ctx->iwidth = srcW;
546 ctx->resampling_ctx->oheight = dstH;
547 ctx->resampling_ctx->owidth = dstW;
548 }
549 ctx->src_pix_fmt = srcFormat;
550 ctx->dst_pix_fmt = dstFormat;
551
552 return ctx;
553 }
554
555 void sws_freeContext(struct SwsContext *ctx)
556 {
557 if (!ctx)
558 return;
559 if ((ctx->resampling_ctx->iwidth != ctx->resampling_ctx->owidth) ||
560 (ctx->resampling_ctx->iheight != ctx->resampling_ctx->oheight)) {
561 img_resample_close(ctx->resampling_ctx);
562 } else {
563 av_free(ctx->resampling_ctx);
564 }
565 av_free(ctx);
566 }
567
568
569 /**
570 * Checks if context is valid or reallocs a new one instead.
571 * If context is NULL, just calls sws_getContext() to get a new one.
572 * Otherwise, checks if the parameters are the same already saved in context.
573 * If that is the case, returns the current context.
574 * Otherwise, frees context and gets a new one.
575 *
576 * Be warned that srcFilter, dstFilter are not checked, they are
577 * asumed to remain valid.
578 */
579 struct SwsContext *sws_getCachedContext(struct SwsContext *ctx,
580 int srcW, int srcH, int srcFormat,
581 int dstW, int dstH, int dstFormat, int flags,
582 SwsFilter *srcFilter, SwsFilter *dstFilter, double *param)
583 {
584 if (ctx != NULL) {
585 if ((ctx->resampling_ctx->iwidth != srcW) ||
586 (ctx->resampling_ctx->iheight != srcH) ||
587 (ctx->src_pix_fmt != srcFormat) ||
588 (ctx->resampling_ctx->owidth != dstW) ||
589 (ctx->resampling_ctx->oheight != dstH) ||
590 (ctx->dst_pix_fmt != dstFormat))
591 {
592 sws_freeContext(ctx);
593 ctx = NULL;
594 }
595 }
596 if (ctx == NULL) {
597 return sws_getContext(srcW, srcH, srcFormat,
598 dstW, dstH, dstFormat, flags,
599 srcFilter, dstFilter, param);
600 }
601 return ctx;
602 }
603
604 int sws_scale(struct SwsContext *ctx, uint8_t* src[], int srcStride[],
605 int srcSliceY, int srcSliceH, uint8_t* dst[], int dstStride[])
606 {
607 AVPicture src_pict, dst_pict;
608 int i, res = 0;
609 AVPicture picture_format_temp;
610 AVPicture picture_resample_temp, *formatted_picture, *resampled_picture;
611 uint8_t *buf1 = NULL, *buf2 = NULL;
612 enum PixelFormat current_pix_fmt;
613
614 for (i = 0; i < 4; i++) {
615 src_pict.data[i] = src[i];
616 src_pict.linesize[i] = srcStride[i];
617 dst_pict.data[i] = dst[i];
618 dst_pict.linesize[i] = dstStride[i];
619 }
620 if ((ctx->resampling_ctx->iwidth != ctx->resampling_ctx->owidth) ||
621 (ctx->resampling_ctx->iheight != ctx->resampling_ctx->oheight)) {
622 /* We have to rescale the picture, but only YUV420P rescaling is supported... */
623
624 if (ctx->src_pix_fmt != PIX_FMT_YUV420P) {
625 int size;
626
627 /* create temporary picture for rescaling input*/
628 size = avpicture_get_size(PIX_FMT_YUV420P, ctx->resampling_ctx->iwidth, ctx->resampling_ctx->iheight);
629 buf1 = av_malloc(size);
630 if (!buf1) {
631 res = -1;
632 goto the_end;
633 }
634 formatted_picture = &picture_format_temp;
635 avpicture_fill((AVPicture*)formatted_picture, buf1,
636 PIX_FMT_YUV420P, ctx->resampling_ctx->iwidth, ctx->resampling_ctx->iheight);
637
638 if (img_convert((AVPicture*)formatted_picture, PIX_FMT_YUV420P,
639 &src_pict, ctx->src_pix_fmt,
640 ctx->resampling_ctx->iwidth, ctx->resampling_ctx->iheight) < 0) {
641
642 av_log(ctx, AV_LOG_ERROR, "pixel format conversion not handled\n");
643 res = -1;
644 goto the_end;
645 }
646 } else {
647 formatted_picture = &src_pict;
648 }
649
650 if (ctx->dst_pix_fmt != PIX_FMT_YUV420P) {
651 int size;
652
653 /* create temporary picture for rescaling output*/
654 size = avpicture_get_size(PIX_FMT_YUV420P, ctx->resampling_ctx->owidth, ctx->resampling_ctx->oheight);
655 buf2 = av_malloc(size);
656 if (!buf2) {
657 res = -1;
658 goto the_end;
659 }
660 resampled_picture = &picture_resample_temp;
661 avpicture_fill((AVPicture*)resampled_picture, buf2,
662 PIX_FMT_YUV420P, ctx->resampling_ctx->owidth, ctx->resampling_ctx->oheight);
663
664 } else {
665 resampled_picture = &dst_pict;
666 }
667
668 /* ...and finally rescale!!! */
669 img_resample(ctx->resampling_ctx, resampled_picture, formatted_picture);
670 current_pix_fmt = PIX_FMT_YUV420P;
671 } else {
672 resampled_picture = &src_pict;
673 current_pix_fmt = ctx->src_pix_fmt;
674 }
675
676 if (current_pix_fmt != ctx->dst_pix_fmt) {
677 if (img_convert(&dst_pict, ctx->dst_pix_fmt,
678 resampled_picture, current_pix_fmt,
679 ctx->resampling_ctx->owidth, ctx->resampling_ctx->oheight) < 0) {
680
681 av_log(ctx, AV_LOG_ERROR, "pixel format conversion not handled\n");
682
683 res = -1;
684 goto the_end;
685 }
686 } else if (resampled_picture != &dst_pict) {
687 av_picture_copy(&dst_pict, resampled_picture, current_pix_fmt,
688 ctx->resampling_ctx->owidth, ctx->resampling_ctx->oheight);
689 }
690
691 the_end:
692 av_free(buf1);
693 av_free(buf2);
694 return res;
695 }
696
697
698 #ifdef TEST
699 #include <stdio.h>
700 #undef exit
701
702 /* input */
703 #define XSIZE 256
704 #define YSIZE 256
705 uint8_t img[XSIZE * YSIZE];
706
707 /* output */
708 #define XSIZE1 512
709 #define YSIZE1 512
710 uint8_t img1[XSIZE1 * YSIZE1];
711 uint8_t img2[XSIZE1 * YSIZE1];
712
713 void save_pgm(const char *filename, uint8_t *img, int xsize, int ysize)
714 {
715 #undef fprintf
716 FILE *f;
717 f=fopen(filename,"w");
718 fprintf(f,"P5\n%d %d\n%d\n", xsize, ysize, 255);
719 fwrite(img,1, xsize * ysize,f);
720 fclose(f);
721 #define fprintf please_use_av_log
722 }
723
724 static void dump_filter(int16_t *filter)
725 {
726 int i, ph;
727
728 for(ph=0;ph<NB_PHASES;ph++) {
729 av_log(NULL, AV_LOG_INFO, "%2d: ", ph);
730 for(i=0;i<NB_TAPS;i++) {
731 av_log(NULL, AV_LOG_INFO, " %5.2f", filter[ph * NB_TAPS + i] / 256.0);
732 }
733 av_log(NULL, AV_LOG_INFO, "\n");
734 }
735 }
736
737 #ifdef HAVE_MMX
738 int mm_flags;
739 #endif
740
741 int main(int argc, char **argv)
742 {
743 int x, y, v, i, xsize, ysize;
744 ImgReSampleContext *s;
745 float fact, factors[] = { 1/2.0, 3.0/4.0, 1.0, 4.0/3.0, 16.0/9.0, 2.0 };
746 char buf[256];
747
748 /* build test image */
749 for(y=0;y<YSIZE;y++) {
750 for(x=0;x<XSIZE;x++) {
751 if (x < XSIZE/2 && y < YSIZE/2) {
752 if (x < XSIZE/4 && y < YSIZE/4) {
753 if ((x % 10) <= 6 &&
754 (y % 10) <= 6)
755 v = 0xff;
756 else
757 v = 0x00;
758 } else if (x < XSIZE/4) {
759 if (x & 1)
760 v = 0xff;
761 else
762 v = 0;
763 } else if (y < XSIZE/4) {
764 if (y & 1)
765 v = 0xff;
766 else
767 v = 0;
768 } else {
769 if (y < YSIZE*3/8) {
770 if ((y+x) & 1)
771 v = 0xff;
772 else
773 v = 0;
774 } else {
775 if (((x+3) % 4) <= 1 &&
776 ((y+3) % 4) <= 1)
777 v = 0xff;
778 else
779 v = 0x00;
780 }
781 }
782 } else if (x < XSIZE/2) {
783 v = ((x - (XSIZE/2)) * 255) / (XSIZE/2);
784 } else if (y < XSIZE/2) {
785 v = ((y - (XSIZE/2)) * 255) / (XSIZE/2);
786 } else {
787 v = ((x + y - XSIZE) * 255) / XSIZE;
788 }
789 img[(YSIZE - y) * XSIZE + (XSIZE - x)] = v;
790 }
791 }
792 save_pgm("/tmp/in.pgm", img, XSIZE, YSIZE);
793 for(i=0;i<FF_ARRAY_ELEMS(factors);i++) {
794 fact = factors[i];
795 xsize = (int)(XSIZE * fact);
796 ysize = (int)((YSIZE - 100) * fact);
797 s = img_resample_full_init(xsize, ysize, XSIZE, YSIZE, 50 ,50, 0, 0, 0, 0, 0, 0);
798 av_log(NULL, AV_LOG_INFO, "Factor=%0.2f\n", fact);
799 dump_filter(&s->h_filters[0][0]);
800 component_resample(s, img1, xsize, xsize, ysize,
801 img + 50 * XSIZE, XSIZE, XSIZE, YSIZE - 100);
802 img_resample_close(s);
803
804 snprintf(buf, sizeof(buf), "/tmp/out%d.pgm", i);
805 save_pgm(buf, img1, xsize, ysize);
806 }
807
808 /* mmx test */
809 #ifdef HAVE_MMX
810 av_log(NULL, AV_LOG_INFO, "MMX test\n");
811 fact = 0.72;
812 xsize = (int)(XSIZE * fact);
813 ysize = (int)(YSIZE * fact);
814 mm_flags = MM_MMX;
815 s = img_resample_init(xsize, ysize, XSIZE, YSIZE);
816 component_resample(s, img1, xsize, xsize, ysize,
817 img, XSIZE, XSIZE, YSIZE);
818
819 mm_flags = 0;
820 s = img_resample_init(xsize, ysize, XSIZE, YSIZE);
821 component_resample(s, img2, xsize, xsize, ysize,
822 img, XSIZE, XSIZE, YSIZE);
823 if (memcmp(img1, img2, xsize * ysize) != 0) {
824 av_log(NULL, AV_LOG_ERROR, "mmx error\n");
825 exit(1);
826 }
827 av_log(NULL, AV_LOG_INFO, "MMX OK\n");
828 #endif /* HAVE_MMX */
829 return 0;
830 }
831
832 #endif /* TEST */