Move yuv2rgb code to subdirs.
[libav.git] / libswscale / swscale_internal.h
1 /*
2 * Copyright (C) 2001-2003 Michael Niedermayer <michaelni@gmx.at>
3 *
4 * This file is part of FFmpeg.
5 *
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.
10 *
11 * FFmpeg is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
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
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21 #ifndef SWSCALE_SWSCALE_INTERNAL_H
22 #define SWSCALE_SWSCALE_INTERNAL_H
23
24 #include "config.h"
25
26 #if HAVE_ALTIVEC_H
27 #include <altivec.h>
28 #endif
29
30 #include "libavutil/avutil.h"
31
32 #define STR(s) AV_TOSTRING(s) //AV_STRINGIFY is too long
33
34 #define MAX_FILTER_SIZE 256
35
36 #define VOFW 2048
37 #define VOF (VOFW*2)
38
39 #ifdef WORDS_BIGENDIAN
40 #define ALT32_CORR (-1)
41 #else
42 #define ALT32_CORR 1
43 #endif
44
45 #if ARCH_X86_64
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
53 #endif
54
55 struct SwsContext;
56
57 typedef int (*SwsFunc)(struct SwsContext *context, uint8_t* src[], int srcStride[], int srcSliceY,
58 int srcSliceH, uint8_t* dst[], int dstStride[]);
59
60 /* This struct should be aligned on at least a 32-byte boundary. */
61 typedef struct SwsContext{
62 /**
63 * info on struct for av_log
64 */
65 const AVClass *av_class;
66
67 /**
68 * Note that src, dst, srcStride, dstStride will be copied in the
69 * sws_scale() wrapper so they can be freely modified here.
70 */
71 SwsFunc swScale;
72 int srcW, srcH, dstH;
73 int chrSrcW, chrSrcH, chrDstW, chrDstH;
74 int lumXInc, chrXInc;
75 int lumYInc, chrYInc;
76 enum PixelFormat dstFormat, srcFormat; ///< format 4:2:0 type is always YV12
77 int origDstFormat, origSrcFormat; ///< format
78 int chrSrcHSubSample, chrSrcVSubSample;
79 int chrIntHSubSample, chrIntVSubSample;
80 int chrDstHSubSample, chrDstVSubSample;
81 int vChrDrop;
82 int sliceDir;
83 double param[2];
84
85 uint32_t pal_yuv[256];
86 uint32_t pal_rgb[256];
87
88 int16_t **lumPixBuf;
89 int16_t **chrPixBuf;
90 int16_t **alpPixBuf;
91 int16_t *hLumFilter;
92 int16_t *hLumFilterPos;
93 int16_t *hChrFilter;
94 int16_t *hChrFilterPos;
95 int16_t *vLumFilter;
96 int16_t *vLumFilterPos;
97 int16_t *vChrFilter;
98 int16_t *vChrFilterPos;
99
100 uint8_t formatConvBuffer[VOF]; //FIXME dynamic allocation, but we have to change a lot of code for this to be useful
101
102 int hLumFilterSize;
103 int hChrFilterSize;
104 int vLumFilterSize;
105 int vChrFilterSize;
106 int vLumBufSize;
107 int vChrBufSize;
108
109 uint8_t *funnyYCode;
110 uint8_t *funnyUVCode;
111 int32_t *lumMmx2FilterPos;
112 int32_t *chrMmx2FilterPos;
113 int16_t *lumMmx2Filter;
114 int16_t *chrMmx2Filter;
115
116 int canMMX2BeUsed;
117
118 int lastInLumBuf;
119 int lastInChrBuf;
120 int lumBufIndex;
121 int chrBufIndex;
122 int dstY;
123 int flags;
124 void * yuvTable; // pointer to the yuv->rgb table start so it can be freed()
125 uint8_t * table_rV[256];
126 uint8_t * table_gU[256];
127 int table_gV[256];
128 uint8_t * table_bU[256];
129
130 //Colorspace stuff
131 int contrast, brightness, saturation; // for sws_getColorspaceDetails
132 int srcColorspaceTable[4];
133 int dstColorspaceTable[4];
134 int srcRange, dstRange;
135 int yuv2rgb_y_offset;
136 int yuv2rgb_y_coeff;
137 int yuv2rgb_v2r_coeff;
138 int yuv2rgb_v2g_coeff;
139 int yuv2rgb_u2g_coeff;
140 int yuv2rgb_u2b_coeff;
141
142 #define RED_DITHER "0*8"
143 #define GREEN_DITHER "1*8"
144 #define BLUE_DITHER "2*8"
145 #define Y_COEFF "3*8"
146 #define VR_COEFF "4*8"
147 #define UB_COEFF "5*8"
148 #define VG_COEFF "6*8"
149 #define UG_COEFF "7*8"
150 #define Y_OFFSET "8*8"
151 #define U_OFFSET "9*8"
152 #define V_OFFSET "10*8"
153 #define LUM_MMX_FILTER_OFFSET "11*8"
154 #define CHR_MMX_FILTER_OFFSET "11*8+4*4*256"
155 #define DSTW_OFFSET "11*8+4*4*256*2" //do not change, it is hardcoded in the ASM
156 #define ESP_OFFSET "11*8+4*4*256*2+8"
157 #define VROUNDER_OFFSET "11*8+4*4*256*2+16"
158 #define U_TEMP "11*8+4*4*256*2+24"
159 #define V_TEMP "11*8+4*4*256*2+32"
160 #define Y_TEMP "11*8+4*4*256*2+40"
161 #define ALP_MMX_FILTER_OFFSET "11*8+4*4*256*2+48"
162
163 uint64_t redDither __attribute__((aligned(8)));
164 uint64_t greenDither __attribute__((aligned(8)));
165 uint64_t blueDither __attribute__((aligned(8)));
166
167 uint64_t yCoeff __attribute__((aligned(8)));
168 uint64_t vrCoeff __attribute__((aligned(8)));
169 uint64_t ubCoeff __attribute__((aligned(8)));
170 uint64_t vgCoeff __attribute__((aligned(8)));
171 uint64_t ugCoeff __attribute__((aligned(8)));
172 uint64_t yOffset __attribute__((aligned(8)));
173 uint64_t uOffset __attribute__((aligned(8)));
174 uint64_t vOffset __attribute__((aligned(8)));
175 int32_t lumMmxFilter[4*MAX_FILTER_SIZE];
176 int32_t chrMmxFilter[4*MAX_FILTER_SIZE];
177 int dstW;
178 uint64_t esp __attribute__((aligned(8)));
179 uint64_t vRounder __attribute__((aligned(8)));
180 uint64_t u_temp __attribute__((aligned(8)));
181 uint64_t v_temp __attribute__((aligned(8)));
182 uint64_t y_temp __attribute__((aligned(8)));
183 int32_t alpMmxFilter[4*MAX_FILTER_SIZE];
184
185 #if HAVE_ALTIVEC
186
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;
194 vector signed short *vYCoeffsBank, *vCCoeffsBank;
195
196 #endif
197
198
199 #if ARCH_BFIN
200 uint32_t oy __attribute__((aligned(4)));
201 uint32_t oc __attribute__((aligned(4)));
202 uint32_t zero __attribute__((aligned(4)));
203 uint32_t cy __attribute__((aligned(4)));
204 uint32_t crv __attribute__((aligned(4)));
205 uint32_t rmask __attribute__((aligned(4)));
206 uint32_t cbu __attribute__((aligned(4)));
207 uint32_t bmask __attribute__((aligned(4)));
208 uint32_t cgu __attribute__((aligned(4)));
209 uint32_t cgv __attribute__((aligned(4)));
210 uint32_t gmask __attribute__((aligned(4)));
211 #endif
212
213 #if HAVE_VIS
214 uint64_t sparc_coeffs[10] __attribute__((aligned(8)));
215 #endif
216
217 } SwsContext;
218 //FIXME check init (where 0)
219
220 SwsFunc ff_yuv2rgb_get_func_ptr(SwsContext *c);
221 int ff_yuv2rgb_c_init_tables(SwsContext *c, const int inv_table[4], int fullRange, int brightness, int contrast, int saturation);
222
223 void ff_yuv2rgb_init_tables_altivec(SwsContext *c, const int inv_table[4], int brightness, int contrast, int saturation);
224 SwsFunc ff_yuv2rgb_init_mmx(SwsContext *c);
225 SwsFunc ff_yuv2rgb_init_vis(SwsContext *c);
226 SwsFunc ff_yuv2rgb_init_mlib(SwsContext *c);
227 SwsFunc ff_yuv2rgb_init_altivec(SwsContext *c);
228 SwsFunc ff_yuv2rgb_get_func_ptr_bfin(SwsContext *c);
229 void ff_yuv2packedX_altivec(SwsContext *c,
230 int16_t *lumFilter, int16_t **lumSrc, int lumFilterSize,
231 int16_t *chrFilter, int16_t **chrSrc, int chrFilterSize,
232 uint8_t *dest, int dstW, int dstY);
233
234 const char *sws_format_name(int format);
235
236 //FIXME replace this with something faster
237 #define isPlanarYUV(x) ( \
238 (x)==PIX_FMT_YUV410P \
239 || (x)==PIX_FMT_YUV420P \
240 || (x)==PIX_FMT_YUVA420P \
241 || (x)==PIX_FMT_YUV411P \
242 || (x)==PIX_FMT_YUV422P \
243 || (x)==PIX_FMT_YUV444P \
244 || (x)==PIX_FMT_YUV440P \
245 || (x)==PIX_FMT_NV12 \
246 || (x)==PIX_FMT_NV21 \
247 )
248 #define isYUV(x) ( \
249 (x)==PIX_FMT_UYVY422 \
250 || (x)==PIX_FMT_YUYV422 \
251 || isPlanarYUV(x) \
252 )
253 #define isGray(x) ( \
254 (x)==PIX_FMT_GRAY8 \
255 || (x)==PIX_FMT_GRAY16BE \
256 || (x)==PIX_FMT_GRAY16LE \
257 )
258 #define isGray16(x) ( \
259 (x)==PIX_FMT_GRAY16BE \
260 || (x)==PIX_FMT_GRAY16LE \
261 )
262 #define isRGB(x) ( \
263 (x)==PIX_FMT_RGB32 \
264 || (x)==PIX_FMT_RGB32_1 \
265 || (x)==PIX_FMT_RGB24 \
266 || (x)==PIX_FMT_RGB565 \
267 || (x)==PIX_FMT_RGB555 \
268 || (x)==PIX_FMT_RGB8 \
269 || (x)==PIX_FMT_RGB4 \
270 || (x)==PIX_FMT_RGB4_BYTE \
271 || (x)==PIX_FMT_MONOBLACK \
272 || (x)==PIX_FMT_MONOWHITE \
273 )
274 #define isBGR(x) ( \
275 (x)==PIX_FMT_BGR32 \
276 || (x)==PIX_FMT_BGR32_1 \
277 || (x)==PIX_FMT_BGR24 \
278 || (x)==PIX_FMT_BGR565 \
279 || (x)==PIX_FMT_BGR555 \
280 || (x)==PIX_FMT_BGR8 \
281 || (x)==PIX_FMT_BGR4 \
282 || (x)==PIX_FMT_BGR4_BYTE \
283 || (x)==PIX_FMT_MONOBLACK \
284 || (x)==PIX_FMT_MONOWHITE \
285 )
286 #define isALPHA(x) ( \
287 (x)==PIX_FMT_BGR32 \
288 || (x)==PIX_FMT_BGR32_1 \
289 || (x)==PIX_FMT_RGB32 \
290 || (x)==PIX_FMT_RGB32_1 \
291 || (x)==PIX_FMT_YUVA420P \
292 )
293
294 static inline int fmt_depth(int fmt)
295 {
296 switch(fmt) {
297 case PIX_FMT_BGRA:
298 case PIX_FMT_ABGR:
299 case PIX_FMT_RGBA:
300 case PIX_FMT_ARGB:
301 return 32;
302 case PIX_FMT_BGR24:
303 case PIX_FMT_RGB24:
304 return 24;
305 case PIX_FMT_BGR565:
306 case PIX_FMT_RGB565:
307 case PIX_FMT_GRAY16BE:
308 case PIX_FMT_GRAY16LE:
309 return 16;
310 case PIX_FMT_BGR555:
311 case PIX_FMT_RGB555:
312 return 15;
313 case PIX_FMT_BGR8:
314 case PIX_FMT_RGB8:
315 return 8;
316 case PIX_FMT_BGR4:
317 case PIX_FMT_RGB4:
318 case PIX_FMT_BGR4_BYTE:
319 case PIX_FMT_RGB4_BYTE:
320 return 4;
321 case PIX_FMT_MONOBLACK:
322 case PIX_FMT_MONOWHITE:
323 return 1;
324 default:
325 return 0;
326 }
327 }
328
329 extern const uint64_t ff_dither4[2];
330 extern const uint64_t ff_dither8[2];
331
332 extern const AVClass sws_context_class;
333
334 #endif /* SWSCALE_SWSCALE_INTERNAL_H */