Mercurial > mplayer.hg
view libvo/csputils.c @ 36578:dd4ceae6520a
mplayer.c: Minor loop simplification.
author | reimar |
---|---|
date | Mon, 20 Jan 2014 22:08:44 +0000 |
parents | 9bcec2bb02b3 |
children |
line wrap: on
line source
/* * Common code related to colorspaces and conversion * * Copyleft (C) 2009 Reimar Döffinger <Reimar.Doeffinger@gmx.de> * * This file is part of MPlayer. * * MPlayer is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * MPlayer is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License along * with MPlayer; if not, write to the Free Software Foundation, Inc., * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. * * You can alternatively redistribute this file and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. */ #include <stdint.h> #include <math.h> #include "libavutil/common.h" #include "csputils.h" /** * \brief little helper function to create a lookup table for gamma * \param map buffer to create map into * \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; } } /** * \brief get the coefficients of the yuv -> rgb conversion matrix * \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 */ void mp_get_yuv2rgb_coeffs(const 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, 1}; 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; 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; } } static float int2pos(int i, int size) { if (i == 0) return 0; if (i == size - 1) return 1; return (i + 0.5) / size; } //! size of gamma map use to avoid slow exp function in gen_yuv2rgb_map #define GMAP_SIZE (1024) /** * \brief generate a 3D YUV -> RGB map * \param params struct containing parameters like brightness, gamma, ... * \param map where to store map. Must provide space for size^3 elements * \param size size of the map */ void mp_gen_yuv2rgb_map(const struct mp_csp_params *params, unsigned char *map, int size) { int i, j, k, l; 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; for (i = 0; i < size; i++) { float v = int2pos(i, size); for (j = 0; j < size; j++) { float u = int2pos(j, size); for (k = 0; k < size; k++) { float y = int2pos(k, size); 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)]; } } } } }