You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
393 lines
9.5 KiB
393 lines
9.5 KiB
/* |
|
* Fast but bad debayer method that scales and rotates by skipping source |
|
* pixels and doesn't interpolate any values at all |
|
*/ |
|
|
|
#include "quickpreview.h" |
|
#include <assert.h> |
|
#include <stdio.h> |
|
|
|
/* Linear -> sRGB lookup table */ |
|
static const int srgb[] = { |
|
0, 12, 21, 28, 33, 38, 42, 46, 49, 52, 55, 58, 61, 63, 66, 68, 70, |
|
73, 75, 77, 79, 81, 82, 84, 86, 88, 89, 91, 93, 94, 96, 97, 99, 100, |
|
102, 103, 104, 106, 107, 109, 110, 111, 112, 114, 115, 116, 117, 118, |
|
120, 121, 122, 123, 124, 125, 126, 127, 129, 130, 131, 132, 133, 134, |
|
135, 136, 137, 138, 139, 140, 141, 142, 142, 143, 144, 145, 146, 147, |
|
148, 149, 150, 151, 151, 152, 153, 154, 155, 156, 157, 157, 158, 159, |
|
160, 161, 161, 162, 163, 164, 165, 165, 166, 167, 168, 168, 169, 170, |
|
171, 171, 172, 173, 174, 174, 175, 176, 176, 177, 178, 179, 179, 180, |
|
181, 181, 182, 183, 183, 184, 185, 185, 186, 187, 187, 188, 189, 189, |
|
190, 191, 191, 192, 193, 193, 194, 194, 195, 196, 196, 197, 197, 198, |
|
199, 199, 200, 201, 201, 202, 202, 203, 204, 204, 205, 205, 206, 206, |
|
207, 208, 208, 209, 209, 210, 210, 211, 212, 212, 213, 213, 214, 214, |
|
215, 215, 216, 217, 217, 218, 218, 219, 219, 220, 220, 221, 221, 222, |
|
222, 223, 223, 224, 224, 225, 226, 226, 227, 227, 228, 228, 229, 229, |
|
230, 230, 231, 231, 232, 232, 233, 233, 234, 234, 235, 235, 236, 236, |
|
237, 237, 237, 238, 238, 239, 239, 240, 240, 241, 241, 242, 242, 243, |
|
243, 244, 244, 245, 245, 245, 246, 246, 247, 247, 248, 248, 249, 249, |
|
250, 250, 251, 251, 251, 252, 252, 253, 253, 254, 254, 255 |
|
}; |
|
|
|
static inline uint32_t pack_rgb(uint8_t r, uint8_t g, uint8_t b) |
|
{ |
|
return (r << 16) | (g << 8) | b; |
|
} |
|
|
|
static inline uint32_t convert_yuv_to_srgb(uint8_t y, uint8_t u, uint8_t v) |
|
{ |
|
uint32_t r = 1.164f * y + 1.596f * (v - 128); |
|
uint32_t g = 1.164f * y - 0.813f * (v - 128) - 0.391f * (u - 128); |
|
uint32_t b = 1.164f * y + 2.018f * (u - 128); |
|
return pack_rgb(r, g, b); |
|
} |
|
|
|
static inline uint32_t apply_colormatrix(uint32_t color, const float *colormatrix) |
|
{ |
|
if (!colormatrix) { |
|
return color; |
|
} |
|
|
|
uint32_t r = (color >> 16) * colormatrix[0] + ((color >> 8) & 0xFF) * colormatrix[1] + (color & 0xFF) * colormatrix[2]; |
|
uint32_t g = (color >> 16) * colormatrix[3] + ((color >> 8) & 0xFF) * colormatrix[4] + (color & 0xFF) * colormatrix[5]; |
|
uint32_t b = (color >> 16) * colormatrix[6] + ((color >> 8) & 0xFF) * colormatrix[7] + (color & 0xFF) * colormatrix[8]; |
|
|
|
// Clip colors |
|
if (r > 0xFF) |
|
r = 0xFF; |
|
if (g > 0xFF) |
|
g = 0xFF; |
|
if (b > 0xFF) |
|
b = 0xFF; |
|
return pack_rgb(r, g, b); |
|
} |
|
|
|
static inline uint32_t coord_map(uint32_t x, uint32_t y, uint32_t width, uint32_t height, int rotation, bool mirrored) |
|
{ |
|
uint32_t x_r, y_r; |
|
if (rotation == 0) { |
|
x_r = x; |
|
y_r = y; |
|
} else if (rotation == 90) { |
|
x_r = y; |
|
y_r = height - x - 1; |
|
} else if (rotation == 270) { |
|
x_r = width - y - 1; |
|
y_r = x; |
|
} else { |
|
x_r = width - x - 1; |
|
y_r = height - y - 1; |
|
} |
|
|
|
if (mirrored) { |
|
x_r = width - x_r - 1; |
|
} |
|
|
|
uint32_t index = y_r * width + x_r; |
|
#ifdef DEBUG |
|
assert(index < width * height); |
|
#endif |
|
return index; |
|
} |
|
|
|
static void quick_preview_rggb8( |
|
uint32_t *dst, |
|
const uint32_t dst_width, |
|
const uint32_t dst_height, |
|
const uint8_t *src, |
|
const uint32_t src_width, |
|
const uint32_t src_height, |
|
const MPPixelFormat format, |
|
const uint32_t rotation, |
|
const bool mirrored, |
|
const float *colormatrix, |
|
const uint8_t blacklevel, |
|
const uint32_t skip) |
|
{ |
|
uint32_t src_y = 0, dst_y = 0; |
|
while (src_y < src_height) { |
|
uint32_t src_x = 0, dst_x = 0; |
|
while (src_x < src_width) { |
|
uint32_t src_i = src_y * src_width + src_x; |
|
|
|
uint8_t b0 = srgb[src[src_i] - blacklevel]; |
|
uint8_t b1 = srgb[src[src_i + 1] - blacklevel]; |
|
uint8_t b2 = srgb[src[src_i + src_width + 1] - blacklevel]; |
|
|
|
uint32_t color; |
|
switch (format) { |
|
case MP_PIXEL_FMT_BGGR8: |
|
color = pack_rgb(b2, b1, b0); |
|
break; |
|
case MP_PIXEL_FMT_GBRG8: |
|
color = pack_rgb(b2, b0, b1); |
|
break; |
|
case MP_PIXEL_FMT_GRBG8: |
|
color = pack_rgb(b1, b0, b2); |
|
break; |
|
case MP_PIXEL_FMT_RGGB8: |
|
color = pack_rgb(b0, b1, b2); |
|
break; |
|
default: |
|
assert(false); |
|
} |
|
|
|
color = apply_colormatrix(color, colormatrix); |
|
|
|
dst[coord_map(dst_x, dst_y, dst_width, dst_height, rotation, mirrored)] = color; |
|
|
|
src_x += 2 + 2 * skip; |
|
++dst_x; |
|
} |
|
|
|
src_y += 2 + 2 * skip; |
|
++dst_y; |
|
} |
|
} |
|
|
|
static void quick_preview_rggb10( |
|
uint32_t *dst, |
|
const uint32_t dst_width, |
|
const uint32_t dst_height, |
|
const uint8_t *src, |
|
const uint32_t src_width, |
|
const uint32_t src_height, |
|
const MPPixelFormat format, |
|
const uint32_t rotation, |
|
const bool mirrored, |
|
const float *colormatrix, |
|
const uint8_t blacklevel, |
|
const uint32_t skip) |
|
{ |
|
assert(src_width % 2 == 0); |
|
|
|
uint32_t width_bytes = mp_pixel_format_width_to_bytes(format, src_width); |
|
|
|
uint32_t src_y = 0, dst_y = 0; |
|
while (src_y < src_height) { |
|
uint32_t src_x = 0, dst_x = 0; |
|
while (src_x < width_bytes) { |
|
uint32_t src_i = src_y * width_bytes + src_x; |
|
|
|
uint8_t b0 = srgb[src[src_i] - blacklevel]; |
|
uint8_t b1 = srgb[src[src_i + 1] - blacklevel]; |
|
uint8_t b2 = srgb[src[src_i + width_bytes + 1] - blacklevel]; |
|
|
|
uint32_t color; |
|
switch (format) { |
|
case MP_PIXEL_FMT_BGGR10P: |
|
color = pack_rgb(b2, b1, b0); |
|
break; |
|
case MP_PIXEL_FMT_GBRG10P: |
|
color = pack_rgb(b2, b0, b1); |
|
break; |
|
case MP_PIXEL_FMT_GRBG10P: |
|
color = pack_rgb(b1, b0, b2); |
|
break; |
|
case MP_PIXEL_FMT_RGGB10P: |
|
color = pack_rgb(b0, b1, b2); |
|
break; |
|
default: |
|
assert(false); |
|
} |
|
|
|
color = apply_colormatrix(color, colormatrix); |
|
|
|
dst[coord_map(dst_x, dst_y, dst_width, dst_height, rotation, mirrored)] = color; |
|
|
|
uint32_t advance = 1 + skip; |
|
if (src_x % 5 == 0) { |
|
src_x += 2 * (advance % 2) + 5 * (advance / 2); |
|
} else { |
|
src_x += 3 * (advance % 2) + 5 * (advance / 2); |
|
} |
|
++dst_x; |
|
} |
|
|
|
src_y += 2 + 2 * skip; |
|
++dst_y; |
|
} |
|
} |
|
|
|
static void quick_preview_yuv( |
|
uint32_t *dst, |
|
const uint32_t dst_width, |
|
const uint32_t dst_height, |
|
const uint8_t *src, |
|
const uint32_t src_width, |
|
const uint32_t src_height, |
|
const MPPixelFormat format, |
|
const uint32_t rotation, |
|
const bool mirrored, |
|
const float *colormatrix, |
|
const uint32_t skip) |
|
{ |
|
assert(src_width % 2 == 0); |
|
|
|
uint32_t width_bytes = src_width * 2; |
|
|
|
uint32_t unrot_dst_width = dst_width; |
|
if (rotation != 0 && rotation != 180) { |
|
unrot_dst_width = dst_height; |
|
} |
|
|
|
uint32_t src_y = 0, dst_y = 0; |
|
while (src_y < src_height) { |
|
uint32_t src_x = 0, dst_x = 0; |
|
while (src_x < width_bytes) { |
|
uint32_t src_i = src_y * width_bytes + src_x; |
|
|
|
uint8_t b0 = src[src_i]; |
|
uint8_t b1 = src[src_i + 1]; |
|
uint8_t b2 = src[src_i + 2]; |
|
uint8_t b3 = src[src_i + 3]; |
|
|
|
uint32_t color1, color2; |
|
switch (format) { |
|
case MP_PIXEL_FMT_UYVY: |
|
color1 = convert_yuv_to_srgb(b1, b0, b2); |
|
color2 = convert_yuv_to_srgb(b3, b0, b2); |
|
break; |
|
case MP_PIXEL_FMT_YUYV: |
|
color1 = convert_yuv_to_srgb(b0, b1, b3); |
|
color2 = convert_yuv_to_srgb(b2, b1, b3); |
|
break; |
|
default: |
|
assert(false); |
|
} |
|
|
|
color1 = apply_colormatrix(color1, colormatrix); |
|
color2 = apply_colormatrix(color2, colormatrix); |
|
|
|
uint32_t dst_i1 = coord_map(dst_x, dst_y, dst_width, dst_height, rotation, mirrored); |
|
dst[dst_i1] = color1; |
|
++dst_x; |
|
|
|
// The last pixel needs to be skipped if we have an odd un-rotated width |
|
if (dst_x < unrot_dst_width) { |
|
uint32_t dst_i2 = coord_map(dst_x, dst_y, dst_width, dst_height, rotation, mirrored); |
|
dst[dst_i2] = color2; |
|
++dst_x; |
|
} |
|
|
|
src_x += 4 + 4 * skip; |
|
} |
|
|
|
src_y += 1 + skip; |
|
++dst_y; |
|
} |
|
} |
|
|
|
void quick_preview( |
|
uint32_t *dst, |
|
const uint32_t dst_width, |
|
const uint32_t dst_height, |
|
const uint8_t *src, |
|
const uint32_t src_width, |
|
const uint32_t src_height, |
|
const MPPixelFormat format, |
|
const uint32_t rotation, |
|
const bool mirrored, |
|
const float *colormatrix, |
|
const uint8_t blacklevel, |
|
const uint32_t skip) |
|
{ |
|
switch (format) { |
|
case MP_PIXEL_FMT_BGGR8: |
|
case MP_PIXEL_FMT_GBRG8: |
|
case MP_PIXEL_FMT_GRBG8: |
|
case MP_PIXEL_FMT_RGGB8: |
|
quick_preview_rggb8( |
|
dst, |
|
dst_width, |
|
dst_height, |
|
src, |
|
src_width, |
|
src_height, |
|
format, |
|
rotation, |
|
mirrored, |
|
colormatrix, |
|
blacklevel, |
|
skip); |
|
break; |
|
case MP_PIXEL_FMT_BGGR10P: |
|
case MP_PIXEL_FMT_GBRG10P: |
|
case MP_PIXEL_FMT_GRBG10P: |
|
case MP_PIXEL_FMT_RGGB10P: |
|
quick_preview_rggb10( |
|
dst, |
|
dst_width, |
|
dst_height, |
|
src, |
|
src_width, |
|
src_height, |
|
format, |
|
rotation, |
|
mirrored, |
|
colormatrix, |
|
blacklevel, |
|
skip); |
|
break; |
|
case MP_PIXEL_FMT_UYVY: |
|
case MP_PIXEL_FMT_YUYV: |
|
quick_preview_yuv( |
|
dst, |
|
dst_width, |
|
dst_height, |
|
src, |
|
src_width, |
|
src_height, |
|
format, |
|
rotation, |
|
mirrored, |
|
colormatrix, |
|
skip); |
|
break; |
|
default: |
|
assert(false); |
|
} |
|
} |
|
|
|
static uint32_t div_ceil(uint32_t x, uint32_t y) |
|
{ |
|
return x/y + !!(x % y); |
|
} |
|
|
|
void quick_preview_size( |
|
uint32_t *dst_width, |
|
uint32_t *dst_height, |
|
uint32_t *skip, |
|
const uint32_t preview_width, |
|
const uint32_t preview_height, |
|
const uint32_t src_width, |
|
const uint32_t src_height, |
|
const MPPixelFormat format, |
|
const int rotation) |
|
{ |
|
uint32_t colors_x = mp_pixel_format_width_to_colors(format, src_width); |
|
uint32_t colors_y = mp_pixel_format_height_to_colors(format, src_height); |
|
|
|
if (rotation != 0 && rotation != 180) { |
|
uint32_t tmp = colors_x; |
|
colors_x = colors_y; |
|
colors_y = tmp; |
|
} |
|
|
|
uint32_t scale_x = colors_x / preview_width; |
|
uint32_t scale_y = colors_y / preview_height; |
|
|
|
if (scale_x > 0) |
|
--scale_x; |
|
if (scale_y > 0) |
|
--scale_y; |
|
*skip = scale_x > scale_y ? scale_x : scale_y; |
|
|
|
*dst_width = div_ceil(colors_x, (1 + *skip)); |
|
if (*dst_width <= 0) |
|
*dst_width = 1; |
|
|
|
*dst_height = div_ceil(colors_y, (1 + *skip)); |
|
if (*dst_height <= 0) |
|
*dst_height = 1; |
|
}
|
|
|