xref: /third_party/ffmpeg/libavutil/colorspace.h (revision cabdff1a)
1/*
2 * Colorspace conversion defines
3 * Copyright (c) 2001, 2002, 2003 Fabrice Bellard
4 *
5 * This file is part of FFmpeg.
6 *
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22/**
23 * @file
24 * Various defines for YUV<->RGB conversion
25 */
26
27#ifndef AVUTIL_COLORSPACE_H
28#define AVUTIL_COLORSPACE_H
29
30#define SCALEBITS 10
31#define ONE_HALF  (1 << (SCALEBITS - 1))
32#define FIX(x)    ((int) ((x) * (1<<SCALEBITS) + 0.5))
33
34#define YUV_TO_RGB1_CCIR(cb1, cr1)\
35{\
36    cb = (cb1) - 128;\
37    cr = (cr1) - 128;\
38    r_add = FIX(1.40200*255.0/224.0) * cr + ONE_HALF;\
39    g_add = - FIX(0.34414*255.0/224.0) * cb - FIX(0.71414*255.0/224.0) * cr + \
40            ONE_HALF;\
41    b_add = FIX(1.77200*255.0/224.0) * cb + ONE_HALF;\
42}
43
44#define YUV_TO_RGB1_CCIR_BT709(cb1, cr1)                      \
45    {                                                         \
46        cb    = (cb1) - 128;                                  \
47        cr    = (cr1) - 128;                                  \
48        r_add = ONE_HALF + FIX(1.5747 * 255.0 / 224.0) * cr;  \
49        g_add = ONE_HALF - FIX(0.1873 * 255.0 / 224.0) * cb - \
50                           FIX(0.4682 * 255.0 / 224.0) * cr;  \
51        b_add = ONE_HALF + FIX(1.8556 * 255.0 / 224.0) * cb;  \
52    }
53
54// To be used for the BT709 variant as well
55#define YUV_TO_RGB2_CCIR(r, g, b, y1)\
56{\
57    y = ((y1) - 16) * FIX(255.0/219.0);\
58    r = cm[(y + r_add) >> SCALEBITS];\
59    g = cm[(y + g_add) >> SCALEBITS];\
60    b = cm[(y + b_add) >> SCALEBITS];\
61}
62
63#define YUV_TO_RGB1(cb1, cr1)\
64{\
65    cb = (cb1) - 128;\
66    cr = (cr1) - 128;\
67    r_add = FIX(1.40200) * cr + ONE_HALF;\
68    g_add = - FIX(0.34414) * cb - FIX(0.71414) * cr + ONE_HALF;\
69    b_add = FIX(1.77200) * cb + ONE_HALF;\
70}
71
72#define YUV_TO_RGB2(r, g, b, y1)\
73{\
74    y = (y1) << SCALEBITS;\
75    r = cm[(y + r_add) >> SCALEBITS];\
76    g = cm[(y + g_add) >> SCALEBITS];\
77    b = cm[(y + b_add) >> SCALEBITS];\
78}
79
80#define Y_CCIR_TO_JPEG(y)\
81 cm[((y) * FIX(255.0/219.0) + (ONE_HALF - 16 * FIX(255.0/219.0))) >> SCALEBITS]
82
83#define Y_JPEG_TO_CCIR(y)\
84 (((y) * FIX(219.0/255.0) + (ONE_HALF + (16 << SCALEBITS))) >> SCALEBITS)
85
86#define C_CCIR_TO_JPEG(y)\
87 cm[(((y) - 128) * FIX(127.0/112.0) + (ONE_HALF + (128 << SCALEBITS))) >> SCALEBITS]
88
89/* NOTE: the clamp is really necessary! */
90static inline int C_JPEG_TO_CCIR(int y) {
91    y = (((y - 128) * FIX(112.0/127.0) + (ONE_HALF + (128 << SCALEBITS))) >> SCALEBITS);
92    if (y < 16)
93        y = 16;
94    return y;
95}
96
97
98#define RGB_TO_Y_CCIR(r, g, b) \
99((FIX(0.29900*219.0/255.0) * (r) + FIX(0.58700*219.0/255.0) * (g) + \
100  FIX(0.11400*219.0/255.0) * (b) + (ONE_HALF + (16 << SCALEBITS))) >> SCALEBITS)
101
102#define RGB_TO_U_CCIR(r1, g1, b1, shift)\
103(((- FIX(0.16874*224.0/255.0) * r1 - FIX(0.33126*224.0/255.0) * g1 +         \
104     FIX(0.50000*224.0/255.0) * b1 + (ONE_HALF << shift) - 1) >> (SCALEBITS + shift)) + 128)
105
106#define RGB_TO_V_CCIR(r1, g1, b1, shift)\
107(((FIX(0.50000*224.0/255.0) * r1 - FIX(0.41869*224.0/255.0) * g1 -           \
108   FIX(0.08131*224.0/255.0) * b1 + (ONE_HALF << shift) - 1) >> (SCALEBITS + shift)) + 128)
109
110#define RGB_TO_Y_JPEG(r, g, b) \
111(FFMIN((FIX(0.29900) * (r) + FIX(0.58700) * (g) + \
112  FIX(0.11400) * (b) + (ONE_HALF)) >> SCALEBITS, 255))
113
114#define RGB_TO_U_JPEG(r1, g1, b1)\
115(((- FIX(0.16874) * r1 - FIX(0.33126) * g1 + \
116     FIX(0.50000) * b1 + (ONE_HALF) - 1) >> (SCALEBITS)) + 128)
117
118#define RGB_TO_V_JPEG(r1, g1, b1)\
119(((FIX(0.50000) * r1 - FIX(0.41869) * g1 - \
120   FIX(0.08131) * b1 + (ONE_HALF) - 1) >> (SCALEBITS)) + 128)
121
122// Conversion macros for 8-bit RGB to YUV
123// Derived from ITU-R BT.709-6 (06/2015) Item 3.5
124// https://www.itu.int/rec/R-REC-BT.709-6-201506-I/en
125
126#define RGB_TO_Y_BT709(r, g, b) \
127((FIX(0.21260*219.0/255.0) * (r) + FIX(0.71520*219.0/255.0) * (g) + \
128  FIX(0.07220*219.0/255.0) * (b) + (ONE_HALF + (16 << SCALEBITS))) >> SCALEBITS)
129
130#define RGB_TO_U_BT709(r1, g1, b1, shift)\
131(((- FIX(0.11457*224.0/255.0) * r1 - FIX(0.38543*224.0/255.0) * g1 +         \
132     FIX(0.50000*224.0/255.0) * b1 + (ONE_HALF << shift) - 1) >> (SCALEBITS + shift)) + 128)
133
134#define RGB_TO_V_BT709(r1, g1, b1, shift)\
135(((FIX(0.50000*224.0/255.0) * r1 - FIX(0.45415*224.0/255.0) * g1 -           \
136   FIX(0.04585*224.0/255.0) * b1 + (ONE_HALF << shift) - 1) >> (SCALEBITS + shift)) + 128)
137
138#define RGB_TO_Y_BT709_FULL(r, g, b) \
139(FFMIN((FIX(0.21260) * (r) + FIX(0.71520) * (g) + \
140  FIX(0.07220) * (b) + (ONE_HALF)) >> SCALEBITS, 255))
141
142#define RGB_TO_U_BT709_FULL(r1, g1, b1)\
143(((- FIX(0.11457) * r1 - FIX(0.38543) * g1 + \
144     FIX(0.50000) * b1 + (ONE_HALF) - 1) >> (SCALEBITS)) + 128)
145
146#define RGB_TO_V_BT709_FULL(r1, g1, b1)\
147(((FIX(0.50000) * r1 - FIX(0.45415) * g1 - \
148   FIX(0.04585) * b1 + (ONE_HALF) - 1) >> (SCALEBITS)) + 128)
149
150#endif /* AVUTIL_COLORSPACE_H */
151