mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-11-19 00:01:23 +00:00
57ce0321c8
Original commit message from CVS:
Added a tarkin encoder/decoder plugin.
I moved the tarking CVS code in here temporarily until they have a
library (hence this plugin is in ext)
test with:
./gst-launch filesrc location=/opt/data/shihad.mpg ! mpegdemux video_00! { queue ! mpeg2dec ! colorspace ! tarkinenc bitrate=3000 ! disksink location=out.ogg }
./gst-launch filesrc location=out.ogg ! tarkindec ! colorspace ! xvideosink
224 lines
6.3 KiB
C
224 lines
6.3 KiB
C
#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]);
|
|
}
|
|
|
|
|