mirror of https://github.com/mpv-player/mpv
cosmetics: csputils.[ch]: reindent
This commit is contained in:
Uoti Urpala
parent
506ab685d4
commit
1478f658f3
242
libvo/csputils.c
242
libvo/csputils.c
|
|
@ -36,105 +36,111 @@
|
|||
* \param size size of buffer
|
||||
* \param gamma gamma value
|
||||
*/
|
||||
void mp_gen_gamma_map(uint8_t *map, int size, float gamma) {
|
||||
int i;
|
||||
if (gamma == 1.0) {
|
||||
for (i = 0; i < size; i++)
|
||||
map[i] = 255 * i / (size - 1);
|
||||
return;
|
||||
}
|
||||
gamma = 1.0 / gamma;
|
||||
for (i = 0; i < size; i++) {
|
||||
float tmp = (float)i / (size - 1.0);
|
||||
tmp = pow(tmp, gamma);
|
||||
if (tmp > 1.0) tmp = 1.0;
|
||||
if (tmp < 0.0) tmp = 0.0;
|
||||
map[i] = 255 * tmp;
|
||||
}
|
||||
void mp_gen_gamma_map(uint8_t *map, int size, float gamma)
|
||||
{
|
||||
if (gamma == 1.0) {
|
||||
for (int i = 0; i < size; i++)
|
||||
map[i] = 255 * i / (size - 1);
|
||||
return;
|
||||
}
|
||||
gamma = 1.0 / gamma;
|
||||
for (int i = 0; i < size; i++) {
|
||||
float tmp = (float)i / (size - 1.0);
|
||||
tmp = pow(tmp, gamma);
|
||||
if (tmp > 1.0)
|
||||
tmp = 1.0;
|
||||
if (tmp < 0.0)
|
||||
tmp = 0.0;
|
||||
map[i] = 255 * tmp;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief get the coefficients of the yuv -> rgb conversion matrix
|
||||
* \param params struct specifying the properties of the conversion like brightness, ...
|
||||
* \param params struct specifying the properties of the conversion like
|
||||
* brightness, ...
|
||||
* \param yuv2rgb array to store coefficients into
|
||||
*
|
||||
* Note: contrast, hue and saturation will only work as expected with YUV formats,
|
||||
* not with e.g. MP_CSP_XYZ
|
||||
* Note: contrast, hue and saturation will only work as expected with YUV
|
||||
* formats, not with e.g. MP_CSP_XYZ
|
||||
*/
|
||||
void mp_get_yuv2rgb_coeffs(struct mp_csp_params *params, float yuv2rgb[3][4]) {
|
||||
float depth_multiplier = params->input_shift >= 0 ?
|
||||
(1 << params->input_shift) :
|
||||
(1.0 / (1 << -params->input_shift));
|
||||
float uvcos = params->saturation * cos(params->hue);
|
||||
float uvsin = params->saturation * sin(params->hue);
|
||||
int format = params->format;
|
||||
int levelconv = params->levelconv;
|
||||
int i;
|
||||
const float (*uv_coeffs)[3];
|
||||
const float *level_adjust;
|
||||
static const float yuv_level_adjust[MP_CSP_LEVELCONV_COUNT][4] = {
|
||||
{-16 / 255.0, -128 / 255.0, -128 / 255.0, 1.164},
|
||||
{ 16 / 255.0 * 1.164, -128 / 255.0, -128 / 255.0, 1.0/1.164},
|
||||
{ 0, -128 / 255.0, -128 / 255.0, 1},
|
||||
};
|
||||
static const float xyz_level_adjust[4] = {0, 0, 0, 0};
|
||||
static const float uv_coeffs_table[MP_CSP_COUNT][3][3] = {
|
||||
[MP_CSP_DEFAULT] = {
|
||||
{1, 0.000, 1.596},
|
||||
{1, -0.391, -0.813},
|
||||
{1, 2.018, 0.000}
|
||||
},
|
||||
[MP_CSP_BT_601] = {
|
||||
{1, 0.000, 1.403},
|
||||
{1, -0.344, -0.714},
|
||||
{1, 1.773, 0.000}
|
||||
},
|
||||
[MP_CSP_BT_709] = {
|
||||
{1, 0.0000, 1.5701},
|
||||
{1, -0.1870, -0.4664},
|
||||
{1, 1.8556, 0.0000}
|
||||
},
|
||||
[MP_CSP_SMPTE_240M] = {
|
||||
{1, 0.0000, 1.5756},
|
||||
{1, -0.2253, -0.5000},
|
||||
{1, 1.8270, 0.0000}
|
||||
},
|
||||
[MP_CSP_EBU] = {
|
||||
{1, 0.000, 1.140},
|
||||
{1, -0.396, -0.581},
|
||||
{1, 2.029, 0.000}
|
||||
},
|
||||
[MP_CSP_XYZ] = {
|
||||
{ 3.2404542, -1.5371385, -0.4985314},
|
||||
{-0.9692660, 1.8760108, 0.0415560},
|
||||
{ 0.0556434, -0.2040259, 1.0572252}
|
||||
},
|
||||
};
|
||||
void mp_get_yuv2rgb_coeffs(struct mp_csp_params *params, float yuv2rgb[3][4])
|
||||
{
|
||||
float depth_multiplier = params->input_shift >= 0 ?
|
||||
(1 << params->input_shift) :
|
||||
(1.0 / (1 << -params->input_shift));
|
||||
float uvcos = params->saturation * cos(params->hue);
|
||||
float uvsin = params->saturation * sin(params->hue);
|
||||
int format = params->format;
|
||||
int levelconv = params->levelconv;
|
||||
int i;
|
||||
const float (*uv_coeffs)[3];
|
||||
const float *level_adjust;
|
||||
static const float yuv_level_adjust[MP_CSP_LEVELCONV_COUNT][4] = {
|
||||
{-16 / 255.0, -128 / 255.0, -128 / 255.0, 1.164},
|
||||
{ 16 / 255.0 * 1.164, -128 / 255.0, -128 / 255.0, 1.0 / 1.164},
|
||||
{ 0, -128 / 255.0, -128 / 255.0, 1},
|
||||
};
|
||||
static const float xyz_level_adjust[4] = {
|
||||
0, 0, 0, 0
|
||||
};
|
||||
static const float uv_coeffs_table[MP_CSP_COUNT][3][3] = {
|
||||
[MP_CSP_DEFAULT] = {
|
||||
{1, 0.000, 1.596},
|
||||
{1, -0.391, -0.813},
|
||||
{1, 2.018, 0.000}
|
||||
},
|
||||
[MP_CSP_BT_601] = {
|
||||
{1, 0.000, 1.403},
|
||||
{1, -0.344, -0.714},
|
||||
{1, 1.773, 0.000}
|
||||
},
|
||||
[MP_CSP_BT_709] = {
|
||||
{1, 0.0000, 1.5701},
|
||||
{1, -0.1870, -0.4664},
|
||||
{1, 1.8556, 0.0000}
|
||||
},
|
||||
[MP_CSP_SMPTE_240M] = {
|
||||
{1, 0.0000, 1.5756},
|
||||
{1, -0.2253, -0.5000},
|
||||
{1, 1.8270, 0.0000}
|
||||
},
|
||||
[MP_CSP_EBU] = {
|
||||
{1, 0.000, 1.140},
|
||||
{1, -0.396, -0.581},
|
||||
{1, 2.029, 0.000}
|
||||
},
|
||||
[MP_CSP_XYZ] = {
|
||||
{ 3.2404542, -1.5371385, -0.4985314},
|
||||
{-0.9692660, 1.8760108, 0.0415560},
|
||||
{ 0.0556434, -0.2040259, 1.0572252}
|
||||
},
|
||||
};
|
||||
|
||||
if (format < 0 || format >= MP_CSP_COUNT)
|
||||
format = MP_CSP_DEFAULT;
|
||||
uv_coeffs = uv_coeffs_table[format];
|
||||
if (levelconv < 0 || levelconv >= MP_CSP_LEVELCONV_COUNT)
|
||||
levelconv = MP_CSP_LEVELCONV_TV_TO_PC;
|
||||
level_adjust = yuv_level_adjust[levelconv];
|
||||
if (format == MP_CSP_XYZ)
|
||||
level_adjust = xyz_level_adjust;
|
||||
if (format < 0 || format >= MP_CSP_COUNT)
|
||||
format = MP_CSP_DEFAULT;
|
||||
uv_coeffs = uv_coeffs_table[format];
|
||||
if (levelconv < 0 || levelconv >= MP_CSP_LEVELCONV_COUNT)
|
||||
levelconv = MP_CSP_LEVELCONV_TV_TO_PC;
|
||||
level_adjust = yuv_level_adjust[levelconv];
|
||||
if (format == MP_CSP_XYZ)
|
||||
level_adjust = xyz_level_adjust;
|
||||
|
||||
for (i = 0; i < 3; i++) {
|
||||
yuv2rgb[i][COL_C] = params->brightness;
|
||||
yuv2rgb[i][COL_Y] = uv_coeffs[i][COL_Y] * level_adjust[COL_C] * params->contrast;
|
||||
yuv2rgb[i][COL_C] += level_adjust[COL_Y] * yuv2rgb[i][COL_Y];
|
||||
yuv2rgb[i][COL_U] = uv_coeffs[i][COL_U] * uvcos + uv_coeffs[i][COL_V] * uvsin;
|
||||
yuv2rgb[i][COL_C] += level_adjust[COL_U] * yuv2rgb[i][COL_U];
|
||||
yuv2rgb[i][COL_V] = uv_coeffs[i][COL_U] * uvsin + uv_coeffs[i][COL_V] * uvcos;
|
||||
yuv2rgb[i][COL_C] += level_adjust[COL_V] * yuv2rgb[i][COL_V];
|
||||
// this "centers" contrast control so that e.g. a contrast of 0
|
||||
// leads to a grey image, not a black one
|
||||
yuv2rgb[i][COL_C] += 0.5 - params->contrast / 2.0;
|
||||
yuv2rgb[i][COL_Y] *= depth_multiplier;
|
||||
yuv2rgb[i][COL_U] *= depth_multiplier;
|
||||
yuv2rgb[i][COL_V] *= depth_multiplier;
|
||||
}
|
||||
for (i = 0; i < 3; i++) {
|
||||
yuv2rgb[i][COL_C] = params->brightness;
|
||||
yuv2rgb[i][COL_Y] = uv_coeffs[i][COL_Y] * level_adjust[COL_C] * params->contrast;
|
||||
yuv2rgb[i][COL_C] += level_adjust[COL_Y] * yuv2rgb[i][COL_Y];
|
||||
yuv2rgb[i][COL_U] = uv_coeffs[i][COL_U] * uvcos + uv_coeffs[i][COL_V] * uvsin;
|
||||
yuv2rgb[i][COL_C] += level_adjust[COL_U] * yuv2rgb[i][COL_U];
|
||||
yuv2rgb[i][COL_V] = uv_coeffs[i][COL_U] * uvsin + uv_coeffs[i][COL_V] * uvcos;
|
||||
yuv2rgb[i][COL_C] += level_adjust[COL_V] * yuv2rgb[i][COL_V];
|
||||
// this "centers" contrast control so that e.g. a contrast of 0
|
||||
// leads to a grey image, not a black one
|
||||
yuv2rgb[i][COL_C] += 0.5 - params->contrast / 2.0;
|
||||
yuv2rgb[i][COL_Y] *= depth_multiplier;
|
||||
yuv2rgb[i][COL_U] *= depth_multiplier;
|
||||
yuv2rgb[i][COL_V] *= depth_multiplier;
|
||||
}
|
||||
}
|
||||
|
||||
//! size of gamma map use to avoid slow exp function in gen_yuv2rgb_map
|
||||
|
|
@ -145,33 +151,35 @@ void mp_get_yuv2rgb_coeffs(struct mp_csp_params *params, float yuv2rgb[3][4]) {
|
|||
* \param map where to store map. Must provide space for (size + 2)^3 elements
|
||||
* \param size size of the map, excluding border
|
||||
*/
|
||||
void mp_gen_yuv2rgb_map(struct mp_csp_params *params, unsigned char *map, int size) {
|
||||
int i, j, k, l;
|
||||
float step = 1.0 / size;
|
||||
float y, u, v;
|
||||
float yuv2rgb[3][4];
|
||||
unsigned char gmaps[3][GMAP_SIZE];
|
||||
mp_gen_gamma_map(gmaps[0], GMAP_SIZE, params->rgamma);
|
||||
mp_gen_gamma_map(gmaps[1], GMAP_SIZE, params->ggamma);
|
||||
mp_gen_gamma_map(gmaps[2], GMAP_SIZE, params->bgamma);
|
||||
mp_get_yuv2rgb_coeffs(params, yuv2rgb);
|
||||
for (i = 0; i < 3; i++)
|
||||
for (j = 0; j < 4; j++)
|
||||
yuv2rgb[i][j] *= GMAP_SIZE - 1;
|
||||
v = 0;
|
||||
for (i = -1; i <= size; i++) {
|
||||
u = 0;
|
||||
for (j = -1; j <= size; j++) {
|
||||
y = 0;
|
||||
for (k = -1; k <= size; k++) {
|
||||
for (l = 0; l < 3; l++) {
|
||||
float rgb = yuv2rgb[l][COL_Y] * y + yuv2rgb[l][COL_U] * u + yuv2rgb[l][COL_V] * v + yuv2rgb[l][COL_C];
|
||||
*map++ = gmaps[l][av_clip(rgb, 0, GMAP_SIZE - 1)];
|
||||
void mp_gen_yuv2rgb_map(struct mp_csp_params *params, unsigned char *map, int size)
|
||||
{
|
||||
int i, j, k, l;
|
||||
float step = 1.0 / size;
|
||||
float y, u, v;
|
||||
float yuv2rgb[3][4];
|
||||
unsigned char gmaps[3][GMAP_SIZE];
|
||||
mp_gen_gamma_map(gmaps[0], GMAP_SIZE, params->rgamma);
|
||||
mp_gen_gamma_map(gmaps[1], GMAP_SIZE, params->ggamma);
|
||||
mp_gen_gamma_map(gmaps[2], GMAP_SIZE, params->bgamma);
|
||||
mp_get_yuv2rgb_coeffs(params, yuv2rgb);
|
||||
for (i = 0; i < 3; i++)
|
||||
for (j = 0; j < 4; j++)
|
||||
yuv2rgb[i][j] *= GMAP_SIZE - 1;
|
||||
v = 0;
|
||||
for (i = -1; i <= size; i++) {
|
||||
u = 0;
|
||||
for (j = -1; j <= size; j++) {
|
||||
y = 0;
|
||||
for (k = -1; k <= size; k++) {
|
||||
for (l = 0; l < 3; l++) {
|
||||
float rgb = yuv2rgb[l][COL_Y] * y + yuv2rgb[l][COL_U] * u +
|
||||
yuv2rgb[l][COL_V] * v + yuv2rgb[l][COL_C];
|
||||
*map++ = gmaps[l][av_clip(rgb, 0, GMAP_SIZE - 1)];
|
||||
}
|
||||
y += (k == -1 || k == size - 1) ? step / 2 : step;
|
||||
}
|
||||
u += (j == -1 || j == size - 1) ? step / 2 : step;
|
||||
}
|
||||
y += (k == -1 || k == size - 1) ? step / 2 : step;
|
||||
}
|
||||
u += (j == -1 || j == size - 1) ? step / 2 : step;
|
||||
v += (i == -1 || i == size - 1) ? step / 2 : step;
|
||||
}
|
||||
v += (i == -1 || i == size - 1) ? step / 2 : step;
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -27,33 +27,33 @@
|
|||
#include <stdint.h>
|
||||
|
||||
enum mp_csp_standard {
|
||||
MP_CSP_DEFAULT,
|
||||
MP_CSP_BT_601,
|
||||
MP_CSP_BT_709,
|
||||
MP_CSP_SMPTE_240M,
|
||||
MP_CSP_EBU,
|
||||
MP_CSP_XYZ,
|
||||
MP_CSP_COUNT
|
||||
MP_CSP_DEFAULT,
|
||||
MP_CSP_BT_601,
|
||||
MP_CSP_BT_709,
|
||||
MP_CSP_SMPTE_240M,
|
||||
MP_CSP_EBU,
|
||||
MP_CSP_XYZ,
|
||||
MP_CSP_COUNT
|
||||
};
|
||||
|
||||
enum mp_csp_levelconv {
|
||||
MP_CSP_LEVELCONV_TV_TO_PC,
|
||||
MP_CSP_LEVELCONV_PC_TO_TV,
|
||||
MP_CSP_LEVELCONV_NONE,
|
||||
MP_CSP_LEVELCONV_COUNT
|
||||
MP_CSP_LEVELCONV_TV_TO_PC,
|
||||
MP_CSP_LEVELCONV_PC_TO_TV,
|
||||
MP_CSP_LEVELCONV_NONE,
|
||||
MP_CSP_LEVELCONV_COUNT
|
||||
};
|
||||
|
||||
struct mp_csp_params {
|
||||
enum mp_csp_standard format;
|
||||
enum mp_csp_levelconv levelconv;
|
||||
float brightness;
|
||||
float contrast;
|
||||
float hue;
|
||||
float saturation;
|
||||
float rgamma;
|
||||
float ggamma;
|
||||
float bgamma;
|
||||
int input_shift;
|
||||
enum mp_csp_standard format;
|
||||
enum mp_csp_levelconv levelconv;
|
||||
float brightness;
|
||||
float contrast;
|
||||
float hue;
|
||||
float saturation;
|
||||
float rgamma;
|
||||
float ggamma;
|
||||
float bgamma;
|
||||
int input_shift;
|
||||
};
|
||||
|
||||
void mp_gen_gamma_map(unsigned char *map, int size, float gamma);
|
||||
|
|
|
|||
Loading…
Reference in New Issue