luboslenco
81 lines
3.1 KiB
Haxe
81 lines
3.1 KiB
Haxe
package arm.node;
|
|
|
|
import iron.object.MeshObject;
|
|
import iron.data.SceneFormat;
|
|
import arm.ui.UITrait;
|
|
import arm.ui.UINodes;
|
|
import arm.node.MaterialShader;
|
|
import arm.Tool;
|
|
|
|
class MakeVoxel {
|
|
|
|
#if rp_voxelao
|
|
public static function run(data:iron.data.ShaderData.ShaderContext) {
|
|
var structure = new kha.graphics4.VertexStructure();
|
|
structure.add("pos", kha.graphics4.VertexData.Short4Norm);
|
|
structure.add("nor", kha.graphics4.VertexData.Short2Norm);
|
|
structure.add("tex", kha.graphics4.VertexData.Short2Norm);
|
|
|
|
var pipeState = data.pipeState;
|
|
pipeState.inputLayout = [structure];
|
|
data.raw.vertex_elements = [{name: "pos", data: 'short4norm'}, {name: "nor", data: 'short2norm'}, {name: "tex", data: 'short2norm'}];
|
|
|
|
// #if arm_skin
|
|
// var isMesh = Std.is(Context.object, MeshObject);
|
|
// var skin = isMesh && cast(Context.object, MeshObject).data.geom.bones != null;
|
|
// if (skin) {
|
|
// structure.add("bone", kha.graphics4.VertexData.Short4Norm);
|
|
// structure.add("weight", kha.graphics4.VertexData.Short4Norm);
|
|
// data.raw.vertex_elements.push({name: "bone", data: 'short4norm'});
|
|
// data.raw.vertex_elements.push({name: "weight", data: 'short4norm'});
|
|
// }
|
|
// #end
|
|
|
|
var ds = MaterialBuilder.getDisplaceStrength();
|
|
pipeState.vertexShader = kha.graphics4.VertexShader.fromSource(
|
|
#if (kha_direct3d11 || kha_direct3d12)
|
|
"#define vec3 float3
|
|
uniform float4x4 W;
|
|
uniform float3x3 N;
|
|
Texture2D<float4> texpaint_pack;
|
|
SamplerState _texpaint_pack_sampler;
|
|
struct SPIRV_Cross_Input { float4 pos : TEXCOORD1; float2 nor : TEXCOORD0; float2 tex : TEXCOORD2; };
|
|
struct SPIRV_Cross_Output { float4 svpos : SV_POSITION; };
|
|
SPIRV_Cross_Output main(SPIRV_Cross_Input stage_input) {
|
|
SPIRV_Cross_Output stage_output;
|
|
" + MaterialBuilder.voxelgiHalfExtents() + "
|
|
stage_output.svpos.xyz = mul(float4(stage_input.pos.xyz, 1.0), W).xyz / voxelgiHalfExtents.xxx;
|
|
float3 wnormal = normalize(mul(float3(stage_input.nor.xy, stage_input.pos.w), N));
|
|
float height = texpaint_pack.SampleLevel(_texpaint_pack_sampler, stage_input.tex, 0.0).a;
|
|
stage_output.svpos.xyz += wnormal * height.xxx * float3(" + ds + "," + ds + "," + ds + ");
|
|
stage_output.svpos.w = 1.0;
|
|
return stage_output;
|
|
}"
|
|
#else
|
|
"#version 450
|
|
in vec4 pos;
|
|
in vec2 nor;
|
|
in vec2 tex;
|
|
out vec3 voxpositionGeom;
|
|
uniform mat4 W;
|
|
uniform mat3 N;
|
|
uniform sampler2D texpaint_pack;
|
|
void main() {
|
|
" + MaterialBuilder.voxelgiHalfExtents() + "
|
|
voxpositionGeom = vec3(W * vec4(pos.xyz, 1.0)) / voxelgiHalfExtents;
|
|
vec3 wnormal = normalize(N * vec3(nor.xy, pos.w));
|
|
float height = textureLod(texpaint_pack, tex, 0.0).a;
|
|
voxpositionGeom += wnormal * vec3(height) * vec3(" + ds + ");
|
|
}"
|
|
#end
|
|
);
|
|
|
|
pipeState.compile();
|
|
data.raw.constants = [{ name: "W", type: "mat4", link: "_worldMatrix" }, { name: "N", type: "mat3", link: "_normalMatrix" }];
|
|
data.constants = [pipeState.getConstantLocation("W"), pipeState.getConstantLocation("N")];
|
|
data.raw.texture_units = [{name: "texpaint_pack"}, {name: "voxels", is_image: true}];
|
|
data.textureUnits = [pipeState.getTextureUnit("texpaint_pack"), pipeState.getTextureUnit("voxels")];
|
|
}
|
|
#end
|
|
}
|