swscale: support AV_PIX_FMT_YA16 as input
[libav.git] / libswscale / swscale_internal.h
CommitLineData
5427e242 1/*
d026b45e
DB
2 * Copyright (C) 2001-2003 Michael Niedermayer <michaelni@gmx.at>
3 *
2912e87a 4 * This file is part of Libav.
d026b45e 5 *
2912e87a 6 * Libav is free software; you can redistribute it and/or
807e0c66
LA
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
d026b45e 10 *
2912e87a 11 * Libav is distributed in the hope that it will be useful,
d026b45e 12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
807e0c66
LA
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
d026b45e 15 *
807e0c66 16 * You should have received a copy of the GNU Lesser General Public
2912e87a 17 * License along with Libav; if not, write to the Free Software
b19bcbaa 18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
d026b45e 19 */
5427e242 20
0eaf5235
SS
21#ifndef SWSCALE_SWSCALE_INTERNAL_H
22#define SWSCALE_SWSCALE_INTERNAL_H
5427e242 23
4706949c
DB
24#include "config.h"
25
b63f641e 26#if HAVE_ALTIVEC_H
f2015f0c
AB
27#include <altivec.h>
28#endif
29
0a7068fa 30#include "libavutil/avassert.h"
83da2c6f 31#include "libavutil/avutil.h"
1d9c2dc8 32#include "libavutil/common.h"
abc78a5a 33#include "libavutil/log.h"
d49ea4af 34#include "libavutil/pixfmt.h"
6b0768e2 35#include "libavutil/pixdesc.h"
94c4def2 36
83890c66 37#define STR(s) AV_TOSTRING(s) // AV_STRINGIFY is too long
1625216e 38
83890c66 39#define FAST_BGR2YV12 // use 7-bit instead of 15-bit coefficients
0e1a5434 40
5427e242
MN
41#define MAX_FILTER_SIZE 256
42
a898cdc9 43#if HAVE_BIGENDIAN
9990e426
MN
44#define ALT32_CORR (-1)
45#else
46#define ALT32_CORR 1
1625216e
MN
47#endif
48
b63f641e 49#if ARCH_X86_64
83890c66 50# define APCK_PTR2 8
1625216e
MN
51# define APCK_COEF 16
52# define APCK_SIZE 24
53#else
83890c66
DB
54# define APCK_PTR2 4
55# define APCK_COEF 8
1625216e 56# define APCK_SIZE 16
9990e426
MN
57#endif
58
1615fb91
DB
59struct SwsContext;
60
83890c66 61typedef int (*SwsFunc)(struct SwsContext *context, const uint8_t *src[],
457eed72 62 int srcStride[], int srcSliceY, int srcSliceH,
83890c66 63 uint8_t *dst[], int dstStride[]);
34e8d147 64
d6cc6ac6 65/**
34e8d147 66 * Write one line of horizontally scaled data to planar output
d6cc6ac6
RB
67 * without any additional vertical scaling (or point-scaling).
68 *
34e8d147 69 * @param src scaled source data, 15bit for 8-10bit output,
28c1115a 70 * 19-bit for 16bit output (in int32_t)
34e8d147 71 * @param dest pointer to the output plane. For >8bit
28c1115a 72 * output, this is in uint16_t
34e8d147
KK
73 * @param dstW width of destination in pixels
74 * @param dither ordered dither array of type int16_t and size 8
75 * @param offset Dither offset
d6cc6ac6 76 */
83890c66
DB
77typedef void (*yuv2planar1_fn)(const int16_t *src, uint8_t *dest, int dstW,
78 const uint8_t *dither, int offset);
34e8d147 79
d6cc6ac6 80/**
ff7913ae 81 * Write one line of horizontally scaled data to planar output
d6cc6ac6
RB
82 * with multi-point vertical scaling between input pixels.
83 *
ff7913ae
KK
84 * @param filter vertical luma/alpha scaling coefficients, 12bit [0,4096]
85 * @param src scaled luma (Y) or alpha (A) source data, 15bit for 8-10bit output,
28c1115a 86 * 19-bit for 16bit output (in int32_t)
ff7913ae
KK
87 * @param filterSize number of vertical input lines to scale
88 * @param dest pointer to output plane. For >8bit
89 * output, this is in uint16_t
90 * @param dstW width of destination pixels
91 * @param offset Dither offset
92 */
83890c66
DB
93typedef void (*yuv2planarX_fn)(const int16_t *filter, int filterSize,
94 const int16_t **src, uint8_t *dest, int dstW,
95 const uint8_t *dither, int offset);
ff7913ae
KK
96
97/**
98 * Write one line of horizontally scaled chroma to interleaved output
99 * with multi-point vertical scaling between input pixels.
100 *
101 * @param c SWS scaling context
d6cc6ac6 102 * @param chrFilter vertical chroma scaling coefficients, 12bit [0,4096]
28c1115a
RB
103 * @param chrUSrc scaled chroma (U) source data, 15bit for 8-10bit output,
104 * 19-bit for 16bit output (in int32_t)
105 * @param chrVSrc scaled chroma (V) source data, 15bit for 8-10bit output,
106 * 19-bit for 16bit output (in int32_t)
d6cc6ac6 107 * @param chrFilterSize number of vertical chroma input lines to scale
ff7913ae 108 * @param dest pointer to the output plane. For >8bit
28c1115a 109 * output, this is in uint16_t
ff7913ae 110 * @param dstW width of chroma planes
d6cc6ac6 111 */
83890c66
DB
112typedef void (*yuv2interleavedX_fn)(struct SwsContext *c,
113 const int16_t *chrFilter,
114 int chrFilterSize,
115 const int16_t **chrUSrc,
116 const int16_t **chrVSrc,
117 uint8_t *dest, int dstW);
ff7913ae 118
d6cc6ac6
RB
119/**
120 * Write one line of horizontally scaled Y/U/V/A to packed-pixel YUV/RGB
121 * output without any additional vertical scaling (or point-scaling). Note
122 * that this function may do chroma scaling, see the "uvalpha" argument.
123 *
124 * @param c SWS scaling context
28c1115a
RB
125 * @param lumSrc scaled luma (Y) source data, 15bit for 8-10bit output,
126 * 19-bit for 16bit output (in int32_t)
127 * @param chrUSrc scaled chroma (U) source data, 15bit for 8-10bit output,
128 * 19-bit for 16bit output (in int32_t)
129 * @param chrVSrc scaled chroma (V) source data, 15bit for 8-10bit output,
130 * 19-bit for 16bit output (in int32_t)
131 * @param alpSrc scaled alpha (A) source data, 15bit for 8-10bit output,
132 * 19-bit for 16bit output (in int32_t)
133 * @param dest pointer to the output plane. For 16bit output, this is
134 * uint16_t
d6cc6ac6
RB
135 * @param dstW width of lumSrc and alpSrc in pixels, number of pixels
136 * to write into dest[]
137 * @param uvalpha chroma scaling coefficient for the second line of chroma
138 * pixels, either 2048 or 0. If 0, one chroma input is used
139 * for 2 output pixels (or if the SWS_FLAG_FULL_CHR_INT flag
140 * is set, it generates 1 output pixel). If 2048, two chroma
141 * input pixels should be averaged for 2 output pixels (this
142 * only happens if SWS_FLAG_FULL_CHR_INT is not set)
143 * @param y vertical line number for this output. This does not need
144 * to be used to calculate the offset in the destination,
145 * but can be used to generate comfort noise using dithering
146 * for some output formats.
147 */
83890c66
DB
148typedef void (*yuv2packed1_fn)(struct SwsContext *c, const int16_t *lumSrc,
149 const int16_t *chrUSrc[2],
150 const int16_t *chrVSrc[2],
151 const int16_t *alpSrc, uint8_t *dest,
152 int dstW, int uvalpha, int y);
d6cc6ac6
RB
153/**
154 * Write one line of horizontally scaled Y/U/V/A to packed-pixel YUV/RGB
155 * output by doing bilinear scaling between two input lines.
156 *
157 * @param c SWS scaling context
28c1115a
RB
158 * @param lumSrc scaled luma (Y) source data, 15bit for 8-10bit output,
159 * 19-bit for 16bit output (in int32_t)
160 * @param chrUSrc scaled chroma (U) source data, 15bit for 8-10bit output,
161 * 19-bit for 16bit output (in int32_t)
162 * @param chrVSrc scaled chroma (V) source data, 15bit for 8-10bit output,
163 * 19-bit for 16bit output (in int32_t)
164 * @param alpSrc scaled alpha (A) source data, 15bit for 8-10bit output,
165 * 19-bit for 16bit output (in int32_t)
166 * @param dest pointer to the output plane. For 16bit output, this is
167 * uint16_t
d6cc6ac6
RB
168 * @param dstW width of lumSrc and alpSrc in pixels, number of pixels
169 * to write into dest[]
170 * @param yalpha luma/alpha scaling coefficients for the second input line.
171 * The first line's coefficients can be calculated by using
172 * 4096 - yalpha
173 * @param uvalpha chroma scaling coefficient for the second input line. The
174 * first line's coefficients can be calculated by using
175 * 4096 - uvalpha
176 * @param y vertical line number for this output. This does not need
177 * to be used to calculate the offset in the destination,
178 * but can be used to generate comfort noise using dithering
179 * for some output formats.
180 */
83890c66
DB
181typedef void (*yuv2packed2_fn)(struct SwsContext *c, const int16_t *lumSrc[2],
182 const int16_t *chrUSrc[2],
183 const int16_t *chrVSrc[2],
184 const int16_t *alpSrc[2],
185 uint8_t *dest,
186 int dstW, int yalpha, int uvalpha, int y);
d6cc6ac6
RB
187/**
188 * Write one line of horizontally scaled Y/U/V/A to packed-pixel YUV/RGB
189 * output by doing multi-point vertical scaling between input pixels.
190 *
191 * @param c SWS scaling context
192 * @param lumFilter vertical luma/alpha scaling coefficients, 12bit [0,4096]
28c1115a
RB
193 * @param lumSrc scaled luma (Y) source data, 15bit for 8-10bit output,
194 * 19-bit for 16bit output (in int32_t)
d6cc6ac6
RB
195 * @param lumFilterSize number of vertical luma/alpha input lines to scale
196 * @param chrFilter vertical chroma scaling coefficients, 12bit [0,4096]
28c1115a
RB
197 * @param chrUSrc scaled chroma (U) source data, 15bit for 8-10bit output,
198 * 19-bit for 16bit output (in int32_t)
199 * @param chrVSrc scaled chroma (V) source data, 15bit for 8-10bit output,
200 * 19-bit for 16bit output (in int32_t)
d6cc6ac6 201 * @param chrFilterSize number of vertical chroma input lines to scale
28c1115a
RB
202 * @param alpSrc scaled alpha (A) source data, 15bit for 8-10bit output,
203 * 19-bit for 16bit output (in int32_t)
204 * @param dest pointer to the output plane. For 16bit output, this is
205 * uint16_t
d6cc6ac6
RB
206 * @param dstW width of lumSrc and alpSrc in pixels, number of pixels
207 * to write into dest[]
208 * @param y vertical line number for this output. This does not need
209 * to be used to calculate the offset in the destination,
210 * but can be used to generate comfort noise using dithering
211 * or some output formats.
212 */
83890c66
DB
213typedef void (*yuv2packedX_fn)(struct SwsContext *c, const int16_t *lumFilter,
214 const int16_t **lumSrc, int lumFilterSize,
215 const int16_t *chrFilter,
216 const int16_t **chrUSrc,
217 const int16_t **chrVSrc, int chrFilterSize,
218 const int16_t **alpSrc, uint8_t *dest,
219 int dstW, int y);
b73fe700 220
4eb93bed
MN
221/**
222 * Write one line of horizontally scaled Y/U/V/A to YUV/RGB
223 * output by doing multi-point vertical scaling between input pixels.
224 *
225 * @param c SWS scaling context
226 * @param lumFilter vertical luma/alpha scaling coefficients, 12bit [0,4096]
227 * @param lumSrc scaled luma (Y) source data, 15bit for 8-10bit output,
228 * 19-bit for 16bit output (in int32_t)
229 * @param lumFilterSize number of vertical luma/alpha input lines to scale
230 * @param chrFilter vertical chroma scaling coefficients, 12bit [0,4096]
231 * @param chrUSrc scaled chroma (U) source data, 15bit for 8-10bit output,
232 * 19-bit for 16bit output (in int32_t)
233 * @param chrVSrc scaled chroma (V) source data, 15bit for 8-10bit output,
234 * 19-bit for 16bit output (in int32_t)
235 * @param chrFilterSize number of vertical chroma input lines to scale
236 * @param alpSrc scaled alpha (A) source data, 15bit for 8-10bit output,
237 * 19-bit for 16bit output (in int32_t)
238 * @param dest pointer to the output planes. For 16bit output, this is
239 * uint16_t
240 * @param dstW width of lumSrc and alpSrc in pixels, number of pixels
241 * to write into dest[]
242 * @param y vertical line number for this output. This does not need
243 * to be used to calculate the offset in the destination,
244 * but can be used to generate comfort noise using dithering
245 * or some output formats.
246 */
247typedef void (*yuv2anyX_fn)(struct SwsContext *c, const int16_t *lumFilter,
248 const int16_t **lumSrc, int lumFilterSize,
249 const int16_t *chrFilter,
250 const int16_t **chrUSrc,
251 const int16_t **chrVSrc, int chrFilterSize,
252 const int16_t **alpSrc, uint8_t **dest,
253 int dstW, int y);
254
8a322796 255/* This struct should be aligned on at least a 32-byte boundary. */
dd68318c 256typedef struct SwsContext {
221b804f
DB
257 /**
258 * info on struct for av_log
259 */
635a8cd2 260 const AVClass *av_class;
221b804f
DB
261
262 /**
56b69633
DB
263 * Note that src, dst, srcStride, dstStride will be copied in the
264 * sws_scale() wrapper so they can be freely modified here.
221b804f 265 */
1909f6b1 266 SwsFunc swscale;
9556124c
RP
267 int srcW; ///< Width of source luma/alpha planes.
268 int srcH; ///< Height of source luma/alpha planes.
269 int dstH; ///< Height of destination luma/alpha planes.
270 int chrSrcW; ///< Width of source chroma planes.
271 int chrSrcH; ///< Height of source chroma planes.
272 int chrDstW; ///< Width of destination chroma planes.
273 int chrDstH; ///< Height of destination chroma planes.
221b804f
DB
274 int lumXInc, chrXInc;
275 int lumYInc, chrYInc;
716d413c
AK
276 enum AVPixelFormat dstFormat; ///< Destination pixel format.
277 enum AVPixelFormat srcFormat; ///< Source pixel format.
30b61475
SS
278 int dstFormatBpp; ///< Number of bits per pixel of the destination pixel format.
279 int srcFormatBpp; ///< Number of bits per pixel of the source pixel format.
28c1115a 280 int dstBpc, srcBpc;
5d4d503f
RP
281 int chrSrcHSubSample; ///< Binary logarithm of horizontal subsampling factor between luma/alpha and chroma planes in source image.
282 int chrSrcVSubSample; ///< Binary logarithm of vertical subsampling factor between luma/alpha and chroma planes in source image.
283 int chrDstHSubSample; ///< Binary logarithm of horizontal subsampling factor between luma/alpha and chroma planes in destination image.
284 int chrDstVSubSample; ///< Binary logarithm of vertical subsampling factor between luma/alpha and chroma planes in destination image.
285 int vChrDrop; ///< Binary logarithm of extra vertical subsampling factor in source image chroma planes specified by user.
286 int sliceDir; ///< Direction that slices are fed to the scaler (1 = top-to-bottom, -1 = bottom-to-top).
9556124c 287 double param[2]; ///< Input parameters for scaling algorithms that need them.
221b804f 288
49004617
VS
289 uint32_t pal_yuv[256];
290 uint32_t pal_rgb[256];
291
5d4d503f
RP
292 /**
293 * @name Scaled horizontal lines ring buffer.
294 * The horizontal scaler keeps just enough scaled lines in a ring buffer
295 * so they may be passed to the vertical scaler. The pointers to the
296 * allocated buffers for each line are duplicated in sequence in the ring
297 * buffer to simplify indexing and avoid wrapping around between lines
298 * inside the vertical scaler code. The wrapping is done before the
299 * vertical scaler is called.
300 */
301 //@{
302 int16_t **lumPixBuf; ///< Ring buffer for scaled horizontal luma plane lines to be fed to the vertical scaler.
b4a224c5
RB
303 int16_t **chrUPixBuf; ///< Ring buffer for scaled horizontal chroma plane lines to be fed to the vertical scaler.
304 int16_t **chrVPixBuf; ///< Ring buffer for scaled horizontal chroma plane lines to be fed to the vertical scaler.
5d4d503f 305 int16_t **alpPixBuf; ///< Ring buffer for scaled horizontal alpha plane lines to be fed to the vertical scaler.
83890c66
DB
306 int vLumBufSize; ///< Number of vertical luma/alpha lines allocated in the ring buffer.
307 int vChrBufSize; ///< Number of vertical chroma lines allocated in the ring buffer.
308 int lastInLumBuf; ///< Last scaled horizontal luma/alpha line from source in the ring buffer.
309 int lastInChrBuf; ///< Last scaled horizontal chroma line from source in the ring buffer.
310 int lumBufIndex; ///< Index in ring buffer of the last scaled horizontal luma/alpha line from source.
311 int chrBufIndex; ///< Index in ring buffer of the last scaled horizontal chroma line from source.
5d4d503f 312 //@}
221b804f 313
69645c02 314 uint8_t *formatConvBuffer;
221b804f 315
5d4d503f
RP
316 /**
317 * @name Horizontal and vertical filters.
318 * To better understand the following fields, here is a pseudo-code of
319 * their usage in filtering a horizontal line:
320 * @code
321 * for (i = 0; i < width; i++) {
322 * dst[i] = 0;
323 * for (j = 0; j < filterSize; j++)
324 * dst[i] += src[ filterPos[i] + j ] * filter[ filterSize * i + j ];
325 * dst[i] >>= FRAC_BITS; // The actual implementation is fixed-point.
326 * }
327 * @endcode
328 */
329 //@{
330 int16_t *hLumFilter; ///< Array of horizontal filter coefficients for luma/alpha planes.
331 int16_t *hChrFilter; ///< Array of horizontal filter coefficients for chroma planes.
332 int16_t *vLumFilter; ///< Array of vertical filter coefficients for luma/alpha planes.
333 int16_t *vChrFilter; ///< Array of vertical filter coefficients for chroma planes.
2254b559
RB
334 int32_t *hLumFilterPos; ///< Array of horizontal filter starting positions for each dst[i] for luma/alpha planes.
335 int32_t *hChrFilterPos; ///< Array of horizontal filter starting positions for each dst[i] for chroma planes.
336 int32_t *vLumFilterPos; ///< Array of vertical filter starting positions for each dst[i] for luma/alpha planes.
337 int32_t *vChrFilterPos; ///< Array of vertical filter starting positions for each dst[i] for chroma planes.
83890c66
DB
338 int hLumFilterSize; ///< Horizontal filter size for luma/alpha pixels.
339 int hChrFilterSize; ///< Horizontal filter size for chroma pixels.
340 int vLumFilterSize; ///< Vertical filter size for luma/alpha pixels.
341 int vChrFilterSize; ///< Vertical filter size for chroma pixels.
5d4d503f 342 //@}
221b804f 343
a65bdceb
DB
344 int lumMmxextFilterCodeSize; ///< Runtime-generated MMXEXT horizontal fast bilinear scaler code size for luma/alpha planes.
345 int chrMmxextFilterCodeSize; ///< Runtime-generated MMXEXT horizontal fast bilinear scaler code size for chroma planes.
346 uint8_t *lumMmxextFilterCode; ///< Runtime-generated MMXEXT horizontal fast bilinear scaler code for luma/alpha planes.
347 uint8_t *chrMmxextFilterCode; ///< Runtime-generated MMXEXT horizontal fast bilinear scaler code for chroma planes.
221b804f 348
a65bdceb 349 int canMMXEXTBeUsed;
221b804f 350
9556124c
RP
351 int dstY; ///< Last destination vertical line output from last slice.
352 int flags; ///< Flags passed by the user to select scaler algorithm, optimizations, subsampling, etc...
83890c66
DB
353 void *yuvTable; // pointer to the yuv->rgb table start so it can be freed()
354 uint8_t *table_rV[256];
355 uint8_t *table_gU[256];
356 int table_gV[256];
357 uint8_t *table_bU[256];
221b804f
DB
358
359 //Colorspace stuff
360 int contrast, brightness, saturation; // for sws_getColorspaceDetails
361 int srcColorspaceTable[4];
362 int dstColorspaceTable[4];
9556124c
RP
363 int srcRange; ///< 0 = MPG YUV range, 1 = JPG YUV range (source image).
364 int dstRange; ///< 0 = MPG YUV range, 1 = JPG YUV range (destination image).
43c16478
MN
365 int yuv2rgb_y_offset;
366 int yuv2rgb_y_coeff;
367 int yuv2rgb_v2r_coeff;
368 int yuv2rgb_v2g_coeff;
369 int yuv2rgb_u2g_coeff;
370 int yuv2rgb_u2b_coeff;
221b804f
DB
371
372#define RED_DITHER "0*8"
373#define GREEN_DITHER "1*8"
374#define BLUE_DITHER "2*8"
375#define Y_COEFF "3*8"
376#define VR_COEFF "4*8"
377#define UB_COEFF "5*8"
378#define VG_COEFF "6*8"
379#define UG_COEFF "7*8"
380#define Y_OFFSET "8*8"
381#define U_OFFSET "9*8"
382#define V_OFFSET "10*8"
8bae9ddc
MN
383#define LUM_MMX_FILTER_OFFSET "11*8"
384#define CHR_MMX_FILTER_OFFSET "11*8+4*4*256"
8a322796 385#define DSTW_OFFSET "11*8+4*4*256*2" //do not change, it is hardcoded in the ASM
221b804f
DB
386#define ESP_OFFSET "11*8+4*4*256*2+8"
387#define VROUNDER_OFFSET "11*8+4*4*256*2+16"
388#define U_TEMP "11*8+4*4*256*2+24"
389#define V_TEMP "11*8+4*4*256*2+32"
6858492e 390#define Y_TEMP "11*8+4*4*256*2+40"
b18f8cbf
RB
391#define ALP_MMX_FILTER_OFFSET "11*8+4*4*256*2+48"
392#define UV_OFF_PX "11*8+4*4*256*3+48"
393#define UV_OFF_BYTE "11*8+4*4*256*3+56"
394#define DITHER16 "11*8+4*4*256*3+64"
395#define DITHER32 "11*8+4*4*256*3+80"
221b804f 396
92db6235
PP
397 DECLARE_ALIGNED(8, uint64_t, redDither);
398 DECLARE_ALIGNED(8, uint64_t, greenDither);
399 DECLARE_ALIGNED(8, uint64_t, blueDither);
400
401 DECLARE_ALIGNED(8, uint64_t, yCoeff);
402 DECLARE_ALIGNED(8, uint64_t, vrCoeff);
403 DECLARE_ALIGNED(8, uint64_t, ubCoeff);
404 DECLARE_ALIGNED(8, uint64_t, vgCoeff);
405 DECLARE_ALIGNED(8, uint64_t, ugCoeff);
406 DECLARE_ALIGNED(8, uint64_t, yOffset);
407 DECLARE_ALIGNED(8, uint64_t, uOffset);
408 DECLARE_ALIGNED(8, uint64_t, vOffset);
83890c66
DB
409 int32_t lumMmxFilter[4 * MAX_FILTER_SIZE];
410 int32_t chrMmxFilter[4 * MAX_FILTER_SIZE];
5d4d503f 411 int dstW; ///< Width of destination luma/alpha planes.
92db6235
PP
412 DECLARE_ALIGNED(8, uint64_t, esp);
413 DECLARE_ALIGNED(8, uint64_t, vRounder);
414 DECLARE_ALIGNED(8, uint64_t, u_temp);
415 DECLARE_ALIGNED(8, uint64_t, v_temp);
416 DECLARE_ALIGNED(8, uint64_t, y_temp);
b18f8cbf 417 int32_t alpMmxFilter[4 * MAX_FILTER_SIZE];
142e76f1
AM
418 // alignment of these values is not necessary, but merely here
419 // to maintain the same offset across x8632 and x86-64. Once we
420 // use proper offset macros in the asm, they can be removed.
5c391a16
RB
421 DECLARE_ALIGNED(8, ptrdiff_t, uv_off_px); ///< offset (in pixels) between u and v planes
422 DECLARE_ALIGNED(8, ptrdiff_t, uv_off_byte); ///< offset (in bytes) between u and v planes
142e76f1
AM
423 DECLARE_ALIGNED(8, uint16_t, dither16)[8];
424 DECLARE_ALIGNED(8, uint32_t, dither32)[8];
4e3e333a
RB
425
426 const uint8_t *chrDither8, *lumDither8;
a31de956 427
b63f641e 428#if HAVE_ALTIVEC
d3243d14
RP
429 vector signed short CY;
430 vector signed short CRV;
431 vector signed short CBU;
432 vector signed short CGU;
433 vector signed short CGV;
434 vector signed short OY;
435 vector unsigned short CSHIFT;
83890c66 436 vector signed short *vYCoeffsBank, *vCCoeffsBank;
a31de956
MN
437#endif
438
1909f6b1 439 /* function pointers for swscale() */
109f62e8
RB
440 yuv2planar1_fn yuv2plane1;
441 yuv2planarX_fn yuv2planeX;
442 yuv2interleavedX_fn yuv2nv12cX;
b73fe700
RB
443 yuv2packed1_fn yuv2packed1;
444 yuv2packed2_fn yuv2packed2;
445 yuv2packedX_fn yuv2packedX;
4eb93bed 446 yuv2anyX_fn yuv2anyX;
40fa5140 447
83890c66 448 /// Unscaled conversion of luma plane to YV12 for horizontal scaler.
bb53e1d1 449 void (*lumToYV12)(uint8_t *dst, const uint8_t *src,
83890c66
DB
450 int width, uint32_t *pal);
451 /// Unscaled conversion of alpha plane to YV12 for horizontal scaler.
bb53e1d1 452 void (*alpToYV12)(uint8_t *dst, const uint8_t *src,
83890c66
DB
453 int width, uint32_t *pal);
454 /// Unscaled conversion of chroma planes to YV12 for horizontal scaler.
bb53e1d1
RP
455 void (*chrToYV12)(uint8_t *dstU, uint8_t *dstV,
456 const uint8_t *src1, const uint8_t *src2,
83890c66 457 int width, uint32_t *pal);
185655c6
RB
458
459 /**
83890c66
DB
460 * Functions to read planar input, such as planar RGB, and convert
461 * internally to Y/UV.
462 */
185655c6
RB
463 /** @{ */
464 void (*readLumPlanar)(uint8_t *dst, const uint8_t *src[4], int width);
83890c66
DB
465 void (*readChrPlanar)(uint8_t *dstU, uint8_t *dstV, const uint8_t *src[4],
466 int width);
185655c6
RB
467 /** @} */
468
4578435f
RB
469 /**
470 * Scale one horizontal line of input data using a bilinear filter
471 * to produce one line of output data. Compared to SwsContext->hScale(),
472 * please take note of the following caveats when using these:
473 * - Scaling is done using only 7bit instead of 14bit coefficients.
474 * - You can use no more than 5 input pixels to produce 4 output
475 * pixels. Therefore, this filter should not be used for downscaling
476 * by more than ~20% in width (because that equals more than 5/4th
477 * downscaling and thus more than 5 pixels input per 4 pixels output).
478 * - In general, bilinear filters create artifacts during downscaling
479 * (even when <20%), because one output pixel will span more than one
480 * input pixel, and thus some pixels will need edges of both neighbor
481 * pixels to interpolate the output pixel. Since you can use at most
482 * two input pixels per output pixel in bilinear scaling, this is
483 * impossible and thus downscaling by any size will create artifacts.
484 * To enable this type of scaling, set SWS_FLAG_FAST_BILINEAR
485 * in SwsContext->flags.
486 */
487 /** @{ */
40fa5140 488 void (*hyscale_fast)(struct SwsContext *c,
b8e89339 489 int16_t *dst, int dstWidth,
40fa5140
RP
490 const uint8_t *src, int srcW, int xInc);
491 void (*hcscale_fast)(struct SwsContext *c,
b8e89339 492 int16_t *dst1, int16_t *dst2, int dstWidth,
457eed72
DB
493 const uint8_t *src1, const uint8_t *src2,
494 int srcW, int xInc);
4578435f 495 /** @} */
40fa5140 496
4578435f
RB
497 /**
498 * Scale one horizontal line of input data using a filter over the input
499 * lines, to produce one (differently sized) line of output data.
500 *
501 * @param dst pointer to destination buffer for horizontally scaled
28c1115a
RB
502 * data. If the number of bits per component of one
503 * destination pixel (SwsContext->dstBpc) is <= 10, data
504 * will be 15bpc in 16bits (int16_t) width. Else (i.e.
505 * SwsContext->dstBpc == 16), data will be 19bpc in
506 * 32bits (int32_t) width.
4578435f 507 * @param dstW width of destination image
28c1115a
RB
508 * @param src pointer to source data to be scaled. If the number of
509 * bits per component of a source pixel (SwsContext->srcBpc)
510 * is 8, this is 8bpc in 8bits (uint8_t) width. Else
511 * (i.e. SwsContext->dstBpc > 8), this is native depth
512 * in 16bits (uint16_t) width. In other words, for 9-bit
513 * YUV input, this is 9bpc, for 10-bit YUV input, this is
514 * 10bpc, and for 16-bit RGB or YUV, this is 16bpc.
4578435f
RB
515 * @param filter filter coefficients to be used per output pixel for
516 * scaling. This contains 14bpp filtering coefficients.
517 * Guaranteed to contain dstW * filterSize entries.
518 * @param filterPos position of the first input pixel to be used for
519 * each output pixel during scaling. Guaranteed to
520 * contain dstW entries.
521 * @param filterSize the number of input coefficients to be used (and
522 * thus the number of input pixels to be used) for
523 * creating a single output pixel. Is aligned to 4
524 * (and input coefficients thus padded with zeroes)
525 * to simplify creating SIMD code.
526 */
3f04ab4f 527 /** @{ */
83890c66
DB
528 void (*hyScale)(struct SwsContext *c, int16_t *dst, int dstW,
529 const uint8_t *src, const int16_t *filter,
2254b559 530 const int32_t *filterPos, int filterSize);
83890c66
DB
531 void (*hcScale)(struct SwsContext *c, int16_t *dst, int dstW,
532 const uint8_t *src, const int16_t *filter,
2254b559 533 const int32_t *filterPos, int filterSize);
3f04ab4f 534 /** @} */
40fa5140 535
83890c66
DB
536 /// Color range conversion function for luma plane if needed.
537 void (*lumConvertRange)(int16_t *dst, int width);
538 /// Color range conversion function for chroma planes if needed.
539 void (*chrConvertRange)(int16_t *dst1, int16_t *dst2, int width);
bae76dc3 540
e7a47515 541 int needs_hcscale; ///< Set if there are chroma planes to be converted.
5427e242
MN
542} SwsContext;
543//FIXME check init (where 0)
5427e242 544
780daf2b 545SwsFunc ff_yuv2rgb_get_func_ptr(SwsContext *c);
457eed72
DB
546int ff_yuv2rgb_c_init_tables(SwsContext *c, const int inv_table[4],
547 int fullRange, int brightness,
548 int contrast, int saturation);
c2503d9c
DB
549void ff_yuv2rgb_init_tables_ppc(SwsContext *c, const int inv_table[4],
550 int brightness, int contrast, int saturation);
5427e242 551
983260b0
RB
552void updateMMXDitherTables(SwsContext *c, int dstY, int lumBufIndex, int chrBufIndex,
553 int lastInLumBuf, int lastInChrBuf);
554
a5195839 555SwsFunc ff_yuv2rgb_init_x86(SwsContext *c);
c2503d9c 556SwsFunc ff_yuv2rgb_init_ppc(SwsContext *c);
c95a3a9f 557
716d413c 558const char *sws_format_name(enum AVPixelFormat format);
94c4def2 559
0a7068fa
AK
560static av_always_inline int is16BPS(enum AVPixelFormat pix_fmt)
561{
562 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(pix_fmt);
563 av_assert0(desc);
564 return desc->comp[0].depth_minus1 == 15;
565}
566
567static av_always_inline int is9_OR_10BPS(enum AVPixelFormat pix_fmt)
568{
569 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(pix_fmt);
570 av_assert0(desc);
571 return desc->comp[0].depth_minus1 == 8 || desc->comp[0].depth_minus1 == 9;
572}
573
574static av_always_inline int isBE(enum AVPixelFormat pix_fmt)
575{
576 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(pix_fmt);
577 av_assert0(desc);
e6c4ac7b 578 return desc->flags & AV_PIX_FMT_FLAG_BE;
0a7068fa
AK
579}
580
581static av_always_inline int isYUV(enum AVPixelFormat pix_fmt)
582{
583 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(pix_fmt);
584 av_assert0(desc);
e6c4ac7b 585 return !(desc->flags & AV_PIX_FMT_FLAG_RGB) && desc->nb_components >= 2;
0a7068fa
AK
586}
587
588static av_always_inline int isPlanarYUV(enum AVPixelFormat pix_fmt)
589{
590 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(pix_fmt);
591 av_assert0(desc);
e6c4ac7b 592 return ((desc->flags & AV_PIX_FMT_FLAG_PLANAR) && isYUV(pix_fmt));
0a7068fa
AK
593}
594
595static av_always_inline int isRGB(enum AVPixelFormat pix_fmt)
596{
597 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(pix_fmt);
598 av_assert0(desc);
e6c4ac7b 599 return (desc->flags & AV_PIX_FMT_FLAG_RGB);
0a7068fa 600}
6b0768e2 601
f7f18352 602#if 0 // FIXME
6b0768e2 603#define isGray(x) \
e6c4ac7b 604 (!(av_pix_fmt_descriptors[x].flags & AV_PIX_FMT_FLAG_PAL) && \
6b0768e2 605 av_pix_fmt_descriptors[x].nb_components <= 2)
f7f18352 606#else
83890c66 607#define isGray(x) \
716d413c 608 ((x) == AV_PIX_FMT_GRAY8 || \
e96c3b81 609 (x) == AV_PIX_FMT_YA8 || \
716d413c 610 (x) == AV_PIX_FMT_GRAY16BE || \
f84a1b59
VG
611 (x) == AV_PIX_FMT_GRAY16LE || \
612 (x) == AV_PIX_FMT_YA16BE || \
613 (x) == AV_PIX_FMT_YA16LE)
f7f18352 614#endif
6b0768e2 615
83890c66 616#define isRGBinInt(x) \
716d413c
AK
617 ((x) == AV_PIX_FMT_RGB48BE || \
618 (x) == AV_PIX_FMT_RGB48LE || \
619 (x) == AV_PIX_FMT_RGB32 || \
620 (x) == AV_PIX_FMT_RGB32_1 || \
621 (x) == AV_PIX_FMT_RGB24 || \
622 (x) == AV_PIX_FMT_RGB565BE || \
623 (x) == AV_PIX_FMT_RGB565LE || \
624 (x) == AV_PIX_FMT_RGB555BE || \
625 (x) == AV_PIX_FMT_RGB555LE || \
626 (x) == AV_PIX_FMT_RGB444BE || \
627 (x) == AV_PIX_FMT_RGB444LE || \
628 (x) == AV_PIX_FMT_RGB8 || \
629 (x) == AV_PIX_FMT_RGB4 || \
630 (x) == AV_PIX_FMT_RGB4_BYTE || \
1481d24c
JF
631 (x) == AV_PIX_FMT_RGBA64BE || \
632 (x) == AV_PIX_FMT_RGBA64LE || \
716d413c
AK
633 (x) == AV_PIX_FMT_MONOBLACK || \
634 (x) == AV_PIX_FMT_MONOWHITE)
83890c66
DB
635
636#define isBGRinInt(x) \
716d413c
AK
637 ((x) == AV_PIX_FMT_BGR48BE || \
638 (x) == AV_PIX_FMT_BGR48LE || \
639 (x) == AV_PIX_FMT_BGR32 || \
640 (x) == AV_PIX_FMT_BGR32_1 || \
641 (x) == AV_PIX_FMT_BGR24 || \
642 (x) == AV_PIX_FMT_BGR565BE || \
643 (x) == AV_PIX_FMT_BGR565LE || \
644 (x) == AV_PIX_FMT_BGR555BE || \
645 (x) == AV_PIX_FMT_BGR555LE || \
646 (x) == AV_PIX_FMT_BGR444BE || \
647 (x) == AV_PIX_FMT_BGR444LE || \
648 (x) == AV_PIX_FMT_BGR8 || \
649 (x) == AV_PIX_FMT_BGR4 || \
650 (x) == AV_PIX_FMT_BGR4_BYTE || \
1481d24c
JF
651 (x) == AV_PIX_FMT_BGRA64BE || \
652 (x) == AV_PIX_FMT_BGRA64LE || \
716d413c
AK
653 (x) == AV_PIX_FMT_MONOBLACK || \
654 (x) == AV_PIX_FMT_MONOWHITE)
83890c66
DB
655
656#define isAnyRGB(x) \
657 (isRGBinInt(x) || \
658 isBGRinInt(x))
659
0a7068fa
AK
660static av_always_inline int isALPHA(enum AVPixelFormat pix_fmt)
661{
662 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(pix_fmt);
663 av_assert0(desc);
664 return desc->nb_components == 2 || desc->nb_components == 4;
665}
666
667static av_always_inline int isPacked(enum AVPixelFormat pix_fmt)
668{
669 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(pix_fmt);
670 av_assert0(desc);
e6c4ac7b 671 return ((desc->nb_components >= 2 && !(desc->flags & AV_PIX_FMT_FLAG_PLANAR)) ||
0a7068fa
AK
672 pix_fmt == AV_PIX_FMT_PAL8);
673}
674
675static av_always_inline int isPlanar(enum AVPixelFormat pix_fmt)
676{
677 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(pix_fmt);
678 av_assert0(desc);
e6c4ac7b 679 return (desc->nb_components >= 2 && (desc->flags & AV_PIX_FMT_FLAG_PLANAR));
0a7068fa
AK
680}
681
682static av_always_inline int isPackedRGB(enum AVPixelFormat pix_fmt)
683{
684 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(pix_fmt);
685 av_assert0(desc);
e6c4ac7b 686 return ((desc->flags & (AV_PIX_FMT_FLAG_PLANAR | AV_PIX_FMT_FLAG_RGB)) == AV_PIX_FMT_FLAG_RGB);
0a7068fa
AK
687}
688
689static av_always_inline int isPlanarRGB(enum AVPixelFormat pix_fmt)
690{
691 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(pix_fmt);
692 av_assert0(desc);
e6c4ac7b
AK
693 return ((desc->flags & (AV_PIX_FMT_FLAG_PLANAR | AV_PIX_FMT_FLAG_RGB)) ==
694 (AV_PIX_FMT_FLAG_PLANAR | AV_PIX_FMT_FLAG_RGB));
0a7068fa
AK
695}
696
697static av_always_inline int usePal(enum AVPixelFormat pix_fmt)
698{
699 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(pix_fmt);
700 av_assert0(desc);
e6c4ac7b 701 return ((desc->flags & AV_PIX_FMT_FLAG_PAL) || (desc->flags & AV_PIX_FMT_FLAG_PSEUDOPAL) ||
e96c3b81 702 pix_fmt == AV_PIX_FMT_YA8);
0a7068fa 703}
e9e12f0e 704
43175f50
DB
705extern const uint64_t ff_dither4[2];
706extern const uint64_t ff_dither8[2];
0cb25594 707
aa2ba8c9
DB
708extern const uint8_t ff_dither_4x4_16[4][8];
709extern const uint8_t ff_dither_8x8_32[8][8];
710extern const uint8_t ff_dither_8x8_73[8][8];
711extern const uint8_t ff_dither_8x8_128[8][8];
712extern const uint8_t ff_dither_8x8_220[8][8];
713
714extern const int32_t ff_yuv2rgb_coeffs[8][4];
715
e40b183d
LB
716extern const AVClass sws_context_class;
717
a4388ebd 718/**
1909f6b1 719 * Set c->swscale to an unscaled converter if one exists for the specific
a4388ebd
RP
720 * source and destination formats, bit depths, flags, etc.
721 */
722void ff_get_unscaled_swscale(SwsContext *c);
3aa682f2 723void ff_get_unscaled_swscale_ppc(SwsContext *c);
0e5d31b1 724
a4388ebd 725/**
58c42af7 726 * Return function pointer to fastest main scaler path function depending
a4388ebd
RP
727 * on architecture and available optimizations.
728 */
729SwsFunc ff_getSwsFunc(SwsContext *c);
730
2dd7a1c0 731void ff_sws_init_input_funcs(SwsContext *c);
21449410
RB
732void ff_sws_init_output_funcs(SwsContext *c,
733 yuv2planar1_fn *yuv2plane1,
734 yuv2planarX_fn *yuv2planeX,
735 yuv2interleavedX_fn *yuv2nv12cX,
736 yuv2packed1_fn *yuv2packed1,
737 yuv2packed2_fn *yuv2packed2,
4eb93bed
MN
738 yuv2packedX_fn *yuv2packedX,
739 yuv2anyX_fn *yuv2anyX);
c2503d9c 740void ff_sws_init_swscale_ppc(SwsContext *c);
a5195839 741void ff_sws_init_swscale_x86(SwsContext *c);
67d80a54 742
0eaf5235 743#endif /* SWSCALE_SWSCALE_INTERNAL_H */