557 lines
16 KiB
C
557 lines
16 KiB
C
|
/*
|
||
|
|
Copyright (C) 1997-2024 Sam Lantinga <slouken@libsdl.org>
|
||
|
|
|
||
|
|
This software is provided 'as-is', without any express or implied
|
||
|
|
warranty. In no event will the authors be held liable for any damages
|
||
|
|
arising from the use of this software.
|
||
|
|
|
||
|
|
Permission is granted to anyone to use this software for any purpose,
|
||
|
|
including commercial applications, and to alter it and redistribute it
|
||
|
|
freely.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <SDL3/SDL.h>
|
||
|
|
|
||
|
|
#include "testyuv_cvt.h"
|
||
|
|
|
||
|
|
#define YUV_SD_THRESHOLD 576
|
||
|
|
|
||
|
|
static YUV_CONVERSION_MODE YUV_ConversionMode = YUV_CONVERSION_BT601;
|
||
|
|
|
||
|
|
void SetYUVConversionMode(YUV_CONVERSION_MODE mode)
|
||
|
|
{
|
||
|
|
YUV_ConversionMode = mode;
|
||
|
|
}
|
||
|
|
|
||
|
|
YUV_CONVERSION_MODE GetYUVConversionMode(void)
|
||
|
|
{
|
||
|
|
return YUV_ConversionMode;
|
||
|
|
}
|
||
|
|
|
||
|
|
YUV_CONVERSION_MODE GetYUVConversionModeForResolution(int width, int height)
|
||
|
|
{
|
||
|
|
YUV_CONVERSION_MODE mode = GetYUVConversionMode();
|
||
|
|
if (mode == YUV_CONVERSION_AUTOMATIC) {
|
||
|
|
if (height <= YUV_SD_THRESHOLD) {
|
||
|
|
mode = YUV_CONVERSION_BT601;
|
||
|
|
} else {
|
||
|
|
mode = YUV_CONVERSION_BT709;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return mode;
|
||
|
|
}
|
||
|
|
|
||
|
|
SDL_Colorspace GetColorspaceForYUVConversionMode(YUV_CONVERSION_MODE mode)
|
||
|
|
{
|
||
|
|
SDL_Colorspace colorspace;
|
||
|
|
|
||
|
|
switch (mode) {
|
||
|
|
case YUV_CONVERSION_JPEG:
|
||
|
|
colorspace = SDL_DEFINE_COLORSPACE(SDL_COLOR_TYPE_YCBCR,
|
||
|
|
SDL_COLOR_RANGE_FULL,
|
||
|
|
SDL_COLOR_PRIMARIES_BT709,
|
||
|
|
SDL_TRANSFER_CHARACTERISTICS_BT601,
|
||
|
|
SDL_MATRIX_COEFFICIENTS_BT601,
|
||
|
|
SDL_CHROMA_LOCATION_CENTER);
|
||
|
|
break;
|
||
|
|
case YUV_CONVERSION_BT601:
|
||
|
|
colorspace = SDL_DEFINE_COLORSPACE(SDL_COLOR_TYPE_YCBCR,
|
||
|
|
SDL_COLOR_RANGE_LIMITED,
|
||
|
|
SDL_COLOR_PRIMARIES_BT709,
|
||
|
|
SDL_TRANSFER_CHARACTERISTICS_BT601,
|
||
|
|
SDL_MATRIX_COEFFICIENTS_BT601,
|
||
|
|
SDL_CHROMA_LOCATION_CENTER);
|
||
|
|
break;
|
||
|
|
case YUV_CONVERSION_BT709:
|
||
|
|
colorspace = SDL_DEFINE_COLORSPACE(SDL_COLOR_TYPE_YCBCR,
|
||
|
|
SDL_COLOR_RANGE_LIMITED,
|
||
|
|
SDL_COLOR_PRIMARIES_BT709,
|
||
|
|
SDL_TRANSFER_CHARACTERISTICS_BT709,
|
||
|
|
SDL_MATRIX_COEFFICIENTS_BT709,
|
||
|
|
SDL_CHROMA_LOCATION_CENTER);
|
||
|
|
break;
|
||
|
|
case YUV_CONVERSION_BT2020:
|
||
|
|
colorspace = SDL_COLORSPACE_BT2020_FULL;
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
colorspace = SDL_COLORSPACE_UNKNOWN;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
return colorspace;
|
||
|
|
}
|
||
|
|
|
||
|
|
static float clip3(float x, float y, float z)
|
||
|
|
{
|
||
|
|
return (z < x) ? x : ((z > y) ? y : z);
|
||
|
|
}
|
||
|
|
|
||
|
|
static float sRGBtoNits(float v)
|
||
|
|
{
|
||
|
|
/* Normalize from 0..255 */
|
||
|
|
v /= 255.0f;
|
||
|
|
|
||
|
|
/* Convert from sRGB */
|
||
|
|
v = v <= 0.04045f ? (v / 12.92f) : SDL_powf(((v + 0.055f) / 1.055f), 2.4f);
|
||
|
|
|
||
|
|
/* Convert to nits, using a default SDR whitepoint of 203 */
|
||
|
|
v *= 203.0f;
|
||
|
|
|
||
|
|
return v;
|
||
|
|
}
|
||
|
|
|
||
|
|
static float PQfromNits(float v)
|
||
|
|
{
|
||
|
|
const float c1 = 0.8359375f;
|
||
|
|
const float c2 = 18.8515625f;
|
||
|
|
const float c3 = 18.6875f;
|
||
|
|
const float m1 = 0.1593017578125f;
|
||
|
|
const float m2 = 78.84375f;
|
||
|
|
|
||
|
|
float y = SDL_clamp(v / 10000.0f, 0.0f, 1.0f);
|
||
|
|
float num = c1 + c2 * SDL_powf(y, m1);
|
||
|
|
float den = 1.0f + c3 * SDL_powf(y, m1);
|
||
|
|
return SDL_powf(num / den, m2);
|
||
|
|
}
|
||
|
|
|
||
|
|
void ConvertRec709toRec2020(float *fR, float *fG, float *fB)
|
||
|
|
{
|
||
|
|
static const float mat709to2020[] = {
|
||
|
|
0.627404f, 0.329283f, 0.043313f,
|
||
|
|
0.069097f, 0.919541f, 0.011362f,
|
||
|
|
0.016391f, 0.088013f, 0.895595f,
|
||
|
|
};
|
||
|
|
const float *matrix = mat709to2020;
|
||
|
|
float v[3];
|
||
|
|
|
||
|
|
v[0] = *fR;
|
||
|
|
v[1] = *fG;
|
||
|
|
v[2] = *fB;
|
||
|
|
|
||
|
|
*fR = matrix[0 * 3 + 0] * v[0] + matrix[0 * 3 + 1] * v[1] + matrix[0 * 3 + 2] * v[2];
|
||
|
|
*fG = matrix[1 * 3 + 0] * v[0] + matrix[1 * 3 + 1] * v[1] + matrix[1 * 3 + 2] * v[2];
|
||
|
|
*fB = matrix[2 * 3 + 0] * v[0] + matrix[2 * 3 + 1] * v[1] + matrix[2 * 3 + 2] * v[2];
|
||
|
|
}
|
||
|
|
|
||
|
|
static void RGBtoYUV(const Uint8 *rgb, int rgb_bits, int *yuv, int yuv_bits, YUV_CONVERSION_MODE mode, int monochrome, int luminance)
|
||
|
|
{
|
||
|
|
/**
|
||
|
|
* This formula is from Microsoft's documentation:
|
||
|
|
* https://msdn.microsoft.com/en-us/library/windows/desktop/dd206750(v=vs.85).aspx
|
||
|
|
* L = Kr * R + Kb * B + (1 - Kr - Kb) * G
|
||
|
|
* Y = floor(2^(M-8) * (219*(L-Z)/S + 16) + 0.5);
|
||
|
|
* U = clip3(0, (2^M)-1, floor(2^(M-8) * (112*(B-L) / ((1-Kb)*S) + 128) + 0.5));
|
||
|
|
* V = clip3(0, (2^M)-1, floor(2^(M-8) * (112*(R-L) / ((1-Kr)*S) + 128) + 0.5));
|
||
|
|
*/
|
||
|
|
SDL_bool studio_RGB = SDL_FALSE;
|
||
|
|
SDL_bool full_range_YUV = SDL_FALSE;
|
||
|
|
float N, M, S, Z, R, G, B, L, Kr, Kb, Y, U, V;
|
||
|
|
|
||
|
|
N = (float)rgb_bits;
|
||
|
|
M = (float)yuv_bits;
|
||
|
|
switch (mode) {
|
||
|
|
case YUV_CONVERSION_JPEG:
|
||
|
|
case YUV_CONVERSION_BT601:
|
||
|
|
/* BT.601 */
|
||
|
|
Kr = 0.299f;
|
||
|
|
Kb = 0.114f;
|
||
|
|
break;
|
||
|
|
case YUV_CONVERSION_BT709:
|
||
|
|
/* BT.709 */
|
||
|
|
Kr = 0.2126f;
|
||
|
|
Kb = 0.0722f;
|
||
|
|
break;
|
||
|
|
case YUV_CONVERSION_BT2020:
|
||
|
|
/* BT.2020 */
|
||
|
|
Kr = 0.2627f;
|
||
|
|
Kb = 0.0593f;
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
/* Invalid */
|
||
|
|
Kr = 1.0f;
|
||
|
|
Kb = 1.0f;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
R = rgb[0];
|
||
|
|
G = rgb[1];
|
||
|
|
B = rgb[2];
|
||
|
|
|
||
|
|
if (mode == YUV_CONVERSION_JPEG || mode == YUV_CONVERSION_BT2020) {
|
||
|
|
full_range_YUV = SDL_TRUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (mode == YUV_CONVERSION_BT2020) {
|
||
|
|
/* Input is sRGB, need to convert to BT.2020 PQ YUV */
|
||
|
|
R = sRGBtoNits(R);
|
||
|
|
G = sRGBtoNits(G);
|
||
|
|
B = sRGBtoNits(B);
|
||
|
|
ConvertRec709toRec2020(&R, &G, &B);
|
||
|
|
R = PQfromNits(R);
|
||
|
|
G = PQfromNits(G);
|
||
|
|
B = PQfromNits(B);
|
||
|
|
S = 1.0f;
|
||
|
|
Z = 0.0f;
|
||
|
|
} else if (studio_RGB) {
|
||
|
|
S = 219.0f * SDL_powf(2.0f, N - 8);
|
||
|
|
Z = 16.0f * SDL_powf(2.0f, N - 8);
|
||
|
|
} else {
|
||
|
|
S = 255.0f;
|
||
|
|
Z = 0.0f;
|
||
|
|
}
|
||
|
|
L = Kr * R + Kb * B + (1 - Kr - Kb) * G;
|
||
|
|
if (monochrome) {
|
||
|
|
R = L;
|
||
|
|
B = L;
|
||
|
|
}
|
||
|
|
if (full_range_YUV) {
|
||
|
|
Y = SDL_floorf((SDL_powf(2.0f, M) - 1) * ((L - Z) / S) + 0.5f);
|
||
|
|
U = clip3(0, SDL_powf(2.0f, M) - 1, SDL_floorf((SDL_powf(2.0f, M) / 2 - 1) * ((B - L) / ((1.0f - Kb) * S)) + SDL_powf(2.0f, M) / 2 + 0.5f));
|
||
|
|
V = clip3(0, SDL_powf(2.0f, M) - 1, SDL_floorf((SDL_powf(2.0f, M) / 2 - 1) * ((R - L) / ((1.0f - Kr) * S)) + SDL_powf(2.0f, M) / 2 + 0.5f));
|
||
|
|
} else {
|
||
|
|
Y = SDL_floorf(SDL_powf(2.0f, (M - 8)) * (219.0f * (L - Z) / S + 16) + 0.5f);
|
||
|
|
U = clip3(0, SDL_powf(2.0f, M) - 1, SDL_floorf(SDL_powf(2.0f, (M - 8)) * (112.0f * (B - L) / ((1.0f - Kb) * S) + 128) + 0.5f));
|
||
|
|
V = clip3(0, SDL_powf(2.0f, M) - 1, SDL_floorf(SDL_powf(2.0f, (M - 8)) * (112.0f * (R - L) / ((1.0f - Kr) * S) + 128) + 0.5f));
|
||
|
|
}
|
||
|
|
|
||
|
|
yuv[0] = (int)Y;
|
||
|
|
yuv[1] = (int)U;
|
||
|
|
yuv[2] = (int)V;
|
||
|
|
|
||
|
|
if (luminance != 100) {
|
||
|
|
yuv[0] = (int)clip3(0, SDL_powf(2.0f, M) - 1, SDL_roundf(yuv[0] * (luminance / 100.0f)));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
static void ConvertRGBtoPlanar2x2(Uint32 format, Uint8 *src, int pitch, Uint8 *out, int w, int h, YUV_CONVERSION_MODE mode, int monochrome, int luminance)
|
||
|
|
{
|
||
|
|
int x, y;
|
||
|
|
int yuv[4][3];
|
||
|
|
Uint8 *Y1, *Y2, *U, *V;
|
||
|
|
Uint8 *rgb1, *rgb2;
|
||
|
|
int rgb_row_advance = (pitch - w * 3) + pitch;
|
||
|
|
int UV_advance;
|
||
|
|
|
||
|
|
rgb1 = src;
|
||
|
|
rgb2 = src + pitch;
|
||
|
|
|
||
|
|
Y1 = out;
|
||
|
|
Y2 = Y1 + w;
|
||
|
|
switch (format) {
|
||
|
|
case SDL_PIXELFORMAT_YV12:
|
||
|
|
V = (Y1 + h * w);
|
||
|
|
U = V + ((h + 1) / 2) * ((w + 1) / 2);
|
||
|
|
UV_advance = 1;
|
||
|
|
break;
|
||
|
|
case SDL_PIXELFORMAT_IYUV:
|
||
|
|
U = (Y1 + h * w);
|
||
|
|
V = U + ((h + 1) / 2) * ((w + 1) / 2);
|
||
|
|
UV_advance = 1;
|
||
|
|
break;
|
||
|
|
case SDL_PIXELFORMAT_NV12:
|
||
|
|
U = (Y1 + h * w);
|
||
|
|
V = U + 1;
|
||
|
|
UV_advance = 2;
|
||
|
|
break;
|
||
|
|
case SDL_PIXELFORMAT_NV21:
|
||
|
|
V = (Y1 + h * w);
|
||
|
|
U = V + 1;
|
||
|
|
UV_advance = 2;
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
SDL_assert(!"Unsupported planar YUV format");
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
for (y = 0; y < (h - 1); y += 2) {
|
||
|
|
for (x = 0; x < (w - 1); x += 2) {
|
||
|
|
RGBtoYUV(rgb1, 8, yuv[0], 8, mode, monochrome, luminance);
|
||
|
|
rgb1 += 3;
|
||
|
|
*Y1++ = (Uint8)yuv[0][0];
|
||
|
|
|
||
|
|
RGBtoYUV(rgb1, 8, yuv[1], 8, mode, monochrome, luminance);
|
||
|
|
rgb1 += 3;
|
||
|
|
*Y1++ = (Uint8)yuv[1][0];
|
||
|
|
|
||
|
|
RGBtoYUV(rgb2, 8, yuv[2], 8, mode, monochrome, luminance);
|
||
|
|
rgb2 += 3;
|
||
|
|
*Y2++ = (Uint8)yuv[2][0];
|
||
|
|
|
||
|
|
RGBtoYUV(rgb2, 8, yuv[3], 8, mode, monochrome, luminance);
|
||
|
|
rgb2 += 3;
|
||
|
|
*Y2++ = (Uint8)yuv[3][0];
|
||
|
|
|
||
|
|
*U = (Uint8)SDL_floorf((yuv[0][1] + yuv[1][1] + yuv[2][1] + yuv[3][1]) / 4.0f + 0.5f);
|
||
|
|
U += UV_advance;
|
||
|
|
|
||
|
|
*V = (Uint8)SDL_floorf((yuv[0][2] + yuv[1][2] + yuv[2][2] + yuv[3][2]) / 4.0f + 0.5f);
|
||
|
|
V += UV_advance;
|
||
|
|
}
|
||
|
|
/* Last column */
|
||
|
|
if (x == (w - 1)) {
|
||
|
|
RGBtoYUV(rgb1, 8, yuv[0], 8, mode, monochrome, luminance);
|
||
|
|
rgb1 += 3;
|
||
|
|
*Y1++ = (Uint8)yuv[0][0];
|
||
|
|
|
||
|
|
RGBtoYUV(rgb2, 8, yuv[2], 8, mode, monochrome, luminance);
|
||
|
|
rgb2 += 3;
|
||
|
|
*Y2++ = (Uint8)yuv[2][0];
|
||
|
|
|
||
|
|
*U = (Uint8)SDL_floorf((yuv[0][1] + yuv[2][1]) / 2.0f + 0.5f);
|
||
|
|
U += UV_advance;
|
||
|
|
|
||
|
|
*V = (Uint8)SDL_floorf((yuv[0][2] + yuv[2][2]) / 2.0f + 0.5f);
|
||
|
|
V += UV_advance;
|
||
|
|
}
|
||
|
|
Y1 += w;
|
||
|
|
Y2 += w;
|
||
|
|
rgb1 += rgb_row_advance;
|
||
|
|
rgb2 += rgb_row_advance;
|
||
|
|
}
|
||
|
|
/* Last row */
|
||
|
|
if (y == (h - 1)) {
|
||
|
|
for (x = 0; x < (w - 1); x += 2) {
|
||
|
|
RGBtoYUV(rgb1, 8, yuv[0], 8, mode, monochrome, luminance);
|
||
|
|
rgb1 += 3;
|
||
|
|
*Y1++ = (Uint8)yuv[0][0];
|
||
|
|
|
||
|
|
RGBtoYUV(rgb1, 8, yuv[1], 8, mode, monochrome, luminance);
|
||
|
|
rgb1 += 3;
|
||
|
|
*Y1++ = (Uint8)yuv[1][0];
|
||
|
|
|
||
|
|
*U = (Uint8)SDL_floorf((yuv[0][1] + yuv[1][1]) / 2.0f + 0.5f);
|
||
|
|
U += UV_advance;
|
||
|
|
|
||
|
|
*V = (Uint8)SDL_floorf((yuv[0][2] + yuv[1][2]) / 2.0f + 0.5f);
|
||
|
|
V += UV_advance;
|
||
|
|
}
|
||
|
|
/* Last column */
|
||
|
|
if (x == (w - 1)) {
|
||
|
|
RGBtoYUV(rgb1, 8, yuv[0], 8, mode, monochrome, luminance);
|
||
|
|
*Y1++ = (Uint8)yuv[0][0];
|
||
|
|
|
||
|
|
*U = (Uint8)yuv[0][1];
|
||
|
|
U += UV_advance;
|
||
|
|
|
||
|
|
*V = (Uint8)yuv[0][2];
|
||
|
|
V += UV_advance;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
static Uint16 Pack10to16(int v)
|
||
|
|
{
|
||
|
|
return (Uint16)(v << 6);
|
||
|
|
}
|
||
|
|
|
||
|
|
static void ConvertRGBtoPlanar2x2_P010(Uint32 format, Uint8 *src, int pitch, Uint8 *out, int w, int h, YUV_CONVERSION_MODE mode, int monochrome, int luminance)
|
||
|
|
{
|
||
|
|
int x, y;
|
||
|
|
int yuv[4][3];
|
||
|
|
Uint16 *Y1, *Y2, *U, *V;
|
||
|
|
Uint8 *rgb1, *rgb2;
|
||
|
|
int rgb_row_advance = (pitch - w * 3) + pitch;
|
||
|
|
int UV_advance;
|
||
|
|
|
||
|
|
rgb1 = src;
|
||
|
|
rgb2 = src + pitch;
|
||
|
|
|
||
|
|
Y1 = (Uint16 *)out;
|
||
|
|
Y2 = Y1 + w;
|
||
|
|
switch (format) {
|
||
|
|
case SDL_PIXELFORMAT_P010:
|
||
|
|
U = (Y1 + h * w);
|
||
|
|
V = U + 1;
|
||
|
|
UV_advance = 2;
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
SDL_assert(!"Unsupported planar YUV format");
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
for (y = 0; y < (h - 1); y += 2) {
|
||
|
|
for (x = 0; x < (w - 1); x += 2) {
|
||
|
|
RGBtoYUV(rgb1, 8, yuv[0], 10, mode, monochrome, luminance);
|
||
|
|
rgb1 += 3;
|
||
|
|
*Y1++ = Pack10to16(yuv[0][0]);
|
||
|
|
|
||
|
|
RGBtoYUV(rgb1, 8, yuv[1], 10, mode, monochrome, luminance);
|
||
|
|
rgb1 += 3;
|
||
|
|
*Y1++ = Pack10to16(yuv[1][0]);
|
||
|
|
|
||
|
|
RGBtoYUV(rgb2, 8, yuv[2], 10, mode, monochrome, luminance);
|
||
|
|
rgb2 += 3;
|
||
|
|
*Y2++ = Pack10to16(yuv[2][0]);
|
||
|
|
|
||
|
|
RGBtoYUV(rgb2, 8, yuv[3], 10, mode, monochrome, luminance);
|
||
|
|
rgb2 += 3;
|
||
|
|
*Y2++ = Pack10to16(yuv[3][0]);
|
||
|
|
|
||
|
|
*U = Pack10to16((int)SDL_floorf((yuv[0][1] + yuv[1][1] + yuv[2][1] + yuv[3][1]) / 4.0f + 0.5f));
|
||
|
|
U += UV_advance;
|
||
|
|
|
||
|
|
*V = Pack10to16((int)SDL_floorf((yuv[0][2] + yuv[1][2] + yuv[2][2] + yuv[3][2]) / 4.0f + 0.5f));
|
||
|
|
V += UV_advance;
|
||
|
|
}
|
||
|
|
/* Last column */
|
||
|
|
if (x == (w - 1)) {
|
||
|
|
RGBtoYUV(rgb1, 8, yuv[0], 10, mode, monochrome, luminance);
|
||
|
|
rgb1 += 3;
|
||
|
|
*Y1++ = Pack10to16(yuv[0][0]);
|
||
|
|
|
||
|
|
RGBtoYUV(rgb2, 8, yuv[2], 10, mode, monochrome, luminance);
|
||
|
|
rgb2 += 3;
|
||
|
|
*Y2++ = Pack10to16(yuv[2][0]);
|
||
|
|
|
||
|
|
*U = Pack10to16((int)SDL_floorf((yuv[0][1] + yuv[2][1]) / 2.0f + 0.5f));
|
||
|
|
U += UV_advance;
|
||
|
|
|
||
|
|
*V = Pack10to16((int)SDL_floorf((yuv[0][2] + yuv[2][2]) / 2.0f + 0.5f));
|
||
|
|
V += UV_advance;
|
||
|
|
}
|
||
|
|
Y1 += w;
|
||
|
|
Y2 += w;
|
||
|
|
rgb1 += rgb_row_advance;
|
||
|
|
rgb2 += rgb_row_advance;
|
||
|
|
}
|
||
|
|
/* Last row */
|
||
|
|
if (y == (h - 1)) {
|
||
|
|
for (x = 0; x < (w - 1); x += 2) {
|
||
|
|
RGBtoYUV(rgb1, 8, yuv[0], 10, mode, monochrome, luminance);
|
||
|
|
rgb1 += 3;
|
||
|
|
*Y1++ = Pack10to16(yuv[0][0]);
|
||
|
|
|
||
|
|
RGBtoYUV(rgb1, 8, yuv[1], 10, mode, monochrome, luminance);
|
||
|
|
rgb1 += 3;
|
||
|
|
*Y1++ = Pack10to16(yuv[1][0]);
|
||
|
|
|
||
|
|
*U = Pack10to16((int)SDL_floorf((yuv[0][1] + yuv[1][1]) / 2.0f + 0.5f));
|
||
|
|
U += UV_advance;
|
||
|
|
|
||
|
|
*V = Pack10to16((int)SDL_floorf((yuv[0][2] + yuv[1][2]) / 2.0f + 0.5f));
|
||
|
|
V += UV_advance;
|
||
|
|
}
|
||
|
|
/* Last column */
|
||
|
|
if (x == (w - 1)) {
|
||
|
|
RGBtoYUV(rgb1, 8, yuv[0], 10, mode, monochrome, luminance);
|
||
|
|
*Y1++ = Pack10to16(yuv[0][0]);
|
||
|
|
|
||
|
|
*U = Pack10to16(yuv[0][1]);
|
||
|
|
U += UV_advance;
|
||
|
|
|
||
|
|
*V = Pack10to16(yuv[0][2]);
|
||
|
|
V += UV_advance;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
static void ConvertRGBtoPacked4(Uint32 format, Uint8 *src, int pitch, Uint8 *out, int w, int h, YUV_CONVERSION_MODE mode, int monochrome, int luminance)
|
||
|
|
{
|
||
|
|
int x, y;
|
||
|
|
int yuv[2][3];
|
||
|
|
Uint8 *Y1, *Y2, *U, *V;
|
||
|
|
Uint8 *rgb;
|
||
|
|
int rgb_row_advance = (pitch - w * 3);
|
||
|
|
|
||
|
|
rgb = src;
|
||
|
|
|
||
|
|
switch (format) {
|
||
|
|
case SDL_PIXELFORMAT_YUY2:
|
||
|
|
Y1 = out;
|
||
|
|
U = out + 1;
|
||
|
|
Y2 = out + 2;
|
||
|
|
V = out + 3;
|
||
|
|
break;
|
||
|
|
case SDL_PIXELFORMAT_UYVY:
|
||
|
|
U = out;
|
||
|
|
Y1 = out + 1;
|
||
|
|
V = out + 2;
|
||
|
|
Y2 = out + 3;
|
||
|
|
break;
|
||
|
|
case SDL_PIXELFORMAT_YVYU:
|
||
|
|
Y1 = out;
|
||
|
|
V = out + 1;
|
||
|
|
Y2 = out + 2;
|
||
|
|
U = out + 3;
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
SDL_assert(!"Unsupported packed YUV format");
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
for (y = 0; y < h; ++y) {
|
||
|
|
for (x = 0; x < (w - 1); x += 2) {
|
||
|
|
RGBtoYUV(rgb, 8, yuv[0], 8, mode, monochrome, luminance);
|
||
|
|
rgb += 3;
|
||
|
|
*Y1 = (Uint8)yuv[0][0];
|
||
|
|
Y1 += 4;
|
||
|
|
|
||
|
|
RGBtoYUV(rgb, 8, yuv[1], 8, mode, monochrome, luminance);
|
||
|
|
rgb += 3;
|
||
|
|
*Y2 = (Uint8)yuv[1][0];
|
||
|
|
Y2 += 4;
|
||
|
|
|
||
|
|
*U = (Uint8)SDL_floorf((yuv[0][1] + yuv[1][1]) / 2.0f + 0.5f);
|
||
|
|
U += 4;
|
||
|
|
|
||
|
|
*V = (Uint8)SDL_floorf((yuv[0][2] + yuv[1][2]) / 2.0f + 0.5f);
|
||
|
|
V += 4;
|
||
|
|
}
|
||
|
|
/* Last column */
|
||
|
|
if (x == (w - 1)) {
|
||
|
|
RGBtoYUV(rgb, 8, yuv[0], 8, mode, monochrome, luminance);
|
||
|
|
rgb += 3;
|
||
|
|
*Y2 = *Y1 = (Uint8)yuv[0][0];
|
||
|
|
Y1 += 4;
|
||
|
|
Y2 += 4;
|
||
|
|
|
||
|
|
*U = (Uint8)yuv[0][1];
|
||
|
|
U += 4;
|
||
|
|
|
||
|
|
*V = (Uint8)yuv[0][2];
|
||
|
|
V += 4;
|
||
|
|
}
|
||
|
|
rgb += rgb_row_advance;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
SDL_bool ConvertRGBtoYUV(Uint32 format, Uint8 *src, int pitch, Uint8 *out, int w, int h, YUV_CONVERSION_MODE mode, int monochrome, int luminance)
|
||
|
|
{
|
||
|
|
switch (format) {
|
||
|
|
case SDL_PIXELFORMAT_P010:
|
||
|
|
ConvertRGBtoPlanar2x2_P010(format, src, pitch, out, w, h, mode, monochrome, luminance);
|
||
|
|
return SDL_TRUE;
|
||
|
|
case SDL_PIXELFORMAT_YV12:
|
||
|
|
case SDL_PIXELFORMAT_IYUV:
|
||
|
|
case SDL_PIXELFORMAT_NV12:
|
||
|
|
case SDL_PIXELFORMAT_NV21:
|
||
|
|
ConvertRGBtoPlanar2x2(format, src, pitch, out, w, h, mode, monochrome, luminance);
|
||
|
|
return SDL_TRUE;
|
||
|
|
case SDL_PIXELFORMAT_YUY2:
|
||
|
|
case SDL_PIXELFORMAT_UYVY:
|
||
|
|
case SDL_PIXELFORMAT_YVYU:
|
||
|
|
ConvertRGBtoPacked4(format, src, pitch, out, w, h, mode, monochrome, luminance);
|
||
|
|
return SDL_TRUE;
|
||
|
|
default:
|
||
|
|
return SDL_FALSE;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
int CalculateYUVPitch(Uint32 format, int width)
|
||
|
|
{
|
||
|
|
switch (format) {
|
||
|
|
case SDL_PIXELFORMAT_P010:
|
||
|
|
return width * 2;
|
||
|
|
case SDL_PIXELFORMAT_YV12:
|
||
|
|
case SDL_PIXELFORMAT_IYUV:
|
||
|
|
case SDL_PIXELFORMAT_NV12:
|
||
|
|
case SDL_PIXELFORMAT_NV21:
|
||
|
|
return width;
|
||
|
|
case SDL_PIXELFORMAT_YUY2:
|
||
|
|
case SDL_PIXELFORMAT_UYVY:
|
||
|
|
case SDL_PIXELFORMAT_YVYU:
|
||
|
|
return 4 * ((width + 1) / 2);
|
||
|
|
default:
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
}
|