1// Copyright (c) 2012 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#include "media/base/simd/convert_yuv_to_rgb.h"
6#include "media/base/simd/yuv_to_rgb_table.h"
7
8namespace media {
9
10#define packuswb(x) ((x) < 0 ? 0 : ((x) > 255 ? 255 : (x)))
11#define paddsw(x, y) (((x) + (y)) < -32768 ? -32768 : \
12    (((x) + (y)) > 32767 ? 32767 : ((x) + (y))))
13
14// On Android, pixel layout is RGBA (see skia/include/core/SkColorPriv.h);
15// however, other Chrome platforms use BGRA (see skia/config/SkUserConfig.h).
16// Ideally, android should not use the functions here due to performance issue
17// (http://crbug.com/249980).
18#if defined(OS_ANDROID)
19#define SK_R32_SHIFT    0
20#define SK_G32_SHIFT    8
21#define SK_B32_SHIFT    16
22#define SK_A32_SHIFT    24
23#define R_INDEX         0
24#define G_INDEX         1
25#define B_INDEX         2
26#define A_INDEX         3
27#else
28#define SK_B32_SHIFT    0
29#define SK_G32_SHIFT    8
30#define SK_R32_SHIFT    16
31#define SK_A32_SHIFT    24
32#define B_INDEX         0
33#define G_INDEX         1
34#define R_INDEX         2
35#define A_INDEX         3
36#endif
37
38static inline void ConvertYUVToRGB32_C(uint8 y,
39                                       uint8 u,
40                                       uint8 v,
41                                       uint8* rgb_buf,
42                                       const int16 convert_table[1024][4]) {
43  int b = convert_table[256+u][B_INDEX];
44  int g = convert_table[256+u][G_INDEX];
45  int r = convert_table[256+u][R_INDEX];
46  int a = convert_table[256+u][A_INDEX];
47
48  b = paddsw(b, convert_table[512+v][B_INDEX]);
49  g = paddsw(g, convert_table[512+v][G_INDEX]);
50  r = paddsw(r, convert_table[512+v][R_INDEX]);
51  a = paddsw(a, convert_table[512+v][A_INDEX]);
52
53  b = paddsw(b, convert_table[y][B_INDEX]);
54  g = paddsw(g, convert_table[y][G_INDEX]);
55  r = paddsw(r, convert_table[y][R_INDEX]);
56  a = paddsw(a, convert_table[y][A_INDEX]);
57
58  b >>= 6;
59  g >>= 6;
60  r >>= 6;
61  a >>= 6;
62
63  *reinterpret_cast<uint32*>(rgb_buf) = (packuswb(b) << SK_B32_SHIFT) |
64                                        (packuswb(g) << SK_G32_SHIFT) |
65                                        (packuswb(r) << SK_R32_SHIFT) |
66                                        (packuswb(a) << SK_A32_SHIFT);
67}
68
69static inline void ConvertYUVAToARGB_C(uint8 y,
70                                       uint8 u,
71                                       uint8 v,
72                                       uint8 a,
73                                       uint8* rgb_buf,
74                                       const int16 convert_table[1024][4]) {
75  int b = convert_table[256+u][0];
76  int g = convert_table[256+u][1];
77  int r = convert_table[256+u][2];
78
79  b = paddsw(b, convert_table[512+v][0]);
80  g = paddsw(g, convert_table[512+v][1]);
81  r = paddsw(r, convert_table[512+v][2]);
82
83  b = paddsw(b, convert_table[y][0]);
84  g = paddsw(g, convert_table[y][1]);
85  r = paddsw(r, convert_table[y][2]);
86
87  b >>= 6;
88  g >>= 6;
89  r >>= 6;
90
91  b = packuswb(b) * a >> 8;
92  g = packuswb(g) * a >> 8;
93  r = packuswb(r) * a >> 8;
94
95  *reinterpret_cast<uint32*>(rgb_buf) = (b << SK_B32_SHIFT) |
96                                        (g << SK_G32_SHIFT) |
97                                        (r << SK_R32_SHIFT) |
98                                        (a << SK_A32_SHIFT);
99}
100
101void ConvertYUVToRGB32Row_C(const uint8* y_buf,
102                            const uint8* u_buf,
103                            const uint8* v_buf,
104                            uint8* rgb_buf,
105                            ptrdiff_t width,
106                            const int16 convert_table[1024][4]) {
107  for (int x = 0; x < width; x += 2) {
108    uint8 u = u_buf[x >> 1];
109    uint8 v = v_buf[x >> 1];
110    uint8 y0 = y_buf[x];
111    ConvertYUVToRGB32_C(y0, u, v, rgb_buf, convert_table);
112    if ((x + 1) < width) {
113      uint8 y1 = y_buf[x + 1];
114      ConvertYUVToRGB32_C(y1, u, v, rgb_buf + 4, convert_table);
115    }
116    rgb_buf += 8;  // Advance 2 pixels.
117  }
118}
119
120void ConvertYUVAToARGBRow_C(const uint8* y_buf,
121                            const uint8* u_buf,
122                            const uint8* v_buf,
123                            const uint8* a_buf,
124                            uint8* rgba_buf,
125                            ptrdiff_t width,
126                            const int16 convert_table[1024][4]) {
127  for (int x = 0; x < width; x += 2) {
128    uint8 u = u_buf[x >> 1];
129    uint8 v = v_buf[x >> 1];
130    uint8 y0 = y_buf[x];
131    uint8 a0 = a_buf[x];
132    ConvertYUVAToARGB_C(y0, u, v, a0, rgba_buf, convert_table);
133    if ((x + 1) < width) {
134      uint8 y1 = y_buf[x + 1];
135      uint8 a1 = a_buf[x + 1];
136      ConvertYUVAToARGB_C(y1, u, v, a1, rgba_buf + 4, convert_table);
137    }
138    rgba_buf += 8;  // Advance 2 pixels.
139  }
140}
141
142// 16.16 fixed point is used.  A shift by 16 isolates the integer.
143// A shift by 17 is used to further subsample the chrominence channels.
144// & 0xffff isolates the fixed point fraction.  >> 2 to get the upper 2 bits,
145// for 1/65536 pixel accurate interpolation.
146void ScaleYUVToRGB32Row_C(const uint8* y_buf,
147                          const uint8* u_buf,
148                          const uint8* v_buf,
149                          uint8* rgb_buf,
150                          ptrdiff_t width,
151                          ptrdiff_t source_dx,
152                          const int16 convert_table[1024][4]) {
153  int x = 0;
154  for (int i = 0; i < width; i += 2) {
155    int y = y_buf[x >> 16];
156    int u = u_buf[(x >> 17)];
157    int v = v_buf[(x >> 17)];
158    ConvertYUVToRGB32_C(y, u, v, rgb_buf, convert_table);
159    x += source_dx;
160    if ((i + 1) < width) {
161      y = y_buf[x >> 16];
162      ConvertYUVToRGB32_C(y, u, v, rgb_buf+4, convert_table);
163      x += source_dx;
164    }
165    rgb_buf += 8;
166  }
167}
168
169void LinearScaleYUVToRGB32Row_C(const uint8* y_buf,
170                                const uint8* u_buf,
171                                const uint8* v_buf,
172                                uint8* rgb_buf,
173                                ptrdiff_t width,
174                                ptrdiff_t source_dx,
175                                const int16 convert_table[1024][4]) {
176  // Avoid point-sampling for down-scaling by > 2:1.
177  int source_x = 0;
178  if (source_dx >= 0x20000)
179    source_x += 0x8000;
180  LinearScaleYUVToRGB32RowWithRange_C(y_buf, u_buf, v_buf, rgb_buf, width,
181                                      source_x, source_dx, convert_table);
182}
183
184void LinearScaleYUVToRGB32RowWithRange_C(const uint8* y_buf,
185                                         const uint8* u_buf,
186                                         const uint8* v_buf,
187                                         uint8* rgb_buf,
188                                         int dest_width,
189                                         int x,
190                                         int source_dx,
191                                         const int16 convert_table[1024][4]) {
192  for (int i = 0; i < dest_width; i += 2) {
193    int y0 = y_buf[x >> 16];
194    int y1 = y_buf[(x >> 16) + 1];
195    int u0 = u_buf[(x >> 17)];
196    int u1 = u_buf[(x >> 17) + 1];
197    int v0 = v_buf[(x >> 17)];
198    int v1 = v_buf[(x >> 17) + 1];
199    int y_frac = (x & 65535);
200    int uv_frac = ((x >> 1) & 65535);
201    int y = (y_frac * y1 + (y_frac ^ 65535) * y0) >> 16;
202    int u = (uv_frac * u1 + (uv_frac ^ 65535) * u0) >> 16;
203    int v = (uv_frac * v1 + (uv_frac ^ 65535) * v0) >> 16;
204    ConvertYUVToRGB32_C(y, u, v, rgb_buf, convert_table);
205    x += source_dx;
206    if ((i + 1) < dest_width) {
207      y0 = y_buf[x >> 16];
208      y1 = y_buf[(x >> 16) + 1];
209      y_frac = (x & 65535);
210      y = (y_frac * y1 + (y_frac ^ 65535) * y0) >> 16;
211      ConvertYUVToRGB32_C(y, u, v, rgb_buf+4, convert_table);
212      x += source_dx;
213    }
214    rgb_buf += 8;
215  }
216}
217
218void ConvertYUVToRGB32_C(const uint8* yplane,
219                         const uint8* uplane,
220                         const uint8* vplane,
221                         uint8* rgbframe,
222                         int width,
223                         int height,
224                         int ystride,
225                         int uvstride,
226                         int rgbstride,
227                         YUVType yuv_type) {
228  unsigned int y_shift = GetVerticalShift(yuv_type);
229  for (int y = 0; y < height; ++y) {
230    uint8* rgb_row = rgbframe + y * rgbstride;
231    const uint8* y_ptr = yplane + y * ystride;
232    const uint8* u_ptr = uplane + (y >> y_shift) * uvstride;
233    const uint8* v_ptr = vplane + (y >> y_shift) * uvstride;
234
235    ConvertYUVToRGB32Row_C(y_ptr,
236                           u_ptr,
237                           v_ptr,
238                           rgb_row,
239                           width,
240                           GetLookupTable(yuv_type));
241  }
242}
243
244void ConvertYUVAToARGB_C(const uint8* yplane,
245                         const uint8* uplane,
246                         const uint8* vplane,
247                         const uint8* aplane,
248                         uint8* rgbaframe,
249                         int width,
250                         int height,
251                         int ystride,
252                         int uvstride,
253                         int astride,
254                         int rgbastride,
255                         YUVType yuv_type) {
256  unsigned int y_shift = yuv_type;
257  for (int y = 0; y < height; y++) {
258    uint8* rgba_row = rgbaframe + y * rgbastride;
259    const uint8* y_ptr = yplane + y * ystride;
260    const uint8* u_ptr = uplane + (y >> y_shift) * uvstride;
261    const uint8* v_ptr = vplane + (y >> y_shift) * uvstride;
262    const uint8* a_ptr = aplane + y * astride;
263
264    ConvertYUVAToARGBRow_C(y_ptr,
265                           u_ptr,
266                           v_ptr,
267                           a_ptr,
268                           rgba_row,
269                           width,
270                           GetLookupTable(yuv_type));
271  }
272}
273
274}  // namespace media
275