mirror of
https://github.com/italicsjenga/slang-shaders.git
synced 2024-11-25 17:01:31 +11:00
305 lines
10 KiB
Plaintext
305 lines
10 KiB
Plaintext
#version 450
|
|
|
|
/*
|
|
|
|
Hyllian's xBR MultiLevel4 Shader - Pass2
|
|
|
|
Copyright (C) 2011-2015 Hyllian - sergiogdb@gmail.com
|
|
|
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
of this software and associated documentation files (the "Software"), to deal
|
|
in the Software without restriction, including without limitation the rights
|
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
copies of the Software, and to permit persons to whom the Software is
|
|
furnished to do so, subject to the following conditions:
|
|
|
|
The above copyright notice and this permission notice shall be included in
|
|
all copies or substantial portions of the Software.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
THE SOFTWARE.
|
|
|
|
*/
|
|
|
|
layout(push_constant) uniform Push
|
|
{
|
|
vec4 SourceSize;
|
|
vec4 OriginalSize;
|
|
vec4 OutputSize;
|
|
uint FrameCount;
|
|
} params;
|
|
|
|
layout(std140, set = 0, binding = 0) uniform UBO
|
|
{
|
|
mat4 MVP;
|
|
} global;
|
|
|
|
// compatibility macros
|
|
#define mul(a,b) (b*a)
|
|
#define float2 vec2
|
|
#define float3 vec3
|
|
#define float4 vec4
|
|
#define bool2 bvec2
|
|
#define bool3 bvec3
|
|
#define bool4 bvec4
|
|
#define float3x3 mat3x3
|
|
#define float4x3 mat4x3
|
|
#define texture_size params.SourceSize
|
|
#define video_size param.SourceSize
|
|
#define output_size params.OutputSize
|
|
|
|
#define round(X) floor((X)+0.5)
|
|
|
|
const float cf2 = 2.0;
|
|
const float cf3 = 4.0;
|
|
const float cf4 = 4.0;
|
|
const float4 eq_threshold = float4(15.0, 15.0, 15.0, 15.0);
|
|
const float4 eq_threshold2 = float4( 5.0, 5.0, 5.0, 5.0);
|
|
const float4 eq_threshold3 = float4(25.0, 25.0, 25.0, 25.0);
|
|
const float y_weight = 48.0;
|
|
const float u_weight = 7.0;
|
|
const float v_weight = 6.0;
|
|
const float3x3 yuv = float3x3(0.299, 0.587, 0.114, -0.169, -0.331, 0.499, 0.499, -0.418, -0.0813);
|
|
const float3x3 yuv_weighted = float3x3(y_weight*yuv[0], u_weight*yuv[1], v_weight*yuv[2]);
|
|
const float4 maximo = float4(255.0, 255.0, 255.0, 255.0);
|
|
|
|
float4 df(float4 A, float4 B)
|
|
{
|
|
return float4(abs(A-B));
|
|
}
|
|
|
|
bool4 rd(float4 A, float4 B, float4 C, float4 D)
|
|
{
|
|
return greaterThan((df(C,D)/(df(A,B)+0.000000001)) , vec4(2.,2.,2.,2.));
|
|
}
|
|
|
|
bool4 id(float4 A, float4 B, float4 C, float4 D)
|
|
{
|
|
return greaterThan(df(C,D) , df(A,B));
|
|
}
|
|
|
|
float4 remapTo01(float4 v, float4 high)
|
|
{
|
|
return (v/high);
|
|
}
|
|
|
|
float4 remapFrom01(float4 v, float4 high)
|
|
{
|
|
return round(high*v);
|
|
}
|
|
|
|
|
|
bool4 eq(float4 A, float4 B)
|
|
{
|
|
return lessThan(df(A, B) , eq_threshold);
|
|
}
|
|
|
|
bool4 eq2(float4 A, float4 B)
|
|
{
|
|
return lessThan(df(A, B) , eq_threshold2);
|
|
}
|
|
|
|
bool4 eq3(float4 A, float4 B)
|
|
{
|
|
return lessThan(df(A, B) , eq_threshold3);
|
|
}
|
|
|
|
float4 weighted_distance(float4 a, float4 b, float4 c, float4 d, float4 e, float4 f, float4 g, float4 h)
|
|
{
|
|
return (df(a,b) + df(a,c) + df(d,e) + df(d,f) + 4.0*df(g,h));
|
|
}
|
|
|
|
#pragma stage vertex
|
|
layout(location = 0) in vec4 Position;
|
|
layout(location = 1) in vec2 TexCoord;
|
|
layout(location = 0) out vec2 vTexCoord;
|
|
layout(location = 1) out vec4 t1;
|
|
layout(location = 2) out vec4 t2;
|
|
layout(location = 3) out vec4 t3;
|
|
layout(location = 4) out vec4 t4;
|
|
layout(location = 5) out vec4 t5;
|
|
layout(location = 6) out vec4 t6;
|
|
layout(location = 7) out vec4 t7;
|
|
|
|
void main()
|
|
{
|
|
gl_Position = global.MVP * Position;
|
|
vTexCoord = TexCoord * 1.0004;
|
|
|
|
float2 ps = float2(1.0/texture_size.x, 1.0/texture_size.y);
|
|
float dx = ps.x;
|
|
float dy = ps.y;
|
|
|
|
// A1 B1 C1
|
|
// A0 A B C C4
|
|
// D0 D E F F4
|
|
// G0 G H I I4
|
|
// G5 H5 I5
|
|
|
|
t1 = vTexCoord.xxxy + float4( -dx, 0., dx,-2.0*dy); // A1 B1 C1
|
|
t2 = vTexCoord.xxxy + float4( -dx, 0., dx, -dy); // A B C
|
|
t3 = vTexCoord.xxxy + float4( -dx, 0., dx, 0.); // D E F
|
|
t4 = vTexCoord.xxxy + float4( -dx, 0., dx, dy); // G H I
|
|
t5 = vTexCoord.xxxy + float4( -dx, 0., dx, 2.0*dy); // G5 H5 I5
|
|
t6 = vTexCoord.xyyy + float4(-2.0*dx,-dy, 0., dy); // A0 D0 G0
|
|
t7 = vTexCoord.xyyy + float4( 2.0*dx,-dy, 0., dy); // C4 F4 I4
|
|
}
|
|
|
|
#pragma stage fragment
|
|
layout(location = 0) in vec2 vTexCoord;
|
|
layout(location = 1) in vec4 t1;
|
|
layout(location = 2) in vec4 t2;
|
|
layout(location = 3) in vec4 t3;
|
|
layout(location = 4) in vec4 t4;
|
|
layout(location = 5) in vec4 t5;
|
|
layout(location = 6) in vec4 t6;
|
|
layout(location = 7) in vec4 t7;
|
|
layout(location = 0) out vec4 FragColor;
|
|
layout(set = 0, binding = 2) uniform sampler2D Source;
|
|
layout(set = 0, binding = 3) uniform sampler2D REF;
|
|
|
|
void main()
|
|
{
|
|
vec4 PA = texture(Source, t2.xw);
|
|
vec4 PB = texture(Source, t2.yw);
|
|
vec4 PC = texture(Source, t2.zw);
|
|
|
|
vec4 PD = texture(Source, t3.xw);
|
|
vec4 PE = texture(Source, t3.yw);
|
|
vec4 PF = texture(Source, t3.zw);
|
|
|
|
vec4 PG = texture(Source, t4.xw);
|
|
vec4 PH = texture(Source, t4.yw);
|
|
vec4 PI = texture(Source, t4.zw);
|
|
|
|
vec3 A1 = texture(REF, t1.xw).rgb;
|
|
vec3 B1 = texture(REF, t1.yw).rgb;
|
|
vec3 C1 = texture(REF, t1.zw).rgb;
|
|
|
|
vec3 A = texture(REF, t2.xw).rgb;
|
|
vec3 B = texture(REF, t2.yw).rgb;
|
|
vec3 C = texture(REF, t2.zw).rgb;
|
|
|
|
vec3 D = texture(REF, t3.xw).rgb;
|
|
vec3 E = texture(REF, t3.yw).rgb;
|
|
vec3 F = texture(REF, t3.zw).rgb;
|
|
|
|
vec3 G = texture(REF, t4.xw).rgb;
|
|
vec3 H = texture(REF, t4.yw).rgb;
|
|
vec3 I = texture(REF, t4.zw).rgb;
|
|
|
|
vec3 G5 = texture(REF, t5.xw).rgb;
|
|
vec3 H5 = texture(REF, t5.yw).rgb;
|
|
vec3 I5 = texture(REF, t5.zw).rgb;
|
|
|
|
vec3 A0 = texture(REF, t6.xy).rgb;
|
|
vec3 D0 = texture(REF, t6.xz).rgb;
|
|
vec3 G0 = texture(REF, t6.xw).rgb;
|
|
|
|
vec3 C4 = texture(REF, t7.xy).rgb;
|
|
vec3 F4 = texture(REF, t7.xz).rgb;
|
|
vec3 I4 = texture(REF, t7.xw).rgb;
|
|
|
|
float4 b = mul( float4x3(B, D, H, F), yuv_weighted[0] );
|
|
float4 c = mul( float4x3(C, A, G, I), yuv_weighted[0] );
|
|
float4 e = mul( float4x3(E, E, E, E), yuv_weighted[0] );
|
|
float4 d = b.yzwx;
|
|
float4 f = b.wxyz;
|
|
float4 g = c.zwxy;
|
|
float4 h = b.zwxy;
|
|
float4 i = c.wxyz;
|
|
|
|
float4 i4 = mul( float4x3(I4, C1, A0, G5), yuv_weighted[0] );
|
|
float4 i5 = mul( float4x3(I5, C4, A1, G0), yuv_weighted[0] );
|
|
float4 h5 = mul( float4x3(H5, F4, B1, D0), yuv_weighted[0] );
|
|
float4 f4 = h5.yzwx;
|
|
|
|
float4 pe = remapFrom01(PE, maximo);
|
|
float4 pf = remapFrom01(PF, maximo);
|
|
float4 ph = remapFrom01(PH, maximo);
|
|
float4 pb = remapFrom01(PB, maximo);
|
|
float4 pd = remapFrom01(PD, maximo);
|
|
|
|
float4 f2 = float4(pf.z, pb.w, pd.x, ph.y);
|
|
float4 h2 = float4(ph.z, pf.w, pb.x, pd.y);
|
|
float4 f1 = float4(pf.y, pb.z, pd.w, ph.x);
|
|
float4 h3 = float4(ph.w, pf.x, pb.y, pd.z);
|
|
|
|
bool4 nbrs;
|
|
// bool4 nbrs = ((pe.yzwx > 1.0) || (pe.wxyz > 1.0)) ? bool4(true) : bool4(false);
|
|
nbrs.x = ((pe.y > 1.0) || (pe.w > 1.0));
|
|
nbrs.y = ((pe.z > 1.0) || (pe.x > 1.0));
|
|
nbrs.z = ((pe.w > 1.0) || (pe.y > 1.0));
|
|
nbrs.w = ((pe.x > 1.0) || (pe.z > 1.0));
|
|
|
|
bool4 jag1;
|
|
// bool4 jag1 = ((f2 > 1.0) || (h2 > 1.0)) ? bool4(true) : bool4(false);
|
|
jag1.x = ((f2.x > 1.0) || (h2.x > 1.0));
|
|
jag1.y = ((f2.y > 1.0) || (h2.y > 1.0));
|
|
jag1.z = ((f2.z > 1.0) || (h2.z > 1.0));
|
|
jag1.w = ((f2.w > 1.0) || (h2.w > 1.0));
|
|
|
|
bool4 jag2;
|
|
// bool4 jag2 = ((f2 > 2.0) || (h2 > 2.0)) ? bool4(true) : bool4(false);
|
|
jag2.x = ((f2.x > 2.0) || (h2.x > 2.0));
|
|
jag2.y = ((f2.y > 2.0) || (h2.y > 2.0));
|
|
jag2.z = ((f2.z > 2.0) || (h2.z > 2.0));
|
|
jag2.w = ((f2.w > 2.0) || (h2.w > 2.0));
|
|
|
|
bool4 jag3;
|
|
// bool4 jag3 = ((f2 > 4.0) || (h2 > 4.0)) ? bool4(true) : bool4(false);
|
|
jag3.x = ((f2.x > 4.0) || (h2.x > 4.0));
|
|
jag3.y = ((f2.y > 4.0) || (h2.y > 4.0));
|
|
jag3.z = ((f2.z > 4.0) || (h2.z > 4.0));
|
|
jag3.w = ((f2.w > 4.0) || (h2.w > 4.0));
|
|
|
|
// pe = (pe == 7.0 || pe == 8.0) ? ((jag3 ) ? pe : (pe - float(2.0))) : pe;
|
|
pe.x = (pe.x == 7.0 || pe.x == 8.0) ? ((jag3.x) ? pe.x : (pe.x - float(2.0))) : pe.x;
|
|
pe.y = (pe.y == 7.0 || pe.y == 8.0) ? ((jag3.y) ? pe.y : (pe.y - float(2.0))) : pe.y;
|
|
pe.z = (pe.z == 7.0 || pe.z == 8.0) ? ((jag3.z) ? pe.z : (pe.z - float(2.0))) : pe.z;
|
|
pe.w = (pe.w == 7.0 || pe.w == 8.0) ? ((jag3.w) ? pe.w : (pe.w - float(2.0))) : pe.w;
|
|
|
|
// pe = (pe == 5.0 || pe == 6.0) ? ((jag2 ) ? pe : (pe - float(2.0))) : pe;
|
|
pe.x = (pe.x == 5.0 || pe.x == 6.0) ? ((jag2.x) ? pe.x : (pe.x - float(2.0))) : pe.x;
|
|
pe.y = (pe.y == 5.0 || pe.y == 6.0) ? ((jag2.y) ? pe.y : (pe.y - float(2.0))) : pe.y;
|
|
pe.z = (pe.z == 5.0 || pe.z == 6.0) ? ((jag2.z) ? pe.z : (pe.z - float(2.0))) : pe.z;
|
|
pe.w = (pe.w == 5.0 || pe.w == 6.0) ? ((jag2.w) ? pe.w : (pe.w - float(2.0))) : pe.w;
|
|
|
|
bool4 jag91;
|
|
// bool4 jag91 = ((id(h,i,e,h) || id(i4,i,f4,i4) ) && (f2 > 1.0) && (f1 > 1.0));
|
|
jag91.x = ((id(h,i,e,h).x || id(i4,i,f4,i4).x) && (f2.x > 1.0) && (f1.x > 1.0));
|
|
jag91.y = ((id(h,i,e,h).y || id(i4,i,f4,i4).y) && (f2.y > 1.0) && (f1.y > 1.0));
|
|
jag91.z = ((id(h,i,e,h).z || id(i4,i,f4,i4).z) && (f2.z > 1.0) && (f1.z > 1.0));
|
|
jag91.w = ((id(h,i,e,h).w || id(i4,i,f4,i4).w) && (f2.w > 1.0) && (f1.w > 1.0));
|
|
|
|
bool4 jag92;
|
|
// bool4 jag92 = ((id(f,i,e,f) || id(i5,i,h5,i5) ) && (h2 > 1.0) && (h3 > 1.0));
|
|
jag92.x = ((id(f,i,e,f).x || id(i5,i,h5,i5).x) && (h2.x > 1.0) && (h3.x > 1.0));
|
|
jag92.y = ((id(f,i,e,f).y || id(i5,i,h5,i5).y) && (h2.y > 1.0) && (h3.y > 1.0));
|
|
jag92.z = ((id(f,i,e,f).z || id(i5,i,h5,i5).z) && (h2.z > 1.0) && (h3.z > 1.0));
|
|
jag92.w = ((id(f,i,e,f).w || id(i5,i,h5,i5).w) && (h2.w > 1.0) && (h3.w > 1.0));
|
|
|
|
bvec4 jag93 = ( rd(h,g,e,g));
|
|
bvec4 jag94 = ( rd(f,c,e,c));
|
|
|
|
bool4 jag9;
|
|
// bool4 jag9 = (!(jag91 && jag93 || jag92 && jag94 ));
|
|
jag9.x = (!(jag91.x && jag93.x || jag92.x && jag94.x));
|
|
jag9.y = (!(jag91.y && jag93.y || jag92.y && jag94.y));
|
|
jag9.z = (!(jag91.z && jag93.z || jag92.z && jag94.z));
|
|
jag9.w = (!(jag91.w && jag93.w || jag92.w && jag94.w));
|
|
|
|
// pe = ((pe == 0.0) || (!nbrs || jag1 ) && jag9 ) ? pe : float4(1.0);
|
|
pe.x = ((pe.x == 0.0) || (!nbrs.x || jag1.x) && jag9.x) ? pe.x : float(1.0);
|
|
pe.y = ((pe.y == 0.0) || (!nbrs.y || jag1.y) && jag9.y) ? pe.y : float(1.0);
|
|
pe.z = ((pe.z == 0.0) || (!nbrs.z || jag1.z) && jag9.z) ? pe.z : float(1.0);
|
|
pe.w = ((pe.w == 0.0) || (!nbrs.w || jag1.w) && jag9.w) ? pe.w : float(1.0);
|
|
|
|
FragColor = vec4(remapTo01(pe, maximo));
|
|
}
|