Files
armorpaint/Sources/arm/data/ConstData.hx
T
2019-06-22 13:57:55 +02:00

319 lines
12 KiB
Haxe

package arm.data;
import kha.graphics4.TextureFormat;
import kha.Blob;
import kha.Image;
import iron.data.Data;
import iron.system.ArmPack;
class ConstData {
// 2D layer view
#if kha_direct3d11
public static var layerViewVert = "
uniform float4x4 projectionMatrix;
struct SPIRV_Cross_Output { float4 color : TEXCOORD0; float2 texCoord : TEXCOORD1; float4 gl_Position : SV_Position; };
SPIRV_Cross_Output main(float3 pos : TEXCOORD0, float2 tex : TEXCOORD1, float4 col : TEXCOORD2) {
SPIRV_Cross_Output stage_output;
stage_output.gl_Position = mul(float4(pos, 1.0f), projectionMatrix);
stage_output.gl_Position.z = (stage_output.gl_Position.z + stage_output.gl_Position.w) * 0.5;
stage_output.texCoord = tex;
stage_output.color = col;
return stage_output;
}
";
public static var layerViewFrag = "
Texture2D<float4> tex;
SamplerState _tex_sampler;
float4 main(float4 color : TEXCOORD0, float2 texCoord : TEXCOORD1) : SV_Target0 {
return tex.SampleLevel(_tex_sampler, texCoord, 0) * color;
}
";
#else // kha_opengl
public static var layerViewVert = "#version 330
in vec3 pos;
in vec2 tex;
in vec4 col;
uniform mat4 projectionMatrix;
out vec2 texCoord;
out vec4 color;
void main() {
gl_Position = projectionMatrix * vec4(pos, 1.0);
texCoord = tex;
color = col;
}
";
public static var layerViewFrag = "#version 330
uniform sampler2D tex;
in vec2 texCoord;
in vec4 color;
out vec4 FragColor;
void main() {
FragColor = textureLod(tex, texCoord, 0) * color;
}
";
#end
// 2D layer merge
#if kha_direct3d11
public static var layerMergeVert = "
struct SPIRV_Cross_Output { float2 texCoord : TEXCOORD0; float4 gl_Position : SV_Position; };
SPIRV_Cross_Output main(float2 pos : TEXCOORD0) {
SPIRV_Cross_Output stage_output;
stage_output.gl_Position = float4(pos.xy, 0.0, 1.0);
stage_output.gl_Position.z = (stage_output.gl_Position.z + stage_output.gl_Position.w) * 0.5;
const float2 madd = float2(0.5, 0.5);
stage_output.texCoord = pos.xy * madd + madd;
stage_output.texCoord.y = 1.0 - stage_output.texCoord.y;
return stage_output;
}
";
public static var layerMergeFrag = "
Texture2D<float4> tex0;
SamplerState _tex0_sampler;
Texture2D<float4> tex1;
SamplerState _tex1_sampler;
Texture2D<float4> tex2;
SamplerState _tex2_sampler;
Texture2D<float4> texa;
SamplerState _texa_sampler;
Texture2D<float4> texb;
SamplerState _texb_sampler;
Texture2D<float4> texc;
SamplerState _texc_sampler;
uniform float opac;
struct SPIRV_Cross_Output { float4 color0 : SV_Target0; float4 color1 : SV_Target1; float4 color2 : SV_Target2; };
SPIRV_Cross_Output main(float2 texCoord : TEXCOORD0) {
float4 col0 = tex0.SampleLevel(_tex0_sampler, texCoord, 0);
float4 col1 = tex1.SampleLevel(_tex1_sampler, texCoord, 0);
float4 col2 = tex2.SampleLevel(_tex2_sampler, texCoord, 0);
float4 cola = texa.SampleLevel(_texa_sampler, texCoord, 0);
float4 colb = texb.SampleLevel(_texb_sampler, texCoord, 0);
float4 colc = texc.SampleLevel(_texc_sampler, texCoord, 0);
float str = col0.a * opac;
SPIRV_Cross_Output stage_output;
stage_output.color0 = float4(lerp(cola.rgb, col0.rgb, str), max(col0.a, cola.a));
stage_output.color1 = float4(lerp(colb, col1, str));
stage_output.color2 = float4(lerp(colc, col2, str));
return stage_output;
}
";
public static var maskMergeFrag = "
Texture2D<float4> tex0;
SamplerState _tex0_sampler;
Texture2D<float4> texa;
SamplerState _texa_sampler;
struct SPIRV_Cross_Output { float4 color0 : SV_Target0; };
SPIRV_Cross_Output main(float2 texCoord : TEXCOORD0) {
float4 col0 = tex0.SampleLevel(_tex0_sampler, texCoord, 0);
float mask = texa.SampleLevel(_texa_sampler, texCoord, 0).r;
SPIRV_Cross_Output stage_output;
stage_output.color0 = float4(col0.rgb, col0.a * mask);
return stage_output;
}
";
#else // kha_opengl
public static var layerMergeVert = "#version 330
in vec2 pos;
out vec2 texCoord;
void main() {
gl_Position = vec4(pos.xy, 0.0, 1.0);
const vec2 madd = vec2(0.5, 0.5);
texCoord = pos.xy * madd + madd;
}
";
public static var layerMergeFrag = "#version 330
uniform sampler2D tex0;
uniform sampler2D tex1;
uniform sampler2D tex2;
uniform sampler2D texa;
uniform sampler2D texb;
uniform sampler2D texc;
uniform float opac;
in vec2 texCoord;
out vec4 FragColor[3];
void main() {
vec4 col0 = textureLod(tex0, texCoord, 0);
vec4 col1 = textureLod(tex1, texCoord, 0);
vec4 col2 = textureLod(tex2, texCoord, 0);
vec4 cola = textureLod(texa, texCoord, 0);
vec4 colb = textureLod(texb, texCoord, 0);
vec4 colc = textureLod(texc, texCoord, 0);
float str = col0.a * opac;
FragColor[0] = vec4(mix(cola.rgb, col0.rgb, str), max(col0.a, cola.a));
FragColor[1] = vec4(mix(colb, col1, str));
FragColor[2] = vec4(mix(colc, col2, str));
}
";
public static var maskMergeFrag = "#version 330
uniform sampler2D tex0;
uniform sampler2D texa;
in vec2 texCoord;
out vec4 FragColor;
void main() {
vec4 col0 = textureLod(tex0, texCoord, 0);
float mask = textureLod(texa, texCoord, 0).r;
FragColor = vec4(col0.rgb, col0.a * mask);
}
";
#end
#if kha_direct3d11
public static var cursorVert = "
uniform float4x4 VP;
uniform float4x4 invVP;
uniform float2 mouse;
uniform float2 step;
uniform float radius;
Texture2D<float4> gbufferD;
SamplerState _gbufferD_sampler;
Texture2D<float4> gbuffer0;
SamplerState _gbuffer0_sampler;
float2 octahedronWrap(float2 v) { return (1.0 - abs(v.yx)) * (float2(v.x >= 0.0 ? 1.0 : -1.0, v.y >= 0.0 ? 1.0 : -1.0)); }
float3x3 rotAxis(float3 axis, float a) {
float c = cos(a);
float3 as = axis * sin(a).xxx;
float3x3 p = float3x3(axis.xxx * axis, axis.yyy * axis, axis.zzz * axis);
float3x3 q = float3x3(c, -as.z, as.y, as.z, c, -as.x, -as.y, as.x, c);
return p * (1.0 - c) + q;
}
float3 getNormal(float2 uv) {
float2 g0 = gbuffer0.SampleLevel(_gbuffer0_sampler, uv, 0.0).rg;
float3 n;
n.z = 1.0 - abs(g0.x) - abs(g0.y);
n.xy = n.z >= 0.0 ? g0.xy : octahedronWrap(g0.xy);
return n;
}
struct SPIRV_Cross_Output { float2 texCoord : TEXCOORD0; float4 gl_Position : SV_Position; };
SPIRV_Cross_Output main(float4 pos : TEXCOORD1, float2 nor : TEXCOORD0, float2 tex : TEXCOORD2) {
SPIRV_Cross_Output stage_output;
stage_output.texCoord = tex;
float2 mouseinv = float2(mouse.x, 1.0 - mouse.y);
float depth = gbufferD.SampleLevel(_gbufferD_sampler, mouseinv, 0).r;
float4 wpos = float4(mouse * 2.0 - 1.0, depth * 2.0 - 1.0, 1.0);
wpos = mul(wpos, invVP);
wpos.xyz /= wpos.w;
float3 n = normalize(
getNormal(mouseinv + float2(step.x, step.y)) +
getNormal(mouseinv + float2(-step.x, step.y)) +
getNormal(mouseinv + float2(-step.x, -step.y)) +
getNormal(mouseinv + float2(step.x, -step.y)) +
getNormal(mouseinv)
);
float ax = acos(dot(float3(1,0,0), float3(n.x,0,0)));
float az = acos(dot(float3(0,0,1), float3(0,0,n.z)));
float sy = -sign(n.y);
wpos.xyz += mul(mul(pos.xyz * radius.xxx, rotAxis(float3(0,0,1), ax + 3.14/2)),
rotAxis(float3(1,0,0), -az * sy + 3.14/2));
stage_output.gl_Position = mul(float4(wpos.xyz, 1.0), VP);
stage_output.gl_Position.z = (stage_output.gl_Position.z + stage_output.gl_Position.w) * 0.5;
return stage_output;
}
";
public static var cursorFrag = "
Texture2D<float4> tex;
SamplerState _tex_sampler;
float4 main(float2 texCoord : TEXCOORD0) : SV_Target0 {
float4 col = tex.Sample(_tex_sampler, texCoord);
return float4(col.rgb / col.a, col.a);
}
";
#else // kha_opengl
public static var cursorVert = "#version 330
uniform mat4 VP;
uniform mat4 invVP;
uniform vec2 mouse;
uniform vec2 step;
uniform float radius;
uniform sampler2D gbufferD;
uniform sampler2D gbuffer0;
in vec4 pos;
in vec2 nor;
in vec2 tex;
out vec2 texCoord;
vec2 octahedronWrap(const vec2 v) { return (1.0 - abs(v.yx)) * (vec2(v.x >= 0.0 ? 1.0 : -1.0, v.y >= 0.0 ? 1.0 : -1.0)); }
mat3 rotAxis(vec3 axis, float a) {
float c = cos(a);
vec3 as = axis * sin(a);
mat3 p = mat3(axis.x * axis, axis.y * axis, axis.z * axis);
mat3 q = mat3(c, -as.z, as.y, as.z, c, -as.x, -as.y, as.x, c);
return p * (1.0 - c) + q;
}
vec3 getNormal(vec2 uv) {
vec2 g0 = textureLod(gbuffer0, mouse, 0.0).rg;
vec3 n;
n.z = 1.0 - abs(g0.x) - abs(g0.y);
n.xy = n.z >= 0.0 ? g0.xy : octahedronWrap(g0.xy);
return n;
}
void main() {
texCoord = tex;
float depth = textureLod(gbufferD, mouse, 0.0).r;
vec4 wpos = vec4(mouse * 2.0 - 1.0, depth * 2.0 - 1.0, 1.0);
wpos = invVP * wpos;
wpos.xyz /= wpos.w;
vec3 n = normalize(
getNormal(mouse + vec2(step.x, step.y)) +
getNormal(mouse + vec2(-step.x, step.y)) +
getNormal(mouse + vec2(-step.x, -step.y)) +
getNormal(mouse + vec2(step.x, -step.y)) +
getNormal(mouse)
);
float ax = acos(dot(vec3(1,0,0), vec3(n.x,0,0)));
float az = acos(dot(vec3(0,0,1), vec3(0,0,n.z)));
float sy = -sign(n.y);
wpos.xyz +=
rotAxis(vec3(1,0,0), -az * sy + 3.14/2) *
rotAxis(vec3(0,0,1), ax + 3.14/2) *
(pos.xyz * radius);
gl_Position = VP * vec4(wpos.xyz, 1.0);
}
";
public static var cursorFrag = "#version 330
uniform sampler2D tex;
in vec2 texCoord;
out vec4 FragColor;
void main() {
vec4 col = texture(tex, texCoord);
FragColor = vec4(col.rgb / col.a, col.a);
}
";
#end
#if kha_krom
public static var font_x0 = [1,2,6,10,18,25,34,42,45,50,55,61,68,71,75,79,85,92,97,104,111,118,1,8,15,22,29,32,35,41,48,55,61,72,81,89,97,105,112,119,1,9,13,20,28,35,45,53,61,69,77,85,93,101,109,1,12,20,28,36,40,46,50,56,62,66,73,80,87,94,101,106,113,120,123,1,8,11,21,28,35,42,49,54,61,66,73,80,90,97,104,111,116,119,1];
public static var font_y0 = [1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,14,14,14,14,14,14,14,14,14,14,14,14,14,14,14,14,14,14,26,26,26,26,26,26,26,26,26,26,26,26,26,26,26,37,37,37,37,37,37,37,37,37,37,37,37,37,37,37,37,37,37,37,37,50,50,50,50,50,50,50,50,50,50,50,50,50,50,50,50,50,50,50,62];
public static var font_x1 = [1,5,9,17,24,33,41,44,49,54,60,67,70,74,78,84,91,96,103,110,117,124,7,14,21,28,31,34,40,47,54,60,71,80,88,96,104,111,118,126,8,12,19,27,34,44,52,60,68,76,84,92,100,108,116,11,19,27,35,39,45,49,55,61,65,72,79,86,93,100,105,112,119,122,126,7,10,20,27,34,41,48,53,60,65,72,79,89,96,103,110,115,118,123,8];
public static var font_y1 = [1,10,5,9,13,11,10,5,13,13,6,8,5,3,4,10,10,9,9,10,9,10,23,22,23,23,21,22,19,18,19,23,25,22,22,23,22,22,22,23,34,34,35,34,34,34,34,35,34,36,34,35,34,35,34,45,45,45,45,49,46,49,42,38,40,44,47,44,47,44,46,46,46,45,48,59,59,56,56,57,59,59,56,57,59,57,56,56,56,59,56,61,60,61,65];
public static var font_xoff = [0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,-1,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0];
public static var font_yoff = [10,2,1,2,0,1,2,1,1,1,2,3,8,6,8,2,2,2,2,2,2,2,2,2,2,2,4,4,4,4,4,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,0,2,0,2,10,1,4,1,4,1,4,1,4,1,2,2,1,1,4,4,4,4,4,4,4,2,4,4,4,4,4,4,1,2,1,5];
public static var font_xadvance = [2.74625,2.85458,3.54791,6.83041,6.22916,8.125,6.89541,1.93375,3.79166,3.85666,4.7775,6.28875,2.1775,3.06041,2.91958,4.57166,6.22916,6.22916,6.22916,6.22916,6.22916,6.22916,6.22916,6.22916,6.22916,6.22916,2.68666,2.34541,5.63875,6.08833,5.79583,5.23791,9.96125,7.23666,6.90625,7.22041,7.27458,6.305,6.13166,7.55625,7.9083,3.01708,6.12083,6.955,5.96916,9.685,7.90833,7.62666,6.99833,7.62666,6.83041,6.58125,6.61916,7.19333,7.05791,9.84208,6.955,6.6625,6.64083,2.94125,4.55,2.94125,4.63666,5.005,3.42875,6.03416,6.22375,5.80666,6.25625,5.87708,3.85125,6.22375,6.11,2.69208,2.64875,5.6225,2.69208,9.72291,6.12083,6.32666,6.22375,6.305,3.75375,5.72,3.62375,6.11541,5.37333,8.33625,5.49791,5.24875,5.49791,3.75375,2.70291,3.75375,7.54541];
#end
#if arm_ltc
public static var ltcMatTex: Image = null;
public static var ltcMagTex: Image = null;
public static function initLTC() {
// Real-Time Polygonal-Light Shading with Linearly Transformed Cosines
// https://eheitzresearch.wordpress.com/415-2/
Data.getBlob("ltc_mat.arm", function(ltc_mat:Blob) {
Data.getBlob("ltc_mag.arm", function(ltc_mag:Blob) {
ltcMatTex = Image.fromBytes(ArmPack.decode(ltc_mat.toBytes()), 64, 64, TextureFormat.RGBA128);
ltcMagTex = Image.fromBytes(ArmPack.decode(ltc_mag.toBytes()), 64, 64, TextureFormat.A32);
});
});
}
#end
}