1 /*
2 * Bayer-to-RGB/YV12 template
3 * Copyright (c) 2011-2014 Peter Ross <pross@xvid.org>
4 *
5 * This file is part of FFmpeg.
6 *
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22 #if defined(BAYER_BGGR) || defined(BAYER_GBRG)
23 #define BAYER_R 0
24 #define BAYER_G 1
25 #define BAYER_B 2
26 #endif
27 #if defined(BAYER_RGGB) || defined(BAYER_GRBG)
28 #define BAYER_R 2
29 #define BAYER_G 1
30 #define BAYER_B 0
31 #endif
32
33 #if defined(BAYER_8)
34 #define BAYER_READ(x) (x)
35 #define BAYER_SIZEOF 1
36 #define BAYER_SHIFT 0
37 #endif
38 #if defined(BAYER_16LE)
39 #define BAYER_READ(x) AV_RL16(&(x))
40 #define BAYER_SIZEOF 2
41 #define BAYER_SHIFT 8
42 #endif
43 #if defined(BAYER_16BE)
44 #define BAYER_READ(x) AV_RB16(&(x))
45 #define BAYER_SIZEOF 2
46 #define BAYER_SHIFT 8
47 #endif
48
49 #define S(y, x) BAYER_READ(src[(y)*src_stride + BAYER_SIZEOF*(x)])
50 #define T(y, x) (unsigned int)S(y, x)
51 #define R(y, x) dst[(y)*dst_stride + (x)*3 + BAYER_R]
52 #define G(y, x) dst[(y)*dst_stride + (x)*3 + BAYER_G]
53 #define B(y, x) dst[(y)*dst_stride + (x)*3 + BAYER_B]
54
55 #if defined(BAYER_BGGR) || defined(BAYER_RGGB)
56 #define BAYER_TO_RGB24_COPY \
57 R(0, 0) = \
58 R(0, 1) = \
59 R(1, 1) = \
60 R(1, 0) = S(1, 1) >> BAYER_SHIFT; \
61 \
62 G(0, 1) = S(0, 1) >> BAYER_SHIFT; \
63 G(0, 0) = \
64 G(1, 1) = (T(0, 1) + T(1, 0)) >> (1 + BAYER_SHIFT); \
65 G(1, 0) = S(1, 0) >> BAYER_SHIFT; \
66 \
67 B(1, 1) = \
68 B(0, 0) = \
69 B(0, 1) = \
70 B(1, 0) = S(0, 0) >> BAYER_SHIFT;
71 #define BAYER_TO_RGB24_INTERPOLATE \
72 R(0, 0) = (T(-1, -1) + T(-1, 1) + T(1, -1) + T(1, 1)) >> (2 + BAYER_SHIFT); \
73 G(0, 0) = (T(-1, 0) + T( 0, -1) + T(0, 1) + T(1, 0)) >> (2 + BAYER_SHIFT); \
74 B(0, 0) = S(0, 0) >> BAYER_SHIFT; \
75 \
76 R(0, 1) = (T(-1, 1) + T(1, 1)) >> (1 + BAYER_SHIFT); \
77 G(0, 1) = S(0, 1) >> BAYER_SHIFT; \
78 B(0, 1) = (T(0, 0) + T(0, 2)) >> (1 + BAYER_SHIFT); \
79 \
80 R(1, 0) = (T(1, -1) + T(1, 1)) >> (1 + BAYER_SHIFT); \
81 G(1, 0) = S(1, 0) >> BAYER_SHIFT; \
82 B(1, 0) = (T(0, 0) + T(2, 0)) >> (1 + BAYER_SHIFT); \
83 \
84 R(1, 1) = S(1, 1) >> BAYER_SHIFT; \
85 G(1, 1) = (T(0, 1) + T(1, 0) + T(1, 2) + T(2, 1)) >> (2 + BAYER_SHIFT); \
86 B(1, 1) = (T(0, 0) + T(0, 2) + T(2, 0) + T(2, 2)) >> (2 + BAYER_SHIFT);
87 #else
88 #define BAYER_TO_RGB24_COPY \
89 R(0, 0) = \
90 R(0, 1) = \
91 R(1, 1) = \
92 R(1, 0) = S(1, 0) >> BAYER_SHIFT; \
93 \
94 G(0, 0) = S(0, 0) >> BAYER_SHIFT; \
95 G(1, 1) = S(1, 1) >> BAYER_SHIFT; \
96 G(0, 1) = \
97 G(1, 0) = (T(0, 0) + T(1, 1)) >> (1 + BAYER_SHIFT); \
98 \
99 B(1, 1) = \
100 B(0, 0) = \
101 B(0, 1) = \
102 B(1, 0) = S(0, 1) >> BAYER_SHIFT;
103 #define BAYER_TO_RGB24_INTERPOLATE \
104 R(0, 0) = (T(-1, 0) + T(1, 0)) >> (1 + BAYER_SHIFT); \
105 G(0, 0) = S(0, 0) >> BAYER_SHIFT; \
106 B(0, 0) = (T(0, -1) + T(0, 1)) >> (1 + BAYER_SHIFT); \
107 \
108 R(0, 1) = (T(-1, 0) + T(-1, 2) + T(1, 0) + T(1, 2)) >> (2 + BAYER_SHIFT); \
109 G(0, 1) = (T(-1, 1) + T(0, 0) + T(0, 2) + T(1, 1)) >> (2 + BAYER_SHIFT); \
110 B(0, 1) = S(0, 1) >> BAYER_SHIFT; \
111 \
112 R(1, 0) = S(1, 0) >> BAYER_SHIFT; \
113 G(1, 0) = (T(0, 0) + T(1, -1) + T(1, 1) + T(2, 0)) >> (2 + BAYER_SHIFT); \
114 B(1, 0) = (T(0, -1) + T(0, 1) + T(2, -1) + T(2, 1)) >> (2 + BAYER_SHIFT); \
115 \
116 R(1, 1) = (T(1, 0) + T(1, 2)) >> (1 + BAYER_SHIFT); \
117 G(1, 1) = S(1, 1) >> BAYER_SHIFT; \
118 B(1, 1) = (T(0, 1) + T(2, 1)) >> (1 + BAYER_SHIFT);
119 #endif
120
121 #if defined(BAYER_BGGR) || defined(BAYER_RGGB)
122 #define BAYER_TO_RGB48_COPY \
123 R(0, 0) = \
124 R(0, 1) = \
125 R(1, 1) = \
126 R(1, 0) = S(1, 1); \
127 \
128 G(0, 1) = S(0, 1); \
129 G(0, 0) = \
130 G(1, 1) = (T(0, 1) + T(1, 0)) >> 1; \
131 G(1, 0) = S(1, 0); \
132 \
133 B(1, 1) = \
134 B(0, 0) = \
135 B(0, 1) = \
136 B(1, 0) = S(0, 0);
137 #define BAYER_TO_RGB48_INTERPOLATE \
138 R(0, 0) = (T(-1, -1) + T(-1, 1) + T(1, -1) + T(1, 1)) >> 2; \
139 G(0, 0) = (T(-1, 0) + T( 0, -1) + T(0, 1) + T(1, 0)) >> 2; \
140 B(0, 0) = S(0, 0); \
141 \
142 R(0, 1) = (T(-1, 1) + T(1, 1)) >> 1; \
143 G(0, 1) = S(0, 1); \
144 B(0, 1) = (T(0, 0) + T(0, 2)) >> 1; \
145 \
146 R(1, 0) = (T(1, -1) + T(1, 1)) >> 1; \
147 G(1, 0) = S(1, 0); \
148 B(1, 0) = (T(0, 0) + T(2, 0)) >> 1; \
149 \
150 R(1, 1) = S(1, 1); \
151 G(1, 1) = (T(0, 1) + T(1, 0) + T(1, 2) + T(2, 1)) >> 2; \
152 B(1, 1) = (T(0, 0) + T(0, 2) + T(2, 0) + T(2, 2)) >> 2;
153 #else
154 #define BAYER_TO_RGB48_COPY \
155 R(0, 0) = \
156 R(0, 1) = \
157 R(1, 1) = \
158 R(1, 0) = S(1, 0); \
159 \
160 G(0, 0) = S(0, 0); \
161 G(1, 1) = S(1, 1); \
162 G(0, 1) = \
163 G(1, 0) = (T(0, 0) + T(1, 1)) >> 1; \
164 \
165 B(1, 1) = \
166 B(0, 0) = \
167 B(0, 1) = \
168 B(1, 0) = S(0, 1);
169 #define BAYER_TO_RGB48_INTERPOLATE \
170 R(0, 0) = (T(-1, 0) + T(1, 0)) >> 1; \
171 G(0, 0) = S(0, 0); \
172 B(0, 0) = (T(0, -1) + T(0, 1)) >> 1; \
173 \
174 R(0, 1) = (T(-1, 0) + T(-1, 2) + T(1, 0) + T(1, 2)) >> 2; \
175 G(0, 1) = (T(-1, 1) + T(0, 0) + T(0, 2) + T(1, 1)) >> 2; \
176 B(0, 1) = S(0, 1); \
177 \
178 R(1, 0) = S(1, 0); \
179 G(1, 0) = (T(0, 0) + T(1, -1) + T(1, 1) + T(2, 0)) >> 2; \
180 B(1, 0) = (T(0, -1) + T(0, 1) + T(2, -1) + T(2, 1)) >> 2; \
181 \
182 R(1, 1) = (T(1, 0) + T(1, 2)) >> 1; \
183 G(1, 1) = S(1, 1); \
184 B(1, 1) = (T(0, 1) + T(2, 1)) >> 1;
185 #endif
186
187 /**
188 * invoke ff_rgb24toyv12 for 2x2 pixels
189 */
190 #define rgb24toyv12_2x2(src, dstY, dstU, dstV, luma_stride, src_stride, rgb2yuv) \
191 ff_rgb24toyv12(src, dstY, dstV, dstU, 2, 2, luma_stride, 0, src_stride, rgb2yuv)
192
BAYER_RENAME(rgb24_copy)193 static void BAYER_RENAME(rgb24_copy)(const uint8_t *src, int src_stride, uint8_t *dst, int dst_stride, int width)
194 {
195 int i;
196 for (i = 0 ; i < width; i+= 2) {
197 BAYER_TO_RGB24_COPY
198 src += 2 * BAYER_SIZEOF;
199 dst += 6;
200 }
201 }
202
BAYER_RENAME(rgb24_interpolate)203 static void BAYER_RENAME(rgb24_interpolate)(const uint8_t *src, int src_stride, uint8_t *dst, int dst_stride, int width)
204 {
205 int i;
206
207 BAYER_TO_RGB24_COPY
208 src += 2 * BAYER_SIZEOF;
209 dst += 6;
210
211 for (i = 2 ; i < width - 2; i+= 2) {
212 BAYER_TO_RGB24_INTERPOLATE
213 src += 2 * BAYER_SIZEOF;
214 dst += 6;
215 }
216
217 if (width > 2) {
218 BAYER_TO_RGB24_COPY
219 }
220 }
221
BAYER_RENAME(rgb48_copy)222 static void BAYER_RENAME(rgb48_copy)(const uint8_t *src, int src_stride, uint8_t *ddst, int dst_stride, int width)
223 {
224 uint16_t *dst = (uint16_t *)ddst;
225 int i;
226
227 dst_stride /= 2;
228 for (i = 0 ; i < width; i+= 2) {
229 BAYER_TO_RGB48_COPY
230 src += 2 * BAYER_SIZEOF;
231 dst += 6;
232 }
233 }
234
BAYER_RENAME(rgb48_interpolate)235 static void BAYER_RENAME(rgb48_interpolate)(const uint8_t *src, int src_stride, uint8_t *ddst, int dst_stride, int width)
236 {
237 uint16_t *dst = (uint16_t *)ddst;
238 int i;
239
240 dst_stride /= 2;
241 BAYER_TO_RGB48_COPY
242 src += 2 * BAYER_SIZEOF;
243 dst += 6;
244
245 for (i = 2 ; i < width - 2; i+= 2) {
246 BAYER_TO_RGB48_INTERPOLATE
247 src += 2 * BAYER_SIZEOF;
248 dst += 6;
249 }
250
251 if (width > 2) {
252 BAYER_TO_RGB48_COPY
253 }
254 }
255
BAYER_RENAME(yv12_copy)256 static void BAYER_RENAME(yv12_copy)(const uint8_t *src, int src_stride, uint8_t *dstY, uint8_t *dstU, uint8_t *dstV, int luma_stride, int width, int32_t *rgb2yuv)
257 {
258 uint8_t dst[12];
259 const int dst_stride = 6;
260 int i;
261 for (i = 0 ; i < width; i+= 2) {
262 BAYER_TO_RGB24_COPY
263 rgb24toyv12_2x2(dst, dstY, dstU, dstV, luma_stride, dst_stride, rgb2yuv);
264 src += 2 * BAYER_SIZEOF;
265 dstY += 2;
266 dstU++;
267 dstV++;
268 }
269 }
270
BAYER_RENAME(yv12_interpolate)271 static void BAYER_RENAME(yv12_interpolate)(const uint8_t *src, int src_stride, uint8_t *dstY, uint8_t *dstU, uint8_t *dstV, int luma_stride, int width, int32_t *rgb2yuv)
272 {
273 uint8_t dst[12];
274 const int dst_stride = 6;
275 int i;
276
277 BAYER_TO_RGB24_COPY
278 rgb24toyv12_2x2(dst, dstY, dstU, dstV, luma_stride, dst_stride, rgb2yuv);
279 src += 2 * BAYER_SIZEOF;
280 dstY += 2;
281 dstU++;
282 dstV++;
283
284 for (i = 2 ; i < width - 2; i+= 2) {
285 BAYER_TO_RGB24_INTERPOLATE
286 rgb24toyv12_2x2(dst, dstY, dstU, dstV, luma_stride, dst_stride, rgb2yuv);
287 src += 2 * BAYER_SIZEOF;
288 dstY += 2;
289 dstU++;
290 dstV++;
291 }
292
293 if (width > 2) {
294 BAYER_TO_RGB24_COPY
295 rgb24toyv12_2x2(dst, dstY, dstU, dstV, luma_stride, dst_stride, rgb2yuv);
296 }
297 }
298
299 #undef S
300 #undef T
301 #undef R
302 #undef G
303 #undef B
304 #undef BAYER_TO_RGB24_COPY
305 #undef BAYER_TO_RGB24_INTERPOLATE
306 #undef BAYER_TO_RGB48_COPY
307 #undef BAYER_TO_RGB48_INTERPOLATE
308
309 #undef BAYER_RENAME
310
311 #undef BAYER_R
312 #undef BAYER_G
313 #undef BAYER_B
314 #undef BAYER_READ
315 #undef BAYER_SIZEOF
316 #undef BAYER_SHIFT
317
318 #if defined(BAYER_BGGR)
319 #undef BAYER_BGGR
320 #endif
321 #if defined(BAYER_RGGB)
322 #undef BAYER_RGGB
323 #endif
324 #if defined(BAYER_GBRG)
325 #undef BAYER_GBRG
326 #endif
327 #if defined(BAYER_GRBG)
328 #undef BAYER_GRBG
329 #endif
330 #if defined(BAYER_8)
331 #undef BAYER_8
332 #endif
333 #if defined(BAYER_16LE)
334 #undef BAYER_16LE
335 #endif
336 #if defined(BAYER_16BE)
337 #undef BAYER_16BE
338 #endif
339