FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
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 /**
122  * invoke ff_rgb24toyv12 for 2x2 pixels
123  */
124 #define rgb24toyv12_2x2(src, dstY, dstU, dstV, luma_stride, src_stride, rgb2yuv) \
125  ff_rgb24toyv12(src, dstY, dstV, dstU, 2, 2, luma_stride, 0, src_stride, rgb2yuv)
126 
127 static void BAYER_RENAME(rgb24_copy)(const uint8_t *src, int src_stride, uint8_t *dst, int dst_stride, int width)
128 {
129  int i;
130  for (i = 0 ; i < width; i+= 2) {
132  src += 2 * BAYER_SIZEOF;
133  dst += 6;
134  }
135 }
136 
137 static void BAYER_RENAME(rgb24_interpolate)(const uint8_t *src, int src_stride, uint8_t *dst, int dst_stride, int width)
138 {
139  int i;
140 
142  src += 2 * BAYER_SIZEOF;
143  dst += 6;
144 
145  for (i = 2 ; i < width - 2; i+= 2) {
147  src += 2 * BAYER_SIZEOF;
148  dst += 6;
149  }
150 
151  if (width > 2) {
153  }
154 }
155 
156 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)
157 {
158  uint8_t dst[12];
159  const int dst_stride = 6;
160  int i;
161  for (i = 0 ; i < width; i+= 2) {
163  rgb24toyv12_2x2(dst, dstY, dstU, dstV, luma_stride, dst_stride, rgb2yuv);
164  src += 2 * BAYER_SIZEOF;
165  dstY += 2;
166  dstU++;
167  dstV++;
168  }
169 }
170 
171 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)
172 {
173  uint8_t dst[12];
174  const int dst_stride = 6;
175  int i;
176 
178  rgb24toyv12_2x2(dst, dstY, dstU, dstV, luma_stride, dst_stride, rgb2yuv);
179  src += 2 * BAYER_SIZEOF;
180  dstY += 2;
181  dstU++;
182  dstV++;
183 
184  for (i = 2 ; i < width - 2; i+= 2) {
186  rgb24toyv12_2x2(dst, dstY, dstU, dstV, luma_stride, dst_stride, rgb2yuv);
187  src += 2 * BAYER_SIZEOF;
188  dstY += 2;
189  dstU++;
190  dstV++;
191  }
192 
193  if (width > 2) {
195  rgb24toyv12_2x2(dst, dstY, dstU, dstV, luma_stride, dst_stride, rgb2yuv);
196  }
197 }
198 
199 #undef S
200 #undef T
201 #undef R
202 #undef G
203 #undef B
204 #undef BAYER_TO_RGB24_COPY
205 #undef BAYER_TO_RGB24_INTERPOLATE
206 
207 #undef BAYER_RENAME
208 
209 #undef BAYER_R
210 #undef BAYER_G
211 #undef BAYER_B
212 #undef BAYER_READ
213 #undef BAYER_SIZEOF
214 #undef BAYER_SHIFT
215 
216 #if defined(BAYER_BGGR)
217 #undef BAYER_BGGR
218 #endif
219 #if defined(BAYER_RGGB)
220 #undef BAYER_RGGB
221 #endif
222 #if defined(BAYER_GBRG)
223 #undef BAYER_GBRG
224 #endif
225 #if defined(BAYER_GRBG)
226 #undef BAYER_GRBG
227 #endif
228 #if defined(BAYER_8)
229 #undef BAYER_8
230 #endif
231 #if defined(BAYER_16LE)
232 #undef BAYER_16LE
233 #endif
234 #if defined(BAYER_16BE)
235 #undef BAYER_16BE
236 #endif