Imported Debian version 2.5.2~trusty
[deb_ffmpeg.git] / ffmpeg / libswscale / bayer_template.c
CommitLineData
2ba45a60
DM
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
127static 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) {
131 BAYER_TO_RGB24_COPY
132 src += 2 * BAYER_SIZEOF;
133 dst += 6;
134 }
135}
136
137static 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
141 BAYER_TO_RGB24_COPY
142 src += 2 * BAYER_SIZEOF;
143 dst += 6;
144
145 for (i = 2 ; i < width - 2; i+= 2) {
146 BAYER_TO_RGB24_INTERPOLATE
147 src += 2 * BAYER_SIZEOF;
148 dst += 6;
149 }
150
151 if (width > 2) {
152 BAYER_TO_RGB24_COPY
153 }
154}
155
156static 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) {
162 BAYER_TO_RGB24_COPY
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
171static 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
177 BAYER_TO_RGB24_COPY
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) {
185 BAYER_TO_RGB24_INTERPOLATE
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) {
194 BAYER_TO_RGB24_COPY
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