mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-13 03:46:34 +00:00
7a778ee4b7
Original commit message from CVS: gst-indent
234 lines
6.1 KiB
C
234 lines
6.1 KiB
C
#ifdef HAVE_CONFIG_H
|
|
#include "config.h"
|
|
#endif
|
|
|
|
#include "yuv.h"
|
|
|
|
/*#define TARKIN_YUV_EXACT*/
|
|
/*#define TARKIN_YUV_LXY*/
|
|
|
|
|
|
static inline uint8_t
|
|
CLAMP (int16_t x)
|
|
{
|
|
return ((x > 255) ? 255 : (x < 0) ? 0 : x);
|
|
}
|
|
|
|
|
|
|
|
void
|
|
rgb24_to_yuv (uint8_t * rgb, Wavelet3DBuf * yuv[], uint32_t frame)
|
|
{
|
|
int count = yuv[0]->width * yuv[0]->height;
|
|
int16_t *y = yuv[0]->data + frame * count;
|
|
int16_t *u = yuv[1]->data + frame * count;
|
|
int16_t *v = yuv[2]->data + frame * count;
|
|
int i;
|
|
|
|
|
|
#if defined(TARKIN_YUV_EXACT)
|
|
for (i = 0; i < count; i++, rgb += 3) {
|
|
y[i] = ((int16_t) 77 * rgb[0] + 150 * rgb[1] + 29 * rgb[2]) / 256;
|
|
u[i] = ((int16_t) - 44 * rgb[0] - 87 * rgb[1] + 131 * rgb[2]) / 256;
|
|
v[i] = ((int16_t) 131 * rgb[0] - 110 * rgb[1] - 21 * rgb[2]) / 256;
|
|
}
|
|
#elif defined(TARKIN_YUV_LXY)
|
|
for (i = 0; i < count; i++, rgb += 3) {
|
|
y[i] = ((int16_t) 54 * rgb[0] + 182 * rgb[1] + 18 * rgb[2]) / 256;
|
|
u[i] = rgb[0] - y[i];
|
|
v[i] = rgb[2] - y[i];
|
|
}
|
|
#else
|
|
for (i = 0; i < count; i++, rgb += 3) {
|
|
v[i] = rgb[0] - rgb[1];
|
|
u[i] = rgb[2] - rgb[1];
|
|
y[i] = rgb[1] + (u[i] + v[i]) / 4;
|
|
}
|
|
#endif
|
|
}
|
|
|
|
|
|
void
|
|
yuv_to_rgb24 (Wavelet3DBuf * yuv[], uint8_t * rgb, uint32_t frame)
|
|
{
|
|
int count = yuv[0]->width * yuv[0]->height;
|
|
int16_t *y = yuv[0]->data + frame * count;
|
|
int16_t *u = yuv[1]->data + frame * count;
|
|
int16_t *v = yuv[2]->data + frame * count;
|
|
int i;
|
|
|
|
#if defined(TARKIN_YUV_EXACT)
|
|
for (i = 0; i < count; i++, rgb += 3) {
|
|
rgb[0] = CLAMP (y[i] + 1.371 * v[i]);
|
|
rgb[1] = CLAMP (y[i] - 0.698 * v[i] - 0.336 * u[i]);
|
|
rgb[2] = CLAMP (y[i] + 1.732 * u[i]);
|
|
}
|
|
#elif defined(TARKIN_YUV_LXY)
|
|
for (i = 0; i < count; i++, rgb += 3) {
|
|
rgb[1] = CLAMP (y[i] - (76 * u[i] - 26 * v[i]) / 256);
|
|
rgb[0] = CLAMP (y[i] + u[i]);
|
|
rgb[2] = CLAMP (y[i] + v[i]);
|
|
}
|
|
#else
|
|
for (i = 0; i < count; i++, rgb += 3) {
|
|
rgb[1] = CLAMP (y[i] - (u[i] + v[i]) / 4);
|
|
rgb[2] = CLAMP (u[i] + rgb[1]);
|
|
rgb[0] = CLAMP (v[i] + rgb[1]);
|
|
}
|
|
#endif
|
|
}
|
|
|
|
|
|
void
|
|
rgb32_to_yuv (uint8_t * rgb, Wavelet3DBuf * yuv[], uint32_t frame)
|
|
{
|
|
int count = yuv[0]->width * yuv[0]->height;
|
|
int16_t *y = yuv[0]->data + frame * count;
|
|
int16_t *u = yuv[1]->data + frame * count;
|
|
int16_t *v = yuv[2]->data + frame * count;
|
|
int i;
|
|
|
|
#if defined(TARKIN_YUV_EXACT)
|
|
for (i = 0; i < count; i++, rgb += 4) {
|
|
y[i] = ((int16_t) 77 * rgb[0] + 150 * rgb[1] + 29 * rgb[2]) / 256;
|
|
u[i] = ((int16_t) - 44 * rgb[0] - 87 * rgb[1] + 131 * rgb[2]) / 256;
|
|
v[i] = ((int16_t) 131 * rgb[0] - 110 * rgb[1] - 21 * rgb[2]) / 256;
|
|
}
|
|
#elif defined(TARKIN_YUV_LXY)
|
|
for (i = 0; i < count; i++, rgb += 4) {
|
|
y[i] = ((int16_t) 54 * rgb[0] + 182 * rgb[1] + 18 * rgb[2]) / 256;
|
|
u[i] = rgb[0] - y[i];
|
|
v[i] = rgb[2] - y[i];
|
|
}
|
|
#else
|
|
for (i = 0; i < count; i++, rgb += 4) {
|
|
v[i] = rgb[0] - rgb[1];
|
|
u[i] = rgb[2] - rgb[1];
|
|
y[i] = rgb[1] + (u[i] + v[i]) / 4;
|
|
}
|
|
#endif
|
|
}
|
|
|
|
|
|
void
|
|
yuv_to_rgb32 (Wavelet3DBuf * yuv[], uint8_t * rgb, uint32_t frame)
|
|
{
|
|
int count = yuv[0]->width * yuv[0]->height;
|
|
int16_t *y = yuv[0]->data + frame * count;
|
|
int16_t *u = yuv[1]->data + frame * count;
|
|
int16_t *v = yuv[2]->data + frame * count;
|
|
int i;
|
|
|
|
#if defined(TARKIN_YUV_EXACT)
|
|
for (i = 0; i < count; i++, rgb += 4) {
|
|
rgb[0] = CLAMP (y[i] + 1.371 * v[i]);
|
|
rgb[1] = CLAMP (y[i] - 0.698 * v[i] - 0.336 * u[i]);
|
|
rgb[2] = CLAMP (y[i] + 1.732 * u[i]);
|
|
}
|
|
#elif defined(TARKIN_YUV_LXY)
|
|
for (i = 0; i < count; i++, rgb += 4) {
|
|
rgb[1] = CLAMP (y[i] - (76 * u[i] - 26 * v[i]) / 256);
|
|
rgb[0] = CLAMP (y[i] + u[i]);
|
|
rgb[2] = CLAMP (y[i] + v[i]);
|
|
}
|
|
#else
|
|
for (i = 0; i < count; i++, rgb += 4) {
|
|
rgb[1] = CLAMP (y[i] - (u[i] + v[i]) / 4);
|
|
rgb[2] = CLAMP (u[i] + rgb[1]);
|
|
rgb[0] = CLAMP (v[i] + rgb[1]);
|
|
}
|
|
#endif
|
|
}
|
|
|
|
|
|
void
|
|
rgba_to_yuv (uint8_t * rgba, Wavelet3DBuf * yuva[], uint32_t frame)
|
|
{
|
|
int count = yuva[0]->width * yuva[0]->height;
|
|
int16_t *y = yuva[0]->data + frame * count;
|
|
int16_t *u = yuva[1]->data + frame * count;
|
|
int16_t *v = yuva[2]->data + frame * count;
|
|
int16_t *a = yuva[3]->data + frame * count;
|
|
int i;
|
|
|
|
#if defined(TARKIN_YUV_EXACT)
|
|
for (i = 0; i < count; i++, rgba += 4) {
|
|
y[i] = ((int16_t) 77 * rgba[0] + 150 * rgba[1] + 29 * rgba[2]) / 256;
|
|
u[i] = ((int16_t) - 44 * rgba[0] - 87 * rgba[1] + 131 * rgba[2]) / 256;
|
|
v[i] = ((int16_t) 131 * rgba[0] - 110 * rgba[1] - 21 * rgba[2]) / 256;
|
|
a[i] = rgba[3];
|
|
}
|
|
#elif defined(TARKIN_YUV_LXY)
|
|
for (i = 0; i < count; i++, rgba += 4) {
|
|
y[i] = ((int16_t) 54 * rgba[0] + 182 * rgba[1] + 18 * rgba[2]) / 256;
|
|
u[i] = rgba[0] - y[i];
|
|
v[i] = rgba[2] - y[i];
|
|
a[i] = rgba[3];
|
|
}
|
|
#else
|
|
for (i = 0; i < count; i++, rgba += 4) {
|
|
v[i] = rgba[0] - rgba[1];
|
|
u[i] = rgba[2] - rgba[1];
|
|
y[i] = rgba[1] + (u[i] + v[i]) / 4;
|
|
a[i] = rgba[3];
|
|
}
|
|
#endif
|
|
}
|
|
|
|
|
|
void
|
|
yuv_to_rgba (Wavelet3DBuf * yuva[], uint8_t * rgba, uint32_t frame)
|
|
{
|
|
int count = yuva[0]->width * yuva[0]->height;
|
|
int16_t *y = yuva[0]->data + frame * count;
|
|
int16_t *u = yuva[1]->data + frame * count;
|
|
int16_t *v = yuva[2]->data + frame * count;
|
|
int16_t *a = yuva[3]->data + frame * count;
|
|
int i;
|
|
|
|
#if defined(TARKIN_YUV_EXACT)
|
|
for (i = 0; i < count; i++, rgba += 4) {
|
|
rgba[0] = CLAMP (y[i] + 1.371 * v[i]);
|
|
rgba[1] = CLAMP (y[i] - 0.698 * v[i] - 0.336 * u[i]);
|
|
rgba[2] = CLAMP (y[i] + 1.732 * u[i]);
|
|
rgba[3] = a[i];
|
|
}
|
|
#elif defined(TARKIN_YUV_LXY)
|
|
for (i = 0; i < count; i++, rgba += 4) {
|
|
rgba[1] = CLAMP (y[i] - (76 * u[i] - 26 * v[i]) / 256);
|
|
rgba[0] = CLAMP (y[i] + u[i]);
|
|
rgba[2] = CLAMP (y[i] + v[i]);
|
|
rgba[3] = a[i];
|
|
}
|
|
#else
|
|
for (i = 0; i < count; i++, rgba += 4) {
|
|
rgba[1] = CLAMP (y[i] - (u[i] + v[i]) / 4);
|
|
rgba[2] = CLAMP (u[i] + rgba[1]);
|
|
rgba[0] = CLAMP (v[i] + rgba[1]);
|
|
rgba[3] = a[i];
|
|
}
|
|
#endif
|
|
}
|
|
|
|
void
|
|
grayscale_to_y (uint8_t * rgba, Wavelet3DBuf * y[], uint32_t frame)
|
|
{
|
|
int count = y[0]->width * y[0]->height;
|
|
int16_t *_y = y[0]->data + frame * count;
|
|
int i;
|
|
|
|
for (i = 0; i < count; i++)
|
|
_y[i] = rgba[i];
|
|
}
|
|
|
|
|
|
void
|
|
y_to_grayscale (Wavelet3DBuf * y[], uint8_t * rgba, uint32_t frame)
|
|
{
|
|
int count = y[0]->width * y[0]->height;
|
|
int16_t *_y = y[0]->data + frame * count;
|
|
int i;
|
|
|
|
for (i = 0; i < count; i++)
|
|
rgba[i] = CLAMP (_y[i]);
|
|
}
|