src/util/format/u_format_yuv.h

   1 /**************************************************************************
   2  *
   3  * Copyright 2010 VMware, Inc.
   4  * All Rights Reserved.
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a
   7  * copy of this software and associated documentation files (the
   8  * "Software"), to deal in the Software without restriction, including
   9  * without limitation the rights to use, copy, modify, merge, publish,
  10  * distribute, sub license, and/or sell copies of the Software, and to
  11  * permit persons to whom the Software is furnished to do so, subject to
  12  * the following conditions:
  13  *
  14  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  15  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  16  * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL
  17  * THE COPYRIGHT HOLDERS, AUTHORS AND/OR ITS SUPPLIERS BE LIABLE FOR ANY CLAIM,
  18  * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
  19  * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
  20  * USE OR OTHER DEALINGS IN THE SOFTWARE.
  21  *
  22  * The above copyright notice and this permission notice (including the
  23  * next paragraph) shall be included in all copies or substantial portions
  24  * of the Software.
  25  *
  26  **************************************************************************/
  27
  28
  29 /**
  30  * @file
  31  * YUV colorspace conversion.
  32  *
  33  * @author Brian Paul <brianp@vmware.com>
  34  * @author Michal Krol <michal@vmware.com>
  35  * @author Jose Fonseca <jfonseca@vmware.com>
  36  *
  37  * See also:
  38  * - http://www.fourcc.org/fccyvrgb.php
  39  * - http://msdn.microsoft.com/en-us/library/ms893078
  40  * - http://en.wikipedia.org/wiki/YUV
  41  */
  42
  43
  44 #ifndef U_FORMAT_YUV_H_
  45 #define U_FORMAT_YUV_H_
  46
  47
  48 #include "pipe/p_compiler.h"
  49 #include "util/u_math.h"
  50
  51
  52 /*
  53  * TODO: Ensure we use consistent and right floating formulas, with enough
  54  * precision in the coefficients.
  55  */
  56
  57 static inline void
  58 util_format_rgb_float_to_yuv(float r, float g, float b,
  59                              uint8_t *y, uint8_t *u, uint8_t *v)
  60 {
  61    const float _r = SATURATE(r);
  62    const float _g = SATURATE(g);
  63    const float _b = SATURATE(b);
  64
  65    const float scale = 255.0f;
  66
  67    const int _y = scale * ( (0.257f * _r) + (0.504f * _g) + (0.098f * _b));
  68    const int _u = scale * (-(0.148f * _r) - (0.291f * _g) + (0.439f * _b));
  69    const int _v = scale * ( (0.439f * _r) - (0.368f * _g) - (0.071f * _b));
  70
  71    *y = _y + 16;
  72    *u = _u + 128;
  73    *v = _v + 128;
  74 }
  75
  76
  77 static inline void
  78 util_format_yuv_to_rgb_float(uint8_t y, uint8_t u, uint8_t v,
  79                              float *r, float *g, float *b)
  80 {
  81    const int _y = y - 16;
  82    const int _u = u - 128;
  83    const int _v = v - 128;
  84
  85    const float y_factor = 255.0f / 219.0f;
  86
  87    const float scale = 1.0f / 255.0f;
  88
  89    *r = scale * (y_factor * _y               + 1.596f * _v);
  90    *g = scale * (y_factor * _y - 0.391f * _u - 0.813f * _v);
  91    *b = scale * (y_factor * _y + 2.018f * _u              );
  92 }
  93
  94
  95 static inline void
  96 util_format_rgb_8unorm_to_yuv(uint8_t r, uint8_t g, uint8_t b,
  97                               uint8_t *y, uint8_t *u, uint8_t *v)
  98 {
  99    *y = ((  66 * r + 129 * g +  25 * b + 128) >> 8) +  16;
 100    *u = (( -38 * r -  74 * g + 112 * b + 128) >> 8) + 128;
 101    *v = (( 112 * r -  94 * g -  18 * b + 128) >> 8) + 128;
 102 }
 103
 104
 105 static inline void
 106 util_format_yuv_to_rgb_8unorm(uint8_t y, uint8_t u, uint8_t v,
 107                               uint8_t *r, uint8_t *g, uint8_t *b)
 108 {
 109    const int _y = y - 16;
 110    const int _u = u - 128;
 111    const int _v = v - 128;
 112
 113    const int _r = (298 * _y            + 409 * _v + 128) >> 8;
 114    const int _g = (298 * _y - 100 * _u - 208 * _v + 128) >> 8;
 115    const int _b = (298 * _y + 516 * _u            + 128) >> 8;
 116
 117    *r = CLAMP(_r, 0, 255);
 118    *g = CLAMP(_g, 0, 255);
 119    *b = CLAMP(_b, 0, 255);
 120 }
 121
 122
 123
 124 void
 125 util_format_uyvy_unpack_rgba_float(void *dst_row, unsigned dst_stride,
 126                               const uint8_t *src_row, unsigned src_stride,
 127                               unsigned width, unsigned height);
 128
 129 void
 130 util_format_uyvy_unpack_rgba_8unorm(uint8_t *dst_row, unsigned dst_stride,
 131                                const uint8_t *src_row, unsigned src_stride,
 132                                unsigned width, unsigned height);
 133
 134 void
 135 util_format_uyvy_pack_rgba_float(uint8_t *dst_row, unsigned dst_stride,
 136                             const float *src_row, unsigned src_stride,
 137                             unsigned width, unsigned height);
 138
 139 void
 140 util_format_uyvy_pack_rgba_8unorm(uint8_t *dst_row, unsigned dst_stride,
 141                              const uint8_t *src_row, unsigned src_stride,
 142                              unsigned width, unsigned height);
 143
 144 void
 145 util_format_uyvy_fetch_rgba(void *dst, const uint8_t *src,
 146                              unsigned i, unsigned j);
 147
 148 void
 149 util_format_yuyv_unpack_rgba_float(void *dst_row, unsigned dst_stride,
 150                               const uint8_t *src_row, unsigned src_stride,
 151                               unsigned width, unsigned height);
 152
 153 void
 154 util_format_yuyv_unpack_rgba_8unorm(uint8_t *dst_row, unsigned dst_stride,
 155                                const uint8_t *src_row, unsigned src_stride,
 156                                unsigned width, unsigned height);
 157
 158 void
 159 util_format_yuyv_pack_rgba_float(uint8_t *dst_row, unsigned dst_stride,
 160                             const float *src_row, unsigned src_stride,
 161                             unsigned width, unsigned height);
 162
 163 void
 164 util_format_yuyv_pack_rgba_8unorm(uint8_t *dst_row, unsigned dst_stride,
 165                              const uint8_t *src_row, unsigned src_stride,
 166                              unsigned width, unsigned height);
 167
 168 void
 169 util_format_yuyv_fetch_rgba(void *dst, const uint8_t *src,
 170                              unsigned i, unsigned j);
 171
 172 void
 173 util_format_r8g8_b8g8_unorm_unpack_rgba_float(void *dst_row, unsigned dst_stride,
 174                                          const uint8_t *src_row, unsigned src_stride,
 175                                          unsigned width, unsigned height);
 176
 177 void
 178 util_format_r8g8_b8g8_unorm_unpack_rgba_8unorm(uint8_t *dst_row, unsigned dst_stride,
 179                                           const uint8_t *src_row, unsigned src_stride,
 180                                           unsigned width, unsigned height);
 181
 182 void
 183 util_format_r8g8_b8g8_unorm_pack_rgba_float(uint8_t *dst_row, unsigned dst_stride,
 184                                        const float *src_row, unsigned src_stride,
 185                                        unsigned width, unsigned height);
 186
 187 void
 188 util_format_r8g8_b8g8_unorm_pack_rgba_8unorm(uint8_t *dst_row, unsigned dst_stride,
 189                                         const uint8_t *src_row, unsigned src_stride,
 190                                         unsigned width, unsigned height);
 191
 192 void
 193 util_format_r8g8_b8g8_unorm_fetch_rgba(void *dst, const uint8_t *src,
 194                                         unsigned i, unsigned j);
 195
 196 void
 197 util_format_g8r8_g8b8_unorm_unpack_rgba_float(void *dst_row, unsigned dst_stride,
 198                                          const uint8_t *src_row, unsigned src_stride,
 199                                          unsigned width, unsigned height);
 200
 201 void
 202 util_format_g8r8_g8b8_unorm_unpack_rgba_8unorm(uint8_t *dst_row, unsigned dst_stride,
 203                                           const uint8_t *src_row, unsigned src_stride,
 204                                           unsigned width, unsigned height);
 205
 206 void
 207 util_format_g8r8_g8b8_unorm_pack_rgba_float(uint8_t *dst_row, unsigned dst_stride,
 208                                        const float *src_row, unsigned src_stride,
 209                                        unsigned width, unsigned height);
 210
 211 void
 212 util_format_g8r8_g8b8_unorm_pack_rgba_8unorm(uint8_t *dst_row, unsigned dst_stride,
 213                                         const uint8_t *src_row, unsigned src_stride,
 214                                         unsigned width, unsigned height);
 215
 216 void
 217 util_format_g8r8_g8b8_unorm_fetch_rgba(void *dst, const uint8_t *src,
 218                                         unsigned i, unsigned j);
 219
 220 #endif /* U_FORMAT_YUV_H_ */