2 * software YUV to RGB converter
4 * Copyright (C) 2001-2007 Michael Niedermayer
5 * (c) 2010 Konstantin Shishkov
7 * This file is part of FFmpeg.
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.
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.
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
28 #if COMPILE_TEMPLATE_MMXEXT
29 #define MOVNTQ "movntq"
30 #define SFENCE "sfence"
33 #define SFENCE " # nop"
41 #define YUV2RGB_LOOP(depth) \
42 h_size = (c->dstW + 7) & ~7; \
43 if (h_size * depth > FFABS(dstStride[0])) \
46 vshift = c->srcFormat != PIX_FMT_YUV422P; \
48 __asm__ volatile ("pxor %mm4, %mm4\n\t"); \
49 for (y = 0; y < srcSliceH; y++) { \
50 uint8_t *image = dst[0] + (y + srcSliceY) * dstStride[0]; \
51 const uint8_t *py = src[0] + y * srcStride[0]; \
52 const uint8_t *pu = src[1] + (y >> vshift) * srcStride[1]; \
53 const uint8_t *pv = src[2] + (y >> vshift) * srcStride[2]; \
54 x86_reg index = -h_size / 2; \
56 #define YUV2RGB_INITIAL_LOAD \
58 "movq (%5, %0, 2), %%mm6\n\t" \
59 "movd (%2, %0), %%mm0\n\t" \
60 "movd (%3, %0), %%mm1\n\t" \
64 * Conversion is performed in usual way:
65 * R = Y' * Ycoef + Vred * V'
66 * G = Y' * Ycoef + Vgreen * V' + Ugreen * U'
67 * B = Y' * Ycoef + Ublue * U'
69 * where X' = X * 8 - Xoffset (multiplication is performed to increase
71 * Since it operates in YUV420 colorspace, Y component is additionally
72 * split into Y1 and Y2 for even and odd pixels.
75 * mm0 - U (4 elems), mm1 - V (4 elems), mm6 - Y (8 elems), mm4 - zero register
77 * mm1 - R, mm2 - G, mm0 - B
80 /* convert Y, U, V into Y1', Y2', U', V' */ \
81 "movq %%mm6, %%mm7\n\t" \
82 "punpcklbw %%mm4, %%mm0\n\t" \
83 "punpcklbw %%mm4, %%mm1\n\t" \
84 "pand "MANGLE(mmx_00ffw)", %%mm6\n\t" \
85 "psrlw $8, %%mm7\n\t" \
86 "psllw $3, %%mm0\n\t" \
87 "psllw $3, %%mm1\n\t" \
88 "psllw $3, %%mm6\n\t" \
89 "psllw $3, %%mm7\n\t" \
90 "psubsw "U_OFFSET"(%4), %%mm0\n\t" \
91 "psubsw "V_OFFSET"(%4), %%mm1\n\t" \
92 "psubw "Y_OFFSET"(%4), %%mm6\n\t" \
93 "psubw "Y_OFFSET"(%4), %%mm7\n\t" \
95 /* multiply by coefficients */ \
96 "movq %%mm0, %%mm2\n\t" \
97 "movq %%mm1, %%mm3\n\t" \
98 "pmulhw "UG_COEFF"(%4), %%mm2\n\t" \
99 "pmulhw "VG_COEFF"(%4), %%mm3\n\t" \
100 "pmulhw "Y_COEFF" (%4), %%mm6\n\t" \
101 "pmulhw "Y_COEFF" (%4), %%mm7\n\t" \
102 "pmulhw "UB_COEFF"(%4), %%mm0\n\t" \
103 "pmulhw "VR_COEFF"(%4), %%mm1\n\t" \
104 "paddsw %%mm3, %%mm2\n\t" \
105 /* now: mm0 = UB, mm1 = VR, mm2 = CG */ \
106 /* mm6 = Y1, mm7 = Y2 */ \
109 "movq %%mm7, %%mm3\n\t" \
110 "movq %%mm7, %%mm5\n\t" \
111 "paddsw %%mm0, %%mm3\n\t" \
112 "paddsw %%mm1, %%mm5\n\t" \
113 "paddsw %%mm2, %%mm7\n\t" \
114 "paddsw %%mm6, %%mm0\n\t" \
115 "paddsw %%mm6, %%mm1\n\t" \
116 "paddsw %%mm6, %%mm2\n\t" \
118 #define RGB_PACK_INTERLEAVE \
119 /* pack and interleave even/odd pixels */ \
120 "packuswb %%mm1, %%mm0\n\t" \
121 "packuswb %%mm5, %%mm3\n\t" \
122 "packuswb %%mm2, %%mm2\n\t" \
123 "movq %%mm0, %%mm1\n\n" \
124 "packuswb %%mm7, %%mm7\n\t" \
125 "punpcklbw %%mm3, %%mm0\n\t" \
126 "punpckhbw %%mm3, %%mm1\n\t" \
127 "punpcklbw %%mm7, %%mm2\n\t" \
129 #define YUV2RGB_ENDLOOP(depth) \
130 "movq 8 (%5, %0, 2), %%mm6\n\t" \
131 "movd 4 (%3, %0), %%mm1\n\t" \
132 "movd 4 (%2, %0), %%mm0\n\t" \
133 "add $"AV_STRINGIFY(depth * 8)", %1\n\t" \
137 #define YUV2RGB_OPERANDS \
138 : "+r" (index), "+r" (image) \
139 : "r" (pu - index), "r" (pv - index), "r"(&c->redDither), \
145 #define YUV2RGB_OPERANDS_ALPHA \
146 : "+r" (index), "+r" (image) \
147 : "r" (pu - index), "r" (pv - index), "r"(&c->redDither), \
148 "r" (py - 2*index), "r" (pa - 2*index) \
153 #define YUV2RGB_ENDFUNC \
154 __asm__ volatile (SFENCE"\n\t" \
161 #define RGB_PACK16(gmask, is15) \
162 "pand "MANGLE(mmx_redmask)", %%mm0\n\t" \
163 "pand "MANGLE(mmx_redmask)", %%mm1\n\t" \
164 "movq %%mm2, %%mm3\n\t" \
165 "psllw $"AV_STRINGIFY(3-is15)", %%mm2\n\t" \
166 "psrlw $"AV_STRINGIFY(5+is15)", %%mm3\n\t" \
167 "psrlw $3, %%mm0\n\t" \
168 IF##is15("psrlw $1, %%mm1\n\t") \
169 "pand "MANGLE(pb_e0)", %%mm2\n\t" \
170 "pand "MANGLE(gmask)", %%mm3\n\t" \
171 "por %%mm2, %%mm0\n\t" \
172 "por %%mm3, %%mm1\n\t" \
173 "movq %%mm0, %%mm2\n\t" \
174 "punpcklbw %%mm1, %%mm0\n\t" \
175 "punpckhbw %%mm1, %%mm2\n\t" \
176 MOVNTQ " %%mm0, (%1)\n\t" \
177 MOVNTQ " %%mm2, 8(%1)\n\t" \
180 "paddusb "BLUE_DITHER"(%4), %%mm0\n\t" \
181 "paddusb "GREEN_DITHER"(%4), %%mm2\n\t" \
182 "paddusb "RED_DITHER"(%4), %%mm1\n\t" \
184 #if !COMPILE_TEMPLATE_MMXEXT
185 static inline int RENAME(yuv420_rgb15)(SwsContext *c, const uint8_t *src[],
187 int srcSliceY, int srcSliceH,
188 uint8_t *dst[], int dstStride[])
190 int y, h_size, vshift;
195 c->blueDither = ff_dither8[y & 1];
196 c->greenDither = ff_dither8[y & 1];
197 c->redDither = ff_dither8[(y + 1) & 1];
213 static inline int RENAME(yuv420_rgb16)(SwsContext *c, const uint8_t *src[],
215 int srcSliceY, int srcSliceH,
216 uint8_t *dst[], int dstStride[])
218 int y, h_size, vshift;
223 c->blueDither = ff_dither8[y & 1];
224 c->greenDither = ff_dither4[y & 1];
225 c->redDither = ff_dither8[(y + 1) & 1];
240 #endif /* !COMPILE_TEMPLATE_MMXEXT */
242 #define RGB_PACK24(blue, red)\
243 "packuswb %%mm3, %%mm0 \n" /* R0 R2 R4 R6 R1 R3 R5 R7 */\
244 "packuswb %%mm5, %%mm1 \n" /* B0 B2 B4 B6 B1 B3 B5 B7 */\
245 "packuswb %%mm7, %%mm2 \n" /* G0 G2 G4 G6 G1 G3 G5 G7 */\
246 "movq %%mm"red", %%mm3 \n"\
247 "movq %%mm"blue", %%mm6 \n"\
248 "psrlq $32, %%mm"red" \n" /* R1 R3 R5 R7 */\
249 "punpcklbw %%mm2, %%mm3 \n" /* R0 G0 R2 G2 R4 G4 R6 G6 */\
250 "punpcklbw %%mm"red", %%mm6 \n" /* B0 R1 B2 R3 B4 R5 B6 R7 */\
251 "movq %%mm3, %%mm5 \n"\
252 "punpckhbw %%mm"blue", %%mm2 \n" /* G1 B1 G3 B3 G5 B5 G7 B7 */\
253 "punpcklwd %%mm6, %%mm3 \n" /* R0 G0 B0 R1 R2 G2 B2 R3 */\
254 "punpckhwd %%mm6, %%mm5 \n" /* R4 G4 B4 R5 R6 G6 B6 R7 */\
257 #if COMPILE_TEMPLATE_MMXEXT
258 DECLARE_ASM_CONST(8, int16_t, mask1101[4]) = {-1,-1, 0,-1};
259 DECLARE_ASM_CONST(8, int16_t, mask0010[4]) = { 0, 0,-1, 0};
260 DECLARE_ASM_CONST(8, int16_t, mask0110[4]) = { 0,-1,-1, 0};
261 DECLARE_ASM_CONST(8, int16_t, mask1001[4]) = {-1, 0, 0,-1};
262 DECLARE_ASM_CONST(8, int16_t, mask0100[4]) = { 0,-1, 0, 0};
264 #define RGB_PACK24_B\
265 "pshufw $0xc6, %%mm2, %%mm1 \n"\
266 "pshufw $0x84, %%mm3, %%mm6 \n"\
267 "pshufw $0x38, %%mm5, %%mm7 \n"\
268 "pand "MANGLE(mask1101)", %%mm6 \n" /* R0 G0 B0 R1 -- -- R2 G2 */\
269 "movq %%mm1, %%mm0 \n"\
270 "pand "MANGLE(mask0110)", %%mm7 \n" /* -- -- R6 G6 B6 R7 -- -- */\
271 "movq %%mm1, %%mm2 \n"\
272 "pand "MANGLE(mask0100)", %%mm1 \n" /* -- -- G3 B3 -- -- -- -- */\
273 "psrlq $48, %%mm3 \n" /* B2 R3 -- -- -- -- -- -- */\
274 "pand "MANGLE(mask0010)", %%mm0 \n" /* -- -- -- -- G1 B1 -- -- */\
275 "psllq $32, %%mm5 \n" /* -- -- -- -- R4 G4 B4 R5 */\
276 "pand "MANGLE(mask1001)", %%mm2 \n" /* G5 B5 -- -- -- -- G7 B7 */\
277 "por %%mm3, %%mm1 \n"\
278 "por %%mm6, %%mm0 \n"\
279 "por %%mm5, %%mm1 \n"\
280 "por %%mm7, %%mm2 \n"\
281 MOVNTQ" %%mm0, (%1) \n"\
282 MOVNTQ" %%mm1, 8(%1) \n"\
283 MOVNTQ" %%mm2, 16(%1) \n"\
287 #define RGB_PACK24_B\
288 "movd %%mm3, (%1) \n" /* R0 G0 B0 R1 */\
289 "movd %%mm2, 4(%1) \n" /* G1 B1 */\
290 "psrlq $32, %%mm3 \n"\
291 "psrlq $16, %%mm2 \n"\
292 "movd %%mm3, 6(%1) \n" /* R2 G2 B2 R3 */\
293 "movd %%mm2, 10(%1) \n" /* G3 B3 */\
294 "psrlq $16, %%mm2 \n"\
295 "movd %%mm5, 12(%1) \n" /* R4 G4 B4 R5 */\
296 "movd %%mm2, 16(%1) \n" /* G5 B5 */\
297 "psrlq $32, %%mm5 \n"\
298 "movd %%mm2, 20(%1) \n" /* -- -- G7 B7 */\
299 "movd %%mm5, 18(%1) \n" /* R6 G6 B6 R7 */\
303 static inline int RENAME(yuv420_rgb24)(SwsContext *c, const uint8_t *src[],
305 int srcSliceY, int srcSliceH,
306 uint8_t *dst[], int dstStride[])
308 int y, h_size, vshift;
314 RGB_PACK24(REG_BLUE, REG_RED)
321 static inline int RENAME(yuv420_bgr24)(SwsContext *c, const uint8_t *src[],
323 int srcSliceY, int srcSliceH,
324 uint8_t *dst[], int dstStride[])
326 int y, h_size, vshift;
332 RGB_PACK24(REG_RED, REG_BLUE)
340 #define SET_EMPTY_ALPHA \
341 "pcmpeqd %%mm"REG_ALPHA", %%mm"REG_ALPHA"\n\t" /* set alpha to 0xFF */ \
344 "movq (%6, %0, 2), %%mm"REG_ALPHA"\n\t" \
346 #define RGB_PACK32(red, green, blue, alpha) \
347 "movq %%mm"blue", %%mm5\n\t" \
348 "movq %%mm"red", %%mm6\n\t" \
349 "punpckhbw %%mm"green", %%mm5\n\t" \
350 "punpcklbw %%mm"green", %%mm"blue"\n\t" \
351 "punpckhbw %%mm"alpha", %%mm6\n\t" \
352 "punpcklbw %%mm"alpha", %%mm"red"\n\t" \
353 "movq %%mm"blue", %%mm"green"\n\t" \
354 "movq %%mm5, %%mm"alpha"\n\t" \
355 "punpcklwd %%mm"red", %%mm"blue"\n\t" \
356 "punpckhwd %%mm"red", %%mm"green"\n\t" \
357 "punpcklwd %%mm6, %%mm5\n\t" \
358 "punpckhwd %%mm6, %%mm"alpha"\n\t" \
359 MOVNTQ " %%mm"blue", 0(%1)\n\t" \
360 MOVNTQ " %%mm"green", 8(%1)\n\t" \
361 MOVNTQ " %%mm5, 16(%1)\n\t" \
362 MOVNTQ " %%mm"alpha", 24(%1)\n\t" \
364 #if !COMPILE_TEMPLATE_MMXEXT
365 static inline int RENAME(yuv420_rgb32)(SwsContext *c, const uint8_t *src[],
367 int srcSliceY, int srcSliceH,
368 uint8_t *dst[], int dstStride[])
370 int y, h_size, vshift;
378 RGB_PACK32(REG_RED, REG_GREEN, REG_BLUE, REG_ALPHA)
385 #if HAVE_7REGS && CONFIG_SWSCALE_ALPHA
386 static inline int RENAME(yuva420_rgb32)(SwsContext *c, const uint8_t *src[],
388 int srcSliceY, int srcSliceH,
389 uint8_t *dst[], int dstStride[])
391 int y, h_size, vshift;
395 const uint8_t *pa = src[3] + y * srcStride[3];
400 RGB_PACK32(REG_RED, REG_GREEN, REG_BLUE, REG_ALPHA)
403 YUV2RGB_OPERANDS_ALPHA
408 static inline int RENAME(yuv420_bgr32)(SwsContext *c, const uint8_t *src[],
410 int srcSliceY, int srcSliceH,
411 uint8_t *dst[], int dstStride[])
413 int y, h_size, vshift;
421 RGB_PACK32(REG_BLUE, REG_GREEN, REG_RED, REG_ALPHA)
428 #if HAVE_7REGS && CONFIG_SWSCALE_ALPHA
429 static inline int RENAME(yuva420_bgr32)(SwsContext *c, const uint8_t *src[],
431 int srcSliceY, int srcSliceH,
432 uint8_t *dst[], int dstStride[])
434 int y, h_size, vshift;
438 const uint8_t *pa = src[3] + y * srcStride[3];
443 RGB_PACK32(REG_BLUE, REG_GREEN, REG_RED, REG_ALPHA)
446 YUV2RGB_OPERANDS_ALPHA
451 #endif /* !COMPILE_TEMPLATE_MMXEXT */