cinelerra-5.1/guicast/bccolors.h

   1
   2 /*
   3  * CINELERRA
   4  * Copyright (C) 1997-2011 Adam Williams <broadcast at earthling dot net>
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software
  18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  *
  20  */
  21
  22 #ifndef __BCCOLORS_H__
  23 #define __BCCOLORS_H__
  24
  25 // Duplicate filename in guicast
  26
  27 #include "bccolors.inc"
  28 #include "clip.h"
  29 #include "vframe.inc"
  30
  31 #include <stdint.h>
  32
  33
  34 /*
  35 Digital YCbCr is derived from analog RGB as follows:
  36 RGB are 0..1 (gamma corrected supposedly),
  37 normalized equations:
  38 Y  Py =   Kr * R  +  Kg * G  +  Kb * B
  39 U  Pb = - 0.5*Kr/(1-Kb)*R - 0.5*Kg/(1-Kb)*G + 0.5*B
  40 V  Pr =   0.5*R - 0.5*Kg/(1-Kr)*G - 0.5*Kb/(1-Kr)*B
  41 inverse:
  42    R = Py + Pr * 2*(1-Kr)
  43    G = Py - Pr * 2*Kr*(1-Kr)/Kg - Pb * 2*Kb*(1-Kb)/Kg
  44    B = Py                       + Pb * 2*(1-Kb)
  45
  46 bt601, white vector= Kr=0.299,Kg=0.587,Kb=0.114
  47    Py = + 0.299000*R + 0.587000*G + 0.114000*B
  48    Pb = - 0.168736*R - 0.331264*G + 0.500000*B
  49    Pr = + 0.500000*R - 0.418698*G - 0.081312*B
  50 inverse:
  51    R  = Py + 1.402000*Pr
  52    G  = Py - 0.714136*Pr - 0.344136*Pb
  53    B  = Py               + 1.772000*Pb
  54
  55 equations zeroed at (0,128,128), range (255,255,255)
  56 bt.601  0..255
  57    Y  = 0   + 0.299000*R + 0.587000*G + 0.114000*B
  58    Cb = 128 - 0.168736*R - 0.331264*G + 0.500000*B
  59    Cr = 128 + 0.500000*R - 0.418698*G - 0.081312*B
  60 inverse:
  61    R  = Y + 1.402000*(Cr-128)
  62    G  = Y - 0.714136*(Cr-128) - 0.344136*(Cb-128)
  63    B  = Y                     + 1.772000*(Cb-128)
  64
  65 equations zeroed at (16,128,128), range (219,224,224)
  66 Y,Cb,Cr = (16,128,128) + (219*Py,224*Pb,224*Pr)
  67
  68 bt.601  16..235
  69    Y  = 16  + 0.256788*R + 0.504129*G + 0.097906*B
  70    Cb = 128 - 0.148227*R - 0.290992*G + 0.439216*B
  71    Cr = 128 + 0.439216*R - 0.367789*G - 0.071426*B
  72 inverse:
  73    R  = (Y-16)*1.164384 + 1.596027*(Cr-128)
  74    G  = (Y-16)*1.164384 - 0.812968*(Cr-128) - 0.391762*(Cb-128)
  75    B  = (Y-16)*1.164384                     + 2.017232*(Cb-128)
  76
  77 bt.709, white vector= Kr=0.2126,Kg=0.7152,Kb=0.0722
  78    Py = + 0.212600*R + 0.715200*G + 0.072200*B
  79    Pb = - 0.114572*R - 0.385428*G + 0.500000*B
  80    Pr = + 0.500000*R - 0.454153*G - 0.045847*B
  81
  82 equations zeroed at (0,128,128), range (255,255,255)
  83 bt.709  0..255
  84    Y  = 0   + 0.212600*R + 0.715200*G + 0.072200*B
  85    Cb = 128 - 0.114572*R - 0.385428*G + 0.500000*B
  86    Cr = 128 + 0.500000*R - 0.454153*G - 0.045847*B
  87 inverse:
  88    R = Y + 1.574800*(Cr-128)
  89    G = Y - 0.468124*(Cr-128) - 0.187324*(Cb-128)
  90    B = Y                     + 1.855600*(Cb-128)
  91
  92 equations zeroed at (16,128,128), range (219,224,224)
  93 Y,Cb,Cr = (16,128,128) + (219*Py,224*Pb,224*Pr)
  94    Y  = 16  + 0.182586*R + 0.614231*G + 0.062007*B
  95    Cb = 128 - 0.100644*R - 0.338572*G + 0.439216*B
  96    Cr = 128 + 0.439216*R - 0.398942*G - 0.040276*B
  97 inverse:
  98    R = (Y-16)*1.164384 + 1.792741*(Cr-128)
  99    G = (Y-16)*1.164384 - 0.532909*(Cr-128) - 0.213249*(Cb-128)
 100    B = (Y-16)*1.164384                     + 2.112402*(Cb-128)
 101
 102 */
 103 // white vector normalized, so:
 104 //  Kg = 1 - Kr - Kb
 105
 106 #define BT601_Kr 0.299
 107 #define BT601_Kb 0.114
 108
 109 #define BT709_Kr 0.2126
 110 #define BT709_Kb 0.0722
 111
 112 #define BT2020_Kr 0.2627
 113 #define BT2020_Kb 0.0593
 114
 115 class YUV
 116 {
 117         int mpeg, yzero, uvzero;
 118         int ymin8, ymax8, ymin16, ymax16;
 119         int uvmin8, uvmax8, uvmin16, uvmax16;
 120         double Kr, Kg, Kb;
 121         float yminf, ymaxf, yrangef;
 122         float uvminf, uvmaxf, uvrangef;
 123         float r_to_y, g_to_y, b_to_y;
 124         float r_to_u, g_to_u, b_to_u;
 125         float r_to_v, g_to_v, b_to_v;
 126         float v_to_r, v_to_g;
 127         float u_to_g, u_to_b;
 128         int *tab;
 129         float *tabf;
 130
 131         void init(double Kr, double Kb, int mpeg);
 132         void init_tables(int len,
 133                 int *rtoy, int *rtou, int *rtov,
 134                 int *gtoy, int *gtou, int *gtov,
 135                 int *btoy, int *btou, int *btov,
 136                 int *ytab, int *vtor, int *vtog, int *utog, int *utob);
 137         void init_tables(int len,
 138                 float *vtorf, float *vtogf, float *utogf, float *utobf);
 139
 140 // dont use pointers,
 141 //  offsets do not require indirect access
 142 #define rtoy16 (tab+0x00000)
 143 #define gtoy16 (tab+0x10000)
 144 #define btoy16 (tab+0x20000)
 145 #define rtou16 (tab+0x30000)
 146 #define gtou16 (tab+0x40000)
 147 #define btou16 (tab+0x50000)
 148 #define rtov16 (tab+0x60000)
 149 #define gtov16 (tab+0x70000)
 150 #define btov16 (tab+0x80000)
 151 #define ytab16 (tab+0x90000)
 152 #define vtor16 (tab+0xa0000)
 153 #define vtog16 (tab+0xb0000)
 154 #define utog16 (tab+0xc0000)
 155 #define utob16 (tab+0xd0000)
 156
 157 #define rtoy8 (tab+0xe0000)
 158 #define gtoy8 (tab+0xe0100)
 159 #define btoy8 (tab+0xe0200)
 160 #define rtou8 (tab+0xe0300)
 161 #define gtou8 (tab+0xe0400)
 162 #define btou8 (tab+0xe0500)
 163 #define rtov8 (tab+0xe0600)
 164 #define gtov8 (tab+0xe0700)
 165 #define btov8 (tab+0xe0800)
 166 #define ytab8 (tab+0xe0900)
 167 #define vtor8 (tab+0xe0a00)
 168 #define vtog8 (tab+0xe0b00)
 169 #define utog8 (tab+0xe0c00)
 170 #define utob8 (tab+0xe0d00)
 171
 172 #define vtor16f (tabf+0x00000)
 173 #define vtog16f (tabf+0x10000)
 174 #define utog16f (tabf+0x20000)
 175 #define utob16f (tabf+0x30000)
 176
 177 #define vtor8f (tabf+0x40000)
 178 #define vtog8f (tabf+0x40100)
 179 #define utog8f (tabf+0x40200)
 180 #define utob8f (tabf+0x40300)
 181
 182 public:
 183         YUV();
 184         ~YUV();
 185         void yuv_set_colors(int color_space, int color_range);
 186         inline int is_mpeg() { return mpeg; }
 187
 188         static YUV yuv;
 189         static float rgb_to_yuv_matrix[10];
 190         static float yuv_to_rgb_matrix[10];
 191
 192 #define YUV_rgb_to_yuv_8(r,g,b, y,u,v) \
 193         y = iclip((rtoy8[r] + gtoy8[g] + btoy8[b] +  yzero) >> 16,  ymin8,  ymax8); \
 194         u = iclip((rtou8[r] + gtou8[g] + btou8[b] + uvzero) >> 16, uvmin8, uvmax8); \
 195         v = iclip((rtov8[r] + gtov8[g] + btov8[b] + uvzero) >> 16, uvmin8, uvmax8)
 196
 197         bc_always_inline void rgb_to_yuv_8(int r, int g, int b, int &y, int &u, int &v) {
 198                 YUV_rgb_to_yuv_8(r,g,b, y,u,v);
 199         }
 200         bc_always_inline void rgb_to_yuv_8(int r, int g, int b, uint8_t &y, uint8_t &u, uint8_t &v) {
 201                 YUV_rgb_to_yuv_8(r,g,b, y,u,v);
 202         }
 203         bc_always_inline void rgb_to_yuv_8(int &y, int &u, int &v) {
 204                 int r = y, g = u, b = v;  YUV_rgb_to_yuv_8(r, g, b, y, u, v);
 205         }
 206
 207 #define YUV_rgb_to_yuv_16(r,g,b, y,u,v) \
 208         y = iclip((rtoy16[r] + gtoy16[g] + btoy16[b] +  yzero) >> 8,  ymin16,  ymax16); \
 209         u = iclip((rtou16[r] + gtou16[g] + btou16[b] + uvzero) >> 8, uvmin16, uvmax16); \
 210         v = iclip((rtov16[r] + gtov16[g] + btov16[b] + uvzero) >> 8, uvmin16, uvmax16)
 211
 212         bc_always_inline void rgb_to_yuv_16(int r, int g, int b, int &y, int &u, int &v) {
 213                 YUV_rgb_to_yuv_16(r,g,b, y,u,v);
 214         }
 215         bc_always_inline void rgb_to_yuv_16(int r, int g, int b, uint16_t &y, uint16_t &u, uint16_t &v) {
 216                 YUV_rgb_to_yuv_16(r,g,b, y,u,v);
 217         }
 218         bc_always_inline void rgb_to_yuv_16(int &y, int &u, int &v) {
 219                 int r = y, g = u, b = v;
 220                 YUV_rgb_to_yuv_16(r, g, b, y, u, v);
 221         }
 222
 223         bc_always_inline void rgb_to_yuv_f(float r, float g, float b, float &y, float &u, float &v) {
 224                 y = r * r_to_y + g * g_to_y + b * b_to_y + yminf;
 225                 u = r * r_to_u + g * g_to_u + b * b_to_u;
 226                 v = r * r_to_v + g * g_to_v + b * b_to_v;
 227         }
 228         bc_always_inline void rgb_to_yuv_f(float r, float g, float b, uint8_t &y, uint8_t &u, uint8_t &v) {
 229                 int ir = iclip(r*0x100, 0, 0xff);
 230                 int ig = iclip(g*0x100, 0, 0xff);
 231                 int ib = iclip(b*0x100, 0, 0xff);
 232                 rgb_to_yuv_8(ir,ig,ib, y,u,v);
 233         }
 234         bc_always_inline void rgb_to_yuv_f(float r, float g, float b, uint16_t &y, uint16_t &u, uint16_t &v) {
 235                 int ir = iclip(r*0x10000, 0, 0xffff);
 236                 int ig = iclip(g*0x10000, 0, 0xffff);
 237                 int ib = iclip(b*0x10000, 0, 0xffff);
 238                 rgb_to_yuv_16(ir,ig,ib, y,u,v);
 239         }
 240
 241 #define YUV_yuv_to_rgb_8(r,g,b, y,u,v) \
 242         r = iclip((ytab8[y] + vtor8[v]) >> 16, 0, 0xff); \
 243         g = iclip((ytab8[y] + utog8[u] + vtog8[v]) >> 16, 0, 0xff); \
 244         b = iclip((ytab8[y] + utob8[u]) >> 16, 0, 0xff)
 245
 246         bc_always_inline void yuv_to_rgb_8(int &r, int &g, int &b, int y, int u, int v) {
 247                 YUV_yuv_to_rgb_8(r,g,b, y,u,v);
 248         }
 249         bc_always_inline void yuv_to_rgb_8(uint8_t &r, uint8_t &g, uint8_t &b, int y, int u, int v) {
 250                 YUV_yuv_to_rgb_8(r,g,b, y,u,v);
 251         }
 252         bc_always_inline void yuv_to_rgb_8(int &r, int &g, int &b) {
 253                 int y = r, u = g, v = b;  YUV_yuv_to_rgb_8(r,g,b, y,u,v);
 254         }
 255         bc_always_inline void yuv_to_rgb_8(float &r, float &g, float &b, int y, int u, int v) {
 256                 int ir, ig, ib;  yuv_to_rgb_8(ir,ig,ib, y,u,v);
 257                 float s = 1/255.f;  r = s*ir;  g = s*ig;  b = s*ib;
 258         }
 259
 260 #define YUV_yuv_to_rgb_16(r,g,b, y,u,v) \
 261         r = iclip((ytab16[y] + vtor16[v]) >> 8, 0, 0xffff); \
 262         g = iclip((ytab16[y] + utog16[u] + vtog16[v]) >> 8, 0, 0xffff); \
 263         b = iclip((ytab16[y] + utob16[u]) >> 8, 0, 0xffff)
 264
 265         bc_always_inline void yuv_to_rgb_16(int &r, int &g, int &b, int y, int u, int v) {
 266                 YUV_yuv_to_rgb_16(r,g,b, y,u,v);
 267         }
 268         bc_always_inline void yuv_to_rgb_16(uint16_t &r, uint16_t &g, uint16_t &b, int y, int u, int v) {
 269                 YUV_yuv_to_rgb_16(r,g,b, y,u,v);
 270         }
 271         bc_always_inline void yuv_to_rgb_16(int &r, int &g, int &b) {
 272                 int y = r, u = g, v = b;  YUV_yuv_to_rgb_16(r,g,b, y,u,v);
 273         }
 274         bc_always_inline void yuv_to_rgb_16(float &r, float &g, float &b, int y, int u, int v) {
 275                 int ir, ig, ib;  YUV_yuv_to_rgb_16(ir,ig,ib, y,u,v);
 276                 float s = 1/65535.f;  r = s*ir;  g = s*ig;  b = s*ib;
 277         }
 278
 279         bc_always_inline void yuv_to_rgb_f(float &r, float &g, float &b, float y, float u, float v) {
 280                 y = (y-yminf) / yrangef;
 281                 r = y + v_to_r * v;
 282                 g = y + u_to_g * u + v_to_g * v;
 283                 b = y + u_to_b * u;
 284         }
 285         bc_always_inline void yuv_to_rgb_f(float &r, float &g, float &b, uint8_t &y, uint8_t &u, uint8_t &v) {
 286                 yuv_to_rgb_8(r,g,b, y,u,v);
 287         }
 288         bc_always_inline void yuv_to_rgb_f(float &r, float &g, float &b, uint16_t &y, uint16_t &u, uint16_t &v) {
 289                 yuv_to_rgb_16(r,g,b, y,u,v);
 290         }
 291
 292         bc_always_inline int rgb_to_y_8(int r, int g, int b) {
 293                 return (rtoy8[r] + gtoy8[g] + btoy8[b] + yzero) >> 16;
 294         }
 295         bc_always_inline int rgb_to_y_16(int r, int g, int b) {
 296                 return (rtoy16[r] + gtoy16[g] + btoy16[b] + yzero) >> 8;
 297         }
 298         bc_always_inline float rgb_to_y_f(float r, float g, float b) {
 299                 return r * r_to_y + g * g_to_y + b * b_to_y + yminf;
 300         }
 301
 302 // For easier programming.  Doesn't do anything.
 303 // unused cases in macro expansions, mismatched argument types
 304         inline void yuv_to_rgb_8(float &r, float &g, float &b, float y, float u, float v) {}
 305         inline void yuv_to_rgb_16(float &r, float &g, float &b, float y, float u, float v) {}
 306         inline void yuv_to_rgb_f(int &r, int &g, int &b, int y, int u, int v) {}
 307         inline void yuv_to_rgb_f(uint8_t &r, uint8_t &g, uint8_t &b, int&y, int u, int v) {}
 308         inline void yuv_to_rgb_f(uint16_t &r, uint16_t &g, uint16_t &b, int y, int u, int v) {}
 309
 310         inline void rgb_to_yuv_8(float r, float g, float b, float &y, float &u, float &v) {}
 311         inline void rgb_to_yuv_16(float r, float g, float b, float &y, float &u, float &v) {}
 312         inline void rgb_to_yuv_f(int r, int g, int b, int &y, int &u, int &v) {}
 313         inline void rgb_to_yuv_f(uint8_t r, uint8_t g, uint8_t b, int &y, int &u, int &v) {}
 314         inline void rgb_to_yuv_f(uint16_t r, uint16_t g, uint16_t b, int &y, int &u, int &v) {}
 315 };
 316
 317
 318 class HSV
 319 {
 320 public:
 321         HSV();
 322         ~HSV();
 323
 324 // All units are 0 - 1
 325         static int rgb_to_hsv(float r, float g, float b, float &h, float &s, float &v);
 326         static int hsv_to_rgb(float &r, float &g, float &b, float h, float s, float v);
 327
 328 // YUV units are 0 - max.  HSV units are 0 - 1
 329         static int yuv_to_hsv(int y, int u, int v, float &h, float &s, float &va, int max);
 330         static int hsv_to_yuv(int &y, int &u, int &v, float h, float s, float va, int max);
 331 // Dummies for macros
 332         static int yuv_to_hsv(float y, float u, float v, float &h, float &s, float &va, float max) { return 0; };
 333         static int hsv_to_yuv(float &y, float &u, float &v, float h, float s, float va, float max) { return 0; };
 334 };
 335
 336
 337 // standard colors
 338 #define BLACK   0x000000
 339 #define WHITE   0xFFFFFF
 340
 341 #define LTBLUE  0x9090FF
 342 #define BLUE    0x0000FF
 343 #define DKBLUE  0x000090
 344
 345 #define LTPINK  0xFFC0C0
 346 #define PINK    0xFF8080
 347 #define RED     0xFF0000
 348
 349 #define LTGREEN 0xC0FFC0
 350 #define GREEN   0x00FF00
 351 #define DKGREEN 0x009000
 352
 353 #define YELLOW  0xFFFF00
 354 #define LTYELLOW 0xFFFFA0
 355 #define MEYELLOW 0xFFFF00
 356 #define MDYELLOW 0xFFFFD2
 357 #define DKYELLOW 0xFFFFB4
 358
 359 #define LTCYAN  0x00CBCB
 360 #define MECYAN  0x009696
 361 #define MDCYAN  0x007E7E
 362 #define DKCYAN  0x004949
 363
 364 #define LTPURPLE 0xFFC0FF
 365 #define MEPURPLE 0xFF00FF
 366 #define MDPURPLE 0xC000C0
 367 #define DKPURPLE 0xA000A0
 368
 369 #define LTGREY  0xE0E0E0
 370 #define MEGREY  0xAFAFAF
 371 #define DMGREY  0x999999
 372 #define MDGREY  0x7D7D7D
 373 #define DKGREY  0x4B4B4B
 374
 375 #define BLOND   0xb4b487
 376 #define SLBLUE  0x6040c0
 377
 378 #define MNGREY  0xe6e6e6
 379 #define FGGREY  0xe3e3e3
 380 #define MNBLUE  0x003cff
 381 #define ORANGE  0xffdd76
 382 #define FTGREY  0xbcbcbc
 383
 384 #endif