Imported Debian version 2.4.3~trusty1
[deb_ffmpeg.git] / ffmpeg / libswscale / x86 / yuv2rgb_template.c
CommitLineData
2ba45a60
DM
1/*
2 * software YUV to RGB converter
3 *
4 * Copyright (C) 2001-2007 Michael Niedermayer
5 * (c) 2010 Konstantin Shishkov
6 *
7 * This file is part of FFmpeg.
8 *
9 * FFmpeg is free software; you can redistribute it and/or
10 * modify it under the terms of the GNU Lesser General Public
11 * License as published by the Free Software Foundation; either
12 * version 2.1 of the License, or (at your option) any later version.
13 *
14 * FFmpeg is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
17 * Lesser General Public License for more details.
18 *
19 * You should have received a copy of the GNU Lesser General Public
20 * License along with FFmpeg; if not, write to the Free Software
21 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
22 */
23
24#include <stdint.h>
25
26#include "libavutil/x86/asm.h"
27#include "libswscale/swscale_internal.h"
28
29#undef MOVNTQ
30#undef EMMS
31#undef SFENCE
32
33#if COMPILE_TEMPLATE_MMXEXT
34#define MOVNTQ "movntq"
35#define SFENCE "sfence"
36#else
37#define MOVNTQ "movq"
38#define SFENCE " # nop"
39#endif
40
41#define REG_BLUE "0"
42#define REG_RED "1"
43#define REG_GREEN "2"
44#define REG_ALPHA "3"
45
46#define YUV2RGB_LOOP(depth) \
47 h_size = (c->dstW + 7) & ~7; \
48 if (h_size * depth > FFABS(dstStride[0])) \
49 h_size -= 8; \
50 \
51 vshift = c->srcFormat != AV_PIX_FMT_YUV422P; \
52 \
53 __asm__ volatile ("pxor %mm4, %mm4\n\t"); \
54 for (y = 0; y < srcSliceH; y++) { \
55 uint8_t *image = dst[0] + (y + srcSliceY) * dstStride[0]; \
56 const uint8_t *py = src[0] + y * srcStride[0]; \
57 const uint8_t *pu = src[1] + (y >> vshift) * srcStride[1]; \
58 const uint8_t *pv = src[2] + (y >> vshift) * srcStride[2]; \
59 x86_reg index = -h_size / 2; \
60
61#define YUV2RGB_INITIAL_LOAD \
62 __asm__ volatile ( \
63 "movq (%5, %0, 2), %%mm6\n\t" \
64 "movd (%2, %0), %%mm0\n\t" \
65 "movd (%3, %0), %%mm1\n\t" \
66 "1: \n\t" \
67
68/* YUV2RGB core
69 * Conversion is performed in usual way:
70 * R = Y' * Ycoef + Vred * V'
71 * G = Y' * Ycoef + Vgreen * V' + Ugreen * U'
72 * B = Y' * Ycoef + Ublue * U'
73 *
74 * where X' = X * 8 - Xoffset (multiplication is performed to increase
75 * precision a bit).
76 * Since it operates in YUV420 colorspace, Y component is additionally
77 * split into Y1 and Y2 for even and odd pixels.
78 *
79 * Input:
80 * mm0 - U (4 elems), mm1 - V (4 elems), mm6 - Y (8 elems), mm4 - zero register
81 * Output:
82 * mm1 - R, mm2 - G, mm0 - B
83 */
84#define YUV2RGB \
85 /* convert Y, U, V into Y1', Y2', U', V' */ \
86 "movq %%mm6, %%mm7\n\t" \
87 "punpcklbw %%mm4, %%mm0\n\t" \
88 "punpcklbw %%mm4, %%mm1\n\t" \
89 "pand "MANGLE(mmx_00ffw)", %%mm6\n\t" \
90 "psrlw $8, %%mm7\n\t" \
91 "psllw $3, %%mm0\n\t" \
92 "psllw $3, %%mm1\n\t" \
93 "psllw $3, %%mm6\n\t" \
94 "psllw $3, %%mm7\n\t" \
95 "psubsw "U_OFFSET"(%4), %%mm0\n\t" \
96 "psubsw "V_OFFSET"(%4), %%mm1\n\t" \
97 "psubw "Y_OFFSET"(%4), %%mm6\n\t" \
98 "psubw "Y_OFFSET"(%4), %%mm7\n\t" \
99\
100 /* multiply by coefficients */ \
101 "movq %%mm0, %%mm2\n\t" \
102 "movq %%mm1, %%mm3\n\t" \
103 "pmulhw "UG_COEFF"(%4), %%mm2\n\t" \
104 "pmulhw "VG_COEFF"(%4), %%mm3\n\t" \
105 "pmulhw "Y_COEFF" (%4), %%mm6\n\t" \
106 "pmulhw "Y_COEFF" (%4), %%mm7\n\t" \
107 "pmulhw "UB_COEFF"(%4), %%mm0\n\t" \
108 "pmulhw "VR_COEFF"(%4), %%mm1\n\t" \
109 "paddsw %%mm3, %%mm2\n\t" \
110 /* now: mm0 = UB, mm1 = VR, mm2 = CG */ \
111 /* mm6 = Y1, mm7 = Y2 */ \
112\
113 /* produce RGB */ \
114 "movq %%mm7, %%mm3\n\t" \
115 "movq %%mm7, %%mm5\n\t" \
116 "paddsw %%mm0, %%mm3\n\t" \
117 "paddsw %%mm1, %%mm5\n\t" \
118 "paddsw %%mm2, %%mm7\n\t" \
119 "paddsw %%mm6, %%mm0\n\t" \
120 "paddsw %%mm6, %%mm1\n\t" \
121 "paddsw %%mm6, %%mm2\n\t" \
122
123#define RGB_PACK_INTERLEAVE \
124 /* pack and interleave even/odd pixels */ \
125 "packuswb %%mm1, %%mm0\n\t" \
126 "packuswb %%mm5, %%mm3\n\t" \
127 "packuswb %%mm2, %%mm2\n\t" \
128 "movq %%mm0, %%mm1\n\n" \
129 "packuswb %%mm7, %%mm7\n\t" \
130 "punpcklbw %%mm3, %%mm0\n\t" \
131 "punpckhbw %%mm3, %%mm1\n\t" \
132 "punpcklbw %%mm7, %%mm2\n\t" \
133
134#define YUV2RGB_ENDLOOP(depth) \
135 "movq 8 (%5, %0, 2), %%mm6\n\t" \
136 "movd 4 (%3, %0), %%mm1\n\t" \
137 "movd 4 (%2, %0), %%mm0\n\t" \
138 "add $"AV_STRINGIFY(depth * 8)", %1\n\t" \
139 "add $4, %0\n\t" \
140 "js 1b\n\t" \
141
142#if COMPILE_TEMPLATE_MMXEXT
143#undef RGB_PACK24_B_OPERANDS
144#define RGB_PACK24_B_OPERANDS NAMED_CONSTRAINTS_ARRAY_ADD(mask1101,mask0110,mask0100,mask0010,mask1001)
145#else
146#undef RGB_PACK24_B_OPERANDS
147#define RGB_PACK24_B_OPERANDS
148#endif
149
150#define YUV2RGB_OPERANDS \
151 : "+r" (index), "+r" (image) \
152 : "r" (pu - index), "r" (pv - index), "r"(&c->redDither), \
153 "r" (py - 2*index) \
154 NAMED_CONSTRAINTS_ADD(mmx_00ffw,pb_03,pb_07,mmx_redmask,pb_e0) \
155 RGB_PACK24_B_OPERANDS \
156 : "memory" \
157 ); \
158 } \
159
160#define YUV2RGB_OPERANDS_ALPHA \
161 : "+r" (index), "+r" (image) \
162 : "r" (pu - index), "r" (pv - index), "r"(&c->redDither), \
163 "r" (py - 2*index), "r" (pa - 2*index) \
164 NAMED_CONSTRAINTS_ADD(mmx_00ffw) \
165 : "memory" \
166 ); \
167 } \
168
169#define YUV2RGB_ENDFUNC \
170 __asm__ volatile (SFENCE"\n\t" \
171 "emms \n\t"); \
172 return srcSliceH; \
173
174#define IF0(x)
175#define IF1(x) x
176
177#define RGB_PACK16(gmask, is15) \
178 "pand "MANGLE(mmx_redmask)", %%mm0\n\t" \
179 "pand "MANGLE(mmx_redmask)", %%mm1\n\t" \
180 "movq %%mm2, %%mm3\n\t" \
181 "psllw $"AV_STRINGIFY(3-is15)", %%mm2\n\t" \
182 "psrlw $"AV_STRINGIFY(5+is15)", %%mm3\n\t" \
183 "psrlw $3, %%mm0\n\t" \
184 IF##is15("psrlw $1, %%mm1\n\t") \
185 "pand "MANGLE(pb_e0)", %%mm2\n\t" \
186 "pand "MANGLE(gmask)", %%mm3\n\t" \
187 "por %%mm2, %%mm0\n\t" \
188 "por %%mm3, %%mm1\n\t" \
189 "movq %%mm0, %%mm2\n\t" \
190 "punpcklbw %%mm1, %%mm0\n\t" \
191 "punpckhbw %%mm1, %%mm2\n\t" \
192 MOVNTQ " %%mm0, (%1)\n\t" \
193 MOVNTQ " %%mm2, 8(%1)\n\t" \
194
195#define DITHER_RGB \
196 "paddusb "BLUE_DITHER"(%4), %%mm0\n\t" \
197 "paddusb "GREEN_DITHER"(%4), %%mm2\n\t" \
198 "paddusb "RED_DITHER"(%4), %%mm1\n\t" \
199
200#if !COMPILE_TEMPLATE_MMXEXT
201static inline int RENAME(yuv420_rgb15)(SwsContext *c, const uint8_t *src[],
202 int srcStride[],
203 int srcSliceY, int srcSliceH,
204 uint8_t *dst[], int dstStride[])
205{
206 int y, h_size, vshift;
207
208 YUV2RGB_LOOP(2)
209
210#ifdef DITHER1XBPP
211 c->blueDither = ff_dither8[y & 1];
212 c->greenDither = ff_dither8[y & 1];
213 c->redDither = ff_dither8[(y + 1) & 1];
214#endif
215
216 YUV2RGB_INITIAL_LOAD
217 YUV2RGB
218 RGB_PACK_INTERLEAVE
219#ifdef DITHER1XBPP
220 DITHER_RGB
221#endif
222 RGB_PACK16(pb_03, 1)
223
224 YUV2RGB_ENDLOOP(2)
225 YUV2RGB_OPERANDS
226 YUV2RGB_ENDFUNC
227}
228
229static inline int RENAME(yuv420_rgb16)(SwsContext *c, const uint8_t *src[],
230 int srcStride[],
231 int srcSliceY, int srcSliceH,
232 uint8_t *dst[], int dstStride[])
233{
234 int y, h_size, vshift;
235
236 YUV2RGB_LOOP(2)
237
238#ifdef DITHER1XBPP
239 c->blueDither = ff_dither8[y & 1];
240 c->greenDither = ff_dither4[y & 1];
241 c->redDither = ff_dither8[(y + 1) & 1];
242#endif
243
244 YUV2RGB_INITIAL_LOAD
245 YUV2RGB
246 RGB_PACK_INTERLEAVE
247#ifdef DITHER1XBPP
248 DITHER_RGB
249#endif
250 RGB_PACK16(pb_07, 0)
251
252 YUV2RGB_ENDLOOP(2)
253 YUV2RGB_OPERANDS
254 YUV2RGB_ENDFUNC
255}
256#endif /* !COMPILE_TEMPLATE_MMXEXT */
257
258#define RGB_PACK24(blue, red)\
259 "packuswb %%mm3, %%mm0 \n" /* R0 R2 R4 R6 R1 R3 R5 R7 */\
260 "packuswb %%mm5, %%mm1 \n" /* B0 B2 B4 B6 B1 B3 B5 B7 */\
261 "packuswb %%mm7, %%mm2 \n" /* G0 G2 G4 G6 G1 G3 G5 G7 */\
262 "movq %%mm"red", %%mm3 \n"\
263 "movq %%mm"blue", %%mm6 \n"\
264 "psrlq $32, %%mm"red" \n" /* R1 R3 R5 R7 */\
265 "punpcklbw %%mm2, %%mm3 \n" /* R0 G0 R2 G2 R4 G4 R6 G6 */\
266 "punpcklbw %%mm"red", %%mm6 \n" /* B0 R1 B2 R3 B4 R5 B6 R7 */\
267 "movq %%mm3, %%mm5 \n"\
268 "punpckhbw %%mm"blue", %%mm2 \n" /* G1 B1 G3 B3 G5 B5 G7 B7 */\
269 "punpcklwd %%mm6, %%mm3 \n" /* R0 G0 B0 R1 R2 G2 B2 R3 */\
270 "punpckhwd %%mm6, %%mm5 \n" /* R4 G4 B4 R5 R6 G6 B6 R7 */\
271 RGB_PACK24_B
272
273#if COMPILE_TEMPLATE_MMXEXT
274DECLARE_ASM_CONST(8, int16_t, mask1101[4]) = {-1,-1, 0,-1};
275DECLARE_ASM_CONST(8, int16_t, mask0010[4]) = { 0, 0,-1, 0};
276DECLARE_ASM_CONST(8, int16_t, mask0110[4]) = { 0,-1,-1, 0};
277DECLARE_ASM_CONST(8, int16_t, mask1001[4]) = {-1, 0, 0,-1};
278DECLARE_ASM_CONST(8, int16_t, mask0100[4]) = { 0,-1, 0, 0};
279#undef RGB_PACK24_B
280#define RGB_PACK24_B\
281 "pshufw $0xc6, %%mm2, %%mm1 \n"\
282 "pshufw $0x84, %%mm3, %%mm6 \n"\
283 "pshufw $0x38, %%mm5, %%mm7 \n"\
284 "pand "MANGLE(mask1101)", %%mm6 \n" /* R0 G0 B0 R1 -- -- R2 G2 */\
285 "movq %%mm1, %%mm0 \n"\
286 "pand "MANGLE(mask0110)", %%mm7 \n" /* -- -- R6 G6 B6 R7 -- -- */\
287 "movq %%mm1, %%mm2 \n"\
288 "pand "MANGLE(mask0100)", %%mm1 \n" /* -- -- G3 B3 -- -- -- -- */\
289 "psrlq $48, %%mm3 \n" /* B2 R3 -- -- -- -- -- -- */\
290 "pand "MANGLE(mask0010)", %%mm0 \n" /* -- -- -- -- G1 B1 -- -- */\
291 "psllq $32, %%mm5 \n" /* -- -- -- -- R4 G4 B4 R5 */\
292 "pand "MANGLE(mask1001)", %%mm2 \n" /* G5 B5 -- -- -- -- G7 B7 */\
293 "por %%mm3, %%mm1 \n"\
294 "por %%mm6, %%mm0 \n"\
295 "por %%mm5, %%mm1 \n"\
296 "por %%mm7, %%mm2 \n"\
297 MOVNTQ" %%mm0, (%1) \n"\
298 MOVNTQ" %%mm1, 8(%1) \n"\
299 MOVNTQ" %%mm2, 16(%1) \n"\
300
301#else
302#undef RGB_PACK24_B
303#define RGB_PACK24_B\
304 "movd %%mm3, (%1) \n" /* R0 G0 B0 R1 */\
305 "movd %%mm2, 4(%1) \n" /* G1 B1 */\
306 "psrlq $32, %%mm3 \n"\
307 "psrlq $16, %%mm2 \n"\
308 "movd %%mm3, 6(%1) \n" /* R2 G2 B2 R3 */\
309 "movd %%mm2, 10(%1) \n" /* G3 B3 */\
310 "psrlq $16, %%mm2 \n"\
311 "movd %%mm5, 12(%1) \n" /* R4 G4 B4 R5 */\
312 "movd %%mm2, 16(%1) \n" /* G5 B5 */\
313 "psrlq $32, %%mm5 \n"\
314 "movd %%mm2, 20(%1) \n" /* -- -- G7 B7 */\
315 "movd %%mm5, 18(%1) \n" /* R6 G6 B6 R7 */\
316
317#endif
318
319static inline int RENAME(yuv420_rgb24)(SwsContext *c, const uint8_t *src[],
320 int srcStride[],
321 int srcSliceY, int srcSliceH,
322 uint8_t *dst[], int dstStride[])
323{
324 int y, h_size, vshift;
325
326 YUV2RGB_LOOP(3)
327
328 YUV2RGB_INITIAL_LOAD
329 YUV2RGB
330 RGB_PACK24(REG_BLUE, REG_RED)
331
332 YUV2RGB_ENDLOOP(3)
333 YUV2RGB_OPERANDS
334 YUV2RGB_ENDFUNC
335}
336
337static inline int RENAME(yuv420_bgr24)(SwsContext *c, const uint8_t *src[],
338 int srcStride[],
339 int srcSliceY, int srcSliceH,
340 uint8_t *dst[], int dstStride[])
341{
342 int y, h_size, vshift;
343
344 YUV2RGB_LOOP(3)
345
346 YUV2RGB_INITIAL_LOAD
347 YUV2RGB
348 RGB_PACK24(REG_RED, REG_BLUE)
349
350 YUV2RGB_ENDLOOP(3)
351 YUV2RGB_OPERANDS
352 YUV2RGB_ENDFUNC
353}
354
355
356#define SET_EMPTY_ALPHA \
357 "pcmpeqd %%mm"REG_ALPHA", %%mm"REG_ALPHA"\n\t" /* set alpha to 0xFF */ \
358
359#define LOAD_ALPHA \
360 "movq (%6, %0, 2), %%mm"REG_ALPHA"\n\t" \
361
362#define RGB_PACK32(red, green, blue, alpha) \
363 "movq %%mm"blue", %%mm5\n\t" \
364 "movq %%mm"red", %%mm6\n\t" \
365 "punpckhbw %%mm"green", %%mm5\n\t" \
366 "punpcklbw %%mm"green", %%mm"blue"\n\t" \
367 "punpckhbw %%mm"alpha", %%mm6\n\t" \
368 "punpcklbw %%mm"alpha", %%mm"red"\n\t" \
369 "movq %%mm"blue", %%mm"green"\n\t" \
370 "movq %%mm5, %%mm"alpha"\n\t" \
371 "punpcklwd %%mm"red", %%mm"blue"\n\t" \
372 "punpckhwd %%mm"red", %%mm"green"\n\t" \
373 "punpcklwd %%mm6, %%mm5\n\t" \
374 "punpckhwd %%mm6, %%mm"alpha"\n\t" \
375 MOVNTQ " %%mm"blue", 0(%1)\n\t" \
376 MOVNTQ " %%mm"green", 8(%1)\n\t" \
377 MOVNTQ " %%mm5, 16(%1)\n\t" \
378 MOVNTQ " %%mm"alpha", 24(%1)\n\t" \
379
380#if !COMPILE_TEMPLATE_MMXEXT
381static inline int RENAME(yuv420_rgb32)(SwsContext *c, const uint8_t *src[],
382 int srcStride[],
383 int srcSliceY, int srcSliceH,
384 uint8_t *dst[], int dstStride[])
385{
386 int y, h_size, vshift;
387
388 YUV2RGB_LOOP(4)
389
390 YUV2RGB_INITIAL_LOAD
391 YUV2RGB
392 RGB_PACK_INTERLEAVE
393 SET_EMPTY_ALPHA
394 RGB_PACK32(REG_RED, REG_GREEN, REG_BLUE, REG_ALPHA)
395
396 YUV2RGB_ENDLOOP(4)
397 YUV2RGB_OPERANDS
398 YUV2RGB_ENDFUNC
399}
400
401#if HAVE_7REGS && CONFIG_SWSCALE_ALPHA
402static inline int RENAME(yuva420_rgb32)(SwsContext *c, const uint8_t *src[],
403 int srcStride[],
404 int srcSliceY, int srcSliceH,
405 uint8_t *dst[], int dstStride[])
406{
407 int y, h_size, vshift;
408
409 YUV2RGB_LOOP(4)
410
411 const uint8_t *pa = src[3] + y * srcStride[3];
412 YUV2RGB_INITIAL_LOAD
413 YUV2RGB
414 RGB_PACK_INTERLEAVE
415 LOAD_ALPHA
416 RGB_PACK32(REG_RED, REG_GREEN, REG_BLUE, REG_ALPHA)
417
418 YUV2RGB_ENDLOOP(4)
419 YUV2RGB_OPERANDS_ALPHA
420 YUV2RGB_ENDFUNC
421}
422#endif
423
424static inline int RENAME(yuv420_bgr32)(SwsContext *c, const uint8_t *src[],
425 int srcStride[],
426 int srcSliceY, int srcSliceH,
427 uint8_t *dst[], int dstStride[])
428{
429 int y, h_size, vshift;
430
431 YUV2RGB_LOOP(4)
432
433 YUV2RGB_INITIAL_LOAD
434 YUV2RGB
435 RGB_PACK_INTERLEAVE
436 SET_EMPTY_ALPHA
437 RGB_PACK32(REG_BLUE, REG_GREEN, REG_RED, REG_ALPHA)
438
439 YUV2RGB_ENDLOOP(4)
440 YUV2RGB_OPERANDS
441 YUV2RGB_ENDFUNC
442}
443
444#if HAVE_7REGS && CONFIG_SWSCALE_ALPHA
445static inline int RENAME(yuva420_bgr32)(SwsContext *c, const uint8_t *src[],
446 int srcStride[],
447 int srcSliceY, int srcSliceH,
448 uint8_t *dst[], int dstStride[])
449{
450 int y, h_size, vshift;
451
452 YUV2RGB_LOOP(4)
453
454 const uint8_t *pa = src[3] + y * srcStride[3];
455 YUV2RGB_INITIAL_LOAD
456 YUV2RGB
457 RGB_PACK_INTERLEAVE
458 LOAD_ALPHA
459 RGB_PACK32(REG_BLUE, REG_GREEN, REG_RED, REG_ALPHA)
460
461 YUV2RGB_ENDLOOP(4)
462 YUV2RGB_OPERANDS_ALPHA
463 YUV2RGB_ENDFUNC
464}
465#endif
466
467#endif /* !COMPILE_TEMPLATE_MMXEXT */