Statistics
| Branch: | Revision:

ffmpeg / libavcodec / imgresample.c @ 0e9fbd35

History | View | Annotate | Download (24.9 KB)

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 "swscale.h"
29
#include "dsputil.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
    AVClass *av_class;
52
    struct ImgReSampleContext *resampling_ctx;
53
    enum PixelFormat src_pix_fmt, dst_pix_fmt;
54
};
55

    
56
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
};
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, owidth,
414
                                &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_init(int owidth, int oheight,
428
                                      int iwidth, int iheight)
429
{
430
    return img_resample_full_init(owidth, oheight, iwidth, iheight,
431
            0, 0, 0, 0, 0, 0, 0, 0);
432
}
433

    
434
ImgReSampleContext *img_resample_full_init(int owidth, int oheight,
435
                                      int iwidth, int iheight,
436
                                      int topBand, int bottomBand,
437
        int leftBand, int rightBand,
438
        int padtop, int padbottom,
439
        int padleft, int padright)
440
{
441
    ImgReSampleContext *s;
442

    
443
    if (!owidth || !oheight || !iwidth || !iheight)
444
        return NULL;
445

    
446
    s = av_mallocz(sizeof(ImgReSampleContext));
447
    if (!s)
448
        return NULL;
449
    if((unsigned)owidth >= UINT_MAX / (LINE_BUF_HEIGHT + NB_TAPS))
450
        return NULL;
451
    s->line_buf = av_mallocz(owidth * (LINE_BUF_HEIGHT + NB_TAPS));
452
    if (!s->line_buf)
453
        goto fail;
454

    
455
    s->owidth = owidth;
456
    s->oheight = oheight;
457
    s->iwidth = iwidth;
458
    s->iheight = iheight;
459

    
460
    s->topBand = topBand;
461
    s->bottomBand = bottomBand;
462
    s->leftBand = leftBand;
463
    s->rightBand = rightBand;
464

    
465
    s->padtop = padtop;
466
    s->padbottom = padbottom;
467
    s->padleft = padleft;
468
    s->padright = padright;
469

    
470
    s->pad_owidth = owidth - (padleft + padright);
471
    s->pad_oheight = oheight - (padtop + padbottom);
472

    
473
    s->h_incr = ((iwidth - leftBand - rightBand) * POS_FRAC) / s->pad_owidth;
474
    s->v_incr = ((iheight - topBand - bottomBand) * POS_FRAC) / s->pad_oheight;
475

    
476
    av_build_filter(&s->h_filters[0][0], (float) s->pad_owidth  /
477
            (float) (iwidth - leftBand - rightBand), NB_TAPS, NB_PHASES, 1<<FILTER_BITS, 0);
478
    av_build_filter(&s->v_filters[0][0], (float) s->pad_oheight /
479
            (float) (iheight - topBand - bottomBand), NB_TAPS, NB_PHASES, 1<<FILTER_BITS, 0);
480

    
481
    return s;
482
fail:
483
    av_free(s);
484
    return NULL;
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 AVClass context_class = { "imgresample", NULL, NULL };
516

    
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));
525
    if (!ctx) {
526
        av_log(NULL, AV_LOG_ERROR, "Cannot allocate a resampling context!\n");
527

    
528
        return NULL;
529
    }
530
    ctx->av_class = &context_class;
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
{
552
    if (!ctx)
553
        return;
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

    
563

    
564
/**
565
 * Checks if context is valid or reallocs a new one instead.
566
 * If context is NULL, just calls sws_getContext() to get a new one.
567
 * Otherwise, checks if the parameters are the same already saved in context.
568
 * If that is the case, returns the current context.
569
 * Otherwise, frees context and gets a new one.
570
 *
571
 * Be warned that srcFilter, dstFilter are not checked, they are
572
 * asumed to remain valid.
573
 */
574
struct SwsContext *sws_getCachedContext(struct SwsContext *ctx,
575
                        int srcW, int srcH, int srcFormat,
576
                        int dstW, int dstH, int dstFormat, int flags,
577
                        SwsFilter *srcFilter, SwsFilter *dstFilter, double *param)
578
{
579
    if (ctx != NULL) {
580
        if ((ctx->resampling_ctx->iwidth != srcW) ||
581
                        (ctx->resampling_ctx->iheight != srcH) ||
582
                        (ctx->src_pix_fmt != srcFormat) ||
583
                        (ctx->resampling_ctx->owidth != dstW) ||
584
                        (ctx->resampling_ctx->oheight != dstH) ||
585
                        (ctx->dst_pix_fmt != dstFormat))
586
        {
587
            sws_freeContext(ctx);
588
            ctx = NULL;
589
        }
590
    }
591
    if (ctx == NULL) {
592
        return sws_getContext(srcW, srcH, srcFormat,
593
                        dstW, dstH, dstFormat, flags,
594
                        srcFilter, dstFilter, param);
595
    }
596
    return ctx;
597
}
598

    
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

    
609
    for (i = 0; i < 4; i++) {
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
        }
681
    } else if (resampled_picture != &dst_pict) {
682
        av_picture_copy(&dst_pict, resampled_picture, current_pix_fmt,
683
                        ctx->resampling_ctx->owidth, ctx->resampling_ctx->oheight);
684
    }
685

    
686
the_end:
687
    av_free(buf1);
688
    av_free(buf2);
689
    return res;
690
}
691

    
692

    
693
#ifdef TEST
694
#include <stdio.h>
695
#undef exit
696

    
697
/* input */
698
#define XSIZE 256
699
#define YSIZE 256
700
uint8_t img[XSIZE * YSIZE];
701

    
702
/* output */
703
#define XSIZE1 512
704
#define YSIZE1 512
705
uint8_t img1[XSIZE1 * YSIZE1];
706
uint8_t img2[XSIZE1 * YSIZE1];
707

    
708
void save_pgm(const char *filename, uint8_t *img, int xsize, int ysize)
709
{
710
#undef fprintf
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);
716
#define fprintf please_use_av_log
717
}
718

    
719
static void dump_filter(int16_t *filter)
720
{
721
    int i, ph;
722

    
723
    for(ph=0;ph<NB_PHASES;ph++) {
724
        av_log(NULL, AV_LOG_INFO, "%2d: ", ph);
725
        for(i=0;i<NB_TAPS;i++) {
726
            av_log(NULL, AV_LOG_INFO, " %5.2f", filter[ph * NB_TAPS + i] / 256.0);
727
        }
728
        av_log(NULL, AV_LOG_INFO, "\n");
729
    }
730
}
731

    
732
#ifdef HAVE_MMX
733
int mm_flags;
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) {
754
                    if (x & 1)
755
                        v = 0xff;
756
                    else
757
                        v = 0;
758
                } else if (y < XSIZE/4) {
759
                    if (y & 1)
760
                        v = 0xff;
761
                    else
762
                        v = 0;
763
                } else {
764
                    if (y < YSIZE*3/8) {
765
                        if ((y+x) & 1)
766
                            v = 0xff;
767
                        else
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
            }
784
            img[(YSIZE - y) * XSIZE + (XSIZE - x)] = v;
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);
791
        ysize = (int)((YSIZE - 100) * fact);
792
        s = img_resample_full_init(xsize, ysize, XSIZE, YSIZE, 50 ,50, 0, 0, 0, 0, 0, 0);
793
        av_log(NULL, AV_LOG_INFO, "Factor=%0.2f\n", fact);
794
        dump_filter(&s->h_filters[0][0]);
795
        component_resample(s, img1, xsize, xsize, ysize,
796
                           img + 50 * XSIZE, XSIZE, XSIZE, YSIZE - 100);
797
        img_resample_close(s);
798

    
799
        snprintf(buf, sizeof(buf), "/tmp/out%d.pgm", i);
800
        save_pgm(buf, img1, xsize, ysize);
801
    }
802

    
803
    /* mmx test */
804
#ifdef HAVE_MMX
805
    av_log(NULL, AV_LOG_INFO, "MMX test\n");
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) {
819
        av_log(NULL, AV_LOG_ERROR, "mmx error\n");
820
        exit(1);
821
    }
822
    av_log(NULL, AV_LOG_INFO, "MMX OK\n");
823
#endif /* HAVE_MMX */
824
    return 0;
825
}
826

    
827
#endif /* TEST */