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
8 namespace 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
ConvertYUVToRGB32_C(uint8 y,uint8 u,uint8 v,uint8 * rgb_buf,const int16 convert_table[1024][4])38 static 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
ConvertYUVAToARGB_C(uint8 y,uint8 u,uint8 v,uint8 a,uint8 * rgb_buf,const int16 convert_table[1024][4])69 static 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
ConvertYUVToRGB32Row_C(const uint8 * y_buf,const uint8 * u_buf,const uint8 * v_buf,uint8 * rgb_buf,ptrdiff_t width,const int16 convert_table[1024][4])101 void 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
ConvertYUVAToARGBRow_C(const uint8 * y_buf,const uint8 * u_buf,const uint8 * v_buf,const uint8 * a_buf,uint8 * rgba_buf,ptrdiff_t width,const int16 convert_table[1024][4])120 void 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.
ScaleYUVToRGB32Row_C(const uint8 * y_buf,const uint8 * u_buf,const uint8 * v_buf,uint8 * rgb_buf,ptrdiff_t width,ptrdiff_t source_dx,const int16 convert_table[1024][4])146 void 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
LinearScaleYUVToRGB32Row_C(const uint8 * y_buf,const uint8 * u_buf,const uint8 * v_buf,uint8 * rgb_buf,ptrdiff_t width,ptrdiff_t source_dx,const int16 convert_table[1024][4])169 void 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
LinearScaleYUVToRGB32RowWithRange_C(const uint8 * y_buf,const uint8 * u_buf,const uint8 * v_buf,uint8 * rgb_buf,int dest_width,int x,int source_dx,const int16 convert_table[1024][4])184 void 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
ConvertYUVToRGB32_C(const uint8 * yplane,const uint8 * uplane,const uint8 * vplane,uint8 * rgbframe,int width,int height,int ystride,int uvstride,int rgbstride,YUVType yuv_type)218 void 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
ConvertYUVAToARGB_C(const uint8 * yplane,const uint8 * uplane,const uint8 * vplane,const uint8 * aplane,uint8 * rgbaframe,int width,int height,int ystride,int uvstride,int astride,int rgbastride,YUVType yuv_type)244 void 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