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#undef MOVNTQ
25#undef EMMS
26#undef SFENCE
27
28#if HAVE_AMD3DNOW
29/* On K6 femms is faster than emms. On K7 femms is directly mapped to emms. */
30#define EMMS   "femms"
31#else
32#define EMMS   "emms"
33#endif
34
35#if HAVE_MMX2
36#define MOVNTQ "movntq"
37#define SFENCE "sfence"
38#else
39#define MOVNTQ "movq"
40#define SFENCE " # nop"
41#endif
42
43#define REG_BLUE  "0"
44#define REG_RED   "1"
45#define REG_GREEN "2"
46#define REG_ALPHA "3"
47
48#define YUV2RGB_LOOP(depth)                                          \
49    h_size = (c->dstW + 7) & ~7;                                     \
50    if (h_size * depth > FFABS(dstStride[0]))                        \
51        h_size -= 8;                                                 \
52                                                                     \
53    if (c->srcFormat == PIX_FMT_YUV422P) {                           \
54        srcStride[1] *= 2;                                           \
55        srcStride[2] *= 2;                                           \
56    }                                                                \
57                                                                     \
58    __asm__ volatile ("pxor %mm4, %mm4\n\t");                        \
59    for (y = 0; y < srcSliceH; y++) {                                \
60        uint8_t *image    = dst[0] + (y + srcSliceY) * dstStride[0]; \
61        const uint8_t *py = src[0] +               y * srcStride[0]; \
62        const uint8_t *pu = src[1] +        (y >> 1) * srcStride[1]; \
63        const uint8_t *pv = src[2] +        (y >> 1) * srcStride[2]; \
64        x86_reg index = -h_size / 2;                                 \
65
66#define YUV2RGB_INITIAL_LOAD          \
67    __asm__ volatile (                \
68        "movq (%5, %0, 2), %%mm6\n\t" \
69        "movd    (%2, %0), %%mm0\n\t" \
70        "movd    (%3, %0), %%mm1\n\t" \
71        "1: \n\t"                     \
72
73/* YUV2RGB core
74 * Conversion is performed in usual way:
75 * R = Y' * Ycoef + Vred * V'
76 * G = Y' * Ycoef + Vgreen * V' + Ugreen * U'
77 * B = Y' * Ycoef               + Ublue * U'
78 *
79 * where X' = X * 8 - Xoffset (multiplication is performed to increase
80 * precision a bit).
81 * Since it operates in YUV420 colorspace, Y component is additionally
82 * split into Y1 and Y2 for even and odd pixels.
83 *
84 * Input:
85 * mm0 - U (4 elems), mm1 - V (4 elems), mm6 - Y (8 elems), mm4 - zero register
86 * Output:
87 * mm1 - R, mm2 - G, mm0 - B
88 */
89#define YUV2RGB                                  \
90    /* convert Y, U, V into Y1', Y2', U', V' */  \
91    "movq      %%mm6, %%mm7\n\t"                 \
92    "punpcklbw %%mm4, %%mm0\n\t"                 \
93    "punpcklbw %%mm4, %%mm1\n\t"                 \
94    "pand     "MANGLE(mmx_00ffw)", %%mm6\n\t"    \
95    "psrlw     $8,    %%mm7\n\t"                 \
96    "psllw     $3,    %%mm0\n\t"                 \
97    "psllw     $3,    %%mm1\n\t"                 \
98    "psllw     $3,    %%mm6\n\t"                 \
99    "psllw     $3,    %%mm7\n\t"                 \
100    "psubsw   "U_OFFSET"(%4), %%mm0\n\t"         \
101    "psubsw   "V_OFFSET"(%4), %%mm1\n\t"         \
102    "psubw    "Y_OFFSET"(%4), %%mm6\n\t"         \
103    "psubw    "Y_OFFSET"(%4), %%mm7\n\t"         \
104\
105     /* multiply by coefficients */              \
106    "movq      %%mm0, %%mm2\n\t"                 \
107    "movq      %%mm1, %%mm3\n\t"                 \
108    "pmulhw   "UG_COEFF"(%4), %%mm2\n\t"         \
109    "pmulhw   "VG_COEFF"(%4), %%mm3\n\t"         \
110    "pmulhw   "Y_COEFF" (%4), %%mm6\n\t"         \
111    "pmulhw   "Y_COEFF" (%4), %%mm7\n\t"         \
112    "pmulhw   "UB_COEFF"(%4), %%mm0\n\t"         \
113    "pmulhw   "VR_COEFF"(%4), %%mm1\n\t"         \
114    "paddsw    %%mm3, %%mm2\n\t"                 \
115    /* now: mm0 = UB, mm1 = VR, mm2 = CG */      \
116    /*      mm6 = Y1, mm7 = Y2 */                \
117\
118    /* produce RGB */                            \
119    "movq      %%mm7, %%mm3\n\t"                 \
120    "movq      %%mm7, %%mm5\n\t"                 \
121    "paddsw    %%mm0, %%mm3\n\t"                 \
122    "paddsw    %%mm1, %%mm5\n\t"                 \
123    "paddsw    %%mm2, %%mm7\n\t"                 \
124    "paddsw    %%mm6, %%mm0\n\t"                 \
125    "paddsw    %%mm6, %%mm1\n\t"                 \
126    "paddsw    %%mm6, %%mm2\n\t"                 \
127\
128    /* pack and interleave even/odd pixels */    \
129    "packuswb  %%mm0, %%mm0\n\t"                 \
130    "packuswb  %%mm1, %%mm1\n\t"                 \
131    "packuswb  %%mm2, %%mm2\n\t"                 \
132    "packuswb  %%mm3, %%mm3\n\t"                 \
133    "packuswb  %%mm5, %%mm5\n\t"                 \
134    "packuswb  %%mm7, %%mm7\n\t"                 \
135    "punpcklbw %%mm3, %%mm0\n\t"                 \
136    "punpcklbw %%mm5, %%mm1\n\t"                 \
137    "punpcklbw %%mm7, %%mm2\n\t"                 \
138
139#define YUV2RGB_ENDLOOP(depth)                   \
140    "movq 8 (%5, %0, 2), %%mm6\n\t"              \
141    "movd 4 (%3, %0),    %%mm1\n\t"              \
142    "movd 4 (%2, %0),    %%mm0\n\t"              \
143    "add $"AV_STRINGIFY(depth * 8)", %1\n\t"     \
144    "add  $4, %0\n\t"                            \
145    "js   1b\n\t"                                \
146
147#define YUV2RGB_OPERANDS                                          \
148        : "+r" (index), "+r" (image)                              \
149        : "r" (pu - index), "r" (pv - index), "r"(&c->redDither), \
150          "r" (py - 2*index)                                      \
151        );                                                        \
152    }                                                             \
153
154#define YUV2RGB_OPERANDS_ALPHA                                    \
155        : "+r" (index), "+r" (image)                              \
156        : "r" (pu - index), "r" (pv - index), "r"(&c->redDither), \
157          "r" (py - 2*index), "r" (pa - 2*index)                  \
158        );                                                        \
159    }                                                             \
160
161#define YUV2RGB_ENDFUNC                          \
162    __asm__ volatile (SFENCE"\n\t"EMMS);         \
163    return srcSliceH;                            \
164
165
166#define RGB_PACK16(gmask, gshift, rshift)        \
167    "pand      "MANGLE(mmx_redmask)", %%mm0\n\t" \
168    "pand      "MANGLE(mmx_redmask)", %%mm1\n\t" \
169    "psrlw     $3,        %%mm0\n\t"             \
170    "pand      "MANGLE(gmask)",       %%mm2\n\t" \
171    "movq      %%mm0,     %%mm5\n\t"             \
172    "movq      %%mm1,     %%mm6\n\t"             \
173    "movq      %%mm2,     %%mm7\n\t"             \
174    "punpcklbw %%mm4,     %%mm0\n\t"             \
175    "punpcklbw %%mm4,     %%mm1\n\t"             \
176    "punpcklbw %%mm4,     %%mm2\n\t"             \
177    "punpckhbw %%mm4,     %%mm5\n\t"             \
178    "punpckhbw %%mm4,     %%mm6\n\t"             \
179    "punpckhbw %%mm4,     %%mm7\n\t"             \
180    "psllw     $"rshift", %%mm1\n\t"             \
181    "psllw     $"rshift", %%mm6\n\t"             \
182    "psllw     $"gshift", %%mm2\n\t"             \
183    "psllw     $"gshift", %%mm7\n\t"             \
184    "por       %%mm1,     %%mm0\n\t"             \
185    "por       %%mm6,     %%mm5\n\t"             \
186    "por       %%mm2,     %%mm0\n\t"             \
187    "por       %%mm7,     %%mm5\n\t"             \
188    MOVNTQ "   %%mm0,      (%1)\n\t"             \
189    MOVNTQ "   %%mm5,     8(%1)\n\t"             \
190
191#define DITHER_RGB                               \
192    "paddusb "BLUE_DITHER"(%4),  %%mm0\n\t"      \
193    "paddusb "GREEN_DITHER"(%4), %%mm2\n\t"      \
194    "paddusb "RED_DITHER"(%4),   %%mm1\n\t"      \
195
196static inline int RENAME(yuv420_rgb15)(SwsContext *c, const uint8_t *src[],
197                                       int srcStride[],
198                                       int srcSliceY, int srcSliceH,
199                                       uint8_t *dst[], int dstStride[])
200{
201    int y, h_size;
202
203    YUV2RGB_LOOP(2)
204
205#ifdef DITHER1XBPP
206        c->blueDither  = ff_dither8[y       & 1];
207        c->greenDither = ff_dither8[y       & 1];
208        c->redDither   = ff_dither8[(y + 1) & 1];
209#endif
210
211        YUV2RGB_INITIAL_LOAD
212        YUV2RGB
213#ifdef DITHER1XBPP
214        DITHER_RGB
215#endif
216        RGB_PACK16(mmx_redmask, "2", "7")
217
218    YUV2RGB_ENDLOOP(2)
219    YUV2RGB_OPERANDS
220    YUV2RGB_ENDFUNC
221}
222
223static inline int RENAME(yuv420_rgb16)(SwsContext *c, const uint8_t *src[],
224                                       int srcStride[],
225                                       int srcSliceY, int srcSliceH,
226                                       uint8_t *dst[], int dstStride[])
227{
228    int y, h_size;
229
230    YUV2RGB_LOOP(2)
231
232#ifdef DITHER1XBPP
233        c->blueDither  = ff_dither8[y       & 1];
234        c->greenDither = ff_dither4[y       & 1];
235        c->redDither   = ff_dither8[(y + 1) & 1];
236#endif
237
238        YUV2RGB_INITIAL_LOAD
239        YUV2RGB
240#ifdef DITHER1XBPP
241        DITHER_RGB
242#endif
243        RGB_PACK16(mmx_grnmask, "3", "8")
244
245    YUV2RGB_ENDLOOP(2)
246    YUV2RGB_OPERANDS
247    YUV2RGB_ENDFUNC
248}
249
250
251#define RGB_PACK24(red, blue)              \
252    /* generate first packed RGB octet */  \
253    "movq      %%mm2,      %%mm5\n\t"      \
254    "movq      %%mm"blue", %%mm6\n\t"      \
255    "movq      %%mm"red",  %%mm7\n\t"      \
256    "punpcklbw %%mm5,      %%mm6\n\t"      \
257    "punpcklbw %%mm4,      %%mm7\n\t"      \
258    "movq      %%mm6,      %%mm3\n\t"      \
259    "punpcklwd %%mm7,      %%mm6\n\t"      \
260    "psrlq     $32,        %%mm3\n\t"      \
261    "movq      %%mm6,      %%mm5\n\t"      \
262    "psllq     $40,        %%mm6\n\t"      \
263    "psllq     $48,        %%mm3\n\t"      \
264    "psrlq     $32,        %%mm5\n\t"      \
265    "psrlq     $40,        %%mm6\n\t"      \
266    "psllq     $24,        %%mm5\n\t"      \
267    "por       %%mm3,      %%mm6\n\t"      \
268    "por       %%mm5,      %%mm6\n\t"      \
269    MOVNTQ "   %%mm6,      (%1)\n\t"       \
270\
271    /* generate second packed RGB octet */ \
272    "movq      %%mm"red",  %%mm7\n\t"      \
273    "movq      %%mm2,      %%mm5\n\t"      \
274    "movq      %%mm"blue", %%mm6\n\t"      \
275    "punpcklbw %%mm4,      %%mm7\n\t"      \
276    "punpcklbw %%mm5,      %%mm6\n\t"      \
277    "movq      %%mm7,      %%mm3\n\t"      \
278    "punpckhwd %%mm7,      %%mm6\n\t"      \
279    "psllq     $16,        %%mm3\n\t"      \
280    "psrlq     $32,        %%mm6\n\t"      \
281    "psrlq     $48,        %%mm3\n\t"      \
282    "psllq     $8,         %%mm6\n\t"      \
283    "movq      %%mm"red",  %%mm7\n\t"      \
284    "por       %%mm6,      %%mm3\n\t"      \
285    "movq      %%mm"blue", %%mm6\n\t"      \
286    "movq      %%mm2,      %%mm5\n\t"      \
287    "punpckhbw %%mm4,      %%mm7\n\t"      \
288    "punpckhbw %%mm5,      %%mm6\n\t"      \
289    "movq      %%mm6,      %%mm5\n\t"      \
290    "punpcklwd %%mm7,      %%mm6\n\t"      \
291    "psrlq     $16,        %%mm5\n\t"      \
292    "psllq     $56,        %%mm5\n\t"      \
293    "por       %%mm5,      %%mm3\n\t"      \
294    "psllq     $32,        %%mm6\n\t"      \
295    "por       %%mm6,      %%mm3\n\t"      \
296    MOVNTQ "   %%mm3,      8(%1)\n\t"      \
297\
298    /* generate third packed RGB octet */  \
299    "movq      %%mm"red",  %%mm7\n\t"      \
300    "movq      %%mm2,      %%mm5\n\t"      \
301    "movq      %%mm2,      %%mm3\n\t"      \
302    "movq      %%mm"blue", %%mm6\n\t"      \
303    "punpckhbw %%mm"red",  %%mm3\n\t"      \
304    "punpckhbw %%mm4,      %%mm7\n\t"      \
305    "psllq     $32,        %%mm3\n\t"      \
306    "punpckhbw %%mm5,      %%mm6\n\t"      \
307    "psrlq     $48,        %%mm3\n\t"      \
308    "punpckhwd %%mm7,      %%mm6\n\t"      \
309    "movq      %%mm6,      %%mm7\n\t"      \
310    "psrlq     $32,        %%mm6\n\t"      \
311    "psllq     $32,        %%mm7\n\t"      \
312    "psllq     $40,        %%mm6\n\t"      \
313    "psrlq     $16,        %%mm7\n\t"      \
314    "por       %%mm6,      %%mm3\n\t"      \
315    "por       %%mm7,      %%mm3\n\t"      \
316    MOVNTQ "   %%mm3,      16(%1)\n\t"     \
317
318static inline int RENAME(yuv420_rgb24)(SwsContext *c, const uint8_t *src[],
319                                       int srcStride[],
320                                       int srcSliceY, int srcSliceH,
321                                       uint8_t *dst[], int dstStride[])
322{
323    int y, h_size;
324
325    YUV2RGB_LOOP(3)
326
327        YUV2RGB_INITIAL_LOAD
328        YUV2RGB
329        RGB_PACK24(REG_BLUE, REG_RED)
330
331    YUV2RGB_ENDLOOP(3)
332    YUV2RGB_OPERANDS
333    YUV2RGB_ENDFUNC
334}
335
336static inline int RENAME(yuv420_bgr24)(SwsContext *c, const uint8_t *src[],
337                                       int srcStride[],
338                                       int srcSliceY, int srcSliceH,
339                                       uint8_t *dst[], int dstStride[])
340{
341    int y, h_size;
342
343    YUV2RGB_LOOP(3)
344
345        YUV2RGB_INITIAL_LOAD
346        YUV2RGB
347        RGB_PACK24(REG_RED, REG_BLUE)
348
349    YUV2RGB_ENDLOOP(3)
350    YUV2RGB_OPERANDS
351    YUV2RGB_ENDFUNC
352}
353
354
355#define SET_EMPTY_ALPHA                                                      \
356    "pcmpeqd   %%mm"REG_ALPHA", %%mm"REG_ALPHA"\n\t" /* set alpha to 0xFF */ \
357
358#define LOAD_ALPHA                                   \
359    "movq      (%6, %0, 2),     %%mm"REG_ALPHA"\n\t" \
360
361#define RGB_PACK32(red, green, blue, alpha)  \
362    "movq      %%mm"blue",  %%mm5\n\t"       \
363    "movq      %%mm"red",   %%mm6\n\t"       \
364    "punpckhbw %%mm"green", %%mm5\n\t"       \
365    "punpcklbw %%mm"green", %%mm"blue"\n\t"  \
366    "punpckhbw %%mm"alpha", %%mm6\n\t"       \
367    "punpcklbw %%mm"alpha", %%mm"red"\n\t"   \
368    "movq      %%mm"blue",  %%mm"green"\n\t" \
369    "movq      %%mm5,       %%mm"alpha"\n\t" \
370    "punpcklwd %%mm"red",   %%mm"blue"\n\t"  \
371    "punpckhwd %%mm"red",   %%mm"green"\n\t" \
372    "punpcklwd %%mm6,       %%mm5\n\t"       \
373    "punpckhwd %%mm6,       %%mm"alpha"\n\t" \
374    MOVNTQ "   %%mm"blue",   0(%1)\n\t"      \
375    MOVNTQ "   %%mm"green",  8(%1)\n\t"      \
376    MOVNTQ "   %%mm5,       16(%1)\n\t"      \
377    MOVNTQ "   %%mm"alpha", 24(%1)\n\t"      \
378
379static inline int RENAME(yuv420_rgb32)(SwsContext *c, const uint8_t *src[],
380                                       int srcStride[],
381                                       int srcSliceY, int srcSliceH,
382                                       uint8_t *dst[], int dstStride[])
383{
384    int y, h_size;
385
386    YUV2RGB_LOOP(4)
387
388        YUV2RGB_INITIAL_LOAD
389        YUV2RGB
390        SET_EMPTY_ALPHA
391        RGB_PACK32(REG_RED, REG_GREEN, REG_BLUE, REG_ALPHA)
392
393    YUV2RGB_ENDLOOP(4)
394    YUV2RGB_OPERANDS
395    YUV2RGB_ENDFUNC
396}
397
398static inline int RENAME(yuva420_rgb32)(SwsContext *c, const uint8_t *src[],
399                                        int srcStride[],
400                                        int srcSliceY, int srcSliceH,
401                                        uint8_t *dst[], int dstStride[])
402{
403#if HAVE_7REGS
404    int y, h_size;
405
406    YUV2RGB_LOOP(4)
407
408        const uint8_t *pa = src[3] + y * srcStride[3];
409        YUV2RGB_INITIAL_LOAD
410        YUV2RGB
411        LOAD_ALPHA
412        RGB_PACK32(REG_RED, REG_GREEN, REG_BLUE, REG_ALPHA)
413
414    YUV2RGB_ENDLOOP(4)
415    YUV2RGB_OPERANDS_ALPHA
416    YUV2RGB_ENDFUNC
417#endif
418}
419
420static inline int RENAME(yuv420_bgr32)(SwsContext *c, const uint8_t *src[],
421                                       int srcStride[],
422                                       int srcSliceY, int srcSliceH,
423                                       uint8_t *dst[], int dstStride[])
424{
425    int y, h_size;
426
427    YUV2RGB_LOOP(4)
428
429        YUV2RGB_INITIAL_LOAD
430        YUV2RGB
431        SET_EMPTY_ALPHA
432        RGB_PACK32(REG_BLUE, REG_GREEN, REG_RED, REG_ALPHA)
433
434    YUV2RGB_ENDLOOP(4)
435    YUV2RGB_OPERANDS
436    YUV2RGB_ENDFUNC
437}
438
439static inline int RENAME(yuva420_bgr32)(SwsContext *c, const uint8_t *src[],
440                                        int srcStride[],
441                                        int srcSliceY, int srcSliceH,
442                                        uint8_t *dst[], int dstStride[])
443{
444#if HAVE_7REGS
445    int y, h_size;
446
447    YUV2RGB_LOOP(4)
448
449        const uint8_t *pa = src[3] + y * srcStride[3];
450        YUV2RGB_INITIAL_LOAD
451        YUV2RGB
452        LOAD_ALPHA
453        RGB_PACK32(REG_BLUE, REG_GREEN, REG_RED, REG_ALPHA)
454
455    YUV2RGB_ENDLOOP(4)
456    YUV2RGB_OPERANDS_ALPHA
457    YUV2RGB_ENDFUNC
458#endif
459}
460