3 "#line 1 \"libavfilter/opencl/colorspace_common.cl\"\n" 5 " * This file is part of FFmpeg.\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" 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" 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" 22 "#define ST2084_MAX_LUMINANCE 10000.0f\n" 23 "#define REFERENCE_WHITE 100.0f\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" 36 " #define chroma_sample(a,b,c,d) (((a) + (b) + (c) + (d)) * 0.25f)\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" 45 "__constant float yuv2rgb_bt2020[] = {\n" 46 " 1.0f, 0.0f, 1.4746f,\n" 47 " 1.0f, -0.16455f, -0.57135f,\n" 48 " 1.0f, 1.8814f, 0.0f\n" 51 "__constant float yuv2rgb_bt709[] = {\n" 52 " 1.0f, 0.0f, 1.5748f,\n" 53 " 1.0f, -0.18732f, -0.46812f,\n" 54 " 1.0f, 1.8556f, 0.0f\n" 57 "__constant float rgb2yuv_bt709[] = {\n" 58 " 0.2126f, 0.7152f, 0.0722f,\n" 59 " -0.11457f, -0.38543f, 0.5f,\n" 60 " 0.5f, -0.45415f, -0.04585f\n" 63 "__constant float rgb2yuv_bt2020[] ={\n" 64 " 0.2627f, 0.678f, 0.0593f,\n" 65 " -0.1396f, -0.36037f, 0.5f,\n" 66 " 0.5f, -0.4598f, -0.0402f,\n" 70 "float get_luma_dst(float3 c) {\n" 71 " return luma_dst.x * c.x + luma_dst.y * c.y + luma_dst.z * c.z;\n" 74 "float get_luma_src(float3 c) {\n" 75 " return luma_src.x * c.x + luma_src.y * c.y + luma_src.z * c.z;\n" 78 "float3 get_chroma_sample(float3 a, float3 b, float3 c, float3 d) {\n" 79 " return chroma_sample(a, b, c, d);\n" 82 "float eotf_st2084(float x) {\n" 83 " float p = powr(x, 1.0f / ST2084_M2);\n" 84 " float a = max(p -ST2084_C1, 0.0f);\n" 85 " float b = max(ST2084_C2 - ST2084_C3 * p, 1e-6f);\n" 86 " float c = powr(a / b, 1.0f / ST2084_M1);\n" 87 " return x > 0.0f ? c * ST2084_MAX_LUMINANCE / REFERENCE_WHITE : 0.0f;\n" 90 "__constant const float HLG_A = 0.17883277f;\n" 91 "__constant const float HLG_B = 0.28466892f;\n" 92 "__constant const float HLG_C = 0.55991073f;\n" 94 "// linearizer for HLG\n" 95 "float inverse_oetf_hlg(float x) {\n" 96 " float a = 4.0f * x * x;\n" 97 " float b = exp((x - HLG_C) / HLG_A) + HLG_B;\n" 98 " return x < 0.5f ? a : b;\n" 101 "// delinearizer for HLG\n" 102 "float oetf_hlg(float x) {\n" 103 " float a = 0.5f * sqrt(x);\n" 104 " float b = HLG_A * log(x - HLG_B) + HLG_C;\n" 105 " return x <= 1.0f ? a : b;\n" 108 "float3 ootf_hlg(float3 c, float peak) {\n" 109 " float luma = get_luma_src(c);\n" 110 " float gamma = 1.2f + 0.42f * log10(peak * REFERENCE_WHITE / 1000.0f);\n" 111 " gamma = max(1.0f, gamma);\n" 112 " float factor = peak * powr(luma, gamma - 1.0f) / powr(12.0f, gamma);\n" 113 " return c * factor;\n" 116 "float3 inverse_ootf_hlg(float3 c, float peak) {\n" 117 " float gamma = 1.2f + 0.42f * log10(peak * REFERENCE_WHITE / 1000.0f);\n" 118 " c *= powr(12.0f, gamma) / peak;\n" 119 " c /= powr(get_luma_dst(c), (gamma - 1.0f) / gamma);\n" 123 "float inverse_eotf_bt1886(float c) {\n" 124 " return c < 0.0f ? 0.0f : powr(c, 1.0f / 2.4f);\n" 127 "float oetf_bt709(float c) {\n" 128 " c = c < 0.0f ? 0.0f : c;\n" 129 " float r1 = 4.5f * c;\n" 130 " float r2 = 1.099f * powr(c, 0.45f) - 0.099f;\n" 131 " return c < 0.018f ? r1 : r2;\n" 133 "float inverse_oetf_bt709(float c) {\n" 134 " float r1 = c / 4.5f;\n" 135 " float r2 = powr((c + 0.099f) / 1.099f, 1.0f / 0.45f);\n" 136 " return c < 0.081f ? r1 : r2;\n" 139 "float3 yuv2rgb(float y, float u, float v) {\n" 140 "#ifdef FULL_RANGE_IN\n" 141 " u -= 0.5f; v -= 0.5f;\n" 143 " y = (y * 255.0f - 16.0f) / 219.0f;\n" 144 " u = (u * 255.0f - 128.0f) / 224.0f;\n" 145 " v = (v * 255.0f - 128.0f) / 224.0f;\n" 147 " float r = y * rgb_matrix[0] + u * rgb_matrix[1] + v * rgb_matrix[2];\n" 148 " float g = y * rgb_matrix[3] + u * rgb_matrix[4] + v * rgb_matrix[5];\n" 149 " float b = y * rgb_matrix[6] + u * rgb_matrix[7] + v * rgb_matrix[8];\n" 150 " return (float3)(r, g, b);\n" 153 "float3 yuv2lrgb(float3 yuv) {\n" 154 " float3 rgb = yuv2rgb(yuv.x, yuv.y, yuv.z);\n" 155 " float r = linearize(rgb.x);\n" 156 " float g = linearize(rgb.y);\n" 157 " float b = linearize(rgb.z);\n" 158 " return (float3)(r, g, b);\n" 161 "float3 rgb2yuv(float r, float g, float b) {\n" 162 " float y = r*yuv_matrix[0] + g*yuv_matrix[1] + b*yuv_matrix[2];\n" 163 " float u = r*yuv_matrix[3] + g*yuv_matrix[4] + b*yuv_matrix[5];\n" 164 " float v = r*yuv_matrix[6] + g*yuv_matrix[7] + b*yuv_matrix[8];\n" 165 "#ifdef FULL_RANGE_OUT\n" 166 " u += 0.5f; v += 0.5f;\n" 168 " y = (219.0f * y + 16.0f) / 255.0f;\n" 169 " u = (224.0f * u + 128.0f) / 255.0f;\n" 170 " v = (224.0f * v + 128.0f) / 255.0f;\n" 172 " return (float3)(y, u, v);\n" 175 "float rgb2y(float r, float g, float b) {\n" 176 " float y = r*yuv_matrix[0] + g*yuv_matrix[1] + b*yuv_matrix[2];\n" 177 " y = (219.0f * y + 16.0f) / 255.0f;\n" 181 "float3 lrgb2yuv(float3 c) {\n" 182 " float r = delinearize(c.x);\n" 183 " float g = delinearize(c.y);\n" 184 " float b = delinearize(c.z);\n" 186 " return rgb2yuv(r, g, b);\n" 189 "float lrgb2y(float3 c) {\n" 190 " float r = delinearize(c.x);\n" 191 " float g = delinearize(c.y);\n" 192 " float b = delinearize(c.z);\n" 194 " return rgb2y(r, g, b);\n" 197 "float3 lrgb2lrgb(float3 c) {\n" 198 "#ifdef RGB2RGB_PASSTHROUGH\n" 201 " float r = c.x, g = c.y, b = c.z;\n" 202 " float rr = rgb2rgb[0] * r + rgb2rgb[1] * g + rgb2rgb[2] * b;\n" 203 " float gg = rgb2rgb[3] * r + rgb2rgb[4] * g + rgb2rgb[5] * b;\n" 204 " float bb = rgb2rgb[6] * r + rgb2rgb[7] * g + rgb2rgb[8] * b;\n" 205 " return (float3)(rr, gg, bb);\n" 209 "float3 ootf(float3 c, float peak) {\n" 211 " return ootf_impl(c, peak);\n" 217 "float3 inverse_ootf(float3 c, float peak) {\n" 218 "#ifdef inverse_ootf_impl\n" 219 " return inverse_ootf_impl(c, peak);\n" const char * ff_opencl_source_colorspace_common