FFmpeg  4.4.5
colorspace_common.c
Go to the documentation of this file.
1 // Generated from libavfilter/opencl/colorspace_common.cl
3 "#line 1 \"libavfilter/opencl/colorspace_common.cl\"\n"
4 "/*\n"
5 " * This file is part of FFmpeg.\n"
6 " *\n"
7 " * FFmpeg is free software; you can redistribute it and/or\n"
8 " * modify it under the terms of the GNU Lesser General Public\n"
9 " * License as published by the Free Software Foundation; either\n"
10 " * version 2.1 of the License, or (at your option) any later version.\n"
11 " *\n"
12 " * FFmpeg is distributed in the hope that it will be useful,\n"
13 " * but WITHOUT ANY WARRANTY; without even the implied warranty of\n"
14 " * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU\n"
15 " * Lesser General Public License for more details.\n"
16 " *\n"
17 " * You should have received a copy of the GNU Lesser General Public\n"
18 " * License along with FFmpeg; if not, write to the Free Software\n"
19 " * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA\n"
20 " */\n"
21 "\n"
22 "#define ST2084_MAX_LUMINANCE 10000.0f\n"
23 "#define REFERENCE_WHITE 100.0f\n"
24 "\n"
25 "#if chroma_loc == 1\n"
26 " #define chroma_sample(a,b,c,d) (((a) + (c)) * 0.5f)\n"
27 "#elif chroma_loc == 3\n"
28 " #define chroma_sample(a,b,c,d) (a)\n"
29 "#elif chroma_loc == 4\n"
30 " #define chroma_sample(a,b,c,d) (((a) + (b)) * 0.5f)\n"
31 "#elif chroma_loc == 5\n"
32 " #define chroma_sample(a,b,c,d) (c)\n"
33 "#elif chroma_loc == 6\n"
34 " #define chroma_sample(a,b,c,d) (((c) + (d)) * 0.5f)\n"
35 "#else\n"
36 " #define chroma_sample(a,b,c,d) (((a) + (b) + (c) + (d)) * 0.25f)\n"
37 "#endif\n"
38 "\n"
39 "constant const float ST2084_M1 = 0.1593017578125f;\n"
40 "constant const float ST2084_M2 = 78.84375f;\n"
41 "constant const float ST2084_C1 = 0.8359375f;\n"
42 "constant const float ST2084_C2 = 18.8515625f;\n"
43 "constant const float ST2084_C3 = 18.6875f;\n"
44 "\n"
45 "float get_luma_dst(float3 c) {\n"
46 " return luma_dst.x * c.x + luma_dst.y * c.y + luma_dst.z * c.z;\n"
47 "}\n"
48 "\n"
49 "float get_luma_src(float3 c) {\n"
50 " return luma_src.x * c.x + luma_src.y * c.y + luma_src.z * c.z;\n"
51 "}\n"
52 "\n"
53 "float3 get_chroma_sample(float3 a, float3 b, float3 c, float3 d) {\n"
54 " return chroma_sample(a, b, c, d);\n"
55 "}\n"
56 "\n"
57 "float eotf_st2084(float x) {\n"
58 " float p = powr(x, 1.0f / ST2084_M2);\n"
59 " float a = max(p -ST2084_C1, 0.0f);\n"
60 " float b = max(ST2084_C2 - ST2084_C3 * p, 1e-6f);\n"
61 " float c = powr(a / b, 1.0f / ST2084_M1);\n"
62 " return x > 0.0f ? c * ST2084_MAX_LUMINANCE / REFERENCE_WHITE : 0.0f;\n"
63 "}\n"
64 "\n"
65 "__constant const float HLG_A = 0.17883277f;\n"
66 "__constant const float HLG_B = 0.28466892f;\n"
67 "__constant const float HLG_C = 0.55991073f;\n"
68 "\n"
69 "// linearizer for HLG\n"
70 "float inverse_oetf_hlg(float x) {\n"
71 " float a = 4.0f * x * x;\n"
72 " float b = exp((x - HLG_C) / HLG_A) + HLG_B;\n"
73 " return x < 0.5f ? a : b;\n"
74 "}\n"
75 "\n"
76 "// delinearizer for HLG\n"
77 "float oetf_hlg(float x) {\n"
78 " float a = 0.5f * sqrt(x);\n"
79 " float b = HLG_A * log(x - HLG_B) + HLG_C;\n"
80 " return x <= 1.0f ? a : b;\n"
81 "}\n"
82 "\n"
83 "float3 ootf_hlg(float3 c, float peak) {\n"
84 " float luma = get_luma_src(c);\n"
85 " float gamma = 1.2f + 0.42f * log10(peak * REFERENCE_WHITE / 1000.0f);\n"
86 " gamma = max(1.0f, gamma);\n"
87 " float factor = peak * powr(luma, gamma - 1.0f) / powr(12.0f, gamma);\n"
88 " return c * factor;\n"
89 "}\n"
90 "\n"
91 "float3 inverse_ootf_hlg(float3 c, float peak) {\n"
92 " float gamma = 1.2f + 0.42f * log10(peak * REFERENCE_WHITE / 1000.0f);\n"
93 " c *= powr(12.0f, gamma) / peak;\n"
94 " c /= powr(get_luma_dst(c), (gamma - 1.0f) / gamma);\n"
95 " return c;\n"
96 "}\n"
97 "\n"
98 "float inverse_eotf_bt1886(float c) {\n"
99 " return c < 0.0f ? 0.0f : powr(c, 1.0f / 2.4f);\n"
100 "}\n"
101 "\n"
102 "float oetf_bt709(float c) {\n"
103 " c = c < 0.0f ? 0.0f : c;\n"
104 " float r1 = 4.5f * c;\n"
105 " float r2 = 1.099f * powr(c, 0.45f) - 0.099f;\n"
106 " return c < 0.018f ? r1 : r2;\n"
107 "}\n"
108 "float inverse_oetf_bt709(float c) {\n"
109 " float r1 = c / 4.5f;\n"
110 " float r2 = powr((c + 0.099f) / 1.099f, 1.0f / 0.45f);\n"
111 " return c < 0.081f ? r1 : r2;\n"
112 "}\n"
113 "\n"
114 "float3 yuv2rgb(float y, float u, float v) {\n"
115 "#ifdef FULL_RANGE_IN\n"
116 " u -= 0.5f; v -= 0.5f;\n"
117 "#else\n"
118 " y = (y * 255.0f - 16.0f) / 219.0f;\n"
119 " u = (u * 255.0f - 128.0f) / 224.0f;\n"
120 " v = (v * 255.0f - 128.0f) / 224.0f;\n"
121 "#endif\n"
122 " float r = y * rgb_matrix[0] + u * rgb_matrix[1] + v * rgb_matrix[2];\n"
123 " float g = y * rgb_matrix[3] + u * rgb_matrix[4] + v * rgb_matrix[5];\n"
124 " float b = y * rgb_matrix[6] + u * rgb_matrix[7] + v * rgb_matrix[8];\n"
125 " return (float3)(r, g, b);\n"
126 "}\n"
127 "\n"
128 "float3 yuv2lrgb(float3 yuv) {\n"
129 " float3 rgb = yuv2rgb(yuv.x, yuv.y, yuv.z);\n"
130 "#ifdef linearize\n"
131 " float r = linearize(rgb.x);\n"
132 " float g = linearize(rgb.y);\n"
133 " float b = linearize(rgb.z);\n"
134 " return (float3)(r, g, b);\n"
135 "#else\n"
136 " return rgb;\n"
137 "#endif\n"
138 "}\n"
139 "\n"
140 "float3 rgb2yuv(float r, float g, float b) {\n"
141 " float y = r*yuv_matrix[0] + g*yuv_matrix[1] + b*yuv_matrix[2];\n"
142 " float u = r*yuv_matrix[3] + g*yuv_matrix[4] + b*yuv_matrix[5];\n"
143 " float v = r*yuv_matrix[6] + g*yuv_matrix[7] + b*yuv_matrix[8];\n"
144 "#ifdef FULL_RANGE_OUT\n"
145 " u += 0.5f; v += 0.5f;\n"
146 "#else\n"
147 " y = (219.0f * y + 16.0f) / 255.0f;\n"
148 " u = (224.0f * u + 128.0f) / 255.0f;\n"
149 " v = (224.0f * v + 128.0f) / 255.0f;\n"
150 "#endif\n"
151 " return (float3)(y, u, v);\n"
152 "}\n"
153 "\n"
154 "float rgb2y(float r, float g, float b) {\n"
155 " float y = r*yuv_matrix[0] + g*yuv_matrix[1] + b*yuv_matrix[2];\n"
156 " y = (219.0f * y + 16.0f) / 255.0f;\n"
157 " return y;\n"
158 "}\n"
159 "\n"
160 "float3 lrgb2yuv(float3 c) {\n"
161 "#ifdef delinearize\n"
162 " float r = delinearize(c.x);\n"
163 " float g = delinearize(c.y);\n"
164 " float b = delinearize(c.z);\n"
165 " return rgb2yuv(r, g, b);\n"
166 "#else\n"
167 " return rgb2yuv(c.x, c.y, c.z);\n"
168 "#endif\n"
169 "}\n"
170 "\n"
171 "float lrgb2y(float3 c) {\n"
172 "#ifdef delinearize\n"
173 " float r = delinearize(c.x);\n"
174 " float g = delinearize(c.y);\n"
175 " float b = delinearize(c.z);\n"
176 " return rgb2y(r, g, b);\n"
177 "#else\n"
178 " return rgb2y(c.x, c.y, c.z);\n"
179 "#endif\n"
180 "}\n"
181 "\n"
182 "float3 lrgb2lrgb(float3 c) {\n"
183 "#ifdef RGB2RGB_PASSTHROUGH\n"
184 " return c;\n"
185 "#else\n"
186 " float r = c.x, g = c.y, b = c.z;\n"
187 " float rr = rgb2rgb[0] * r + rgb2rgb[1] * g + rgb2rgb[2] * b;\n"
188 " float gg = rgb2rgb[3] * r + rgb2rgb[4] * g + rgb2rgb[5] * b;\n"
189 " float bb = rgb2rgb[6] * r + rgb2rgb[7] * g + rgb2rgb[8] * b;\n"
190 " return (float3)(rr, gg, bb);\n"
191 "#endif\n"
192 "}\n"
193 "\n"
194 "float3 ootf(float3 c, float peak) {\n"
195 "#ifdef ootf_impl\n"
196 " return ootf_impl(c, peak);\n"
197 "#else\n"
198 " return c;\n"
199 "#endif\n"
200 "}\n"
201 "\n"
202 "float3 inverse_ootf(float3 c, float peak) {\n"
203 "#ifdef inverse_ootf_impl\n"
204 " return inverse_ootf_impl(c, peak);\n"
205 "#else\n"
206 " return c;\n"
207 "#endif\n"
208 "}\n"
209 ;
const char * ff_opencl_source_colorspace_common