00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024 #include <fvutils/color/bayer.h>
00025 #include <fvutils/color/colorspaces.h>
00026 #include <fvutils/color/yuv.h>
00027 #include <fvutils/color/rgbyuv.h>
00028
00029 namespace firevision {
00030 #if 0
00031 }
00032 #endif
00033
00034
00035
00036
00037
00038 #define assign(y, u, v, y1, u1, v1, y2, u2, v2) { \
00039 *y++ = y1; \
00040 *y++ = y2; \
00041 *u++ = ((u1 + u2) >> 1); \
00042 *v++ = ((v1 + v2) >> 1); }
00043
00044
00045 void
00046 bayerGBRG_to_yuv422planar_nearest_neighbour(const unsigned char *bayer, unsigned char *yuv,
00047 unsigned int width, unsigned int height)
00048 {
00049 unsigned char *y = yuv;
00050 unsigned char *u = YUV422_PLANAR_U_PLANE(yuv, width, height);
00051 unsigned char *v = YUV422_PLANAR_V_PLANE(yuv, width, height);
00052 const unsigned char *b = bayer;
00053
00054 int y1, u1, v1, y2, u2, v2;
00055 int t1, t2;
00056
00057 for ( unsigned int h = 0; h < height; h += 2) {
00058
00059
00060 for (unsigned int w = 0; w < width; w += 2) {
00061 t1 = b[width];
00062 t2 = b[1];
00063 RGB2YUV(t1, *b, t2, y1, u1, v1);
00064 ++b;
00065
00066 t1 = b[width - 1];
00067 t2 = b[-1];
00068 RGB2YUV(t1, t2, *b, y2, u2, v2);
00069 ++b;
00070
00071 assign(y, u, v, y1, u1, v1, y2, u2, v2);
00072 }
00073
00074
00075 for (unsigned int w = 0; w < width; w += 2) {
00076 t1 = b[1];
00077 t2 = b[-width+1];
00078 RGB2YUV(*b, t1, t2, y1, u1, v1);
00079 ++b;
00080
00081 t1 = b[-1];
00082 t2 = b[-width];
00083 RGB2YUV(t1, *b, t2, y2, u2, v2);
00084 ++b;
00085
00086 assign(y, u, v, y1, u1, v1, y2, u2, v2);
00087 }
00088 }
00089 }
00090
00091
00092
00093 void
00094 bayerGBRG_to_yuv422planar_bilinear(const unsigned char *bayer, unsigned char *yuv,
00095 unsigned int width, unsigned int height)
00096 {
00097 unsigned char *y = yuv;
00098 unsigned char *u = YUV422_PLANAR_U_PLANE(yuv, width, height);
00099 unsigned char *v = YUV422_PLANAR_V_PLANE(yuv, width, height);
00100 const unsigned char *bf = bayer;
00101
00102 int y1, u1, v1, y2, u2, v2;
00103 int r, g, b;
00104
00105
00106
00107
00108 RGB2YUV(bf[width], *bf, bf[1], y1, u1, v1);
00109 ++bf;
00110
00111 r = (bf[width - 1] + bf[width + 1]) >> 1;
00112
00113
00114
00115 g = (bf[-1] + bf[1]) >> 1;
00116 RGB2YUV(r, g, *bf, y2, u2, v2);
00117 ++bf;
00118
00119 assign(y, u, v, y1, u1, v1, y2, u2, v2);
00120
00121
00122 for (unsigned int w = 2; w < width - 2; w += 2) {
00123 b = (bf[-1] + bf[1]) >> 1;
00124 RGB2YUV(bf[width], *bf, b, y1, u1, v1);
00125 ++bf;
00126
00127 r = (bf[width - 1] + bf[width + 1]) >> 1;
00128
00129
00130
00131 g = (bf[-1] + bf[1]) >> 1;
00132 RGB2YUV(r, g, *bf, y2, u2, v2);
00133 ++bf;
00134
00135 assign(y, u, v, y1, u1, v1, y2, u2, v2);
00136 }
00137
00138
00139 b = (bf[-1] + bf[1]) >> 1;
00140 RGB2YUV(bf[width], *bf, b, y1, u1, v1);
00141 ++bf;
00142
00143 g = (bf[-1] + bf[width]) >> 1;
00144 RGB2YUV(bf[width - 1], g, *bf, y2, u2, v2);
00145 ++bf;
00146
00147 assign(y, u, v, y1, u1, v1, y2, u2, v2);
00148
00149 for ( unsigned int h = 1; h < height - 1; h += 2) {
00150
00151
00152
00153
00154 g = (bf[-width] + bf[1]) >> 1;
00155 b = (bf[width-1] + bf[width+1]) >> 1;
00156 RGB2YUV(*bf, g, b, y1, u1, v1);
00157 ++bf;
00158
00159 r = (bf[-1] + bf[1]) >> 1;
00160 b = (bf[-width] + bf[width]) >> 1;
00161 RGB2YUV(r, *bf, g, y2, u2, v2);
00162 ++bf;
00163
00164 assign(y, u, v, y1, u1, v1, y2, u2, v2);
00165
00166 for (unsigned int w = 2; w < width - 2; w += 2) {
00167 g = (bf[-width] + bf[1] + bf[width] + bf[-1]) >> 2;
00168 b = (bf[-width-1] + bf[-width+1] + bf[width-1] + bf[width+1]) >> 2;
00169 RGB2YUV(*bf, g, b, y1, u1, v1);
00170 ++bf;
00171
00172 r = (bf[-1] + bf[1]) >> 1;
00173 b = (bf[-width] + bf[width]) >> 1;
00174 RGB2YUV(r, *bf, b, y2, u2, v2);
00175 ++bf;
00176
00177 assign(y, u, v, y1, u1, v1, y2, u2, v2);
00178 }
00179
00180 g = (bf[-width] + bf[1] + bf[width] + bf[-1]) >> 2;
00181 b = (bf[-width-1] + bf[-width+1] + bf[width-1] + bf[width+1]) >> 2;
00182 RGB2YUV(*bf, g, b, y1, u1, v1);
00183 ++bf;
00184
00185 b = (bf[-width] + bf[width]) >> 1;
00186 RGB2YUV(bf[-1], *bf, g, y2, u2, v2);
00187 ++bf;
00188
00189 assign(y, u, v, y1, u1, v1, y2, u2, v2);
00190
00191
00192
00193
00194 r = (bf[width] + bf[-width]) >> 1;
00195 RGB2YUV(r, *bf, bf[1], y1, u1, v1);
00196 ++bf;
00197
00198 r = (bf[-width-1] + bf[-width+1] + bf[width - 1] + bf[width + 1]) >> 2;
00199 g = (bf[-width] + bf[1] + bf[width] + bf[-1]) >> 2;
00200 RGB2YUV(r, g, *bf, y2, u2, v2);
00201 ++bf;
00202
00203 assign(y, u, v, y1, u1, v1, y2, u2, v2);
00204
00205 for (unsigned int w = 2; w < width - 2; w += 2) {
00206 r = (bf[width] + bf[-width]) >> 1;
00207 b = (bf[-1] + bf[1]) >> 1;
00208 RGB2YUV(r, *bf, b, y1, u1, v1);
00209 ++bf;
00210
00211 r = (bf[-width-1] + bf[-width+1] + bf[width-1] + bf[width+1]) >> 2;
00212 g = (bf[-width] + bf[1] + bf[width] + bf[-1]) >> 2;
00213 RGB2YUV(r, g, *bf, y2, u2, v2);
00214 ++bf;
00215
00216 assign(y, u, v, y1, u1, v1, y2, u2, v2);
00217 }
00218
00219 r = (bf[width] + bf[-width]) >> 1;
00220 b = (bf[-1] + bf[1]) >> 1;
00221 RGB2YUV(r, *bf, b, y1, u1, v1);
00222 ++bf;
00223
00224 r = (bf[-width-1] + bf[width-1]) >> 1;
00225
00226
00227 g = (bf[-width] + bf[-1]) >> 1;
00228 RGB2YUV(r, g, *bf, y2, u2, v2);
00229 ++bf;
00230
00231 assign(y, u, v, y1, u1, v1, y2, u2, v2);
00232 }
00233
00234
00235
00236
00237 g = (bf[-width] + bf[1]) >> 1;
00238 b = bf[-width+1];
00239 RGB2YUV(*bf, g, b, y1, u1, v1);
00240 ++bf;
00241
00242 r = (bf[-1] + bf[1]) >> 1;
00243 b = bf[-width];
00244 RGB2YUV(r, g, *bf, y2, u2, v2);
00245 ++bf;
00246
00247 assign(y, u, v, y1, u1, v1, y2, u2, v2);
00248
00249 for (unsigned int w = 2; w < width - 2; w += 2) {
00250
00251
00252 g = (bf[-width] + bf[-1]) >> 1;
00253 b = (bf[-width-1] + bf[-width+1]) >> 1;
00254 RGB2YUV(*bf, g, b, y1, u1, v1);
00255 ++bf;
00256
00257 r = (bf[-1] + bf[1]) >> 1;
00258 b = bf[-width];
00259 RGB2YUV(r, *bf, b, y2, u2, v2);
00260 ++bf;
00261
00262 assign(y, u, v, y1, u1, v1, y2, u2, v2);
00263 }
00264
00265
00266
00267 g = (bf[-width] + bf[-1]) >> 1;
00268 b = (bf[-width-1] + bf[-width+1]) >> 1;
00269 RGB2YUV(*bf, g, b, y1, u1, v1);
00270 ++bf;
00271
00272 b = bf[-width];
00273 RGB2YUV(bf[-1], *bf, b, y2, u2, v2);
00274 ++bf;
00275
00276 assign(y, u, v, y1, u1, v1, y2, u2, v2);
00277
00278 }
00279
00280
00281 void
00282 bayerGBRG_to_yuv422planar_bilinear2(const unsigned char *bayer, unsigned char *yuv,
00283 unsigned int width, unsigned int height)
00284 {
00285 unsigned char *y = yuv;
00286 unsigned char *u = YUV422_PLANAR_U_PLANE(yuv, width, height);
00287 unsigned char *v = YUV422_PLANAR_V_PLANE(yuv, width, height);
00288 const unsigned char *bf = bayer;
00289
00290 int y1, u1, v1, y2, u2, v2;
00291 int r, g, b;
00292
00293
00294 bf += width;
00295 y += width;
00296 u += width >> 1;
00297 v += width >> 1;
00298
00299 for ( unsigned int h = 1; h < height - 1; h += 2) {
00300
00301
00302
00303 ++bf; ++bf;
00304 ++y; ++y;
00305 ++u; ++v;
00306
00307 for (unsigned int w = 2; w < width - 2; w += 2) {
00308 g = (bf[1] + bf[-1]) >> 1;
00309 b = (bf[width-1] + bf[width+1]) >> 1;
00310 RGB2YUV(*bf, g, b, y1, u1, v1);
00311 ++bf;
00312
00313 r = (bf[-1] + bf[1]) >> 1;
00314 b = (bf[-width] + bf[width]) >> 1;
00315 RGB2YUV(r, *bf, b, y2, u2, v2);
00316 ++bf;
00317
00318 assign(y, u, v, y1, u1, v1, y2, u2, v2);
00319 }
00320
00321
00322 ++bf; ++bf;
00323 ++y; ++y;
00324 ++u; ++v;
00325
00326
00327
00328 ++bf; ++bf;
00329 ++y; ++y;
00330 ++u; ++v;
00331
00332 for (unsigned int w = 2; w < width - 2; w += 2) {
00333 r = (bf[width] + bf[-width]) >> 1;
00334 b = (bf[-1] + bf[1]) >> 1;
00335 RGB2YUV(r, *bf, b, y1, u1, v1);
00336 ++bf;
00337
00338 r = (bf[width-1] + bf[width+1]) >> 1;
00339 g = (bf[1] + bf[width]) >> 1;
00340 RGB2YUV(r, g, *bf, y2, u2, v2);
00341 ++bf;
00342
00343 assign(y, u, v, y1, u1, v1, y2, u2, v2);
00344 }
00345
00346
00347 ++bf; ++bf;
00348 ++y; ++y;
00349 ++u; ++v;
00350 }
00351
00352
00353
00354 }
00355
00356
00357
00358
00359
00360
00361
00362
00363
00364
00365
00366
00367
00368
00369
00370
00371
00372
00373
00374
00375
00376
00377
00378
00379
00380
00381
00382
00383
00384
00385
00386
00387
00388
00389
00390
00391
00392
00393
00394
00395
00396
00397
00398
00399
00400
00401
00402
00403
00404
00405
00406
00407
00408
00409
00410
00411
00412
00413
00414
00415
00416
00417
00418
00419
00420
00421
00422
00423
00424
00425
00426
00427
00428
00429
00430
00431
00432
00433
00434
00435
00436
00437
00438
00439
00440
00441
00442
00443
00444
00445
00446
00447
00448
00449
00450
00451
00452
00453
00454
00455
00456
00457
00458
00459
00460
00461
00462
00463
00464
00465
00466
00467
00468
00469
00470
00471
00472
00473
00474
00475
00476
00477
00478
00479
00480
00481
00482
00483
00484
00485
00486
00487
00488
00489
00490
00491
00492
00493
00494
00495
00496
00497
00498
00499
00500
00501
00502
00503
00504
00505
00506
00507
00508
00509
00510
00511
00512
00513
00514
00515
00516
00517
00518
00519
00520
00521
00522
00523
00524
00525
00526
00527
00528
00529
00530
00531
00532
00533
00534
00535
00536
00537
00538
00539
00540
00541
00542
00543
00544
00545 }