Coverage Report

Created: 2026-08-17 07:50

next uncovered line (L), next uncovered region (R), next uncovered branch (B)
/src/ffmpeg/libswscale/utils.c
Line
Count
Source
1
/*
2
 * Copyright (C) 2024 Niklas Haas
3
 * Copyright (C) 2001-2003 Michael Niedermayer <michaelni@gmx.at>
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
#include "config.h"
23
24
#define _DEFAULT_SOURCE
25
#include <inttypes.h>
26
#include <math.h>
27
#include <stdio.h>
28
#include <string.h>
29
30
#include "libavutil/attributes.h"
31
#include "libavutil/avassert.h"
32
#include "libavutil/cpu.h"
33
#include "libavutil/csp.h"
34
#include "libavutil/emms.h"
35
#include "libavutil/imgutils.h"
36
#include "libavutil/intreadwrite.h"
37
#include "libavutil/libm.h"
38
#include "libavutil/mathematics.h"
39
#include "libavutil/mem.h"
40
#include "libavutil/opt.h"
41
#include "libavutil/pixdesc.h"
42
#include "libavutil/refstruct.h"
43
#include "libavutil/slicethread.h"
44
#include "libavutil/thread.h"
45
#include "libavutil/aarch64/cpu.h"
46
#include "libavutil/ppc/cpu.h"
47
#include "libavutil/x86/cpu.h"
48
#include "libavutil/loongarch/cpu.h"
49
50
#include "rgb2rgb.h"
51
#include "swscale.h"
52
#include "swscale_internal.h"
53
#include "graph.h"
54
#include "jit.h"
55
56
#if CONFIG_VULKAN
57
#include "vulkan/ops.h"
58
#endif
59
60
SwsBackend ff_sws_enabled_backends(const SwsContext *ctx)
61
0
{
62
0
    if (ctx->backends)
63
0
        return ctx->backends;
64
65
0
    SwsBackend fallback = SWS_BACKEND_STABLE;
66
0
    if (ctx->flags & SWS_UNSTABLE)
67
0
        fallback |= SWS_BACKEND_UNSTABLE;
68
69
0
    return fallback;
70
0
}
71
72
/**
73
 * Allocate and return an SwsContext without performing initialization.
74
 */
75
static SwsContext *alloc_set_opts(int srcW, int srcH, enum AVPixelFormat srcFormat,
76
                                  int dstW, int dstH, enum AVPixelFormat dstFormat,
77
                                  int flags, const double *param)
78
0
{
79
0
    SwsContext *sws = sws_alloc_context();
80
0
    if (!sws)
81
0
        return NULL;
82
83
0
    sws->flags = flags;
84
0
    sws->src_w = srcW;
85
0
    sws->src_h = srcH;
86
0
    sws->dst_w = dstW;
87
0
    sws->dst_h      = dstH;
88
0
    sws->src_format = srcFormat;
89
0
    sws->dst_format = dstFormat;
90
91
0
    for (int i = 0; param && i < SWS_NUM_SCALER_PARAMS; i++)
92
0
        sws->scaler_params[i] = param[i];
93
94
0
    return sws;
95
0
}
96
97
int ff_shuffle_filter_coefficients(SwsInternal *c, int *filterPos,
98
                                   int filterSize, int16_t *filter,
99
                                   int dstW)
100
0
{
101
#if ARCH_X86_64
102
    int i, j, k;
103
    int cpu_flags = av_get_cpu_flags();
104
    if (!filter)
105
        return 0;
106
    if (EXTERNAL_AVX2_FAST(cpu_flags) && !(cpu_flags & AV_CPU_FLAG_SLOW_GATHER)) {
107
        if ((c->srcBpc == 8) && (c->dstBpc <= 14)) {
108
           int16_t *filterCopy = NULL;
109
           if (filterSize > 4) {
110
               filterCopy = av_malloc_array(dstW, filterSize * sizeof(*filterCopy));
111
               if (!filterCopy)
112
                   return AVERROR(ENOMEM);
113
               memcpy(filterCopy, filter, dstW * filterSize * sizeof(int16_t));
114
           }
115
           // Do not swap filterPos for pixels which won't be processed by
116
           // the main loop.
117
           for (i = 0; i + 16 <= dstW; i += 16) {
118
               FFSWAP(int, filterPos[i + 2], filterPos[i + 4]);
119
               FFSWAP(int, filterPos[i + 3], filterPos[i + 5]);
120
               FFSWAP(int, filterPos[i + 10], filterPos[i + 12]);
121
               FFSWAP(int, filterPos[i + 11], filterPos[i + 13]);
122
           }
123
           if (filterSize > 4) {
124
               // 16 pixels are processed at a time.
125
               for (i = 0; i + 16 <= dstW; i += 16) {
126
                   // 4 filter coeffs are processed at a time.
127
                   for (k = 0; k + 4 <= filterSize; k += 4) {
128
                       for (j = 0; j < 16; ++j) {
129
                           int from = (i + j) * filterSize + k;
130
                           int to = i * filterSize + j * 4 + k * 16;
131
                           memcpy(&filter[to], &filterCopy[from], 4 * sizeof(int16_t));
132
                       }
133
                   }
134
               }
135
               // 4 pixels are processed at a time in the tail.
136
               for (; i < dstW; i += 4) {
137
                   // 4 filter coeffs are processed at a time.
138
                   int rem = dstW - i >= 4 ? 4 : dstW - i;
139
                   for (k = 0; k + 4 <= filterSize; k += 4) {
140
                       for (j = 0; j < rem; ++j) {
141
                           int from = (i + j) * filterSize + k;
142
                           int to = i * filterSize + j * 4 + k * 4;
143
                           memcpy(&filter[to], &filterCopy[from], 4 * sizeof(int16_t));
144
                       }
145
                   }
146
               }
147
           }
148
           av_free(filterCopy);
149
        }
150
    }
151
#endif
152
0
    return 0;
153
0
}
154
155
static double getSplineCoeff(double a, double b, double c, double d,
156
                             double dist)
157
0
{
158
0
    if (dist <= 1.0)
159
0
        return ((d * dist + c) * dist + b) * dist + a;
160
0
    else
161
0
        return getSplineCoeff(0.0,
162
0
                               b + 2.0 * c + 3.0 * d,
163
0
                               c + 3.0 * d,
164
0
                              -b - 3.0 * c - 6.0 * d,
165
0
                              dist - 1.0);
166
0
}
167
168
static av_cold int get_local_pos(SwsInternal *s, int chr_subsample, int pos, int dir)
169
0
{
170
0
    if (pos == -1 || pos <= -513) {
171
0
        pos = (128 << chr_subsample) - 128;
172
0
    }
173
0
    pos += 128; // relative to ideal left edge
174
0
    return pos >> chr_subsample;
175
0
}
176
177
typedef struct {
178
    int flag;                   ///< flag associated to the algorithm
179
    const char *description;    ///< human-readable description
180
    int size_factor;            ///< size factor used when initing the filters
181
} ScaleAlgorithm;
182
183
static const ScaleAlgorithm scale_algorithms[] = {
184
    { SWS_AREA,          "area averaging",                  1 /* downscale only, for upscale it is bilinear */ },
185
    { SWS_BICUBIC,       "bicubic",                         4 },
186
    { SWS_BICUBLIN,      "luma bicubic / chroma bilinear", -1 },
187
    { SWS_BILINEAR,      "bilinear",                        2 },
188
    { SWS_FAST_BILINEAR, "fast bilinear",                  -1 },
189
    { SWS_GAUSS,         "Gaussian",                        8 /* infinite ;) */ },
190
    { SWS_LANCZOS,       "Lanczos",                        -1 /* custom */ },
191
    { SWS_POINT,         "nearest neighbor / point",       -1 },
192
    { SWS_SINC,          "sinc",                           20 /* infinite ;) */ },
193
    { SWS_SPLINE,        "bicubic spline",                 20 /* infinite :)*/ },
194
    { SWS_X,             "experimental",                    8 },
195
};
196
197
static av_cold int initFilter(int16_t **outFilter, int32_t **filterPos,
198
                              int *outFilterSize, int xInc, int srcW,
199
                              int dstW, int filterAlign, int one,
200
                              int scaler, int flags, int cpu_flags,
201
                              SwsVector *srcFilter, SwsVector *dstFilter,
202
                              double param[SWS_NUM_SCALER_PARAMS], int srcPos, int dstPos)
203
0
{
204
0
    int i;
205
0
    int filterSize;
206
0
    int filter2Size;
207
0
    int minFilterSize;
208
0
    int64_t *filter    = NULL;
209
0
    int64_t *filter2   = NULL;
210
0
    const int64_t fone = 1LL << (54 - FFMIN(av_log2(srcW/dstW), 8));
211
0
    int ret            = -1;
212
213
0
    emms_c(); // FIXME should not be required but IS (even for non-MMX versions)
214
215
    // NOTE: the +3 is for the MMX(+1) / SSE(+3) scaler which reads over the end
216
0
    if (!FF_ALLOC_TYPED_ARRAY(*filterPos, dstW + 3))
217
0
        goto nomem;
218
219
0
    if (FFABS(xInc - 0x10000) < 10 && srcPos == dstPos) { // unscaled
220
0
        int i;
221
0
        filterSize = 1;
222
0
        if (!FF_ALLOCZ_TYPED_ARRAY(filter, dstW * filterSize))
223
0
            goto nomem;
224
225
0
        for (i = 0; i < dstW; i++) {
226
0
            filter[i * filterSize] = fone;
227
0
            (*filterPos)[i]        = i;
228
0
        }
229
0
    } else if (scaler == SWS_POINT) { // lame looking point sampling mode
230
0
        int i;
231
0
        int64_t xDstInSrc;
232
0
        filterSize = 1;
233
0
        if (!FF_ALLOC_TYPED_ARRAY(filter, dstW * filterSize))
234
0
            goto nomem;
235
236
0
        xDstInSrc = ((dstPos*(int64_t)xInc)>>8) - ((srcPos*0x8000LL)>>7);
237
0
        for (i = 0; i < dstW; i++) {
238
0
            int xx = (xDstInSrc - ((filterSize - 1) << 15) + (1 << 15)) >> 16;
239
240
0
            (*filterPos)[i] = xx;
241
0
            filter[i]       = fone;
242
0
            xDstInSrc      += xInc;
243
0
        }
244
0
    } else if ((xInc <= (1 << 16) && (scaler == SWS_AREA)) ||
245
0
               (scaler == SWS_FAST_BILINEAR)) { // bilinear upscale
246
0
        int i;
247
0
        int64_t xDstInSrc;
248
0
        filterSize = 2;
249
0
        if (!FF_ALLOC_TYPED_ARRAY(filter, dstW * filterSize))
250
0
            goto nomem;
251
252
0
        xDstInSrc = ((dstPos*(int64_t)xInc)>>8) - ((srcPos*0x8000LL)>>7);
253
0
        for (i = 0; i < dstW; i++) {
254
0
            int xx = (xDstInSrc - ((filterSize - 1) << 15) + (1 << 15)) >> 16;
255
0
            int j;
256
257
0
            (*filterPos)[i] = xx;
258
            // bilinear upscale / linear interpolate / area averaging
259
0
            for (j = 0; j < filterSize; j++) {
260
0
                int64_t coeff = fone - FFABS((int64_t)xx * (1 << 16) - xDstInSrc) * (fone >> 16);
261
0
                if (coeff < 0)
262
0
                    coeff = 0;
263
0
                filter[i * filterSize + j] = coeff;
264
0
                xx++;
265
0
            }
266
0
            xDstInSrc += xInc;
267
0
        }
268
0
    } else {
269
0
        int64_t xDstInSrc;
270
0
        int sizeFactor = -1;
271
272
0
        for (i = 0; i < FF_ARRAY_ELEMS(scale_algorithms); i++) {
273
0
            if (scaler == scale_algorithms[i].flag && scale_algorithms[i].size_factor > 0) {
274
0
                sizeFactor = scale_algorithms[i].size_factor;
275
0
                break;
276
0
            }
277
0
        }
278
0
        if (scaler == SWS_LANCZOS)
279
0
            sizeFactor = param[0] != SWS_PARAM_DEFAULT ? ceil(2 * param[0]) : 6;
280
0
        av_assert0(sizeFactor > 0);
281
282
0
        if (sizeFactor > 50) {
283
0
            ret = AVERROR(EINVAL);
284
0
            goto fail;
285
0
        }
286
287
0
        if (xInc <= 1 << 16)
288
0
            filterSize = 1 + sizeFactor;    // upscale
289
0
        else
290
0
            filterSize = 1 + (sizeFactor * srcW + dstW - 1) / dstW;
291
292
0
        filterSize = FFMIN(filterSize, srcW - 2);
293
0
        filterSize = FFMAX(filterSize, 1);
294
295
0
        filter = av_malloc_array(dstW, filterSize * sizeof(*filter));
296
0
        if (!filter)
297
0
            goto nomem;
298
0
        xDstInSrc = ((dstPos*(int64_t)xInc)>>7) - ((srcPos*0x10000LL)>>7);
299
0
        for (i = 0; i < dstW; i++) {
300
0
            int xx = (xDstInSrc - (filterSize - 2) * (1LL<<16)) / (1 << 17);
301
0
            int j;
302
0
            (*filterPos)[i] = xx;
303
0
            for (j = 0; j < filterSize; j++) {
304
0
                int64_t d = (FFABS(((int64_t)xx * (1 << 17)) - xDstInSrc)) << 13;
305
0
                double floatd;
306
0
                int64_t coeff;
307
308
0
                if (xInc > 1 << 16)
309
0
                    d = d * dstW / srcW;
310
0
                floatd = d * (1.0 / (1 << 30));
311
312
0
                if (scaler == SWS_BICUBIC) {
313
0
                    int64_t B = (param[0] != SWS_PARAM_DEFAULT ? param[0] :   0) * (1 << 24);
314
0
                    int64_t C = (param[1] != SWS_PARAM_DEFAULT ? param[1] : 0.6) * (1 << 24);
315
316
0
                    if (d >= 1LL << 31) {
317
0
                        coeff = 0.0;
318
0
                    } else {
319
0
                        int64_t dd  = (d  * d) >> 30;
320
0
                        int64_t ddd = (dd * d) >> 30;
321
322
0
                        if (d < 1LL << 30)
323
0
                            coeff =  (12 * (1 << 24) -  9 * B - 6 * C) * ddd +
324
0
                                    (-18 * (1 << 24) + 12 * B + 6 * C) *  dd +
325
0
                                      (6 * (1 << 24) -  2 * B)         * (1 << 30);
326
0
                        else
327
0
                            coeff =      (-B -  6 * C) * ddd +
328
0
                                      (6 * B + 30 * C) * dd  +
329
0
                                    (-12 * B - 48 * C) * d   +
330
0
                                      (8 * B + 24 * C) * (1 << 30);
331
0
                    }
332
0
                    coeff /= (1LL<<54)/fone;
333
0
                } else if (scaler == SWS_X) {
334
0
                    double A = param[0] != SWS_PARAM_DEFAULT ? param[0] : 1.0;
335
0
                    double c;
336
337
0
                    if (floatd < 1.0)
338
0
                        c = cos(floatd * M_PI);
339
0
                    else
340
0
                        c = -1.0;
341
0
                    if (c < 0.0)
342
0
                        c = -pow(-c, A);
343
0
                    else
344
0
                        c = pow(c, A);
345
0
                    coeff = (c * 0.5 + 0.5) * fone;
346
0
                } else if (scaler == SWS_AREA) {
347
0
                    int64_t d2 = d - (1 << 29);
348
0
                    if (d2 * xInc < -(1LL << (29 + 16)))
349
0
                        coeff = 1.0 * (1LL << (30 + 16));
350
0
                    else if (d2 * xInc < (1LL << (29 + 16)))
351
0
                        coeff = -d2 * xInc + (1LL << (29 + 16));
352
0
                    else
353
0
                        coeff = 0.0;
354
0
                    coeff *= fone >> (30 + 16);
355
0
                } else if (scaler == SWS_GAUSS) {
356
0
                    double p = param[0] != SWS_PARAM_DEFAULT ? param[0] : 3.0;
357
0
                    coeff = exp2(-p * floatd * floatd) * fone;
358
0
                } else if (scaler == SWS_SINC) {
359
0
                    coeff = (d ? sin(floatd * M_PI) / (floatd * M_PI) : 1.0) * fone;
360
0
                } else if (scaler == SWS_LANCZOS) {
361
0
                    double p = param[0] != SWS_PARAM_DEFAULT ? param[0] : 3.0;
362
0
                    coeff = (d ? sin(floatd * M_PI) * sin(floatd * M_PI / p) /
363
0
                             (floatd * floatd * M_PI * M_PI / p) : 1.0) * fone;
364
0
                    if (floatd > p)
365
0
                        coeff = 0;
366
0
                } else if (scaler == SWS_BILINEAR) {
367
0
                    coeff = (1 << 30) - d;
368
0
                    if (coeff < 0)
369
0
                        coeff = 0;
370
0
                    coeff *= fone >> 30;
371
0
                } else if (scaler == SWS_SPLINE) {
372
0
                    double p = -2.196152422706632;
373
0
                    coeff = getSplineCoeff(1.0, 0.0, p, -p - 1.0, floatd) * fone;
374
0
                } else {
375
0
                    av_assert0(0);
376
0
                }
377
378
0
                filter[i * filterSize + j] = coeff;
379
0
                xx++;
380
0
            }
381
0
            xDstInSrc += 2LL * xInc;
382
0
        }
383
0
    }
384
385
    /* apply src & dst Filter to filter -> filter2
386
     * av_free(filter);
387
     */
388
0
    av_assert0(filterSize > 0);
389
0
    filter2Size = filterSize;
390
0
    if (srcFilter)
391
0
        filter2Size += srcFilter->length - 1;
392
0
    if (dstFilter)
393
0
        filter2Size += dstFilter->length - 1;
394
0
    av_assert0(filter2Size > 0);
395
0
    filter2 = av_calloc(dstW, filter2Size * sizeof(*filter2));
396
0
    if (!filter2)
397
0
        goto nomem;
398
0
    for (i = 0; i < dstW; i++) {
399
0
        int j, k;
400
401
0
        if (srcFilter) {
402
0
            for (k = 0; k < srcFilter->length; k++) {
403
0
                for (j = 0; j < filterSize; j++)
404
0
                    filter2[i * filter2Size + k + j] +=
405
0
                        srcFilter->coeff[k] * filter[i * filterSize + j];
406
0
            }
407
0
        } else {
408
0
            for (j = 0; j < filterSize; j++)
409
0
                filter2[i * filter2Size + j] = filter[i * filterSize + j];
410
0
        }
411
        // FIXME dstFilter
412
413
0
        (*filterPos)[i] += (filterSize - 1) / 2 - (filter2Size - 1) / 2;
414
0
    }
415
0
    av_freep(&filter);
416
417
    /* try to reduce the filter-size (step1 find size and shift left) */
418
    // Assume it is near normalized (*0.5 or *2.0 is OK but * 0.001 is not).
419
0
    minFilterSize = 0;
420
0
    for (i = dstW - 1; i >= 0; i--) {
421
0
        int min = filter2Size;
422
0
        int j;
423
0
        int64_t cutOff = 0.0;
424
425
        /* get rid of near zero elements on the left by shifting left */
426
0
        for (j = 0; j < filter2Size; j++) {
427
0
            int k;
428
0
            cutOff += FFABS(filter2[i * filter2Size]);
429
430
0
            if (cutOff > SWS_MAX_REDUCE_CUTOFF * fone)
431
0
                break;
432
433
            /* preserve monotonicity because the core can't handle the
434
             * filter otherwise */
435
0
            if (i < dstW - 1 && (*filterPos)[i] >= (*filterPos)[i + 1])
436
0
                break;
437
438
            // move filter coefficients left
439
0
            for (k = 1; k < filter2Size; k++)
440
0
                filter2[i * filter2Size + k - 1] = filter2[i * filter2Size + k];
441
0
            filter2[i * filter2Size + k - 1] = 0;
442
0
            (*filterPos)[i]++;
443
0
        }
444
445
0
        cutOff = 0;
446
        /* count near zeros on the right */
447
0
        for (j = filter2Size - 1; j > 0; j--) {
448
0
            cutOff += FFABS(filter2[i * filter2Size + j]);
449
450
0
            if (cutOff > SWS_MAX_REDUCE_CUTOFF * fone)
451
0
                break;
452
0
            min--;
453
0
        }
454
455
0
        if (min > minFilterSize)
456
0
            minFilterSize = min;
457
0
    }
458
459
0
    if (PPC_ALTIVEC(cpu_flags)) {
460
        // we can handle the special case 4, so we don't want to go the full 8
461
0
        if (minFilterSize < 5)
462
0
            filterAlign = 4;
463
464
        /* We really don't want to waste our time doing useless computation, so
465
         * fall back on the scalar C code for very small filters.
466
         * Vectorizing is worth it only if you have a decent-sized vector. */
467
0
        if (minFilterSize < 3)
468
0
            filterAlign = 1;
469
0
    }
470
471
0
    if (HAVE_MMX && cpu_flags & AV_CPU_FLAG_MMX || have_neon(cpu_flags)) {
472
        // special case for unscaled vertical filtering
473
0
        if (minFilterSize == 1 && filterAlign == 2)
474
0
            filterAlign = 1;
475
0
    }
476
477
0
    if (have_lasx(cpu_flags) || have_lsx(cpu_flags)) {
478
0
        int reNum = minFilterSize & (0x07);
479
480
0
        if (minFilterSize < 5)
481
0
            filterAlign = 4;
482
0
        if (reNum < 3)
483
0
            filterAlign = 1;
484
0
    }
485
486
0
    av_assert0(minFilterSize > 0);
487
0
    filterSize = (minFilterSize + (filterAlign - 1)) & (~(filterAlign - 1));
488
0
    av_assert0(filterSize > 0);
489
0
    filter = av_malloc_array(dstW, filterSize * sizeof(*filter));
490
0
    if (!filter)
491
0
        goto nomem;
492
0
    if (filterSize >= MAX_FILTER_SIZE * 16 /
493
0
                      ((flags & SWS_ACCURATE_RND) ? APCK_SIZE : 16)) {
494
0
        ret = RETCODE_USE_CASCADE;
495
0
        goto fail;
496
0
    }
497
0
    *outFilterSize = filterSize;
498
499
0
    if (flags & SWS_PRINT_INFO)
500
0
        av_log(NULL, AV_LOG_VERBOSE,
501
0
               "SwScaler: reducing / aligning filtersize %d -> %d\n",
502
0
               filter2Size, filterSize);
503
    /* try to reduce the filter-size (step2 reduce it) */
504
0
    for (i = 0; i < dstW; i++) {
505
0
        int j;
506
507
0
        for (j = 0; j < filterSize; j++) {
508
0
            if (j >= filter2Size)
509
0
                filter[i * filterSize + j] = 0;
510
0
            else
511
0
                filter[i * filterSize + j] = filter2[i * filter2Size + j];
512
0
            if ((flags & SWS_BITEXACT) && j >= minFilterSize)
513
0
                filter[i * filterSize + j] = 0;
514
0
        }
515
0
    }
516
517
    // FIXME try to align filterPos if possible
518
519
    // fix borders
520
0
    for (i = 0; i < dstW; i++) {
521
0
        int j;
522
0
        if ((*filterPos)[i] < 0) {
523
            // move filter coefficients left to compensate for filterPos
524
0
            for (j = 1; j < filterSize; j++) {
525
0
                int left = FFMAX(j + (*filterPos)[i], 0);
526
0
                filter[i * filterSize + left] += filter[i * filterSize + j];
527
0
                filter[i * filterSize + j]     = 0;
528
0
            }
529
0
            (*filterPos)[i]= 0;
530
0
        }
531
532
0
        if ((*filterPos)[i] + filterSize > srcW) {
533
0
            int shift = (*filterPos)[i] + FFMIN(filterSize - srcW, 0);
534
0
            int64_t acc = 0;
535
536
0
            for (j = filterSize - 1; j >= 0; j--) {
537
0
                if ((*filterPos)[i] + j >= srcW) {
538
0
                    acc += filter[i * filterSize + j];
539
0
                    filter[i * filterSize + j] = 0;
540
0
                }
541
0
            }
542
0
            for (j = filterSize - 1; j >= 0; j--) {
543
0
                if (j < shift) {
544
0
                    filter[i * filterSize + j] = 0;
545
0
                } else {
546
0
                    filter[i * filterSize + j] = filter[i * filterSize + j - shift];
547
0
                }
548
0
            }
549
550
0
            (*filterPos)[i]-= shift;
551
0
            filter[i * filterSize + srcW - 1 - (*filterPos)[i]] += acc;
552
0
        }
553
0
        av_assert0((*filterPos)[i] >= 0);
554
0
        av_assert0((*filterPos)[i] < srcW);
555
0
        if ((*filterPos)[i] + filterSize > srcW) {
556
0
            for (j = 0; j < filterSize; j++) {
557
0
                av_assert0((*filterPos)[i] + j < srcW || !filter[i * filterSize + j]);
558
0
            }
559
0
        }
560
0
    }
561
562
    // Note the +1 is for the MMX scaler which reads over the end
563
    /* align at 16 for AltiVec (needed by hScale_altivec_real) */
564
0
    *outFilter = av_calloc(dstW + 3, *outFilterSize * sizeof(**outFilter));
565
0
    if (!*outFilter)
566
0
        goto nomem;
567
568
    /* normalize & store in outFilter */
569
0
    for (i = 0; i < dstW; i++) {
570
0
        int j;
571
0
        int64_t error = 0;
572
0
        int64_t sum   = 0;
573
574
0
        for (j = 0; j < filterSize; j++) {
575
0
            sum += filter[i * filterSize + j];
576
0
        }
577
0
        sum = (sum + one / 2) / one;
578
0
        if (!sum) {
579
0
            av_log(NULL, AV_LOG_WARNING, "SwScaler: zero vector in scaling\n");
580
0
            sum = 1;
581
0
        }
582
0
        for (j = 0; j < *outFilterSize; j++) {
583
0
            int64_t v = filter[i * filterSize + j] + error;
584
0
            int intV  = ROUNDED_DIV(v, sum);
585
0
            (*outFilter)[i * (*outFilterSize) + j] = intV;
586
0
            error                                  = v - intV * sum;
587
0
        }
588
0
    }
589
590
0
    (*filterPos)[dstW + 0] =
591
0
    (*filterPos)[dstW + 1] =
592
0
    (*filterPos)[dstW + 2] = (*filterPos)[dstW - 1]; /* the MMX/SSE scaler will
593
                                                      * read over the end */
594
0
    for (i = 0; i < *outFilterSize; i++) {
595
0
        int k = (dstW - 1) * (*outFilterSize) + i;
596
0
        (*outFilter)[k + 1 * (*outFilterSize)] =
597
0
        (*outFilter)[k + 2 * (*outFilterSize)] =
598
0
        (*outFilter)[k + 3 * (*outFilterSize)] = (*outFilter)[k];
599
0
    }
600
601
0
    ret = 0;
602
0
    goto done;
603
0
nomem:
604
0
    ret = AVERROR(ENOMEM);
605
0
fail:
606
0
    if(ret < 0)
607
0
        av_log(NULL, ret == RETCODE_USE_CASCADE ? AV_LOG_DEBUG : AV_LOG_ERROR, "sws: initFilter failed\n");
608
0
done:
609
0
    av_free(filter);
610
0
    av_free(filter2);
611
0
    return ret;
612
0
}
613
614
static void fill_rgb2yuv_table(SwsInternal *c, const int table[4], int dstRange)
615
0
{
616
0
    int64_t W, V, Z, Cy, Cu, Cv;
617
0
    int64_t vr =  table[0];
618
0
    int64_t ub =  table[1];
619
0
    int64_t ug = -table[2];
620
0
    int64_t vg = -table[3];
621
0
    int64_t ONE = 65536;
622
0
    int64_t cy = ONE;
623
0
    uint8_t *p = (uint8_t*)c->input_rgb2yuv_table;
624
0
    int i;
625
0
    static const int8_t map[] = {
626
0
    BY_IDX, GY_IDX, -1    , BY_IDX, BY_IDX, GY_IDX, -1    , BY_IDX,
627
0
    RY_IDX, -1    , GY_IDX, RY_IDX, RY_IDX, -1    , GY_IDX, RY_IDX,
628
0
    RY_IDX, GY_IDX, -1    , RY_IDX, RY_IDX, GY_IDX, -1    , RY_IDX,
629
0
    BY_IDX, -1    , GY_IDX, BY_IDX, BY_IDX, -1    , GY_IDX, BY_IDX,
630
0
    BU_IDX, GU_IDX, -1    , BU_IDX, BU_IDX, GU_IDX, -1    , BU_IDX,
631
0
    RU_IDX, -1    , GU_IDX, RU_IDX, RU_IDX, -1    , GU_IDX, RU_IDX,
632
0
    RU_IDX, GU_IDX, -1    , RU_IDX, RU_IDX, GU_IDX, -1    , RU_IDX,
633
0
    BU_IDX, -1    , GU_IDX, BU_IDX, BU_IDX, -1    , GU_IDX, BU_IDX,
634
0
    BV_IDX, GV_IDX, -1    , BV_IDX, BV_IDX, GV_IDX, -1    , BV_IDX,
635
0
    RV_IDX, -1    , GV_IDX, RV_IDX, RV_IDX, -1    , GV_IDX, RV_IDX,
636
0
    RV_IDX, GV_IDX, -1    , RV_IDX, RV_IDX, GV_IDX, -1    , RV_IDX,
637
0
    BV_IDX, -1    , GV_IDX, BV_IDX, BV_IDX, -1    , GV_IDX, BV_IDX,
638
0
    RY_IDX, BY_IDX, RY_IDX, BY_IDX, RY_IDX, BY_IDX, RY_IDX, BY_IDX,
639
0
    BY_IDX, RY_IDX, BY_IDX, RY_IDX, BY_IDX, RY_IDX, BY_IDX, RY_IDX,
640
0
    GY_IDX, -1    , GY_IDX, -1    , GY_IDX, -1    , GY_IDX, -1    ,
641
0
    -1    , GY_IDX, -1    , GY_IDX, -1    , GY_IDX, -1    , GY_IDX,
642
0
    RU_IDX, BU_IDX, RU_IDX, BU_IDX, RU_IDX, BU_IDX, RU_IDX, BU_IDX,
643
0
    BU_IDX, RU_IDX, BU_IDX, RU_IDX, BU_IDX, RU_IDX, BU_IDX, RU_IDX,
644
0
    GU_IDX, -1    , GU_IDX, -1    , GU_IDX, -1    , GU_IDX, -1    ,
645
0
    -1    , GU_IDX, -1    , GU_IDX, -1    , GU_IDX, -1    , GU_IDX,
646
0
    RV_IDX, BV_IDX, RV_IDX, BV_IDX, RV_IDX, BV_IDX, RV_IDX, BV_IDX,
647
0
    BV_IDX, RV_IDX, BV_IDX, RV_IDX, BV_IDX, RV_IDX, BV_IDX, RV_IDX,
648
0
    GV_IDX, -1    , GV_IDX, -1    , GV_IDX, -1    , GV_IDX, -1    ,
649
0
    -1    , GV_IDX, -1    , GV_IDX, -1    , GV_IDX, -1    , GV_IDX, //23
650
0
    -1    , -1    , -1    , -1    , -1    , -1    , -1    , -1    , //24
651
0
    -1    , -1    , -1    , -1    , -1    , -1    , -1    , -1    , //25
652
0
    -1    , -1    , -1    , -1    , -1    , -1    , -1    , -1    , //26
653
0
    -1    , -1    , -1    , -1    , -1    , -1    , -1    , -1    , //27
654
0
    -1    , -1    , -1    , -1    , -1    , -1    , -1    , -1    , //28
655
0
    -1    , -1    , -1    , -1    , -1    , -1    , -1    , -1    , //29
656
0
    -1    , -1    , -1    , -1    , -1    , -1    , -1    , -1    , //30
657
0
    -1    , -1    , -1    , -1    , -1    , -1    , -1    , -1    , //31
658
0
    BY_IDX, GY_IDX, RY_IDX, -1    , -1    , -1    , -1    , -1    , //32
659
0
    BU_IDX, GU_IDX, RU_IDX, -1    , -1    , -1    , -1    , -1    , //33
660
0
    BV_IDX, GV_IDX, RV_IDX, -1    , -1    , -1    , -1    , -1    , //34
661
0
    };
662
663
0
    dstRange = 0; //FIXME range = 1 is handled elsewhere
664
665
0
    if (!dstRange) {
666
0
        cy = cy * 255 / 219;
667
0
    } else {
668
0
        vr = vr * 224 / 255;
669
0
        ub = ub * 224 / 255;
670
0
        ug = ug * 224 / 255;
671
0
        vg = vg * 224 / 255;
672
0
    }
673
0
    W = ROUNDED_DIV(ONE*ONE*ug, ub);
674
0
    V = ROUNDED_DIV(ONE*ONE*vg, vr);
675
0
    Z = ONE*ONE-W-V;
676
677
0
    Cy = ROUNDED_DIV(cy*Z, ONE);
678
0
    Cu = ROUNDED_DIV(ub*Z, ONE);
679
0
    Cv = ROUNDED_DIV(vr*Z, ONE);
680
681
0
    c->input_rgb2yuv_table[RY_IDX] = -ROUNDED_DIV((1 << RGB2YUV_SHIFT)*V        , Cy);
682
0
    c->input_rgb2yuv_table[GY_IDX] =  ROUNDED_DIV((1 << RGB2YUV_SHIFT)*ONE*ONE  , Cy);
683
0
    c->input_rgb2yuv_table[BY_IDX] = -ROUNDED_DIV((1 << RGB2YUV_SHIFT)*W        , Cy);
684
685
0
    c->input_rgb2yuv_table[RU_IDX] =  ROUNDED_DIV((1 << RGB2YUV_SHIFT)*V        , Cu);
686
0
    c->input_rgb2yuv_table[GU_IDX] = -ROUNDED_DIV((1 << RGB2YUV_SHIFT)*ONE*ONE  , Cu);
687
0
    c->input_rgb2yuv_table[BU_IDX] =  ROUNDED_DIV((1 << RGB2YUV_SHIFT)*(Z+W)    , Cu);
688
689
0
    c->input_rgb2yuv_table[RV_IDX] =  ROUNDED_DIV((1 << RGB2YUV_SHIFT)*(V+Z)    , Cv);
690
0
    c->input_rgb2yuv_table[GV_IDX] = -ROUNDED_DIV((1 << RGB2YUV_SHIFT)*ONE*ONE  , Cv);
691
0
    c->input_rgb2yuv_table[BV_IDX] =  ROUNDED_DIV((1 << RGB2YUV_SHIFT)*W        , Cv);
692
693
0
    if(/*!dstRange && */!memcmp(table, ff_yuv2rgb_coeffs[SWS_CS_DEFAULT], sizeof(ff_yuv2rgb_coeffs[SWS_CS_DEFAULT]))) {
694
0
        c->input_rgb2yuv_table[BY_IDX] =  ((int)(0.114 * 219 / 255 * (1 << RGB2YUV_SHIFT) + 0.5));
695
0
        c->input_rgb2yuv_table[BV_IDX] = (-(int)(0.081 * 224 / 255 * (1 << RGB2YUV_SHIFT) + 0.5));
696
0
        c->input_rgb2yuv_table[BU_IDX] =  ((int)(0.500 * 224 / 255 * (1 << RGB2YUV_SHIFT) + 0.5));
697
0
        c->input_rgb2yuv_table[GY_IDX] =  ((int)(0.587 * 219 / 255 * (1 << RGB2YUV_SHIFT) + 0.5));
698
0
        c->input_rgb2yuv_table[GV_IDX] = (-(int)(0.419 * 224 / 255 * (1 << RGB2YUV_SHIFT) + 0.5));
699
0
        c->input_rgb2yuv_table[GU_IDX] = (-(int)(0.331 * 224 / 255 * (1 << RGB2YUV_SHIFT) + 0.5));
700
0
        c->input_rgb2yuv_table[RY_IDX] =  ((int)(0.299 * 219 / 255 * (1 << RGB2YUV_SHIFT) + 0.5));
701
0
        c->input_rgb2yuv_table[RV_IDX] =  ((int)(0.500 * 224 / 255 * (1 << RGB2YUV_SHIFT) + 0.5));
702
0
        c->input_rgb2yuv_table[RU_IDX] = (-(int)(0.169 * 224 / 255 * (1 << RGB2YUV_SHIFT) + 0.5));
703
0
    }
704
0
    for(i=0; i<FF_ARRAY_ELEMS(map); i++)
705
0
        AV_WL16(p + 16*4 + 2*i, map[i] >= 0 ? c->input_rgb2yuv_table[map[i]] : 0);
706
0
}
707
708
#if CONFIG_SMALL
709
static void init_xyz_tables(uint16_t xyzgamma_tab[4096],  uint16_t xyzgammainv_tab[65536],
710
                            uint16_t rgbgamma_tab[65536], uint16_t rgbgammainv_tab[4096])
711
#else
712
static uint16_t xyzgamma_tab[4096],  rgbgammainv_tab[4096];
713
static uint16_t rgbgamma_tab[65536], xyzgammainv_tab[65536];
714
static av_cold void init_xyz_tables(void)
715
#endif
716
0
{
717
0
    double xyzgamma    = XYZ_GAMMA;
718
0
    double rgbgamma    = 1.0 / RGB_GAMMA;
719
0
    double xyzgammainv = 1.0 / XYZ_GAMMA;
720
0
    double rgbgammainv = RGB_GAMMA;
721
722
    /* set input gamma vectors */
723
0
    for (int i = 0; i < 4096; i++) {
724
0
        xyzgamma_tab[i]    = lrint(pow(i / 4095.0, xyzgamma)    * 65535.0);
725
0
        rgbgammainv_tab[i] = lrint(pow(i / 4095.0, rgbgammainv) * 65535.0);
726
0
    }
727
728
    /* set output gamma vectors */
729
0
    for (int i = 0; i < 65536; i++) {
730
0
        rgbgamma_tab[i]    = lrint(pow(i / 65535.0, rgbgamma)    * 4095.0);
731
0
        xyzgammainv_tab[i] = lrint(pow(i / 65535.0, xyzgammainv) * 4095.0);
732
0
    }
733
0
}
734
735
av_cold int ff_sws_fill_xyztables(SwsInternal *c)
736
0
{
737
0
    static const int16_t xyz2rgb_matrix[3][3] = {
738
0
        {13270, -6295, -2041},
739
0
        {-3969,  7682,   170},
740
0
        {  228,  -835,  4329} };
741
0
    static const int16_t rgb2xyz_matrix[3][3] = {
742
0
        {1689, 1464,  739},
743
0
        { 871, 2929,  296},
744
0
        {  79,  488, 3891} };
745
746
0
    if (c->xyz2rgb.gamma.in)
747
0
        return 0;
748
749
0
    memcpy(c->xyz2rgb.mat, xyz2rgb_matrix, sizeof(c->xyz2rgb.mat));
750
0
    memcpy(c->rgb2xyz.mat, rgb2xyz_matrix, sizeof(c->rgb2xyz.mat));
751
752
#if CONFIG_SMALL
753
    c->xyz2rgb.gamma.in = av_malloc(sizeof(uint16_t) * 2 * (4096 + 65536));
754
    if (!c->xyz2rgb.gamma.in)
755
        return AVERROR(ENOMEM);
756
    c->rgb2xyz.gamma.in  = c->xyz2rgb.gamma.in  + 4096;
757
    c->xyz2rgb.gamma.out = c->rgb2xyz.gamma.in  + 4096;
758
    c->rgb2xyz.gamma.out = c->xyz2rgb.gamma.out + 65536;
759
    init_xyz_tables(c->xyz2rgb.gamma.in,  c->rgb2xyz.gamma.out,
760
                    c->xyz2rgb.gamma.out, c->rgb2xyz.gamma.in);
761
#else
762
0
    c->xyz2rgb.gamma.in  = xyzgamma_tab;
763
0
    c->xyz2rgb.gamma.out = rgbgamma_tab;
764
0
    c->rgb2xyz.gamma.in  = rgbgammainv_tab;
765
0
    c->rgb2xyz.gamma.out = xyzgammainv_tab;
766
767
0
    static AVOnce xyz_init_static_once = AV_ONCE_INIT;
768
0
    ff_thread_once(&xyz_init_static_once, init_xyz_tables);
769
0
#endif
770
0
    return 0;
771
0
}
772
773
static int handle_jpeg(/* enum AVPixelFormat */ int *format)
774
0
{
775
0
    switch (*format) {
776
0
    case AV_PIX_FMT_YUVJ420P:
777
0
        *format = AV_PIX_FMT_YUV420P;
778
0
        return 1;
779
0
    case AV_PIX_FMT_YUVJ411P:
780
0
        *format = AV_PIX_FMT_YUV411P;
781
0
        return 1;
782
0
    case AV_PIX_FMT_YUVJ422P:
783
0
        *format = AV_PIX_FMT_YUV422P;
784
0
        return 1;
785
0
    case AV_PIX_FMT_YUVJ444P:
786
0
        *format = AV_PIX_FMT_YUV444P;
787
0
        return 1;
788
0
    case AV_PIX_FMT_YUVJ440P:
789
0
        *format = AV_PIX_FMT_YUV440P;
790
0
        return 1;
791
0
    case AV_PIX_FMT_GRAY8:
792
0
    case AV_PIX_FMT_YA8:
793
0
    case AV_PIX_FMT_GRAY9LE:
794
0
    case AV_PIX_FMT_GRAY9BE:
795
0
    case AV_PIX_FMT_GRAY10LE:
796
0
    case AV_PIX_FMT_GRAY10BE:
797
0
    case AV_PIX_FMT_GRAY12LE:
798
0
    case AV_PIX_FMT_GRAY12BE:
799
0
    case AV_PIX_FMT_GRAY14LE:
800
0
    case AV_PIX_FMT_GRAY14BE:
801
0
    case AV_PIX_FMT_GRAY16LE:
802
0
    case AV_PIX_FMT_GRAY16BE:
803
0
    case AV_PIX_FMT_YA16BE:
804
0
    case AV_PIX_FMT_YA16LE:
805
0
        return 1;
806
0
    default:
807
0
        return 0;
808
0
    }
809
0
}
810
811
static int handle_0alpha(/* enum AVPixelFormat */ int *format)
812
0
{
813
0
    switch (*format) {
814
0
    case AV_PIX_FMT_0BGR    : *format = AV_PIX_FMT_ABGR   ; return 1;
815
0
    case AV_PIX_FMT_BGR0    : *format = AV_PIX_FMT_BGRA   ; return 4;
816
0
    case AV_PIX_FMT_0RGB    : *format = AV_PIX_FMT_ARGB   ; return 1;
817
0
    case AV_PIX_FMT_RGB0    : *format = AV_PIX_FMT_RGBA   ; return 4;
818
0
    default:                                          return 0;
819
0
    }
820
0
}
821
822
static int handle_xyz(/* enum AVPixelFormat */ int *format)
823
0
{
824
0
    switch (*format) {
825
0
    case AV_PIX_FMT_XYZ12BE : *format = AV_PIX_FMT_RGB48BE; return 1;
826
0
    case AV_PIX_FMT_XYZ12LE : *format = AV_PIX_FMT_RGB48LE; return 1;
827
0
    default:                                                return 0;
828
0
    }
829
0
}
830
831
static int handle_formats(SwsContext *sws)
832
0
{
833
0
    SwsInternal *c = sws_internal(sws);
834
0
    c->src0Alpha |= handle_0alpha(&sws->src_format);
835
0
    c->dst0Alpha |= handle_0alpha(&sws->dst_format);
836
0
    c->srcXYZ    |= handle_xyz(&sws->src_format);
837
0
    c->dstXYZ    |= handle_xyz(&sws->dst_format);
838
0
    if (c->srcXYZ || c->dstXYZ)
839
0
        return ff_sws_fill_xyztables(c);
840
0
    else
841
0
        return 0;
842
0
}
843
844
static int range_override_needed(enum AVPixelFormat format)
845
0
{
846
0
    return !isYUV(format) && !isGray(format);
847
0
}
848
849
int sws_setColorspaceDetails(SwsContext *sws, const int inv_table[4],
850
                             int srcRange, const int table[4], int dstRange,
851
                             int brightness, int contrast, int saturation)
852
0
{
853
0
    SwsInternal *c = sws_internal(sws);
854
0
    const AVPixFmtDescriptor *desc_dst;
855
0
    const AVPixFmtDescriptor *desc_src;
856
0
    int ret, need_reinit = 0;
857
858
0
    if (c->nb_slice_ctx) {
859
0
        int parent_ret = 0;
860
0
        for (int i = 0; i < c->nb_slice_ctx; i++) {
861
0
            int ret = sws_setColorspaceDetails(c->slice_ctx[i], inv_table,
862
0
                                               srcRange, table, dstRange,
863
0
                                               brightness, contrast, saturation);
864
0
            if (ret < 0)
865
0
                parent_ret = ret;
866
0
        }
867
868
0
        return parent_ret;
869
0
    }
870
871
0
    ret = handle_formats(sws);
872
0
    if (ret < 0)
873
0
        return ret;
874
0
    desc_dst = av_pix_fmt_desc_get(sws->dst_format);
875
0
    desc_src = av_pix_fmt_desc_get(sws->src_format);
876
877
0
    if(range_override_needed(sws->dst_format))
878
0
        dstRange = 0;
879
0
    if(range_override_needed(sws->src_format))
880
0
        srcRange = 0;
881
882
0
    if (sws->src_range != srcRange ||
883
0
        sws->dst_range != dstRange ||
884
0
        c->brightness != brightness ||
885
0
        c->contrast   != contrast ||
886
0
        c->saturation != saturation ||
887
0
        memcmp(c->srcColorspaceTable, inv_table, sizeof(int) * 4) ||
888
0
        memcmp(c->dstColorspaceTable,     table, sizeof(int) * 4)
889
0
    )
890
0
        need_reinit = 1;
891
892
0
    memmove(c->srcColorspaceTable, inv_table, sizeof(int) * 4);
893
0
    memmove(c->dstColorspaceTable, table, sizeof(int) * 4);
894
895
896
897
0
    c->brightness = brightness;
898
0
    c->contrast   = contrast;
899
0
    c->saturation = saturation;
900
0
    sws->src_range = srcRange;
901
0
    sws->dst_range = dstRange;
902
903
0
    if (need_reinit)
904
0
        ff_sws_init_range_convert(c);
905
906
0
    c->dstFormatBpp = av_get_bits_per_pixel(desc_dst);
907
0
    c->srcFormatBpp = av_get_bits_per_pixel(desc_src);
908
909
0
    if (c->cascaded_context[c->cascaded_mainindex])
910
0
        return sws_setColorspaceDetails(c->cascaded_context[c->cascaded_mainindex],inv_table, srcRange,table, dstRange, brightness,  contrast, saturation);
911
912
0
    if (!need_reinit)
913
0
        return 0;
914
915
0
    if ((isYUV(sws->dst_format) || isGray(sws->dst_format)) && (isYUV(sws->src_format) || isGray(sws->src_format))) {
916
0
        if (!c->cascaded_context[0] &&
917
0
            memcmp(c->dstColorspaceTable, c->srcColorspaceTable, sizeof(int) * 4) &&
918
0
            sws->src_w && sws->src_h && sws->dst_w && sws->dst_h) {
919
0
            enum AVPixelFormat tmp_format;
920
0
            int tmp_width, tmp_height;
921
0
            int srcW = sws->src_w;
922
0
            int srcH = sws->src_h;
923
0
            int dstW = sws->dst_w;
924
0
            int dstH = sws->dst_h;
925
0
            int ret;
926
0
            av_log(c, AV_LOG_VERBOSE, "YUV color matrix differs for YUV->YUV, using intermediate RGB to convert\n");
927
928
0
            if (isNBPS(sws->dst_format) || is16BPS(sws->dst_format)) {
929
0
                if (isALPHA(sws->src_format) && isALPHA(sws->dst_format)) {
930
0
                    tmp_format = AV_PIX_FMT_BGRA64;
931
0
                } else {
932
0
                    tmp_format = AV_PIX_FMT_BGR48;
933
0
                }
934
0
            } else {
935
0
                if (isALPHA(sws->src_format) && isALPHA(sws->dst_format)) {
936
0
                    tmp_format = AV_PIX_FMT_BGRA;
937
0
                } else {
938
0
                    tmp_format = AV_PIX_FMT_BGR24;
939
0
                }
940
0
            }
941
942
0
            if (srcW*srcH > dstW*dstH) {
943
0
                tmp_width  = dstW;
944
0
                tmp_height = dstH;
945
0
            } else {
946
0
                tmp_width  = srcW;
947
0
                tmp_height = srcH;
948
0
            }
949
950
0
            ret = av_image_alloc(c->cascaded_tmp[0], c->cascaded_tmpStride[0],
951
0
                                tmp_width, tmp_height, tmp_format, 64);
952
0
            if (ret < 0)
953
0
                return ret;
954
955
0
            c->cascaded_context[0] = alloc_set_opts(srcW, srcH, sws->src_format,
956
0
                                                    tmp_width, tmp_height, tmp_format,
957
0
                                                    sws->flags, sws->scaler_params);
958
0
            if (!c->cascaded_context[0])
959
0
                return -1;
960
961
0
            c->cascaded_context[0]->alpha_blend = sws->alpha_blend;
962
0
            ret = sws_init_context(c->cascaded_context[0], NULL , NULL);
963
0
            if (ret < 0)
964
0
                return ret;
965
            //we set both src and dst depending on that the RGB side will be ignored
966
0
            sws_setColorspaceDetails(c->cascaded_context[0], inv_table,
967
0
                                     srcRange, table, dstRange,
968
0
                                     brightness, contrast, saturation);
969
970
0
            c->cascaded_context[1] = alloc_set_opts(tmp_width, tmp_height, tmp_format,
971
0
                                                    dstW, dstH, sws->dst_format,
972
0
                                                    sws->flags, sws->scaler_params);
973
0
            if (!c->cascaded_context[1])
974
0
                return -1;
975
0
            c->cascaded_context[1]->src_range = srcRange;
976
0
            c->cascaded_context[1]->dst_range = dstRange;
977
0
            ret = sws_init_context(c->cascaded_context[1], NULL , NULL);
978
0
            if (ret < 0)
979
0
                return ret;
980
0
            sws_setColorspaceDetails(c->cascaded_context[1], inv_table,
981
0
                                     srcRange, table, dstRange,
982
0
                                     0, 1 << 16, 1 << 16);
983
0
            return 0;
984
0
        }
985
        //We do not support this combination currently, we need to cascade more contexts to compensate
986
0
        if (c->cascaded_context[0] && memcmp(c->dstColorspaceTable, c->srcColorspaceTable, sizeof(int) * 4))
987
0
            return -1; //AVERROR_PATCHWELCOME;
988
0
        return 0;
989
0
    }
990
991
0
    if (!isYUV(sws->dst_format) && !isGray(sws->dst_format)) {
992
0
        ff_yuv2rgb_c_init_tables(c, inv_table, srcRange, brightness,
993
0
                                 contrast, saturation);
994
        // FIXME factorize
995
996
#if ARCH_PPC
997
        ff_yuv2rgb_init_tables_ppc(c, inv_table, brightness,
998
                                   contrast, saturation);
999
#endif
1000
0
    }
1001
1002
0
    fill_rgb2yuv_table(c, table, dstRange);
1003
1004
0
    return 0;
1005
0
}
1006
1007
int sws_getColorspaceDetails(SwsContext *sws, int **inv_table,
1008
                             int *srcRange, int **table, int *dstRange,
1009
                             int *brightness, int *contrast, int *saturation)
1010
0
{
1011
0
    SwsInternal *c = sws_internal(sws);
1012
0
    if (!c)
1013
0
        return -1;
1014
1015
0
    if (c->nb_slice_ctx) {
1016
0
        return sws_getColorspaceDetails(c->slice_ctx[0], inv_table, srcRange,
1017
0
                                        table, dstRange, brightness, contrast,
1018
0
                                        saturation);
1019
0
    }
1020
1021
0
    *inv_table  = c->srcColorspaceTable;
1022
0
    *table      = c->dstColorspaceTable;
1023
0
    *srcRange   = range_override_needed(sws->src_format) ? 1 : sws->src_range;
1024
0
    *dstRange   = range_override_needed(sws->dst_format) ? 1 : sws->dst_range;
1025
0
    *brightness = c->brightness;
1026
0
    *contrast   = c->contrast;
1027
0
    *saturation = c->saturation;
1028
1029
0
    return 0;
1030
0
}
1031
1032
SwsContext *sws_alloc_context(void)
1033
0
{
1034
0
    SwsInternal *c = av_mallocz(sizeof(*c) + SWSINTERNAL_ADDITIONAL_ASM_SIZE);
1035
0
    if (!c)
1036
0
        return NULL;
1037
1038
0
    c->opts.av_class = &ff_sws_context_class;
1039
0
    av_opt_set_defaults(c);
1040
0
    atomic_init(&c->stride_unaligned_warned, 0);
1041
0
    atomic_init(&c->data_unaligned_warned,   0);
1042
1043
0
    return &c->opts;
1044
0
}
1045
1046
static uint16_t * alloc_gamma_tbl(double e)
1047
0
{
1048
0
    int i = 0;
1049
0
    uint16_t * tbl;
1050
0
    tbl = (uint16_t*)av_malloc(sizeof(uint16_t) * 1 << 16);
1051
0
    if (!tbl)
1052
0
        return NULL;
1053
1054
0
    for (i = 0; i < 65536; ++i) {
1055
0
        tbl[i] = pow(i / 65535.0, e) * 65535.0;
1056
0
    }
1057
0
    return tbl;
1058
0
}
1059
1060
static enum AVPixelFormat alphaless_fmt(enum AVPixelFormat fmt)
1061
0
{
1062
0
    switch(fmt) {
1063
0
    case AV_PIX_FMT_ARGB:       return AV_PIX_FMT_RGB24;
1064
0
    case AV_PIX_FMT_RGBA:       return AV_PIX_FMT_RGB24;
1065
0
    case AV_PIX_FMT_ABGR:       return AV_PIX_FMT_BGR24;
1066
0
    case AV_PIX_FMT_BGRA:       return AV_PIX_FMT_BGR24;
1067
0
    case AV_PIX_FMT_YA8:        return AV_PIX_FMT_GRAY8;
1068
1069
0
    case AV_PIX_FMT_YUVA420P:   return AV_PIX_FMT_YUV420P;
1070
0
    case AV_PIX_FMT_YUVA422P:   return AV_PIX_FMT_YUV422P;
1071
0
    case AV_PIX_FMT_YUVA444P:           return AV_PIX_FMT_YUV444P;
1072
1073
0
    case AV_PIX_FMT_GBRAP:              return AV_PIX_FMT_GBRP;
1074
1075
0
    case AV_PIX_FMT_GBRAP10LE:          return AV_PIX_FMT_GBRP10;
1076
0
    case AV_PIX_FMT_GBRAP10BE:          return AV_PIX_FMT_GBRP10;
1077
1078
0
    case AV_PIX_FMT_GBRAP12LE:          return AV_PIX_FMT_GBRP12;
1079
0
    case AV_PIX_FMT_GBRAP12BE:          return AV_PIX_FMT_GBRP12;
1080
1081
0
    case AV_PIX_FMT_GBRAP14LE:          return AV_PIX_FMT_GBRP14;
1082
0
    case AV_PIX_FMT_GBRAP14BE:          return AV_PIX_FMT_GBRP14;
1083
1084
0
    case AV_PIX_FMT_GBRAP16LE:          return AV_PIX_FMT_GBRP16;
1085
0
    case AV_PIX_FMT_GBRAP16BE:          return AV_PIX_FMT_GBRP16;
1086
1087
0
    case AV_PIX_FMT_RGBA64LE:   return AV_PIX_FMT_RGB48;
1088
0
    case AV_PIX_FMT_RGBA64BE:   return AV_PIX_FMT_RGB48;
1089
0
    case AV_PIX_FMT_BGRA64LE:   return AV_PIX_FMT_BGR48;
1090
0
    case AV_PIX_FMT_BGRA64BE:   return AV_PIX_FMT_BGR48;
1091
1092
0
    case AV_PIX_FMT_YA16BE:             return AV_PIX_FMT_GRAY16;
1093
0
    case AV_PIX_FMT_YA16LE:             return AV_PIX_FMT_GRAY16;
1094
1095
0
    case AV_PIX_FMT_YUVA420P9BE:        return AV_PIX_FMT_YUV420P9;
1096
0
    case AV_PIX_FMT_YUVA422P9BE:        return AV_PIX_FMT_YUV422P9;
1097
0
    case AV_PIX_FMT_YUVA444P9BE:        return AV_PIX_FMT_YUV444P9;
1098
0
    case AV_PIX_FMT_YUVA420P9LE:        return AV_PIX_FMT_YUV420P9;
1099
0
    case AV_PIX_FMT_YUVA422P9LE:        return AV_PIX_FMT_YUV422P9;
1100
0
    case AV_PIX_FMT_YUVA444P9LE:        return AV_PIX_FMT_YUV444P9;
1101
0
    case AV_PIX_FMT_YUVA420P10BE:       return AV_PIX_FMT_YUV420P10;
1102
0
    case AV_PIX_FMT_YUVA422P10BE:       return AV_PIX_FMT_YUV422P10;
1103
0
    case AV_PIX_FMT_YUVA444P10BE:       return AV_PIX_FMT_YUV444P10;
1104
0
    case AV_PIX_FMT_YUVA420P10LE:       return AV_PIX_FMT_YUV420P10;
1105
0
    case AV_PIX_FMT_YUVA422P10LE:       return AV_PIX_FMT_YUV422P10;
1106
0
    case AV_PIX_FMT_YUVA444P10LE:       return AV_PIX_FMT_YUV444P10;
1107
0
    case AV_PIX_FMT_YUVA420P16BE:       return AV_PIX_FMT_YUV420P16;
1108
0
    case AV_PIX_FMT_YUVA422P16BE:       return AV_PIX_FMT_YUV422P16;
1109
0
    case AV_PIX_FMT_YUVA444P16BE:       return AV_PIX_FMT_YUV444P16;
1110
0
    case AV_PIX_FMT_YUVA420P16LE:       return AV_PIX_FMT_YUV420P16;
1111
0
    case AV_PIX_FMT_YUVA422P16LE:       return AV_PIX_FMT_YUV422P16;
1112
0
    case AV_PIX_FMT_YUVA444P16LE:       return AV_PIX_FMT_YUV444P16;
1113
1114
//     case AV_PIX_FMT_AYUV64LE:
1115
//     case AV_PIX_FMT_AYUV64BE:
1116
//     case AV_PIX_FMT_PAL8:
1117
0
    default: return AV_PIX_FMT_NONE;
1118
0
    }
1119
0
}
1120
1121
static int scaler_flag(SwsScaler scaler, int fallback)
1122
0
{
1123
0
    switch (scaler) {
1124
0
    case SWS_SCALE_BILINEAR: return SWS_BILINEAR; break;
1125
0
    case SWS_SCALE_BICUBIC:  return SWS_BICUBIC;  break;
1126
0
    case SWS_SCALE_POINT:    return SWS_POINT;    break;
1127
0
    case SWS_SCALE_AREA:     return SWS_AREA;     break;
1128
0
    case SWS_SCALE_GAUSSIAN: return SWS_GAUSS;    break;
1129
0
    case SWS_SCALE_SINC:     return SWS_SINC;     break;
1130
0
    case SWS_SCALE_LANCZOS:  return SWS_LANCZOS;  break;
1131
0
    case SWS_SCALE_SPLINE:   return SWS_SPLINE;   break;
1132
0
    default:
1133
0
        return fallback;
1134
0
    }
1135
0
}
1136
1137
av_cold int ff_sws_init_single_context(SwsContext *sws, SwsFilter *srcFilter,
1138
                                       SwsFilter *dstFilter)
1139
0
{
1140
0
    int i;
1141
0
    int usesVFilter, usesHFilter;
1142
0
    int unscaled;
1143
0
    SwsInternal *c        = sws_internal(sws);
1144
0
    SwsFilter dummyFilter = { NULL, NULL, NULL, NULL };
1145
0
    int srcW              = sws->src_w;
1146
0
    int srcH              = sws->src_h;
1147
0
    int dstW              = sws->dst_w;
1148
0
    int dstH              = sws->dst_h;
1149
0
    int dst_stride        = FFALIGN(dstW * sizeof(int16_t) + 66, 16);
1150
0
    int flags, cpu_flags;
1151
0
    enum AVPixelFormat srcFormat, dstFormat;
1152
0
    const AVPixFmtDescriptor *desc_src;
1153
0
    const AVPixFmtDescriptor *desc_dst;
1154
0
    int ret = 0;
1155
0
    enum AVPixelFormat tmpFmt;
1156
0
    static const float float_mult = 1.0f / 255.0f;
1157
1158
0
    cpu_flags = av_get_cpu_flags();
1159
0
    flags     = sws->flags;
1160
0
    emms_c();
1161
1162
0
    unscaled = (srcW == dstW && srcH == dstH);
1163
1164
0
    if (!c->contrast && !c->saturation && !c->dstFormatBpp)
1165
0
        sws_setColorspaceDetails(sws, ff_yuv2rgb_coeffs[SWS_CS_DEFAULT], sws->src_range,
1166
0
                                 ff_yuv2rgb_coeffs[SWS_CS_DEFAULT],
1167
0
                                 sws->dst_range, 0, 1 << 16, 1 << 16);
1168
1169
0
    ret = handle_formats(sws);
1170
0
    if (ret < 0)
1171
0
        return ret;
1172
0
    srcFormat = sws->src_format;
1173
0
    dstFormat = sws->dst_format;
1174
0
    desc_src = av_pix_fmt_desc_get(srcFormat);
1175
0
    desc_dst = av_pix_fmt_desc_get(dstFormat);
1176
1177
    // If the source has no alpha then disable alpha blendaway
1178
0
    if (c->src0Alpha)
1179
0
        sws->alpha_blend = SWS_ALPHA_BLEND_NONE;
1180
1181
0
    if (!(unscaled && sws_isSupportedEndiannessConversion(srcFormat) &&
1182
0
          av_pix_fmt_swap_endianness(srcFormat) == dstFormat)) {
1183
0
        if (!sws_isSupportedInput(srcFormat)) {
1184
0
            av_log(c, AV_LOG_ERROR, "%s is not supported as input pixel format\n",
1185
0
                   av_get_pix_fmt_name(srcFormat));
1186
0
            return AVERROR(EINVAL);
1187
0
        }
1188
0
        if (!sws_isSupportedOutput(dstFormat)) {
1189
0
            av_log(c, AV_LOG_ERROR, "%s is not supported as output pixel format\n",
1190
0
                   av_get_pix_fmt_name(dstFormat));
1191
0
            return AVERROR(EINVAL);
1192
0
        }
1193
0
    }
1194
0
    av_assert2(desc_src && desc_dst);
1195
1196
0
    i = flags & (SWS_POINT         |
1197
0
                 SWS_AREA          |
1198
0
                 SWS_BILINEAR      |
1199
0
                 SWS_FAST_BILINEAR |
1200
0
                 SWS_BICUBIC       |
1201
0
                 SWS_X             |
1202
0
                 SWS_GAUSS         |
1203
0
                 SWS_LANCZOS       |
1204
0
                 SWS_SINC          |
1205
0
                 SWS_SPLINE        |
1206
0
                 SWS_BICUBLIN);
1207
1208
    /* provide a default scaler if not set by caller */
1209
0
    if (!i) {
1210
0
        if (dstW < srcW && dstH < srcH)
1211
0
            i = SWS_BICUBIC;
1212
0
        else if (dstW > srcW && dstH > srcH)
1213
0
            i = SWS_BICUBIC;
1214
0
        else
1215
0
            i = SWS_BICUBIC;
1216
0
        flags |= i;
1217
0
        sws->flags = flags;
1218
0
    } else if (i & (i - 1)) {
1219
0
        av_log(c, AV_LOG_ERROR,
1220
0
               "Exactly one scaler algorithm must be chosen, got %X\n", i);
1221
0
        return AVERROR(EINVAL);
1222
0
    }
1223
1224
0
    if (i == SWS_FAST_BILINEAR) {
1225
0
        if (srcW < 8 || dstW <= 8) {
1226
0
            i = SWS_BILINEAR;
1227
0
            flags ^= SWS_FAST_BILINEAR | i;
1228
0
            sws->flags = flags;
1229
0
        }
1230
0
    }
1231
1232
0
    SwsScaler scaler_sub = sws->scaler_sub ? sws->scaler_sub : sws->scaler;
1233
0
    int lum_scaler = scaler_flag(sws->scaler, i == SWS_BICUBLIN ? SWS_BICUBIC  : i);
1234
0
    int chr_scaler = scaler_flag(scaler_sub,  i == SWS_BICUBLIN ? SWS_BILINEAR : i);
1235
1236
    /* sanity check */
1237
0
    if (srcW < 1 || srcH < 1 || dstW < 1 || dstH < 1) {
1238
        /* FIXME check if these are enough and try to lower them after
1239
         * fixing the relevant parts of the code */
1240
0
        av_log(c, AV_LOG_ERROR, "%dx%d -> %dx%d is invalid scaling dimension\n",
1241
0
               srcW, srcH, dstW, dstH);
1242
0
        return AVERROR(EINVAL);
1243
0
    }
1244
1245
0
    if (!dstFilter)
1246
0
        dstFilter = &dummyFilter;
1247
0
    if (!srcFilter)
1248
0
        srcFilter = &dummyFilter;
1249
1250
0
    int64_t lumXInc      = (((int64_t)srcW << 16) + (dstW >> 1)) / dstW;
1251
0
    int64_t lumYInc      = (((int64_t)srcH << 16) + (dstH >> 1)) / dstH;
1252
0
    c->dstFormatBpp = av_get_bits_per_pixel(desc_dst);
1253
0
    c->srcFormatBpp = av_get_bits_per_pixel(desc_src);
1254
0
    c->vRounder     = 4 * 0x0001000100010001ULL;
1255
1256
0
    usesVFilter = (srcFilter->lumV && srcFilter->lumV->length > 1) ||
1257
0
                  (srcFilter->chrV && srcFilter->chrV->length > 1) ||
1258
0
                  (dstFilter->lumV && dstFilter->lumV->length > 1) ||
1259
0
                  (dstFilter->chrV && dstFilter->chrV->length > 1);
1260
0
    usesHFilter = (srcFilter->lumH && srcFilter->lumH->length > 1) ||
1261
0
                  (srcFilter->chrH && srcFilter->chrH->length > 1) ||
1262
0
                  (dstFilter->lumH && dstFilter->lumH->length > 1) ||
1263
0
                  (dstFilter->chrH && dstFilter->chrH->length > 1);
1264
1265
0
    av_pix_fmt_get_chroma_sub_sample(srcFormat, &c->chrSrcHSubSample, &c->chrSrcVSubSample);
1266
0
    av_pix_fmt_get_chroma_sub_sample(dstFormat, &c->chrDstHSubSample, &c->chrDstVSubSample);
1267
1268
0
    c->dst_slice_align = 1 << c->chrDstVSubSample;
1269
1270
0
    if (isAnyRGB(dstFormat) && !(flags&SWS_FULL_CHR_H_INT)) {
1271
0
        if (dstW&1) {
1272
0
            av_log(c, AV_LOG_DEBUG, "Forcing full internal H chroma due to odd output size\n");
1273
0
            flags |= SWS_FULL_CHR_H_INT;
1274
0
            sws->flags = flags;
1275
0
        }
1276
1277
0
        if (   c->chrSrcHSubSample == 0
1278
0
            && c->chrSrcVSubSample == 0
1279
0
            && sws->dither != SWS_DITHER_BAYER //SWS_FULL_CHR_H_INT is currently not supported with SWS_DITHER_BAYER
1280
0
            && !(sws->flags & SWS_FAST_BILINEAR)
1281
0
        ) {
1282
0
            av_log(c, AV_LOG_DEBUG, "Forcing full internal H chroma due to input having non subsampled chroma\n");
1283
0
            flags |= SWS_FULL_CHR_H_INT;
1284
0
            sws->flags = flags;
1285
0
        }
1286
0
    }
1287
1288
0
    if (sws->dither == SWS_DITHER_AUTO) {
1289
0
        if (flags & SWS_ERROR_DIFFUSION)
1290
0
            sws->dither = SWS_DITHER_ED;
1291
0
    }
1292
1293
0
    if(dstFormat == AV_PIX_FMT_BGR4_BYTE ||
1294
0
       dstFormat == AV_PIX_FMT_RGB4_BYTE ||
1295
0
       dstFormat == AV_PIX_FMT_BGR8 ||
1296
0
       dstFormat == AV_PIX_FMT_RGB8) {
1297
0
        if (sws->dither == SWS_DITHER_AUTO)
1298
0
            sws->dither = (flags & SWS_FULL_CHR_H_INT) ? SWS_DITHER_ED : SWS_DITHER_BAYER;
1299
0
        if (!(flags & SWS_FULL_CHR_H_INT)) {
1300
0
            if (sws->dither == SWS_DITHER_ED || sws->dither == SWS_DITHER_A_DITHER || sws->dither == SWS_DITHER_X_DITHER || sws->dither == SWS_DITHER_NONE) {
1301
0
                av_log(c, AV_LOG_DEBUG,
1302
0
                    "Desired dithering only supported in full chroma interpolation for destination format '%s'\n",
1303
0
                    av_get_pix_fmt_name(dstFormat));
1304
0
                flags   |= SWS_FULL_CHR_H_INT;
1305
0
                sws->flags = flags;
1306
0
            }
1307
0
        }
1308
0
        if (flags & SWS_FULL_CHR_H_INT) {
1309
0
            if (sws->dither == SWS_DITHER_BAYER) {
1310
0
                av_log(c, AV_LOG_DEBUG,
1311
0
                    "Ordered dither is not supported in full chroma interpolation for destination format '%s'\n",
1312
0
                    av_get_pix_fmt_name(dstFormat));
1313
0
                sws->dither = SWS_DITHER_ED;
1314
0
            }
1315
0
        }
1316
0
    }
1317
0
    if (isPlanarRGB(dstFormat)) {
1318
0
        if (!(flags & SWS_FULL_CHR_H_INT)) {
1319
0
            av_log(c, AV_LOG_DEBUG,
1320
0
                   "%s output is not supported with half chroma resolution, switching to full\n",
1321
0
                   av_get_pix_fmt_name(dstFormat));
1322
0
            flags   |= SWS_FULL_CHR_H_INT;
1323
0
            sws->flags = flags;
1324
0
        }
1325
0
    }
1326
1327
    /* reuse chroma for 2 pixels RGB/BGR unless user wants full
1328
     * chroma interpolation */
1329
0
    if (flags & SWS_FULL_CHR_H_INT &&
1330
0
        isAnyRGB(dstFormat)        &&
1331
0
        !isPlanarRGB(dstFormat)    &&
1332
0
        dstFormat != AV_PIX_FMT_RGBA64LE &&
1333
0
        dstFormat != AV_PIX_FMT_RGBA64BE &&
1334
0
        dstFormat != AV_PIX_FMT_BGRA64LE &&
1335
0
        dstFormat != AV_PIX_FMT_BGRA64BE &&
1336
0
        dstFormat != AV_PIX_FMT_RGB48LE &&
1337
0
        dstFormat != AV_PIX_FMT_RGB48BE &&
1338
0
        dstFormat != AV_PIX_FMT_BGR48LE &&
1339
0
        dstFormat != AV_PIX_FMT_BGR48BE &&
1340
0
        dstFormat != AV_PIX_FMT_RGBA  &&
1341
0
        dstFormat != AV_PIX_FMT_ARGB  &&
1342
0
        dstFormat != AV_PIX_FMT_BGRA  &&
1343
0
        dstFormat != AV_PIX_FMT_ABGR  &&
1344
0
        dstFormat != AV_PIX_FMT_RGB24 &&
1345
0
        dstFormat != AV_PIX_FMT_BGR24 &&
1346
0
        dstFormat != AV_PIX_FMT_BGR4_BYTE &&
1347
0
        dstFormat != AV_PIX_FMT_RGB4_BYTE &&
1348
0
        dstFormat != AV_PIX_FMT_BGR8 &&
1349
0
        dstFormat != AV_PIX_FMT_RGB8 &&
1350
0
        dstFormat != AV_PIX_FMT_X2RGB10LE &&
1351
0
        dstFormat != AV_PIX_FMT_X2BGR10LE
1352
0
    ) {
1353
0
        av_log(c, AV_LOG_WARNING,
1354
0
               "full chroma interpolation for destination format '%s' not yet implemented\n",
1355
0
               av_get_pix_fmt_name(dstFormat));
1356
0
        flags   &= ~SWS_FULL_CHR_H_INT;
1357
0
        sws->flags = flags;
1358
0
    }
1359
0
    if (isAnyRGB(dstFormat) && !(flags & SWS_FULL_CHR_H_INT))
1360
0
        c->chrDstHSubSample = 1;
1361
1362
    // drop some chroma lines if the user wants it
1363
0
    c->vChrDrop          = (flags & SWS_SRC_V_CHR_DROP_MASK) >>
1364
0
                           SWS_SRC_V_CHR_DROP_SHIFT;
1365
0
    c->chrSrcVSubSample += c->vChrDrop;
1366
1367
    /* drop every other pixel for chroma calculation unless user
1368
     * wants full chroma */
1369
0
    if (isAnyRGB(srcFormat) && !(srcW & 1) && !(flags & SWS_FULL_CHR_H_INP) &&
1370
0
        srcFormat != AV_PIX_FMT_RGB8 && srcFormat != AV_PIX_FMT_BGR8 &&
1371
0
        srcFormat != AV_PIX_FMT_RGB4 && srcFormat != AV_PIX_FMT_BGR4 &&
1372
0
        srcFormat != AV_PIX_FMT_RGB4_BYTE && srcFormat != AV_PIX_FMT_BGR4_BYTE &&
1373
0
        srcFormat != AV_PIX_FMT_GBRP9BE   && srcFormat != AV_PIX_FMT_GBRP9LE  &&
1374
0
        srcFormat != AV_PIX_FMT_GBRP10BE  && srcFormat != AV_PIX_FMT_GBRP10LE &&
1375
0
        srcFormat != AV_PIX_FMT_GBRP10MSBBE  && srcFormat != AV_PIX_FMT_GBRP10MSBLE &&
1376
0
        srcFormat != AV_PIX_FMT_GBRAP10BE && srcFormat != AV_PIX_FMT_GBRAP10LE &&
1377
0
        srcFormat != AV_PIX_FMT_GBRP12BE  && srcFormat != AV_PIX_FMT_GBRP12LE &&
1378
0
        srcFormat != AV_PIX_FMT_GBRP12MSBBE && srcFormat != AV_PIX_FMT_GBRP12MSBLE &&
1379
0
        srcFormat != AV_PIX_FMT_GBRAP12BE && srcFormat != AV_PIX_FMT_GBRAP12LE &&
1380
0
        srcFormat != AV_PIX_FMT_GBRAP14BE && srcFormat != AV_PIX_FMT_GBRAP14LE &&
1381
0
        srcFormat != AV_PIX_FMT_GBRP14BE  && srcFormat != AV_PIX_FMT_GBRP14LE &&
1382
0
        srcFormat != AV_PIX_FMT_GBRP16BE  && srcFormat != AV_PIX_FMT_GBRP16LE &&
1383
0
        srcFormat != AV_PIX_FMT_GBRAP16BE  && srcFormat != AV_PIX_FMT_GBRAP16LE &&
1384
0
        srcFormat != AV_PIX_FMT_GBRPF32BE  && srcFormat != AV_PIX_FMT_GBRPF32LE &&
1385
0
        srcFormat != AV_PIX_FMT_GBRAPF32BE && srcFormat != AV_PIX_FMT_GBRAPF32LE &&
1386
0
        srcFormat != AV_PIX_FMT_GBRPF16BE  && srcFormat != AV_PIX_FMT_GBRPF16LE &&
1387
0
        srcFormat != AV_PIX_FMT_GBRAPF16BE && srcFormat != AV_PIX_FMT_GBRAPF16LE &&
1388
0
        ((dstW >> c->chrDstHSubSample) <= (srcW >> 1) ||
1389
0
         (flags & SWS_FAST_BILINEAR)))
1390
0
        c->chrSrcHSubSample = 1;
1391
1392
    // Note the AV_CEIL_RSHIFT is so that we always round toward +inf.
1393
0
    c->chrSrcW = AV_CEIL_RSHIFT(srcW, c->chrSrcHSubSample);
1394
0
    c->chrSrcH = AV_CEIL_RSHIFT(srcH, c->chrSrcVSubSample);
1395
0
    c->chrDstW = AV_CEIL_RSHIFT(dstW, c->chrDstHSubSample);
1396
0
    c->chrDstH = AV_CEIL_RSHIFT(dstH, c->chrDstVSubSample);
1397
1398
0
    if (!FF_ALLOCZ_TYPED_ARRAY(c->formatConvBuffer, FFALIGN(srcW * 2 + 78, 16) * 2))
1399
0
        goto nomem;
1400
1401
0
    c->srcBpc = desc_src->comp[0].depth;
1402
0
    if (c->srcBpc < 8)
1403
0
        c->srcBpc = 8;
1404
0
    c->dstBpc = desc_dst->comp[0].depth;
1405
0
    if (c->dstBpc < 8)
1406
0
        c->dstBpc = 8;
1407
0
    if (isAnyRGB(srcFormat) || srcFormat == AV_PIX_FMT_PAL8)
1408
0
        c->srcBpc = 16;
1409
0
    if (c->dstBpc == 16)
1410
0
        dst_stride <<= 1;
1411
1412
0
    if (INLINE_MMXEXT(cpu_flags) && c->srcBpc == 8 && c->dstBpc <= 14) {
1413
0
        c->canMMXEXTBeUsed = dstW >= srcW && (dstW & 31) == 0 &&
1414
0
                             c->chrDstW >= c->chrSrcW &&
1415
0
                             (srcW & 15) == 0;
1416
0
        if (!c->canMMXEXTBeUsed && dstW >= srcW && c->chrDstW >= c->chrSrcW && (srcW & 15) == 0
1417
1418
0
            && (flags & SWS_FAST_BILINEAR)) {
1419
0
            if (flags & SWS_PRINT_INFO)
1420
0
                av_log(c, AV_LOG_INFO,
1421
0
                       "output width is not a multiple of 32 -> no MMXEXT scaler\n");
1422
0
        }
1423
0
        if (usesHFilter || isNBPS(sws->src_format) || is16BPS(sws->src_format) || isAnyRGB(sws->src_format))
1424
0
            c->canMMXEXTBeUsed = 0;
1425
0
    } else
1426
0
        c->canMMXEXTBeUsed = 0;
1427
1428
0
    int64_t chrXInc = (((int64_t)c->chrSrcW << 16) + (c->chrDstW >> 1)) / c->chrDstW;
1429
0
    int64_t chrYInc = (((int64_t)c->chrSrcH << 16) + (c->chrDstH >> 1)) / c->chrDstH;
1430
1431
    /* Match pixel 0 of the src to pixel 0 of dst and match pixel n-2 of src
1432
     * to pixel n-2 of dst, but only for the FAST_BILINEAR mode otherwise do
1433
     * correct scaling.
1434
     * n-2 is the last chrominance sample available.
1435
     * This is not perfect, but no one should notice the difference, the more
1436
     * correct variant would be like the vertical one, but that would require
1437
     * some special code for the first and last pixel */
1438
0
    if (flags & SWS_FAST_BILINEAR) {
1439
0
        if (c->canMMXEXTBeUsed) {
1440
0
            lumXInc += 20;
1441
0
            chrXInc += 20;
1442
0
        }
1443
        // we don't use the x86 asm scaler if MMX is available
1444
0
        else if (INLINE_MMX(cpu_flags) && c->dstBpc <= 14) {
1445
0
            lumXInc = ((int64_t)(srcW       - 2) << 16) / (dstW       - 2) - 20;
1446
0
            chrXInc = ((int64_t)(c->chrSrcW - 2) << 16) / (c->chrDstW - 2) - 20;
1447
0
        }
1448
0
    }
1449
0
    if (chrXInc < 10 || chrXInc > INT_MAX ||
1450
0
        chrYInc < 10 || chrYInc > INT_MAX ||
1451
0
        lumXInc < 10 || lumXInc > INT_MAX ||
1452
0
        lumYInc < 10 || lumYInc > INT_MAX)
1453
0
        return AVERROR_PATCHWELCOME;
1454
1455
0
    c->lumXInc = lumXInc;
1456
0
    c->lumYInc = lumYInc;
1457
0
    c->chrXInc = chrXInc;
1458
0
    c->chrYInc = chrYInc;
1459
1460
1461
    // hardcoded for now
1462
0
    c->gamma_value = 2.2;
1463
0
    tmpFmt = AV_PIX_FMT_RGBA64LE;
1464
1465
0
    if (!unscaled && sws->gamma_flag && (srcFormat != tmpFmt || dstFormat != tmpFmt)) {
1466
0
        SwsInternal *c2;
1467
0
        c->cascaded_context[0] = NULL;
1468
1469
0
        ret = av_image_alloc(c->cascaded_tmp[0], c->cascaded_tmpStride[0],
1470
0
                            srcW, srcH, tmpFmt, 64);
1471
0
        if (ret < 0)
1472
0
            return ret;
1473
1474
0
        c->cascaded_context[0] = sws_getContext(srcW, srcH, srcFormat,
1475
0
                                                srcW, srcH, tmpFmt,
1476
0
                                                flags, NULL, NULL,
1477
0
                                                sws->scaler_params);
1478
0
        if (!c->cascaded_context[0]) {
1479
0
            return AVERROR(ENOMEM);
1480
0
        }
1481
1482
0
        c->cascaded_context[1] = sws_getContext(srcW, srcH, tmpFmt,
1483
0
                                                dstW, dstH, tmpFmt,
1484
0
                                                flags, srcFilter, dstFilter,
1485
0
                                                sws->scaler_params);
1486
1487
0
        if (!c->cascaded_context[1])
1488
0
            return AVERROR(ENOMEM);
1489
1490
0
        c2 = sws_internal(c->cascaded_context[1]);
1491
0
        c2->is_internal_gamma = 1;
1492
0
        c2->gamma     = alloc_gamma_tbl(    c->gamma_value);
1493
0
        c2->inv_gamma = alloc_gamma_tbl(1.f/c->gamma_value);
1494
0
        if (!c2->gamma || !c2->inv_gamma)
1495
0
            return AVERROR(ENOMEM);
1496
1497
        // is_internal_flag is set after creating the context
1498
        // to properly create the gamma convert FilterDescriptor
1499
        // we have to re-initialize it
1500
0
        ff_free_filters(c2);
1501
0
        if ((ret = ff_init_filters(c2)) < 0) {
1502
0
            sws_freeContext(c->cascaded_context[1]);
1503
0
            c->cascaded_context[1] = NULL;
1504
0
            return ret;
1505
0
        }
1506
1507
0
        c->cascaded_context[2] = NULL;
1508
0
        if (dstFormat != tmpFmt) {
1509
0
            ret = av_image_alloc(c->cascaded_tmp[1], c->cascaded_tmpStride[1],
1510
0
                                dstW, dstH, tmpFmt, 64);
1511
0
            if (ret < 0)
1512
0
                return ret;
1513
1514
0
            c->cascaded_context[2] = sws_getContext(dstW, dstH, tmpFmt,
1515
0
                                                    dstW, dstH, dstFormat,
1516
0
                                                    flags, NULL, NULL,
1517
0
                                                    sws->scaler_params);
1518
0
            if (!c->cascaded_context[2])
1519
0
                return AVERROR(ENOMEM);
1520
0
        }
1521
0
        return 0;
1522
0
    }
1523
1524
0
    if (isBayer(srcFormat)) {
1525
0
        if (!unscaled ||
1526
0
            (dstFormat != AV_PIX_FMT_RGB24 && dstFormat != AV_PIX_FMT_YUV420P &&
1527
0
             dstFormat != AV_PIX_FMT_RGB48)) {
1528
0
            enum AVPixelFormat tmpFormat = isBayer16BPS(srcFormat) ? AV_PIX_FMT_RGB48 : AV_PIX_FMT_RGB24;
1529
1530
0
            ret = av_image_alloc(c->cascaded_tmp[0], c->cascaded_tmpStride[0],
1531
0
                                srcW, srcH, tmpFormat, 64);
1532
0
            if (ret < 0)
1533
0
                return ret;
1534
1535
0
            c->cascaded_context[0] = sws_getContext(srcW, srcH, srcFormat,
1536
0
                                                    srcW, srcH, tmpFormat,
1537
0
                                                    flags, srcFilter, NULL,
1538
0
                                                    sws->scaler_params);
1539
0
            if (!c->cascaded_context[0])
1540
0
                return AVERROR(ENOMEM);
1541
1542
0
            c->cascaded_context[1] = sws_getContext(srcW, srcH, tmpFormat,
1543
0
                                                    dstW, dstH, dstFormat,
1544
0
                                                    flags, NULL, dstFilter,
1545
0
                                                    sws->scaler_params);
1546
0
            if (!c->cascaded_context[1])
1547
0
                return AVERROR(ENOMEM);
1548
0
            return 0;
1549
0
        }
1550
0
    }
1551
1552
0
    if (unscaled && c->srcBpc == 8 && dstFormat == AV_PIX_FMT_GRAYF32){
1553
0
        for (i = 0; i < 256; ++i){
1554
0
            c->uint2float_lut[i] = (float)i * float_mult;
1555
0
        }
1556
0
    }
1557
1558
    // float will be converted to uint16_t
1559
0
    if (isFloat(srcFormat) && !isAnyRGB(srcFormat) &&
1560
0
        (!unscaled || unscaled && dstFormat != srcFormat && (srcFormat != AV_PIX_FMT_GRAYF32 ||
1561
0
        dstFormat != AV_PIX_FMT_GRAY8))){
1562
0
        c->srcBpc = 16;
1563
0
    }
1564
1565
0
    if (CONFIG_SWSCALE_ALPHA && isALPHA(srcFormat) && !isALPHA(dstFormat)) {
1566
0
        enum AVPixelFormat tmpFormat = alphaless_fmt(srcFormat);
1567
1568
0
        if (tmpFormat != AV_PIX_FMT_NONE && sws->alpha_blend != SWS_ALPHA_BLEND_NONE) {
1569
0
            if (!unscaled ||
1570
0
                dstFormat != tmpFormat ||
1571
0
                usesHFilter || usesVFilter ||
1572
0
                sws->src_range != sws->dst_range
1573
0
            ) {
1574
0
                c->cascaded_mainindex = 1;
1575
0
                ret = av_image_alloc(c->cascaded_tmp[0], c->cascaded_tmpStride[0],
1576
0
                                     srcW, srcH, tmpFormat, 64);
1577
0
                if (ret < 0)
1578
0
                    return ret;
1579
1580
0
                c->cascaded_context[0] = alloc_set_opts(srcW, srcH, srcFormat,
1581
0
                                                        srcW, srcH, tmpFormat,
1582
0
                                                        flags, sws->scaler_params);
1583
0
                if (!c->cascaded_context[0])
1584
0
                    return AVERROR(EINVAL);
1585
0
                c->cascaded_context[0]->alpha_blend = sws->alpha_blend;
1586
0
                ret = sws_init_context(c->cascaded_context[0], NULL , NULL);
1587
0
                if (ret < 0)
1588
0
                    return ret;
1589
1590
0
                c->cascaded_context[1] = alloc_set_opts(srcW, srcH, tmpFormat,
1591
0
                                                        dstW, dstH, dstFormat,
1592
0
                                                        flags, sws->scaler_params);
1593
0
                if (!c->cascaded_context[1])
1594
0
                    return AVERROR(EINVAL);
1595
1596
0
                c->cascaded_context[1]->src_range = sws->src_range;
1597
0
                c->cascaded_context[1]->dst_range = sws->dst_range;
1598
0
                ret = sws_init_context(c->cascaded_context[1], srcFilter , dstFilter);
1599
0
                if (ret < 0)
1600
0
                    return ret;
1601
1602
0
                return 0;
1603
0
            }
1604
0
        }
1605
0
    }
1606
1607
    /* alpha blend special case, note this has been split via cascaded contexts if its scaled */
1608
0
    if (unscaled && !usesHFilter && !usesVFilter &&
1609
0
        sws->alpha_blend != SWS_ALPHA_BLEND_NONE &&
1610
0
        isALPHA(srcFormat) &&
1611
0
        (sws->src_range == sws->dst_range || isAnyRGB(dstFormat)) &&
1612
0
        alphaless_fmt(srcFormat) == dstFormat
1613
0
    ) {
1614
0
        c->convert_unscaled = ff_sws_alphablendaway;
1615
1616
0
        if (flags & SWS_PRINT_INFO)
1617
0
            av_log(c, AV_LOG_INFO,
1618
0
                    "using alpha blendaway %s -> %s special converter\n",
1619
0
                    av_get_pix_fmt_name(srcFormat), av_get_pix_fmt_name(dstFormat));
1620
0
        return 0;
1621
0
    }
1622
1623
    /* unscaled special cases */
1624
0
    if (unscaled && !usesHFilter && !usesVFilter &&
1625
0
        (sws->src_range == sws->dst_range || isAnyRGB(dstFormat) ||
1626
0
         isFloat(srcFormat) || isFloat(dstFormat) || isBayer(srcFormat))){
1627
1628
0
        ff_get_unscaled_swscale(c);
1629
1630
0
        if (c->convert_unscaled) {
1631
0
            if (flags & SWS_PRINT_INFO)
1632
0
                av_log(c, AV_LOG_INFO,
1633
0
                       "using unscaled %s -> %s special converter\n",
1634
0
                       av_get_pix_fmt_name(srcFormat), av_get_pix_fmt_name(dstFormat));
1635
0
            return 0;
1636
0
        }
1637
0
    }
1638
1639
    /* precalculate horizontal scaler filter coefficients */
1640
0
    {
1641
#if HAVE_MMXEXT_INLINE
1642
// can't downscale !!!
1643
        if (c->canMMXEXTBeUsed && (flags & SWS_FAST_BILINEAR)) {
1644
            c->lumMmxextFilterCodeSize = ff_init_hscaler_mmxext(dstW, c->lumXInc, NULL,
1645
                                                             NULL, NULL, 8);
1646
            c->chrMmxextFilterCodeSize = ff_init_hscaler_mmxext(c->chrDstW, c->chrXInc,
1647
                                                             NULL, NULL, NULL, 4);
1648
1649
            c->lumMmxextFilterCode = ff_sws_jit_alloc(c->lumMmxextFilterCodeSize);
1650
            c->chrMmxextFilterCode = ff_sws_jit_alloc(c->chrMmxextFilterCodeSize);
1651
            if (!c->lumMmxextFilterCode || !c->chrMmxextFilterCode) {
1652
                av_log(c, AV_LOG_ERROR, "Failed to allocate MMX2FilterCode\n");
1653
                return AVERROR(ENOMEM);
1654
            }
1655
1656
            if (!FF_ALLOCZ_TYPED_ARRAY(c->hLumFilter,    dstW           / 8 + 8) ||
1657
                !FF_ALLOCZ_TYPED_ARRAY(c->hChrFilter,    c->chrDstW     / 4 + 8) ||
1658
                !FF_ALLOCZ_TYPED_ARRAY(c->hLumFilterPos, dstW       / 2 / 8 + 8) ||
1659
                !FF_ALLOCZ_TYPED_ARRAY(c->hChrFilterPos, c->chrDstW / 2 / 4 + 8))
1660
                goto nomem;
1661
1662
            ff_init_hscaler_mmxext(      dstW, c->lumXInc, c->lumMmxextFilterCode,
1663
                                c->hLumFilter, (uint32_t*)c->hLumFilterPos, 8);
1664
            ff_init_hscaler_mmxext(c->chrDstW, c->chrXInc, c->chrMmxextFilterCode,
1665
                                c->hChrFilter, (uint32_t*)c->hChrFilterPos, 4);
1666
1667
            if ((ret = ff_sws_jit_protect(c->lumMmxextFilterCode, c->lumMmxextFilterCodeSize)) < 0 ||
1668
                (ret = ff_sws_jit_protect(c->chrMmxextFilterCode, c->chrMmxextFilterCodeSize)) < 0) {
1669
                av_log(c, AV_LOG_ERROR, "mprotect failed, cannot use fast bilinear scaler\n");
1670
                goto fail;
1671
            }
1672
        } else
1673
#endif /* HAVE_MMXEXT_INLINE */
1674
0
        {
1675
0
            const int filterAlign = X86_MMX(cpu_flags)     ? 4 :
1676
0
                                    PPC_ALTIVEC(cpu_flags) ? 8 :
1677
0
                                    have_neon(cpu_flags)   ? 4 :
1678
0
                                    have_lsx(cpu_flags)    ? 8 :
1679
0
                                    have_lasx(cpu_flags)   ? 8 : 1;
1680
1681
0
            if ((ret = initFilter(&c->hLumFilter, &c->hLumFilterPos,
1682
0
                           &c->hLumFilterSize, c->lumXInc,
1683
0
                           srcW, dstW, filterAlign, 1 << 14,
1684
0
                           lum_scaler, flags,
1685
0
                           cpu_flags, srcFilter->lumH, dstFilter->lumH,
1686
0
                           sws->scaler_params,
1687
0
                           get_local_pos(c, 0, 0, 0),
1688
0
                           get_local_pos(c, 0, 0, 0))) < 0)
1689
0
                goto fail;
1690
0
            if (ff_shuffle_filter_coefficients(c, c->hLumFilterPos, c->hLumFilterSize, c->hLumFilter, dstW) < 0)
1691
0
                goto nomem;
1692
0
            if ((ret = initFilter(&c->hChrFilter, &c->hChrFilterPos,
1693
0
                           &c->hChrFilterSize, c->chrXInc,
1694
0
                           c->chrSrcW, c->chrDstW, filterAlign, 1 << 14,
1695
0
                           chr_scaler, flags,
1696
0
                           cpu_flags, srcFilter->chrH, dstFilter->chrH,
1697
0
                           sws->scaler_params,
1698
0
                           get_local_pos(c, c->chrSrcHSubSample, sws->src_h_chr_pos, 0),
1699
0
                           get_local_pos(c, c->chrDstHSubSample, sws->dst_h_chr_pos, 0))) < 0)
1700
0
                goto fail;
1701
0
            if (ff_shuffle_filter_coefficients(c, c->hChrFilterPos, c->hChrFilterSize, c->hChrFilter, c->chrDstW) < 0)
1702
0
                goto nomem;
1703
0
        }
1704
0
    } // initialize horizontal stuff
1705
1706
    /* precalculate vertical scaler filter coefficients */
1707
0
    {
1708
0
        const int filterAlign = X86_MMX(cpu_flags)     ? 2 :
1709
0
                                PPC_ALTIVEC(cpu_flags) ? 8 :
1710
0
                                have_neon(cpu_flags)   ? 2 : 1;
1711
1712
0
        ret = initFilter(&c->vLumFilter, &c->vLumFilterPos, &c->vLumFilterSize,
1713
0
                       c->lumYInc, srcH, dstH, filterAlign, (1 << 12),
1714
0
                       lum_scaler, flags,
1715
0
                       cpu_flags, srcFilter->lumV, dstFilter->lumV,
1716
0
                       sws->scaler_params,
1717
0
                       get_local_pos(c, 0, 0, 1),
1718
0
                       get_local_pos(c, 0, 0, 1));
1719
0
        int usecascade = (ret == RETCODE_USE_CASCADE);
1720
0
        if (ret < 0 && !usecascade)
1721
0
            goto fail;
1722
0
        if ((ret = initFilter(&c->vChrFilter, &c->vChrFilterPos, &c->vChrFilterSize,
1723
0
                       c->chrYInc, c->chrSrcH, c->chrDstH,
1724
0
                       filterAlign, (1 << 12),
1725
0
                       chr_scaler, flags,
1726
0
                       cpu_flags, srcFilter->chrV, dstFilter->chrV,
1727
0
                       sws->scaler_params,
1728
0
                       get_local_pos(c, c->chrSrcVSubSample, sws->src_v_chr_pos, 1),
1729
0
                       get_local_pos(c, c->chrDstVSubSample, sws->dst_v_chr_pos, 1))) < 0)
1730
1731
0
            goto fail;
1732
0
        if (usecascade) {
1733
0
            ret = RETCODE_USE_CASCADE;
1734
0
            goto fail;
1735
0
        }
1736
1737
#if HAVE_ALTIVEC
1738
        ret = ff_sws_init_altivec_bufs(c);
1739
        if (ret < 0)
1740
            goto fail;
1741
#endif
1742
0
    }
1743
1744
0
    for (i = 0; i < 4; i++)
1745
0
        if (!FF_ALLOCZ_TYPED_ARRAY(c->dither_error[i], sws->dst_w + 3))
1746
0
            goto nomem;
1747
1748
0
    c->needAlpha = (CONFIG_SWSCALE_ALPHA && isALPHA(sws->src_format) && isALPHA(sws->dst_format)) ? 1 : 0;
1749
1750
    // 64 / c->scalingBpp is the same as 16 / sizeof(scaling_intermediate)
1751
0
    c->uv_off   = (dst_stride>>1) + 64 / (c->dstBpc &~ 7);
1752
0
    c->uv_offx2 = dst_stride + 16;
1753
1754
0
    av_assert0(c->chrDstH <= dstH);
1755
1756
0
    if (flags & SWS_PRINT_INFO) {
1757
0
        const char *scaler = NULL, *cpucaps;
1758
1759
0
        for (i = 0; i < FF_ARRAY_ELEMS(scale_algorithms); i++) {
1760
0
            if (flags & scale_algorithms[i].flag) {
1761
0
                scaler = scale_algorithms[i].description;
1762
0
                break;
1763
0
            }
1764
0
        }
1765
0
        if (!scaler)
1766
0
            scaler =  "ehh flags invalid?!";
1767
0
        av_log(c, AV_LOG_INFO, "%s scaler, from %s to %s%s ",
1768
0
               scaler,
1769
0
               av_get_pix_fmt_name(srcFormat),
1770
0
               dstFormat == AV_PIX_FMT_BGR555   || dstFormat == AV_PIX_FMT_BGR565   ||
1771
0
               dstFormat == AV_PIX_FMT_RGB444BE || dstFormat == AV_PIX_FMT_RGB444LE ||
1772
0
               dstFormat == AV_PIX_FMT_BGR444BE || dstFormat == AV_PIX_FMT_BGR444LE ?
1773
0
                                                             "dithered " : "",
1774
0
               av_get_pix_fmt_name(dstFormat));
1775
1776
0
        if (INLINE_MMXEXT(cpu_flags))
1777
0
            cpucaps = "MMXEXT";
1778
0
        else if (INLINE_MMX(cpu_flags))
1779
0
            cpucaps = "MMX";
1780
0
        else if (PPC_ALTIVEC(cpu_flags))
1781
0
            cpucaps = "AltiVec";
1782
0
        else
1783
0
            cpucaps = "C";
1784
1785
0
        av_log(c, AV_LOG_INFO, "using %s\n", cpucaps);
1786
1787
0
        av_log(c, AV_LOG_VERBOSE, "%dx%d -> %dx%d\n", srcW, srcH, dstW, dstH);
1788
0
        av_log(c, AV_LOG_DEBUG,
1789
0
               "lum srcW=%d srcH=%d dstW=%d dstH=%d xInc=%d yInc=%d\n",
1790
0
               sws->src_w, sws->src_h, sws->dst_w, sws->dst_h, c->lumXInc, c->lumYInc);
1791
0
        av_log(c, AV_LOG_DEBUG,
1792
0
               "chr srcW=%d srcH=%d dstW=%d dstH=%d xInc=%d yInc=%d\n",
1793
0
               c->chrSrcW, c->chrSrcH, c->chrDstW, c->chrDstH,
1794
0
               c->chrXInc, c->chrYInc);
1795
0
    }
1796
1797
0
    ff_sws_init_scale(c);
1798
1799
0
    return ff_init_filters(c);
1800
0
nomem:
1801
0
    ret = AVERROR(ENOMEM);
1802
0
fail: // FIXME replace things by appropriate error codes
1803
0
    if (ret == RETCODE_USE_CASCADE)  {
1804
0
        int tmpW = sqrt(srcW * (int64_t)dstW);
1805
0
        int tmpH = sqrt(srcH * (int64_t)dstH);
1806
0
        enum AVPixelFormat tmpFormat = AV_PIX_FMT_YUV420P;
1807
1808
0
        if (isALPHA(srcFormat))
1809
0
            tmpFormat = AV_PIX_FMT_YUVA420P;
1810
1811
0
        if (srcW*(int64_t)srcH <= 4LL*dstW*dstH)
1812
0
            return AVERROR(EINVAL);
1813
1814
0
        ret = av_image_alloc(c->cascaded_tmp[0], c->cascaded_tmpStride[0],
1815
0
                             tmpW, tmpH, tmpFormat, 64);
1816
0
        if (ret < 0)
1817
0
            return ret;
1818
1819
0
        c->cascaded_context[0] = sws_getContext(srcW, srcH, srcFormat,
1820
0
                                                tmpW, tmpH, tmpFormat,
1821
0
                                                flags, srcFilter, NULL,
1822
0
                                                sws->scaler_params);
1823
0
        if (!c->cascaded_context[0])
1824
0
            return AVERROR(ENOMEM);
1825
1826
0
        c->cascaded_context[1] = sws_getContext(tmpW, tmpH, tmpFormat,
1827
0
                                                dstW, dstH, dstFormat,
1828
0
                                                flags, NULL, dstFilter,
1829
0
                                                sws->scaler_params);
1830
0
        if (!c->cascaded_context[1])
1831
0
            return AVERROR(ENOMEM);
1832
0
        return 0;
1833
0
    }
1834
0
    return ret;
1835
0
}
1836
1837
static int context_init_threaded(SwsContext *sws,
1838
                                 SwsFilter *src_filter, SwsFilter *dst_filter)
1839
0
{
1840
0
    SwsInternal *c = sws_internal(sws);
1841
0
    int ret;
1842
1843
0
    ret = avpriv_slicethread_create2(&c->slicethread, (void*) sws,
1844
0
                                     ff_sws_slice_worker, NULL, sws->threads);
1845
0
    if (ret == AVERROR(ENOSYS)) {
1846
0
        sws->threads = 1;
1847
0
        return 0;
1848
0
    } else if (ret < 0)
1849
0
        return ret;
1850
1851
0
    sws->threads = ret;
1852
1853
0
    c->slice_ctx = av_calloc(sws->threads, sizeof(*c->slice_ctx));
1854
0
    if (!c->slice_ctx)
1855
0
        return AVERROR(ENOMEM);
1856
1857
0
    for (int i = 0; i < sws->threads; i++) {
1858
0
        SwsContext *slice;
1859
0
        slice = c->slice_ctx[i] = sws_alloc_context();
1860
0
        if (!slice)
1861
0
            return AVERROR(ENOMEM);
1862
0
        sws_internal(slice)->parent = sws;
1863
0
        c->nb_slice_ctx++;
1864
1865
0
        ret = av_opt_copy(slice, sws);
1866
0
        if (ret < 0)
1867
0
            return ret;
1868
0
        slice->threads = 1;
1869
1870
0
        ret = ff_sws_init_single_context(slice, src_filter, dst_filter);
1871
0
        if (ret < 0)
1872
0
            return ret;
1873
1874
0
        if (slice->dither == SWS_DITHER_ED) {
1875
0
            av_log(c, AV_LOG_VERBOSE,
1876
0
                   "Error-diffusion dither is in use, scaling will be single-threaded.");
1877
0
            break;
1878
0
        }
1879
0
    }
1880
1881
0
    return 0;
1882
0
}
1883
1884
av_cold int sws_init_context(SwsContext *sws, SwsFilter *srcFilter,
1885
                             SwsFilter *dstFilter)
1886
0
{
1887
0
    SwsInternal *c = sws_internal(sws);
1888
0
    static AVOnce rgb2rgb_once = AV_ONCE_INIT;
1889
0
    enum AVPixelFormat src_format, dst_format;
1890
0
    int ret;
1891
1892
0
    c->is_legacy_init = 1;
1893
0
    c->frame_src = av_frame_alloc();
1894
0
    c->frame_dst = av_frame_alloc();
1895
0
    if (!c->frame_src || !c->frame_dst)
1896
0
        return AVERROR(ENOMEM);
1897
1898
0
    if (ff_thread_once(&rgb2rgb_once, ff_sws_rgb2rgb_init) != 0)
1899
0
        return AVERROR_UNKNOWN;
1900
1901
0
    src_format = sws->src_format;
1902
0
    dst_format = sws->dst_format;
1903
0
    sws->src_range |= handle_jpeg(&sws->src_format);
1904
0
    sws->dst_range |= handle_jpeg(&sws->dst_format);
1905
1906
0
    if (src_format != sws->src_format || dst_format != sws->dst_format)
1907
0
        av_log(c, AV_LOG_WARNING, "deprecated pixel format used, make sure you did set range correctly\n");
1908
1909
0
    if (sws->threads != 1) {
1910
0
        ret = context_init_threaded(sws, srcFilter, dstFilter);
1911
0
        if (ret < 0 || sws->threads > 1)
1912
0
            return ret;
1913
        // threading disabled in this build, init as single-threaded
1914
0
    }
1915
1916
0
    return ff_sws_init_single_context(sws, srcFilter, dstFilter);
1917
0
}
1918
1919
SwsContext *sws_getContext(int srcW, int srcH, enum AVPixelFormat srcFormat,
1920
                           int dstW, int dstH, enum AVPixelFormat dstFormat,
1921
                           int flags, SwsFilter *srcFilter,
1922
                           SwsFilter *dstFilter, const double *param)
1923
0
{
1924
0
    SwsContext *sws;
1925
1926
0
    sws = alloc_set_opts(srcW, srcH, srcFormat,
1927
0
                         dstW, dstH, dstFormat,
1928
0
                         flags, param);
1929
0
    if (!sws)
1930
0
        return NULL;
1931
1932
0
    if (sws_init_context(sws, srcFilter, dstFilter) < 0) {
1933
0
        sws_freeContext(sws);
1934
0
        return NULL;
1935
0
    }
1936
1937
0
    return sws;
1938
0
}
1939
1940
static int isnan_vec(SwsVector *a)
1941
0
{
1942
0
    int i;
1943
0
    for (i=0; i<a->length; i++)
1944
0
        if (isnan(a->coeff[i]))
1945
0
            return 1;
1946
0
    return 0;
1947
0
}
1948
1949
static void makenan_vec(SwsVector *a)
1950
0
{
1951
0
    int i;
1952
0
    for (i=0; i<a->length; i++)
1953
0
        a->coeff[i] = NAN;
1954
0
}
1955
1956
SwsVector *sws_allocVec(int length)
1957
0
{
1958
0
    SwsVector *vec;
1959
1960
0
    if(length <= 0 || length > INT_MAX/ sizeof(double))
1961
0
        return NULL;
1962
1963
0
    vec = av_malloc(sizeof(SwsVector));
1964
0
    if (!vec)
1965
0
        return NULL;
1966
0
    vec->length = length;
1967
0
    vec->coeff  = av_malloc(sizeof(double) * length);
1968
0
    if (!vec->coeff)
1969
0
        av_freep(&vec);
1970
0
    return vec;
1971
0
}
1972
1973
SwsVector *sws_getGaussianVec(double variance, double quality)
1974
0
{
1975
0
    const int length = (int)(variance * quality + 0.5) | 1;
1976
0
    int i;
1977
0
    double middle  = (length - 1) * 0.5;
1978
0
    SwsVector *vec;
1979
1980
0
    if(variance < 0 || quality < 0)
1981
0
        return NULL;
1982
1983
0
    vec = sws_allocVec(length);
1984
1985
0
    if (!vec)
1986
0
        return NULL;
1987
1988
0
    for (i = 0; i < length; i++) {
1989
0
        double dist = i - middle;
1990
0
        vec->coeff[i] = exp(-dist * dist / (2 * variance * variance)) /
1991
0
                        sqrt(2 * variance * M_PI);
1992
0
    }
1993
1994
0
    sws_normalizeVec(vec, 1.0);
1995
1996
0
    return vec;
1997
0
}
1998
1999
/**
2000
 * Allocate and return a vector with length coefficients, all
2001
 * with the same value c.
2002
 */
2003
static
2004
SwsVector *sws_getConstVec(double c, int length)
2005
0
{
2006
0
    int i;
2007
0
    SwsVector *vec = sws_allocVec(length);
2008
2009
0
    if (!vec)
2010
0
        return NULL;
2011
2012
0
    for (i = 0; i < length; i++)
2013
0
        vec->coeff[i] = c;
2014
2015
0
    return vec;
2016
0
}
2017
2018
/**
2019
 * Allocate and return a vector with just one coefficient, with
2020
 * value 1.0.
2021
 */
2022
static
2023
SwsVector *sws_getIdentityVec(void)
2024
0
{
2025
0
    return sws_getConstVec(1.0, 1);
2026
0
}
2027
2028
static double sws_dcVec(SwsVector *a)
2029
0
{
2030
0
    int i;
2031
0
    double sum = 0;
2032
2033
0
    for (i = 0; i < a->length; i++)
2034
0
        sum += a->coeff[i];
2035
2036
0
    return sum;
2037
0
}
2038
2039
void sws_scaleVec(SwsVector *a, double scalar)
2040
0
{
2041
0
    int i;
2042
2043
0
    for (i = 0; i < a->length; i++)
2044
0
        a->coeff[i] *= scalar;
2045
0
}
2046
2047
void sws_normalizeVec(SwsVector *a, double height)
2048
0
{
2049
0
    sws_scaleVec(a, height / sws_dcVec(a));
2050
0
}
2051
2052
static SwsVector *sws_sumVec(SwsVector *a, SwsVector *b)
2053
0
{
2054
0
    int length = FFMAX(a->length, b->length);
2055
0
    int i;
2056
0
    SwsVector *vec = sws_getConstVec(0.0, length);
2057
2058
0
    if (!vec)
2059
0
        return NULL;
2060
2061
0
    for (i = 0; i < a->length; i++)
2062
0
        vec->coeff[i + (length - 1) / 2 - (a->length - 1) / 2] += a->coeff[i];
2063
0
    for (i = 0; i < b->length; i++)
2064
0
        vec->coeff[i + (length - 1) / 2 - (b->length - 1) / 2] += b->coeff[i];
2065
2066
0
    return vec;
2067
0
}
2068
2069
/* shift left / or right if "shift" is negative */
2070
static SwsVector *sws_getShiftedVec(SwsVector *a, int shift)
2071
0
{
2072
0
    int length = a->length + FFABS(shift) * 2;
2073
0
    int i;
2074
0
    SwsVector *vec = sws_getConstVec(0.0, length);
2075
2076
0
    if (!vec)
2077
0
        return NULL;
2078
2079
0
    for (i = 0; i < a->length; i++) {
2080
0
        vec->coeff[i + (length    - 1) / 2 -
2081
0
                       (a->length - 1) / 2 - shift] = a->coeff[i];
2082
0
    }
2083
2084
0
    return vec;
2085
0
}
2086
2087
static
2088
void sws_shiftVec(SwsVector *a, int shift)
2089
0
{
2090
0
    SwsVector *shifted = sws_getShiftedVec(a, shift);
2091
0
    if (!shifted) {
2092
0
        makenan_vec(a);
2093
0
        return;
2094
0
    }
2095
0
    av_free(a->coeff);
2096
0
    a->coeff  = shifted->coeff;
2097
0
    a->length = shifted->length;
2098
0
    av_free(shifted);
2099
0
}
2100
2101
static
2102
void sws_addVec(SwsVector *a, SwsVector *b)
2103
0
{
2104
0
    SwsVector *sum = sws_sumVec(a, b);
2105
0
    if (!sum) {
2106
0
        makenan_vec(a);
2107
0
        return;
2108
0
    }
2109
0
    av_free(a->coeff);
2110
0
    a->coeff  = sum->coeff;
2111
0
    a->length = sum->length;
2112
0
    av_free(sum);
2113
0
}
2114
2115
/**
2116
 * Print with av_log() a textual representation of the vector a
2117
 * if log_level <= av_log_level.
2118
 */
2119
static
2120
void sws_printVec2(SwsVector *a, AVClass *log_ctx, int log_level)
2121
0
{
2122
0
    int i;
2123
0
    double max = 0;
2124
0
    double min = 0;
2125
0
    double range;
2126
2127
0
    for (i = 0; i < a->length; i++)
2128
0
        if (a->coeff[i] > max)
2129
0
            max = a->coeff[i];
2130
2131
0
    for (i = 0; i < a->length; i++)
2132
0
        if (a->coeff[i] < min)
2133
0
            min = a->coeff[i];
2134
2135
0
    range = max - min;
2136
2137
0
    for (i = 0; i < a->length; i++) {
2138
0
        int x = (int)((a->coeff[i] - min) * 60.0 / range + 0.5);
2139
0
        av_log(log_ctx, log_level, "%1.3f ", a->coeff[i]);
2140
0
        for (; x > 0; x--)
2141
0
            av_log(log_ctx, log_level, " ");
2142
0
        av_log(log_ctx, log_level, "|\n");
2143
0
    }
2144
0
}
2145
2146
void sws_freeVec(SwsVector *a)
2147
0
{
2148
0
    if (!a)
2149
0
        return;
2150
0
    av_freep(&a->coeff);
2151
0
    a->length = 0;
2152
0
    av_free(a);
2153
0
}
2154
2155
void sws_freeFilter(SwsFilter *filter)
2156
0
{
2157
0
    if (!filter)
2158
0
        return;
2159
2160
0
    sws_freeVec(filter->lumH);
2161
0
    sws_freeVec(filter->lumV);
2162
0
    sws_freeVec(filter->chrH);
2163
0
    sws_freeVec(filter->chrV);
2164
0
    av_free(filter);
2165
0
}
2166
2167
SwsFilter *sws_getDefaultFilter(float lumaGBlur, float chromaGBlur,
2168
                                float lumaSharpen, float chromaSharpen,
2169
                                float chromaHShift, float chromaVShift,
2170
                                int verbose)
2171
0
{
2172
0
    SwsFilter *filter = av_malloc(sizeof(SwsFilter));
2173
0
    if (!filter)
2174
0
        return NULL;
2175
2176
0
    if (lumaGBlur != 0.0) {
2177
0
        filter->lumH = sws_getGaussianVec(lumaGBlur, 3.0);
2178
0
        filter->lumV = sws_getGaussianVec(lumaGBlur, 3.0);
2179
0
    } else {
2180
0
        filter->lumH = sws_getIdentityVec();
2181
0
        filter->lumV = sws_getIdentityVec();
2182
0
    }
2183
2184
0
    if (chromaGBlur != 0.0) {
2185
0
        filter->chrH = sws_getGaussianVec(chromaGBlur, 3.0);
2186
0
        filter->chrV = sws_getGaussianVec(chromaGBlur, 3.0);
2187
0
    } else {
2188
0
        filter->chrH = sws_getIdentityVec();
2189
0
        filter->chrV = sws_getIdentityVec();
2190
0
    }
2191
2192
0
    if (!filter->lumH || !filter->lumV || !filter->chrH || !filter->chrV)
2193
0
        goto fail;
2194
2195
0
    if (chromaSharpen != 0.0) {
2196
0
        SwsVector *id = sws_getIdentityVec();
2197
0
        if (!id)
2198
0
            goto fail;
2199
0
        sws_scaleVec(filter->chrH, -chromaSharpen);
2200
0
        sws_scaleVec(filter->chrV, -chromaSharpen);
2201
0
        sws_addVec(filter->chrH, id);
2202
0
        sws_addVec(filter->chrV, id);
2203
0
        sws_freeVec(id);
2204
0
    }
2205
2206
0
    if (lumaSharpen != 0.0) {
2207
0
        SwsVector *id = sws_getIdentityVec();
2208
0
        if (!id)
2209
0
            goto fail;
2210
0
        sws_scaleVec(filter->lumH, -lumaSharpen);
2211
0
        sws_scaleVec(filter->lumV, -lumaSharpen);
2212
0
        sws_addVec(filter->lumH, id);
2213
0
        sws_addVec(filter->lumV, id);
2214
0
        sws_freeVec(id);
2215
0
    }
2216
2217
0
    if (chromaHShift != 0.0)
2218
0
        sws_shiftVec(filter->chrH, (int)(chromaHShift + 0.5));
2219
2220
0
    if (chromaVShift != 0.0)
2221
0
        sws_shiftVec(filter->chrV, (int)(chromaVShift + 0.5));
2222
2223
0
    sws_normalizeVec(filter->chrH, 1.0);
2224
0
    sws_normalizeVec(filter->chrV, 1.0);
2225
0
    sws_normalizeVec(filter->lumH, 1.0);
2226
0
    sws_normalizeVec(filter->lumV, 1.0);
2227
2228
0
    if (isnan_vec(filter->chrH) ||
2229
0
        isnan_vec(filter->chrV) ||
2230
0
        isnan_vec(filter->lumH) ||
2231
0
        isnan_vec(filter->lumV))
2232
0
        goto fail;
2233
2234
0
    if (verbose)
2235
0
        sws_printVec2(filter->chrH, NULL, AV_LOG_DEBUG);
2236
0
    if (verbose)
2237
0
        sws_printVec2(filter->lumH, NULL, AV_LOG_DEBUG);
2238
2239
0
    return filter;
2240
2241
0
fail:
2242
0
    sws_freeVec(filter->lumH);
2243
0
    sws_freeVec(filter->lumV);
2244
0
    sws_freeVec(filter->chrH);
2245
0
    sws_freeVec(filter->chrV);
2246
0
    av_freep(&filter);
2247
0
    return NULL;
2248
0
}
2249
2250
void sws_freeContext(SwsContext *sws)
2251
0
{
2252
0
    SwsInternal *c = sws_internal(sws);
2253
0
    int i;
2254
0
    if (!c)
2255
0
        return;
2256
2257
0
    av_refstruct_unref(&c->hw_priv);
2258
2259
0
    for (i = 0; i < FF_ARRAY_ELEMS(c->graph); i++)
2260
0
        ff_sws_graph_free(&c->graph[i]);
2261
0
    ff_frame_pool_uninit(&c->frame_pool);
2262
2263
0
    for (i = 0; i < c->nb_slice_ctx; i++)
2264
0
        sws_freeContext(c->slice_ctx[i]);
2265
0
    av_freep(&c->slice_ctx);
2266
2267
0
    avpriv_slicethread_free(&c->slicethread);
2268
2269
0
    for (i = 0; i < 4; i++)
2270
0
        av_freep(&c->dither_error[i]);
2271
2272
0
    av_frame_free(&c->frame_src);
2273
0
    av_frame_free(&c->frame_dst);
2274
2275
0
    av_freep(&c->src_ranges.ranges);
2276
2277
0
    av_freep(&c->vLumFilter);
2278
0
    av_freep(&c->vChrFilter);
2279
0
    av_freep(&c->hLumFilter);
2280
0
    av_freep(&c->hChrFilter);
2281
#if HAVE_ALTIVEC
2282
    ff_sws_free_altivec_bufs(c);
2283
#endif
2284
2285
0
    av_freep(&c->vLumFilterPos);
2286
0
    av_freep(&c->vChrFilterPos);
2287
0
    av_freep(&c->hLumFilterPos);
2288
0
    av_freep(&c->hChrFilterPos);
2289
2290
#if HAVE_MMX_INLINE
2291
    ff_sws_jit_free(c->lumMmxextFilterCode, c->lumMmxextFilterCodeSize);
2292
    ff_sws_jit_free(c->chrMmxextFilterCode, c->chrMmxextFilterCodeSize);
2293
    c->lumMmxextFilterCode = NULL;
2294
    c->chrMmxextFilterCode = NULL;
2295
#endif /* HAVE_MMX_INLINE */
2296
2297
0
    av_freep(&c->yuvTable);
2298
0
    av_freep(&c->formatConvBuffer);
2299
2300
0
    sws_freeContext(c->cascaded_context[0]);
2301
0
    sws_freeContext(c->cascaded_context[1]);
2302
0
    sws_freeContext(c->cascaded_context[2]);
2303
0
    memset(c->cascaded_context, 0, sizeof(c->cascaded_context));
2304
0
    av_freep(&c->cascaded_tmp[0][0]);
2305
0
    av_freep(&c->cascaded_tmp[1][0]);
2306
2307
0
    av_freep(&c->gamma);
2308
0
    av_freep(&c->inv_gamma);
2309
#if CONFIG_SMALL
2310
    av_freep(&c->xyz2rgb.gamma.in);
2311
#endif
2312
2313
0
    av_freep(&c->rgb0_scratch);
2314
0
    av_freep(&c->xyz_scratch);
2315
2316
0
    ff_free_filters(c);
2317
2318
0
    av_free(c);
2319
0
}
2320
2321
void sws_free_context(SwsContext **pctx)
2322
0
{
2323
0
    SwsContext *ctx = *pctx;
2324
0
    if (!ctx)
2325
0
        return;
2326
2327
0
    sws_freeContext(ctx);
2328
0
    *pctx = NULL;
2329
0
}
2330
2331
SwsContext *sws_getCachedContext(SwsContext *prev, int srcW,
2332
                                 int srcH, enum AVPixelFormat srcFormat,
2333
                                 int dstW, int dstH,
2334
                                 enum AVPixelFormat dstFormat, int flags,
2335
                                 SwsFilter *srcFilter,
2336
                                 SwsFilter *dstFilter,
2337
                                 const double *param)
2338
0
{
2339
0
    SwsContext *sws;
2340
0
    static const double default_param[2] = { SWS_PARAM_DEFAULT,
2341
0
                                             SWS_PARAM_DEFAULT };
2342
2343
0
    if (!param)
2344
0
        param = default_param;
2345
2346
0
    if (prev && (prev->src_w            == srcW      &&
2347
0
                 prev->src_h            == srcH      &&
2348
0
                 prev->src_format       == srcFormat &&
2349
0
                 prev->dst_w            == dstW      &&
2350
0
                 prev->dst_h            == dstH      &&
2351
0
                 prev->dst_format       == dstFormat &&
2352
0
                 prev->flags            == flags     &&
2353
0
                 !memcmp(prev->scaler_params, param,
2354
0
                         sizeof(prev->scaler_params)))) {
2355
0
        return prev;
2356
0
    }
2357
2358
0
    if (!(sws = sws_alloc_context())) {
2359
0
        sws_free_context(&prev);
2360
0
        return NULL;
2361
0
    }
2362
2363
0
    if (prev) {
2364
0
        av_opt_copy(sws, prev);
2365
0
        sws_free_context(&prev);
2366
0
    }
2367
2368
0
    sws->src_w            = srcW;
2369
0
    sws->src_h            = srcH;
2370
0
    sws->src_format       = srcFormat;
2371
0
    sws->dst_w            = dstW;
2372
0
    sws->dst_h            = dstH;
2373
0
    sws->dst_format       = dstFormat;
2374
0
    sws->flags            = flags;
2375
0
    for (int i = 0; i < SWS_NUM_SCALER_PARAMS; i++)
2376
0
        sws->scaler_params[i] = param[i];
2377
2378
0
    if (sws_init_context(sws, srcFilter, dstFilter) < 0)
2379
0
        sws_free_context(&sws);
2380
2381
0
    return sws;
2382
0
}
2383
2384
int ff_range_add(RangeList *rl, unsigned int start, unsigned int len)
2385
0
{
2386
0
    Range *tmp;
2387
0
    unsigned int idx;
2388
2389
    /* find the first existing range after the new one */
2390
0
    for (idx = 0; idx < rl->nb_ranges; idx++)
2391
0
        if (rl->ranges[idx].start > start)
2392
0
            break;
2393
2394
    /* check for overlap */
2395
0
    if (idx > 0) {
2396
0
        Range *prev = &rl->ranges[idx - 1];
2397
0
        if (prev->start + prev->len > start)
2398
0
            return AVERROR(EINVAL);
2399
0
    }
2400
0
    if (idx < rl->nb_ranges) {
2401
0
        Range *next = &rl->ranges[idx];
2402
0
        if (start + len > next->start)
2403
0
            return AVERROR(EINVAL);
2404
0
    }
2405
2406
0
    tmp = av_fast_realloc(rl->ranges, &rl->ranges_allocated,
2407
0
                          (rl->nb_ranges + 1) * sizeof(*rl->ranges));
2408
0
    if (!tmp)
2409
0
        return AVERROR(ENOMEM);
2410
0
    rl->ranges = tmp;
2411
2412
0
    memmove(rl->ranges + idx + 1, rl->ranges + idx,
2413
0
            sizeof(*rl->ranges) * (rl->nb_ranges - idx));
2414
0
    rl->ranges[idx].start = start;
2415
0
    rl->ranges[idx].len   = len;
2416
0
    rl->nb_ranges++;
2417
2418
    /* merge ranges */
2419
0
    if (idx > 0) {
2420
0
        Range *prev = &rl->ranges[idx - 1];
2421
0
        Range *cur  = &rl->ranges[idx];
2422
0
        if (prev->start + prev->len == cur->start) {
2423
0
            prev->len += cur->len;
2424
0
            memmove(rl->ranges + idx - 1, rl->ranges + idx,
2425
0
                    sizeof(*rl->ranges) * (rl->nb_ranges - idx));
2426
0
            rl->nb_ranges--;
2427
0
            idx--;
2428
0
        }
2429
0
    }
2430
0
    if (idx < rl->nb_ranges - 1) {
2431
0
        Range *cur  = &rl->ranges[idx];
2432
0
        Range *next = &rl->ranges[idx + 1];
2433
0
        if (cur->start + cur->len == next->start) {
2434
0
            cur->len += next->len;
2435
0
            memmove(rl->ranges + idx, rl->ranges + idx + 1,
2436
0
                    sizeof(*rl->ranges) * (rl->nb_ranges - idx - 1));
2437
0
            rl->nb_ranges--;
2438
0
        }
2439
0
    }
2440
2441
0
    return 0;
2442
0
}
2443
2444
int ff_sws_thread_exec(void *priv,
2445
                       int (*func)(void *priv, int jobnr, int threadnr, int nb_jobs, int nb_threads),
2446
                       int nb_threads, int nb_jobs)
2447
0
{
2448
0
    AVSliceThread *slicethread;
2449
0
    int ret = avpriv_slicethread_create2(&slicethread, priv, func, NULL, nb_threads);
2450
0
    if (ret == AVERROR(ENOSYS)) {
2451
        /* Fallback for build configurations without threading */
2452
0
        for (int i = 0; i < nb_jobs; i++) {
2453
0
            int ret = func(priv, i, 0, nb_jobs, 1);
2454
0
            if (ret)
2455
0
                return ret;
2456
0
        }
2457
0
        return 0;
2458
0
    } else if (ret < 0) {
2459
0
        return ret;
2460
0
    }
2461
2462
0
    ret = avpriv_slicethread_execute2(slicethread, nb_jobs, 0);
2463
0
    avpriv_slicethread_free(&slicethread);
2464
0
    return ret;
2465
0
}