431 lines
10 KiB
GLSL
431 lines
10 KiB
GLSL
// ST 2094-10:2021 - SMPTE Standard - Dynamic Metadata for Color Volume Transform - Application #1
|
|
// https://ieeexplore.ieee.org/document/9405553
|
|
|
|
//!PARAM min_luma
|
|
//!TYPE float
|
|
0.0
|
|
|
|
//!PARAM max_luma
|
|
//!TYPE float
|
|
0.0
|
|
|
|
//!PARAM max_cll
|
|
//!TYPE float
|
|
0.0
|
|
|
|
//!PARAM max_fall
|
|
//!TYPE float
|
|
0.0
|
|
|
|
//!PARAM scene_max_r
|
|
//!TYPE float
|
|
0.0
|
|
|
|
//!PARAM scene_max_g
|
|
//!TYPE float
|
|
0.0
|
|
|
|
//!PARAM scene_max_b
|
|
//!TYPE float
|
|
0.0
|
|
|
|
//!PARAM scene_avg
|
|
//!TYPE float
|
|
0.0
|
|
|
|
//!PARAM max_pq_y
|
|
//!TYPE float
|
|
0.0
|
|
|
|
//!PARAM avg_pq_y
|
|
//!TYPE float
|
|
0.0
|
|
|
|
//!PARAM reference_white
|
|
//!TYPE float
|
|
//!MINIMUM 0.0
|
|
//!MAXIMUM 1000.0
|
|
203.0
|
|
|
|
//!PARAM chroma_correction_scaling
|
|
//!TYPE float
|
|
//!MINIMUM 0.0
|
|
//!MAXIMUM 1.0
|
|
0.5
|
|
|
|
//!BUFFER METERED
|
|
//!VAR float metered_avg_l
|
|
//!STORAGE
|
|
|
|
//!HOOK OUTPUT
|
|
//!BIND HOOKED
|
|
//!SAVE AVG
|
|
//!COMPONENTS 1
|
|
//!WIDTH 1024
|
|
//!HEIGHT 1024
|
|
//!WHEN avg_pq_y 0 = scene_avg 0 = *
|
|
//!DESC tone mapping (st2094-10, average, 1024)
|
|
|
|
const vec3 y_coef = vec3(0.2627002120112671, 0.6779980715188708, 0.05930171646986196);
|
|
|
|
const float m1 = 2610.0 / 4096.0 / 4.0;
|
|
const float m2 = 2523.0 / 4096.0 * 128.0;
|
|
const float c1 = 3424.0 / 4096.0;
|
|
const float c2 = 2413.0 / 4096.0 * 32.0;
|
|
const float c3 = 2392.0 / 4096.0 * 32.0;
|
|
const float pw = 10000.0;
|
|
|
|
float pq_eotf_inv(float x) {
|
|
float t = pow(x / pw, m1);
|
|
return pow((c1 + c2 * t) / (1.0 + c3 * t), m2);
|
|
}
|
|
|
|
vec4 hook() {
|
|
vec4 color = HOOKED_tex(HOOKED_pos);
|
|
float l = dot(color.rgb * reference_white, y_coef);
|
|
float i = pq_eotf_inv(l);
|
|
return vec4(i, vec3(0.0));
|
|
}
|
|
|
|
//!HOOK OUTPUT
|
|
//!BIND AVG
|
|
//!SAVE AVG
|
|
//!WIDTH AVG.w 2 /
|
|
//!HEIGHT AVG.h 2 /
|
|
//!WHEN avg_pq_y 0 = scene_avg 0 = *
|
|
//!DESC tone mapping (st2094-10, average, 512)
|
|
vec4 hook() { return AVG_tex(AVG_pos); }
|
|
|
|
//!HOOK OUTPUT
|
|
//!BIND AVG
|
|
//!SAVE AVG
|
|
//!WIDTH AVG.w 2 /
|
|
//!HEIGHT AVG.h 2 /
|
|
//!WHEN avg_pq_y 0 = scene_avg 0 = *
|
|
//!DESC tone mapping (st2094-10, average, 256)
|
|
vec4 hook() { return AVG_tex(AVG_pos); }
|
|
|
|
//!HOOK OUTPUT
|
|
//!BIND AVG
|
|
//!SAVE AVG
|
|
//!WIDTH AVG.w 2 /
|
|
//!HEIGHT AVG.h 2 /
|
|
//!WHEN avg_pq_y 0 = scene_avg 0 = *
|
|
//!DESC tone mapping (st2094-10, average, 128)
|
|
vec4 hook() { return AVG_tex(AVG_pos); }
|
|
|
|
//!HOOK OUTPUT
|
|
//!BIND AVG
|
|
//!SAVE AVG
|
|
//!WIDTH AVG.w 2 /
|
|
//!HEIGHT AVG.h 2 /
|
|
//!WHEN avg_pq_y 0 = scene_avg 0 = *
|
|
//!DESC tone mapping (st2094-10, average, 64)
|
|
vec4 hook() { return AVG_tex(AVG_pos); }
|
|
|
|
//!HOOK OUTPUT
|
|
//!BIND AVG
|
|
//!SAVE AVG
|
|
//!WIDTH AVG.w 2 /
|
|
//!HEIGHT AVG.h 2 /
|
|
//!WHEN avg_pq_y 0 = scene_avg 0 = *
|
|
//!DESC tone mapping (st2094-10, average, 32)
|
|
vec4 hook() { return AVG_tex(AVG_pos); }
|
|
|
|
//!HOOK OUTPUT
|
|
//!BIND AVG
|
|
//!SAVE AVG
|
|
//!WIDTH AVG.w 2 /
|
|
//!HEIGHT AVG.h 2 /
|
|
//!WHEN avg_pq_y 0 = scene_avg 0 = *
|
|
//!DESC tone mapping (st2094-10, average, 16)
|
|
vec4 hook() { return AVG_tex(AVG_pos); }
|
|
|
|
//!HOOK OUTPUT
|
|
//!BIND AVG
|
|
//!SAVE AVG
|
|
//!WIDTH AVG.w 2 /
|
|
//!HEIGHT AVG.h 2 /
|
|
//!WHEN avg_pq_y 0 = scene_avg 0 = *
|
|
//!DESC tone mapping (st2094-10, average, 8)
|
|
vec4 hook() { return AVG_tex(AVG_pos); }
|
|
|
|
//!HOOK OUTPUT
|
|
//!BIND AVG
|
|
//!SAVE AVG
|
|
//!WIDTH AVG.w 2 /
|
|
//!HEIGHT AVG.h 2 /
|
|
//!WHEN avg_pq_y 0 = scene_avg 0 = *
|
|
//!DESC tone mapping (st2094-10, average, 4)
|
|
vec4 hook() { return AVG_tex(AVG_pos); }
|
|
|
|
//!HOOK OUTPUT
|
|
//!BIND AVG
|
|
//!SAVE AVG
|
|
//!WIDTH AVG.w 2 /
|
|
//!HEIGHT AVG.h 2 /
|
|
//!WHEN avg_pq_y 0 = scene_avg 0 = *
|
|
//!DESC tone mapping (st2094-10, average, 2)
|
|
vec4 hook() { return AVG_tex(AVG_pos); }
|
|
|
|
//!HOOK OUTPUT
|
|
//!BIND AVG
|
|
//!BIND METERED
|
|
//!SAVE AVG
|
|
//!WIDTH 1
|
|
//!HEIGHT 1
|
|
//!COMPUTE 1 1
|
|
//!WHEN avg_pq_y 0 = scene_avg 0 = *
|
|
//!DESC tone mapping (st2094-10, average, 1)
|
|
|
|
void hook() {
|
|
metered_avg_l = AVG_tex(AVG_pos).x;
|
|
}
|
|
|
|
//!HOOK OUTPUT
|
|
//!BIND HOOKED
|
|
//!BIND METERED
|
|
//!DESC tone mapping (st2094-10)
|
|
|
|
const float m1 = 2610.0 / 4096.0 / 4.0;
|
|
const float m2 = 2523.0 / 4096.0 * 128.0;
|
|
const float c1 = 3424.0 / 4096.0;
|
|
const float c2 = 2413.0 / 4096.0 * 32.0;
|
|
const float c3 = 2392.0 / 4096.0 * 32.0;
|
|
const float pw = 10000.0;
|
|
|
|
float pq_eotf_inv(float x) {
|
|
float t = pow(x / pw, m1);
|
|
return pow((c1 + c2 * t) / (1.0 + c3 * t), m2);
|
|
}
|
|
|
|
vec3 pq_eotf_inv(vec3 color) {
|
|
return vec3(
|
|
pq_eotf_inv(color.r),
|
|
pq_eotf_inv(color.g),
|
|
pq_eotf_inv(color.b)
|
|
);
|
|
}
|
|
|
|
float pq_eotf(float x) {
|
|
float t = pow(x, 1.0 / m2);
|
|
return pow(max(t - c1, 0.0) / (c2 - c3 * t), 1.0 / m1) * pw;
|
|
}
|
|
|
|
vec3 pq_eotf(vec3 color) {
|
|
return vec3(
|
|
pq_eotf(color.r),
|
|
pq_eotf(color.g),
|
|
pq_eotf(color.b)
|
|
);
|
|
}
|
|
|
|
vec3 RGB_to_XYZ(vec3 RGB) {
|
|
return RGB * mat3(
|
|
0.6369580483012914, 0.14461690358620832, 0.1688809751641721,
|
|
0.2627002120112671, 0.6779980715188708, 0.05930171646986196,
|
|
0.0 , 0.028072693049087428, 1.060985057710791
|
|
);
|
|
}
|
|
|
|
vec3 XYZ_to_RGB(vec3 XYZ) {
|
|
return XYZ * mat3(
|
|
1.716651187971268, -0.355670783776392, -0.25336628137366,
|
|
-0.666684351832489, 1.616481236634939, 0.0157685458139111,
|
|
0.017639857445311, -0.042770613257809, 0.942103121235474
|
|
);
|
|
}
|
|
|
|
vec3 XYZ_to_LMS(vec3 XYZ) {
|
|
return XYZ * mat3(
|
|
0.3592832590121217, 0.6976051147779502, -0.0358915932320290,
|
|
-0.1920808463704993, 1.1004767970374321, 0.0753748658519118,
|
|
0.0070797844607479, 0.0748396662186362, 0.8433265453898765
|
|
);
|
|
}
|
|
|
|
vec3 LMS_to_XYZ(vec3 LMS) {
|
|
return LMS * mat3(
|
|
2.0701522183894223, -1.3263473389671563, 0.2066510476294053,
|
|
0.3647385209748072, 0.6805660249472273, -0.0453045459220347,
|
|
-0.0497472075358123, -0.0492609666966131, 1.1880659249923042
|
|
);
|
|
}
|
|
|
|
vec3 LMS_to_ICtCp(vec3 LMS) {
|
|
return LMS * mat3(
|
|
2048.0 / 4096.0, 2048.0 / 4096.0, 0.0 / 4096.0,
|
|
6610.0 / 4096.0, -13613.0 / 4096.0, 7003.0 / 4096.0,
|
|
17933.0 / 4096.0, -17390.0 / 4096.0, -543.0 / 4096.0
|
|
);
|
|
}
|
|
|
|
vec3 ICtCp_to_LMS(vec3 ICtCp) {
|
|
return ICtCp * mat3(
|
|
1.0, 0.0086090370379328, 0.1110296250030260,
|
|
1.0, -0.0086090370379328, -0.1110296250030260,
|
|
1.0, 0.5600313357106791, -0.3206271749873189
|
|
);
|
|
}
|
|
|
|
vec3 RGB_to_ICtCp(vec3 color) {
|
|
color *= reference_white;
|
|
color = RGB_to_XYZ(color);
|
|
color = XYZ_to_LMS(color);
|
|
color = pq_eotf_inv(color);
|
|
color = LMS_to_ICtCp(color);
|
|
return color;
|
|
}
|
|
|
|
vec3 ICtCp_to_RGB(vec3 color) {
|
|
color = ICtCp_to_LMS(color);
|
|
color = pq_eotf(color);
|
|
color = LMS_to_XYZ(color);
|
|
color = XYZ_to_RGB(color);
|
|
color /= reference_white;
|
|
return color;
|
|
}
|
|
|
|
float get_max_l() {
|
|
if (max_pq_y > 0.0)
|
|
return pq_eotf(max_pq_y);
|
|
|
|
if (scene_max_r > 0.0 || scene_max_g > 0.0 || scene_max_b > 0.0) {
|
|
vec3 scene_max_rgb = vec3(scene_max_r, scene_max_g, scene_max_b);
|
|
return RGB_to_XYZ(scene_max_rgb).y;
|
|
}
|
|
|
|
if (max_cll > 0.0)
|
|
return max_cll;
|
|
|
|
if (max_luma > 0.0)
|
|
return max_luma;
|
|
|
|
return 1000.0;
|
|
}
|
|
|
|
float get_min_l() {
|
|
if (min_luma > 0.0)
|
|
return min_luma;
|
|
|
|
return 0.001;
|
|
}
|
|
|
|
float get_avg_l() {
|
|
if (avg_pq_y > 0.0)
|
|
return pq_eotf(avg_pq_y);
|
|
|
|
if (scene_avg > 0.0)
|
|
return scene_avg;
|
|
|
|
if (metered_avg_l > 0.0)
|
|
return pq_eotf(clamp(metered_avg_l, 0.1, 0.5));
|
|
|
|
if (max_fall > 0.0)
|
|
return max_fall;
|
|
|
|
return pq_eotf(0.3);
|
|
}
|
|
|
|
// n: contrast [0.5, 1.5]
|
|
// o: offset [-0.5, 0.5]
|
|
// g: gain [0.5, 1.5]
|
|
// p: gamma [0.5, 1.5]
|
|
float f(
|
|
float x,
|
|
float iw, float ib, float ow, float ob, float adapt,
|
|
float n, float o, float g, float p
|
|
) {
|
|
float x1 = ib;
|
|
float y1 = ob;
|
|
|
|
float x3 = iw;
|
|
float y3 = ow;
|
|
|
|
float x2 = adapt;
|
|
float y2 = sqrt(x2 * sqrt(y3 * y1));
|
|
|
|
x2 = clamp(x2, x1 + 0.001, 0.9 * x3);
|
|
y2 = clamp(y2, y1 + 0.001, 0.9 * y3);
|
|
|
|
float a = x3 * y3 * (x1 - x2) + x2 * y2 * (x3 - x1) + x1 * y1 * (x2 - x3);
|
|
|
|
mat3 cmat = mat3(
|
|
x2 * x3 * (y2 - y3), x1 * x3 * (y3 - y1), x1 * x2 * (y1 - y2),
|
|
x3 * y3 - x2 * y2 , x1 * y1 - x3 * y3 , x2 * y2 - x1 * y1 ,
|
|
x3 - x2 , x1 - x3 , x2 - x1
|
|
);
|
|
|
|
vec3 coeffs = vec3(y1, y2, y3) * cmat / a;
|
|
|
|
float c1 = coeffs.r;
|
|
float c2 = coeffs.g;
|
|
float c3 = coeffs.b;
|
|
|
|
x = clamp(x, x1, x3);
|
|
x = pow(x, n);
|
|
x = (c1 + c2 * x) / (1.0 + c3 * x);
|
|
x = pow(min(max(0.0, ((x / y3) * g) + o), 1.0), p) * y3;
|
|
x = clamp(x, y1, y3);
|
|
|
|
return x;
|
|
}
|
|
|
|
float f(float x, float iw, float ib, float ow, float ob, float adapt) {
|
|
return f(x, iw, ib, ow, ob, adapt, 1.0, 0.0, 1.0, 1.0);
|
|
}
|
|
|
|
float curve(float x) {
|
|
float ow = 1.0;
|
|
float ob = 0.001;
|
|
float iw = max(get_max_l() / reference_white, ow + 1e-3);
|
|
float ib = min(get_min_l() / reference_white, ob - 1e-3);
|
|
float avg = get_avg_l() / reference_white;
|
|
return f(x, iw, ib, ow, ob, avg);
|
|
}
|
|
|
|
vec2 chroma_correction(vec2 ab, float i1, float i2) {
|
|
float r1 = i1 / max(i2, 1e-6);
|
|
float r2 = i2 / max(i1, 1e-6);
|
|
return ab * mix(1.0, min(r1, r2), chroma_correction_scaling);
|
|
}
|
|
|
|
vec3 tone_mapping(vec3 iab) {
|
|
float i2 = pq_eotf_inv(curve(pq_eotf(iab.x) / reference_white) * reference_white);
|
|
vec2 ab2 = chroma_correction(iab.yz, iab.x, i2);
|
|
return vec3(i2, ab2);
|
|
}
|
|
|
|
// c: chroma compensation weight [-0.5, 0.5]
|
|
// s: saturation gain [-0.5, 0.5]
|
|
vec3 gamut_adjustment(vec3 f, float c, float s) {
|
|
float y = RGB_to_XYZ(f).y;
|
|
return f * pow((1.0 + c) * f / y, vec3(s));
|
|
}
|
|
|
|
vec3 gamut_adjustment(vec3 f) {
|
|
return gamut_adjustment(f, 0.0, 0.0);
|
|
}
|
|
|
|
// t: tone detail factor [0, 1];
|
|
vec3 detail_managenment(vec3 p, float t) {
|
|
// TODO: do what?
|
|
vec3 q = p;
|
|
return p * (1.0 - t) + q * t;
|
|
}
|
|
|
|
vec3 detail_managenment(vec3 p) {
|
|
return detail_managenment(p, 0.0);
|
|
}
|
|
|
|
vec4 hook() {
|
|
vec4 color = HOOKED_tex(HOOKED_pos);
|
|
|
|
color.rgb = RGB_to_ICtCp(color.rgb);
|
|
color.rgb = tone_mapping(color.rgb);
|
|
color.rgb = ICtCp_to_RGB(color.rgb);
|
|
color.rgb = gamut_adjustment(color.rgb);
|
|
color.rgb = detail_managenment(color.rgb);
|
|
|
|
return color;
|
|
}
|