Add 420,422 and 444 planar 16bit per component pix formats.
[libav.git] / libswscale / swscale_internal.h
CommitLineData
5427e242 1/*
d026b45e
DB
2 * Copyright (C) 2001-2003 Michael Niedermayer <michaelni@gmx.at>
3 *
4 * This file is part of FFmpeg.
5 *
807e0c66
LA
6 * FFmpeg is free software; you can redistribute it and/or
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
DB
10 *
11 * FFmpeg is distributed in the hope that it will be useful,
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
LA
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with FFmpeg; 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
83da2c6f 30#include "libavutil/avutil.h"
94c4def2 31
59ded10c 32#define STR(s) AV_TOSTRING(s) //AV_STRINGIFY is too long
1625216e 33
5427e242
MN
34#define MAX_FILTER_SIZE 256
35
8b2fce0d 36#define VOFW 2048
710af50e 37#define VOF (VOFW*2)
8b2fce0d 38
9990e426
MN
39#ifdef WORDS_BIGENDIAN
40#define ALT32_CORR (-1)
41#else
42#define ALT32_CORR 1
1625216e
MN
43#endif
44
b63f641e 45#if ARCH_X86_64
1625216e
MN
46# define APCK_PTR2 8
47# define APCK_COEF 16
48# define APCK_SIZE 24
49#else
50# define APCK_PTR2 4
51# define APCK_COEF 8
52# define APCK_SIZE 16
9990e426
MN
53#endif
54
1615fb91
DB
55struct SwsContext;
56
457eed72
DB
57typedef int (*SwsFunc)(struct SwsContext *context, uint8_t* src[],
58 int srcStride[], int srcSliceY, int srcSliceH,
59 uint8_t* dst[], int dstStride[]);
5427e242 60
8a322796 61/* This struct should be aligned on at least a 32-byte boundary. */
5427e242 62typedef struct SwsContext{
221b804f
DB
63 /**
64 * info on struct for av_log
65 */
635a8cd2 66 const AVClass *av_class;
221b804f
DB
67
68 /**
56b69633
DB
69 * Note that src, dst, srcStride, dstStride will be copied in the
70 * sws_scale() wrapper so they can be freely modified here.
221b804f
DB
71 */
72 SwsFunc swScale;
73 int srcW, srcH, dstH;
74 int chrSrcW, chrSrcH, chrDstW, chrDstH;
75 int lumXInc, chrXInc;
76 int lumYInc, chrYInc;
58e4b706 77 enum PixelFormat dstFormat, srcFormat; ///< format 4:2:0 type is always YV12
221b804f
DB
78 int origDstFormat, origSrcFormat; ///< format
79 int chrSrcHSubSample, chrSrcVSubSample;
80 int chrIntHSubSample, chrIntVSubSample;
81 int chrDstHSubSample, chrDstVSubSample;
82 int vChrDrop;
83 int sliceDir;
84 double param[2];
85
49004617
VS
86 uint32_t pal_yuv[256];
87 uint32_t pal_rgb[256];
88
221b804f
DB
89 int16_t **lumPixBuf;
90 int16_t **chrPixBuf;
6858492e 91 int16_t **alpPixBuf;
221b804f
DB
92 int16_t *hLumFilter;
93 int16_t *hLumFilterPos;
94 int16_t *hChrFilter;
95 int16_t *hChrFilterPos;
96 int16_t *vLumFilter;
97 int16_t *vLumFilterPos;
98 int16_t *vChrFilter;
99 int16_t *vChrFilterPos;
100
8a322796 101 uint8_t formatConvBuffer[VOF]; //FIXME dynamic allocation, but we have to change a lot of code for this to be useful
221b804f
DB
102
103 int hLumFilterSize;
104 int hChrFilterSize;
105 int vLumFilterSize;
106 int vChrFilterSize;
107 int vLumBufSize;
108 int vChrBufSize;
109
110 uint8_t *funnyYCode;
111 uint8_t *funnyUVCode;
112 int32_t *lumMmx2FilterPos;
113 int32_t *chrMmx2FilterPos;
114 int16_t *lumMmx2Filter;
115 int16_t *chrMmx2Filter;
116
117 int canMMX2BeUsed;
118
119 int lastInLumBuf;
120 int lastInChrBuf;
121 int lumBufIndex;
122 int chrBufIndex;
123 int dstY;
124 int flags;
125 void * yuvTable; // pointer to the yuv->rgb table start so it can be freed()
126 uint8_t * table_rV[256];
127 uint8_t * table_gU[256];
128 int table_gV[256];
129 uint8_t * table_bU[256];
130
131 //Colorspace stuff
132 int contrast, brightness, saturation; // for sws_getColorspaceDetails
133 int srcColorspaceTable[4];
134 int dstColorspaceTable[4];
135 int srcRange, dstRange;
43c16478
MN
136 int yuv2rgb_y_offset;
137 int yuv2rgb_y_coeff;
138 int yuv2rgb_v2r_coeff;
139 int yuv2rgb_v2g_coeff;
140 int yuv2rgb_u2g_coeff;
141 int yuv2rgb_u2b_coeff;
221b804f
DB
142
143#define RED_DITHER "0*8"
144#define GREEN_DITHER "1*8"
145#define BLUE_DITHER "2*8"
146#define Y_COEFF "3*8"
147#define VR_COEFF "4*8"
148#define UB_COEFF "5*8"
149#define VG_COEFF "6*8"
150#define UG_COEFF "7*8"
151#define Y_OFFSET "8*8"
152#define U_OFFSET "9*8"
153#define V_OFFSET "10*8"
8bae9ddc
MN
154#define LUM_MMX_FILTER_OFFSET "11*8"
155#define CHR_MMX_FILTER_OFFSET "11*8+4*4*256"
8a322796 156#define DSTW_OFFSET "11*8+4*4*256*2" //do not change, it is hardcoded in the ASM
221b804f
DB
157#define ESP_OFFSET "11*8+4*4*256*2+8"
158#define VROUNDER_OFFSET "11*8+4*4*256*2+16"
159#define U_TEMP "11*8+4*4*256*2+24"
160#define V_TEMP "11*8+4*4*256*2+32"
6858492e
CS
161#define Y_TEMP "11*8+4*4*256*2+40"
162#define ALP_MMX_FILTER_OFFSET "11*8+4*4*256*2+48"
221b804f
DB
163
164 uint64_t redDither __attribute__((aligned(8)));
165 uint64_t greenDither __attribute__((aligned(8)));
166 uint64_t blueDither __attribute__((aligned(8)));
167
168 uint64_t yCoeff __attribute__((aligned(8)));
169 uint64_t vrCoeff __attribute__((aligned(8)));
170 uint64_t ubCoeff __attribute__((aligned(8)));
171 uint64_t vgCoeff __attribute__((aligned(8)));
172 uint64_t ugCoeff __attribute__((aligned(8)));
173 uint64_t yOffset __attribute__((aligned(8)));
174 uint64_t uOffset __attribute__((aligned(8)));
175 uint64_t vOffset __attribute__((aligned(8)));
176 int32_t lumMmxFilter[4*MAX_FILTER_SIZE];
177 int32_t chrMmxFilter[4*MAX_FILTER_SIZE];
178 int dstW;
179 uint64_t esp __attribute__((aligned(8)));
180 uint64_t vRounder __attribute__((aligned(8)));
181 uint64_t u_temp __attribute__((aligned(8)));
182 uint64_t v_temp __attribute__((aligned(8)));
6858492e
CS
183 uint64_t y_temp __attribute__((aligned(8)));
184 int32_t alpMmxFilter[4*MAX_FILTER_SIZE];
a31de956 185
b63f641e 186#if HAVE_ALTIVEC
a31de956
MN
187 vector signed short CY;
188 vector signed short CRV;
189 vector signed short CBU;
190 vector signed short CGU;
191 vector signed short CGV;
192 vector signed short OY;
193 vector unsigned short CSHIFT;
221b804f 194 vector signed short *vYCoeffsBank, *vCCoeffsBank;
a31de956
MN
195#endif
196
b63f641e 197#if ARCH_BFIN
d3f3eea9
MH
198 uint32_t oy __attribute__((aligned(4)));
199 uint32_t oc __attribute__((aligned(4)));
200 uint32_t zero __attribute__((aligned(4)));
201 uint32_t cy __attribute__((aligned(4)));
202 uint32_t crv __attribute__((aligned(4)));
203 uint32_t rmask __attribute__((aligned(4)));
204 uint32_t cbu __attribute__((aligned(4)));
205 uint32_t bmask __attribute__((aligned(4)));
206 uint32_t cgu __attribute__((aligned(4)));
207 uint32_t cgv __attribute__((aligned(4)));
208 uint32_t gmask __attribute__((aligned(4)));
209#endif
210
b63f641e 211#if HAVE_VIS
90c1d7b4
DB
212 uint64_t sparc_coeffs[10] __attribute__((aligned(8)));
213#endif
d3f3eea9 214
40fa5140
RP
215 /* function pointers for swScale() */
216 void (*yuv2nv12X )(struct SwsContext *c,
217 const int16_t *lumFilter, const int16_t **lumSrc, int lumFilterSize,
218 const int16_t *chrFilter, const int16_t **chrSrc, int chrFilterSize,
219 uint8_t *dest, uint8_t *uDest,
220 int dstW, int chrDstW, int dstFormat);
221 void (*yuv2yuv1 )(struct SwsContext *c,
222 const int16_t *lumSrc, const int16_t *chrSrc, const int16_t *alpSrc,
223 uint8_t *dest,
224 uint8_t *uDest, uint8_t *vDest, uint8_t *aDest,
225 long dstW, long chrDstW);
226 void (*yuv2yuvX )(struct SwsContext *c,
227 const int16_t *lumFilter, const int16_t **lumSrc, int lumFilterSize,
228 const int16_t *chrFilter, const int16_t **chrSrc, int chrFilterSize,
229 const int16_t **alpSrc,
230 uint8_t *dest,
231 uint8_t *uDest, uint8_t *vDest, uint8_t *aDest,
232 long dstW, long chrDstW);
233 void (*yuv2packed1)(struct SwsContext *c,
234 const uint16_t *buf0,
235 const uint16_t *uvbuf0, const uint16_t *uvbuf1,
236 const uint16_t *abuf0,
237 uint8_t *dest,
238 int dstW, int uvalpha, int dstFormat, int flags, int y);
239 void (*yuv2packed2)(struct SwsContext *c,
240 const uint16_t *buf0, const uint16_t *buf1,
241 const uint16_t *uvbuf0, const uint16_t *uvbuf1,
242 const uint16_t *abuf0, const uint16_t *abuf1,
243 uint8_t *dest,
244 int dstW, int yalpha, int uvalpha, int y);
245 void (*yuv2packedX)(struct SwsContext *c,
246 const int16_t *lumFilter, const int16_t **lumSrc, int lumFilterSize,
247 const int16_t *chrFilter, const int16_t **chrSrc, int chrFilterSize,
248 const int16_t **alpSrc, uint8_t *dest,
249 long dstW, long dstY);
250
251 void (*hyscale_internal)(uint8_t *dst, const uint8_t *src,
252 long width, uint32_t *pal);
39e5f87b
CS
253 void (*hascale_internal)(uint8_t *dst, const uint8_t *src,
254 long width, uint32_t *pal);
40fa5140
RP
255 void (*hcscale_internal)(uint8_t *dstU, uint8_t *dstV,
256 const uint8_t *src1, const uint8_t *src2,
257 long width, uint32_t *pal);
258 void (*hyscale_fast)(struct SwsContext *c,
259 int16_t *dst, int dstWidth,
260 const uint8_t *src, int srcW, int xInc);
261 void (*hcscale_fast)(struct SwsContext *c,
262 int16_t *dst, int dstWidth,
457eed72
DB
263 const uint8_t *src1, const uint8_t *src2,
264 int srcW, int xInc);
40fa5140
RP
265
266 void (*hScale)(int16_t *dst, int dstW, const uint8_t *src, int srcW,
457eed72
DB
267 int xInc, const int16_t *filter, const int16_t *filterPos,
268 long filterSize);
40fa5140 269
5427e242
MN
270} SwsContext;
271//FIXME check init (where 0)
5427e242 272
780daf2b 273SwsFunc ff_yuv2rgb_get_func_ptr(SwsContext *c);
457eed72
DB
274int ff_yuv2rgb_c_init_tables(SwsContext *c, const int inv_table[4],
275 int fullRange, int brightness,
276 int contrast, int saturation);
5427e242 277
457eed72
DB
278void ff_yuv2rgb_init_tables_altivec(SwsContext *c, const int inv_table[4],
279 int brightness, int contrast, int saturation);
befa8e66
RP
280SwsFunc ff_yuv2rgb_init_mmx(SwsContext *c);
281SwsFunc ff_yuv2rgb_init_vis(SwsContext *c);
282SwsFunc ff_yuv2rgb_init_mlib(SwsContext *c);
780daf2b 283SwsFunc ff_yuv2rgb_init_altivec(SwsContext *c);
befa8e66 284SwsFunc ff_yuv2rgb_get_func_ptr_bfin(SwsContext *c);
d5e9bc06 285void ff_bfin_get_unscaled_swscale(SwsContext *c);
780daf2b 286void ff_yuv2packedX_altivec(SwsContext *c,
457eed72
DB
287 int16_t *lumFilter, int16_t **lumSrc, int lumFilterSize,
288 int16_t *chrFilter, int16_t **chrSrc, int chrFilterSize,
289 uint8_t *dest, int dstW, int dstY);
c95a3a9f 290
8055ede6 291const char *sws_format_name(int format);
94c4def2 292
e9e12f0e 293//FIXME replace this with something faster
714f32bf
MN
294#define is16BPS(x) ( \
295 (x)==PIX_FMT_GRAY16BE \
296 || (x)==PIX_FMT_GRAY16LE \
297 )
c3dc6c5a 298#define isBE(x) ((x)&1)
9d9de37d
IP
299#define isPlanarYUV(x) ( \
300 (x)==PIX_FMT_YUV410P \
301 || (x)==PIX_FMT_YUV420P \
9621f2f5 302 || (x)==PIX_FMT_YUVA420P \
9d9de37d
IP
303 || (x)==PIX_FMT_YUV411P \
304 || (x)==PIX_FMT_YUV422P \
305 || (x)==PIX_FMT_YUV444P \
9ba7fe6d 306 || (x)==PIX_FMT_YUV440P \
9d9de37d
IP
307 || (x)==PIX_FMT_NV12 \
308 || (x)==PIX_FMT_NV21 \
309 )
310#define isYUV(x) ( \
311 (x)==PIX_FMT_UYVY422 \
312 || (x)==PIX_FMT_YUYV422 \
313 || isPlanarYUV(x) \
314 )
315#define isGray(x) ( \
316 (x)==PIX_FMT_GRAY8 \
317 || (x)==PIX_FMT_GRAY16BE \
318 || (x)==PIX_FMT_GRAY16LE \
319 )
320#define isGray16(x) ( \
321 (x)==PIX_FMT_GRAY16BE \
322 || (x)==PIX_FMT_GRAY16LE \
323 )
324#define isRGB(x) ( \
d52337af 325 (x)==PIX_FMT_RGB32 \
9990e426 326 || (x)==PIX_FMT_RGB32_1 \
9d9de37d
IP
327 || (x)==PIX_FMT_RGB24 \
328 || (x)==PIX_FMT_RGB565 \
329 || (x)==PIX_FMT_RGB555 \
330 || (x)==PIX_FMT_RGB8 \
331 || (x)==PIX_FMT_RGB4 \
332 || (x)==PIX_FMT_RGB4_BYTE \
333 || (x)==PIX_FMT_MONOBLACK \
ec1bca2a 334 || (x)==PIX_FMT_MONOWHITE \
9d9de37d
IP
335 )
336#define isBGR(x) ( \
d52337af 337 (x)==PIX_FMT_BGR32 \
9990e426 338 || (x)==PIX_FMT_BGR32_1 \
9d9de37d
IP
339 || (x)==PIX_FMT_BGR24 \
340 || (x)==PIX_FMT_BGR565 \
341 || (x)==PIX_FMT_BGR555 \
342 || (x)==PIX_FMT_BGR8 \
343 || (x)==PIX_FMT_BGR4 \
344 || (x)==PIX_FMT_BGR4_BYTE \
345 || (x)==PIX_FMT_MONOBLACK \
ec1bca2a 346 || (x)==PIX_FMT_MONOWHITE \
9d9de37d 347 )
c2751600
CS
348#define isALPHA(x) ( \
349 (x)==PIX_FMT_BGR32 \
350 || (x)==PIX_FMT_BGR32_1 \
351 || (x)==PIX_FMT_RGB32 \
352 || (x)==PIX_FMT_RGB32_1 \
353 || (x)==PIX_FMT_YUVA420P \
354 )
e9e12f0e
LA
355
356static inline int fmt_depth(int fmt)
357{
358 switch(fmt) {
359 case PIX_FMT_BGRA:
360 case PIX_FMT_ABGR:
361 case PIX_FMT_RGBA:
362 case PIX_FMT_ARGB:
363 return 32;
364 case PIX_FMT_BGR24:
365 case PIX_FMT_RGB24:
366 return 24;
367 case PIX_FMT_BGR565:
368 case PIX_FMT_RGB565:
4884b9e5
KS
369 case PIX_FMT_GRAY16BE:
370 case PIX_FMT_GRAY16LE:
e9e12f0e
LA
371 return 16;
372 case PIX_FMT_BGR555:
373 case PIX_FMT_RGB555:
374 return 15;
375 case PIX_FMT_BGR8:
376 case PIX_FMT_RGB8:
377 return 8;
378 case PIX_FMT_BGR4:
379 case PIX_FMT_RGB4:
380 case PIX_FMT_BGR4_BYTE:
381 case PIX_FMT_RGB4_BYTE:
382 return 4;
383 case PIX_FMT_MONOBLACK:
ec1bca2a 384 case PIX_FMT_MONOWHITE:
e9e12f0e
LA
385 return 1;
386 default:
387 return 0;
388 }
389}
390
43175f50
DB
391extern const uint64_t ff_dither4[2];
392extern const uint64_t ff_dither8[2];
0cb25594 393
e40b183d
LB
394extern const AVClass sws_context_class;
395
0eaf5235 396#endif /* SWSCALE_SWSCALE_INTERNAL_H */