1 // Copyright 2010 Google Inc. All Rights Reserved.
2 //
3 // Use of this source code is governed by a BSD-style license
4 // that can be found in the COPYING file in the root of the source
5 // tree. An additional intellectual property rights grant can be found
6 // in the file PATENTS. All contributing project authors may
7 // be found in the AUTHORS file in the root of the source tree.
8 // -----------------------------------------------------------------------------
9 //
10 // YUV->RGB conversion functions
11 //
12 // Author: Skal (pascal.massimino@gmail.com)
13
14 #include "src/dsp/yuv.h"
15
16 #include <assert.h>
17 #include <stdlib.h>
18
19 //-----------------------------------------------------------------------------
20 // Plain-C version
21
22 #define ROW_FUNC(FUNC_NAME, FUNC, XSTEP) \
23 static void FUNC_NAME(const uint8_t* y, \
24 const uint8_t* u, const uint8_t* v, \
25 uint8_t* dst, int len) { \
26 const uint8_t* const end = dst + (len & ~1) * (XSTEP); \
27 while (dst != end) { \
28 FUNC(y[0], u[0], v[0], dst); \
29 FUNC(y[1], u[0], v[0], dst + (XSTEP)); \
30 y += 2; \
31 ++u; \
32 ++v; \
33 dst += 2 * (XSTEP); \
34 } \
35 if (len & 1) { \
36 FUNC(y[0], u[0], v[0], dst); \
37 } \
38 } \
39
40 // All variants implemented.
41 ROW_FUNC(YuvToRgbRow, VP8YuvToRgb, 3)
42 ROW_FUNC(YuvToBgrRow, VP8YuvToBgr, 3)
43 ROW_FUNC(YuvToRgbaRow, VP8YuvToRgba, 4)
44 ROW_FUNC(YuvToBgraRow, VP8YuvToBgra, 4)
45 ROW_FUNC(YuvToArgbRow, VP8YuvToArgb, 4)
46 ROW_FUNC(YuvToRgba4444Row, VP8YuvToRgba4444, 2)
47 ROW_FUNC(YuvToRgb565Row, VP8YuvToRgb565, 2)
48
49 #undef ROW_FUNC
50
51 // Main call for processing a plane with a WebPSamplerRowFunc function:
WebPSamplerProcessPlane(const uint8_t * y,int y_stride,const uint8_t * u,const uint8_t * v,int uv_stride,uint8_t * dst,int dst_stride,int width,int height,WebPSamplerRowFunc func)52 void WebPSamplerProcessPlane(const uint8_t* y, int y_stride,
53 const uint8_t* u, const uint8_t* v, int uv_stride,
54 uint8_t* dst, int dst_stride,
55 int width, int height, WebPSamplerRowFunc func) {
56 int j;
57 for (j = 0; j < height; ++j) {
58 func(y, u, v, dst, width);
59 y += y_stride;
60 if (j & 1) {
61 u += uv_stride;
62 v += uv_stride;
63 }
64 dst += dst_stride;
65 }
66 }
67
68 //-----------------------------------------------------------------------------
69 // Main call
70
71 WebPSamplerRowFunc WebPSamplers[MODE_LAST];
72
73 extern void WebPInitSamplersSSE2(void);
74 extern void WebPInitSamplersMIPS32(void);
75 extern void WebPInitSamplersMIPSdspR2(void);
76
77 static volatile VP8CPUInfo yuv_last_cpuinfo_used =
78 (VP8CPUInfo)&yuv_last_cpuinfo_used;
79
WebPInitSamplers(void)80 WEBP_TSAN_IGNORE_FUNCTION void WebPInitSamplers(void) {
81 if (yuv_last_cpuinfo_used == VP8GetCPUInfo) return;
82
83 WebPSamplers[MODE_RGB] = YuvToRgbRow;
84 WebPSamplers[MODE_RGBA] = YuvToRgbaRow;
85 WebPSamplers[MODE_BGR] = YuvToBgrRow;
86 WebPSamplers[MODE_BGRA] = YuvToBgraRow;
87 WebPSamplers[MODE_ARGB] = YuvToArgbRow;
88 WebPSamplers[MODE_RGBA_4444] = YuvToRgba4444Row;
89 WebPSamplers[MODE_RGB_565] = YuvToRgb565Row;
90 WebPSamplers[MODE_rgbA] = YuvToRgbaRow;
91 WebPSamplers[MODE_bgrA] = YuvToBgraRow;
92 WebPSamplers[MODE_Argb] = YuvToArgbRow;
93 WebPSamplers[MODE_rgbA_4444] = YuvToRgba4444Row;
94
95 // If defined, use CPUInfo() to overwrite some pointers with faster versions.
96 if (VP8GetCPUInfo != NULL) {
97 #if defined(WEBP_USE_SSE2)
98 if (VP8GetCPUInfo(kSSE2)) {
99 WebPInitSamplersSSE2();
100 }
101 #endif // WEBP_USE_SSE2
102 #if defined(WEBP_USE_MIPS32)
103 if (VP8GetCPUInfo(kMIPS32)) {
104 WebPInitSamplersMIPS32();
105 }
106 #endif // WEBP_USE_MIPS32
107 #if defined(WEBP_USE_MIPS_DSP_R2)
108 if (VP8GetCPUInfo(kMIPSdspR2)) {
109 WebPInitSamplersMIPSdspR2();
110 }
111 #endif // WEBP_USE_MIPS_DSP_R2
112 }
113 yuv_last_cpuinfo_used = VP8GetCPUInfo;
114 }
115
116 //-----------------------------------------------------------------------------
117 // ARGB -> YUV converters
118
ConvertARGBToY_C(const uint32_t * argb,uint8_t * y,int width)119 static void ConvertARGBToY_C(const uint32_t* argb, uint8_t* y, int width) {
120 int i;
121 for (i = 0; i < width; ++i) {
122 const uint32_t p = argb[i];
123 y[i] = VP8RGBToY((p >> 16) & 0xff, (p >> 8) & 0xff, (p >> 0) & 0xff,
124 YUV_HALF);
125 }
126 }
127
WebPConvertARGBToUV_C(const uint32_t * argb,uint8_t * u,uint8_t * v,int src_width,int do_store)128 void WebPConvertARGBToUV_C(const uint32_t* argb, uint8_t* u, uint8_t* v,
129 int src_width, int do_store) {
130 // No rounding. Last pixel is dealt with separately.
131 const int uv_width = src_width >> 1;
132 int i;
133 for (i = 0; i < uv_width; ++i) {
134 const uint32_t v0 = argb[2 * i + 0];
135 const uint32_t v1 = argb[2 * i + 1];
136 // VP8RGBToU/V expects four accumulated pixels. Hence we need to
137 // scale r/g/b value by a factor 2. We just shift v0/v1 one bit less.
138 const int r = ((v0 >> 15) & 0x1fe) + ((v1 >> 15) & 0x1fe);
139 const int g = ((v0 >> 7) & 0x1fe) + ((v1 >> 7) & 0x1fe);
140 const int b = ((v0 << 1) & 0x1fe) + ((v1 << 1) & 0x1fe);
141 const int tmp_u = VP8RGBToU(r, g, b, YUV_HALF << 2);
142 const int tmp_v = VP8RGBToV(r, g, b, YUV_HALF << 2);
143 if (do_store) {
144 u[i] = tmp_u;
145 v[i] = tmp_v;
146 } else {
147 // Approximated average-of-four. But it's an acceptable diff.
148 u[i] = (u[i] + tmp_u + 1) >> 1;
149 v[i] = (v[i] + tmp_v + 1) >> 1;
150 }
151 }
152 if (src_width & 1) { // last pixel
153 const uint32_t v0 = argb[2 * i + 0];
154 const int r = (v0 >> 14) & 0x3fc;
155 const int g = (v0 >> 6) & 0x3fc;
156 const int b = (v0 << 2) & 0x3fc;
157 const int tmp_u = VP8RGBToU(r, g, b, YUV_HALF << 2);
158 const int tmp_v = VP8RGBToV(r, g, b, YUV_HALF << 2);
159 if (do_store) {
160 u[i] = tmp_u;
161 v[i] = tmp_v;
162 } else {
163 u[i] = (u[i] + tmp_u + 1) >> 1;
164 v[i] = (v[i] + tmp_v + 1) >> 1;
165 }
166 }
167 }
168
169 //-----------------------------------------------------------------------------
170
ConvertRGB24ToY_C(const uint8_t * rgb,uint8_t * y,int width)171 static void ConvertRGB24ToY_C(const uint8_t* rgb, uint8_t* y, int width) {
172 int i;
173 for (i = 0; i < width; ++i, rgb += 3) {
174 y[i] = VP8RGBToY(rgb[0], rgb[1], rgb[2], YUV_HALF);
175 }
176 }
177
ConvertBGR24ToY_C(const uint8_t * bgr,uint8_t * y,int width)178 static void ConvertBGR24ToY_C(const uint8_t* bgr, uint8_t* y, int width) {
179 int i;
180 for (i = 0; i < width; ++i, bgr += 3) {
181 y[i] = VP8RGBToY(bgr[2], bgr[1], bgr[0], YUV_HALF);
182 }
183 }
184
WebPConvertRGBA32ToUV_C(const uint16_t * rgb,uint8_t * u,uint8_t * v,int width)185 void WebPConvertRGBA32ToUV_C(const uint16_t* rgb,
186 uint8_t* u, uint8_t* v, int width) {
187 int i;
188 for (i = 0; i < width; i += 1, rgb += 4) {
189 const int r = rgb[0], g = rgb[1], b = rgb[2];
190 u[i] = VP8RGBToU(r, g, b, YUV_HALF << 2);
191 v[i] = VP8RGBToV(r, g, b, YUV_HALF << 2);
192 }
193 }
194
195 //-----------------------------------------------------------------------------
196
197 #if !WEBP_NEON_OMIT_C_CODE
198 #define MAX_Y ((1 << 10) - 1) // 10b precision over 16b-arithmetic
clip_y(int v)199 static uint16_t clip_y(int v) {
200 return (v < 0) ? 0 : (v > MAX_Y) ? MAX_Y : (uint16_t)v;
201 }
202
SharpYUVUpdateY_C(const uint16_t * ref,const uint16_t * src,uint16_t * dst,int len)203 static uint64_t SharpYUVUpdateY_C(const uint16_t* ref, const uint16_t* src,
204 uint16_t* dst, int len) {
205 uint64_t diff = 0;
206 int i;
207 for (i = 0; i < len; ++i) {
208 const int diff_y = ref[i] - src[i];
209 const int new_y = (int)dst[i] + diff_y;
210 dst[i] = clip_y(new_y);
211 diff += (uint64_t)abs(diff_y);
212 }
213 return diff;
214 }
215
SharpYUVUpdateRGB_C(const int16_t * ref,const int16_t * src,int16_t * dst,int len)216 static void SharpYUVUpdateRGB_C(const int16_t* ref, const int16_t* src,
217 int16_t* dst, int len) {
218 int i;
219 for (i = 0; i < len; ++i) {
220 const int diff_uv = ref[i] - src[i];
221 dst[i] += diff_uv;
222 }
223 }
224
SharpYUVFilterRow_C(const int16_t * A,const int16_t * B,int len,const uint16_t * best_y,uint16_t * out)225 static void SharpYUVFilterRow_C(const int16_t* A, const int16_t* B, int len,
226 const uint16_t* best_y, uint16_t* out) {
227 int i;
228 for (i = 0; i < len; ++i, ++A, ++B) {
229 const int v0 = (A[0] * 9 + A[1] * 3 + B[0] * 3 + B[1] + 8) >> 4;
230 const int v1 = (A[1] * 9 + A[0] * 3 + B[1] * 3 + B[0] + 8) >> 4;
231 out[2 * i + 0] = clip_y(best_y[2 * i + 0] + v0);
232 out[2 * i + 1] = clip_y(best_y[2 * i + 1] + v1);
233 }
234 }
235 #endif // !WEBP_NEON_OMIT_C_CODE
236
237 #undef MAX_Y
238
239 //-----------------------------------------------------------------------------
240
241 void (*WebPConvertRGB24ToY)(const uint8_t* rgb, uint8_t* y, int width);
242 void (*WebPConvertBGR24ToY)(const uint8_t* bgr, uint8_t* y, int width);
243 void (*WebPConvertRGBA32ToUV)(const uint16_t* rgb,
244 uint8_t* u, uint8_t* v, int width);
245
246 void (*WebPConvertARGBToY)(const uint32_t* argb, uint8_t* y, int width);
247 void (*WebPConvertARGBToUV)(const uint32_t* argb, uint8_t* u, uint8_t* v,
248 int src_width, int do_store);
249
250 uint64_t (*WebPSharpYUVUpdateY)(const uint16_t* ref, const uint16_t* src,
251 uint16_t* dst, int len);
252 void (*WebPSharpYUVUpdateRGB)(const int16_t* ref, const int16_t* src,
253 int16_t* dst, int len);
254 void (*WebPSharpYUVFilterRow)(const int16_t* A, const int16_t* B, int len,
255 const uint16_t* best_y, uint16_t* out);
256
257 static volatile VP8CPUInfo rgba_to_yuv_last_cpuinfo_used =
258 (VP8CPUInfo)&rgba_to_yuv_last_cpuinfo_used;
259
260 extern void WebPInitConvertARGBToYUVSSE2(void);
261 extern void WebPInitConvertARGBToYUVNEON(void);
262 extern void WebPInitSharpYUVSSE2(void);
263 extern void WebPInitSharpYUVNEON(void);
264
WebPInitConvertARGBToYUV(void)265 WEBP_TSAN_IGNORE_FUNCTION void WebPInitConvertARGBToYUV(void) {
266 if (rgba_to_yuv_last_cpuinfo_used == VP8GetCPUInfo) return;
267
268 WebPConvertARGBToY = ConvertARGBToY_C;
269 WebPConvertARGBToUV = WebPConvertARGBToUV_C;
270
271 WebPConvertRGB24ToY = ConvertRGB24ToY_C;
272 WebPConvertBGR24ToY = ConvertBGR24ToY_C;
273
274 WebPConvertRGBA32ToUV = WebPConvertRGBA32ToUV_C;
275
276 #if !WEBP_NEON_OMIT_C_CODE
277 WebPSharpYUVUpdateY = SharpYUVUpdateY_C;
278 WebPSharpYUVUpdateRGB = SharpYUVUpdateRGB_C;
279 WebPSharpYUVFilterRow = SharpYUVFilterRow_C;
280 #endif
281
282 if (VP8GetCPUInfo != NULL) {
283 #if defined(WEBP_USE_SSE2)
284 if (VP8GetCPUInfo(kSSE2)) {
285 WebPInitConvertARGBToYUVSSE2();
286 WebPInitSharpYUVSSE2();
287 }
288 #endif // WEBP_USE_SSE2
289 }
290
291 #if defined(WEBP_USE_NEON)
292 if (WEBP_NEON_OMIT_C_CODE ||
293 (VP8GetCPUInfo != NULL && VP8GetCPUInfo(kNEON))) {
294 WebPInitConvertARGBToYUVNEON();
295 WebPInitSharpYUVNEON();
296 }
297 #endif // WEBP_USE_NEON
298
299 assert(WebPConvertARGBToY != NULL);
300 assert(WebPConvertARGBToUV != NULL);
301 assert(WebPConvertRGB24ToY != NULL);
302 assert(WebPConvertBGR24ToY != NULL);
303 assert(WebPConvertRGBA32ToUV != NULL);
304 assert(WebPSharpYUVUpdateY != NULL);
305 assert(WebPSharpYUVUpdateRGB != NULL);
306 assert(WebPSharpYUVFilterRow != NULL);
307
308 rgba_to_yuv_last_cpuinfo_used = VP8GetCPUInfo;
309 }
310