summaryrefslogtreecommitdiffstats
path: root/libvo/csputils.c
diff options
context:
space:
mode:
Diffstat (limited to 'libvo/csputils.c')
-rw-r--r--libvo/csputils.c242
1 files changed, 125 insertions, 117 deletions
diff --git a/libvo/csputils.c b/libvo/csputils.c
index 0a58aa5a0e..7c488f77b0 100644
--- a/libvo/csputils.c
+++ b/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;
- }
}