FFmpeg
Loading...
Searching...
No Matches
bayer_template.c
Go to the documentation of this file.
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
193static 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
203static 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
222static 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
235static 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
256static 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, const 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
271static 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, const 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
dst
uint8_t ptrdiff_t const uint8_t ptrdiff_t int intptr_t intptr_t int int16_t * dst
Definition dsp.h:87
int32_t
int32_t
Definition audioconvert.c:56
yv12_copy
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, const int32_t *rgb2yuv)
Definition bayer_template.c:256
BAYER_TO_RGB24_COPY
#define BAYER_TO_RGB24_COPY
Definition bayer_template.c:88
yv12_interpolate
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, const int32_t *rgb2yuv)
Definition bayer_template.c:271
rgb48_copy
static void BAYER_RENAME rgb48_copy(const uint8_t *src, int src_stride, uint8_t *ddst, int dst_stride, int width)
Definition bayer_template.c:222
rgb24toyv12_2x2
#define rgb24toyv12_2x2(src, dstY, dstU, dstV, luma_stride, src_stride, rgb2yuv)
invoke ff_rgb24toyv12 for 2x2 pixels
Definition bayer_template.c:190
BAYER_TO_RGB48_INTERPOLATE
#define BAYER_TO_RGB48_INTERPOLATE
Definition bayer_template.c:169
rgb48_interpolate
static void BAYER_RENAME rgb48_interpolate(const uint8_t *src, int src_stride, uint8_t *ddst, int dst_stride, int width)
Definition bayer_template.c:235
rgb24_interpolate
static void BAYER_RENAME rgb24_interpolate(const uint8_t *src, int src_stride, uint8_t *dst, int dst_stride, int width)
Definition bayer_template.c:203
rgb24_copy
static void BAYER_RENAME rgb24_copy(const uint8_t *src, int src_stride, uint8_t *dst, int dst_stride, int width)
Definition bayer_template.c:193
BAYER_TO_RGB48_COPY
#define BAYER_TO_RGB48_COPY
Definition bayer_template.c:154
BAYER_TO_RGB24_INTERPOLATE
#define BAYER_TO_RGB24_INTERPOLATE
Definition bayer_template.c:103
i
#define i(width, name, range_min, range_max)
Definition cbs_h264.c:63
rgb2yuv
static void fn rgb2yuv(uint8_t *_yuv[3], const ptrdiff_t yuv_stride[3], int16_t *rgb[3], ptrdiff_t s, int w, int h, const int16_t rgb2yuv_coeffs[3][3][8], const int16_t yuv_offset[8])
Definition colorspacedsp_template.c:130
BAYER_RENAME
#define BAYER_RENAME(x)
Definition swscale_unscaled.c:1594
src
#define src
Definition vp8dsp.c:248
width
#define width
Definition dsp.h:89
Generated on Mon Aug 17 2026 20:21:13 for FFmpeg by doxygen 1.13.2