forked from LeenkxTeam/LNXSDK
Update Files
This commit is contained in:
401
leenkx/Sources/iron/data/ShaderData.hx
Normal file
401
leenkx/Sources/iron/data/ShaderData.hx
Normal file
@ -0,0 +1,401 @@
|
||||
package iron.data;
|
||||
|
||||
import kha.graphics4.PipelineState;
|
||||
import kha.graphics4.ConstantLocation;
|
||||
import kha.graphics4.TextureUnit;
|
||||
import kha.graphics4.VertexStructure;
|
||||
import kha.graphics4.VertexData;
|
||||
import kha.graphics4.CompareMode;
|
||||
import kha.graphics4.CullMode;
|
||||
import kha.graphics4.BlendingOperation;
|
||||
import kha.graphics4.BlendingFactor;
|
||||
import kha.graphics4.TextureAddressing;
|
||||
import kha.graphics4.TextureFilter;
|
||||
import kha.graphics4.MipMapFilter;
|
||||
import kha.graphics4.VertexShader;
|
||||
import kha.graphics4.FragmentShader;
|
||||
import kha.graphics4.TextureFormat;
|
||||
import kha.graphics4.DepthStencilFormat;
|
||||
import iron.data.SceneFormat;
|
||||
using StringTools;
|
||||
|
||||
class ShaderData {
|
||||
|
||||
public var name: String;
|
||||
public var raw: TShaderData;
|
||||
public var contexts: Array<ShaderContext> = [];
|
||||
|
||||
#if (lnx_noembed && kha_krom)
|
||||
public static var shaderPath = "../krom-resources/";
|
||||
public static inline var shaderExt = #if kha_vulkan ".spirv" #elseif (krom_android || krom_wasm) ".essl" #elseif kha_opengl ".glsl" #elseif kha_metal ".metal" #else ".d3d11" #end ;
|
||||
#end
|
||||
|
||||
public function new(raw: TShaderData, done: ShaderData->Void, overrideContext: TShaderOverride = null) {
|
||||
this.raw = raw;
|
||||
this.name = raw.name;
|
||||
|
||||
for (c in raw.contexts) contexts.push(null);
|
||||
var contextsLoaded = 0;
|
||||
|
||||
for (i in 0...raw.contexts.length) {
|
||||
var c = raw.contexts[i];
|
||||
new ShaderContext(c, function(con: ShaderContext) {
|
||||
contexts[i] = con;
|
||||
contextsLoaded++;
|
||||
if (contextsLoaded == raw.contexts.length) done(this);
|
||||
}, overrideContext);
|
||||
}
|
||||
}
|
||||
|
||||
public static function parse(file: String, name: String, done: ShaderData->Void, overrideContext: TShaderOverride = null) {
|
||||
Data.getSceneRaw(file, function(format: TSceneFormat) {
|
||||
var raw: TShaderData = Data.getShaderRawByName(format.shader_datas, name);
|
||||
if (raw == null) {
|
||||
trace('Shader data "$name" not found!');
|
||||
done(null);
|
||||
}
|
||||
new ShaderData(raw, done, overrideContext);
|
||||
});
|
||||
}
|
||||
|
||||
public function delete() {
|
||||
for (c in contexts) c.delete();
|
||||
}
|
||||
|
||||
public function getContext(name: String): ShaderContext {
|
||||
for (c in contexts) if (c.raw.name == name) return c;
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
class ShaderContext {
|
||||
public var raw: TShaderContext;
|
||||
public var pipeState: PipelineState;
|
||||
public var constants: Array<ConstantLocation>;
|
||||
public var textureUnits: Array<TextureUnit>;
|
||||
public var overrideContext: TShaderOverride;
|
||||
|
||||
var structure: VertexStructure;
|
||||
var instancingType = 0;
|
||||
|
||||
public function new(raw: TShaderContext, done: ShaderContext->Void, overrideContext: TShaderOverride = null) {
|
||||
this.raw = raw;
|
||||
#if (!rp_voxels)
|
||||
if (raw.name == "voxel") {
|
||||
done(this);
|
||||
return;
|
||||
}
|
||||
#end
|
||||
this.overrideContext = overrideContext;
|
||||
parseVertexStructure();
|
||||
compile(done);
|
||||
}
|
||||
|
||||
public function compile(done: ShaderContext->Void) {
|
||||
if (pipeState != null) pipeState.delete();
|
||||
pipeState = new PipelineState();
|
||||
constants = [];
|
||||
textureUnits = [];
|
||||
|
||||
if (instancingType > 0) { // Instancing
|
||||
var instStruct = new VertexStructure();
|
||||
instStruct.add("ipos", VertexData.Float3);
|
||||
if (instancingType == 2 || instancingType == 4) {
|
||||
instStruct.add("irot", VertexData.Float3);
|
||||
}
|
||||
if (instancingType == 3 || instancingType == 4) {
|
||||
instStruct.add("iscl", VertexData.Float3);
|
||||
}
|
||||
instStruct.instanced = true;
|
||||
pipeState.inputLayout = [structure, instStruct];
|
||||
}
|
||||
else { // Regular
|
||||
pipeState.inputLayout = [structure];
|
||||
}
|
||||
|
||||
// Depth
|
||||
pipeState.depthWrite = raw.depth_write;
|
||||
pipeState.depthMode = getCompareMode(raw.compare_mode);
|
||||
|
||||
// Cull
|
||||
pipeState.cullMode = getCullMode(raw.cull_mode);
|
||||
|
||||
// Blending
|
||||
if (raw.blend_source != null) pipeState.blendSource = getBlendingFactor(raw.blend_source);
|
||||
if (raw.blend_destination != null) pipeState.blendDestination = getBlendingFactor(raw.blend_destination);
|
||||
if (raw.blend_operation != null) pipeState.blendOperation = getBlendingOperation(raw.blend_operation);
|
||||
if (raw.alpha_blend_source != null) pipeState.alphaBlendSource = getBlendingFactor(raw.alpha_blend_source);
|
||||
if (raw.alpha_blend_destination != null) pipeState.alphaBlendDestination = getBlendingFactor(raw.alpha_blend_destination);
|
||||
if (raw.alpha_blend_operation != null) pipeState.alphaBlendOperation = getBlendingOperation(raw.alpha_blend_operation);
|
||||
|
||||
// Per-target color write mask
|
||||
if (raw.color_writes_red != null) for (i in 0...raw.color_writes_red.length) pipeState.colorWriteMasksRed[i] = raw.color_writes_red[i];
|
||||
if (raw.color_writes_green != null) for (i in 0...raw.color_writes_green.length) pipeState.colorWriteMasksGreen[i] = raw.color_writes_green[i];
|
||||
if (raw.color_writes_blue != null) for (i in 0...raw.color_writes_blue.length) pipeState.colorWriteMasksBlue[i] = raw.color_writes_blue[i];
|
||||
if (raw.color_writes_alpha != null) for (i in 0...raw.color_writes_alpha.length) pipeState.colorWriteMasksAlpha[i] = raw.color_writes_alpha[i];
|
||||
|
||||
// Color attachment format
|
||||
if (raw.color_attachments != null) {
|
||||
pipeState.colorAttachmentCount = raw.color_attachments.length;
|
||||
for (i in 0...raw.color_attachments.length) pipeState.colorAttachments[i] = getTextureFormat(raw.color_attachments[i]);
|
||||
}
|
||||
|
||||
// Depth attachment format
|
||||
if (raw.depth_attachment != null) {
|
||||
#if (krom_windows || krom_linux || krom_darwin || krom_android || krom_ios)
|
||||
pipeState.depthStencilAttachment = getDepthStencilFormat(raw.depth_attachment);
|
||||
#end
|
||||
}
|
||||
|
||||
// Conservative raster for voxelization
|
||||
if (raw.conservative_raster != null) pipeState.conservativeRasterization = raw.conservative_raster;
|
||||
|
||||
// Shaders
|
||||
if (raw.shader_from_source) {
|
||||
pipeState.vertexShader = VertexShader.fromSource(raw.vertex_shader);
|
||||
pipeState.fragmentShader = FragmentShader.fromSource(raw.fragment_shader);
|
||||
|
||||
#if kha_krom
|
||||
// Shader compile error
|
||||
if (pipeState.vertexShader.shader == null || pipeState.fragmentShader.shader == null) {
|
||||
done(null);
|
||||
return;
|
||||
}
|
||||
#end
|
||||
|
||||
finishCompile(done);
|
||||
}
|
||||
else {
|
||||
|
||||
#if (lnx_noembed && kha_krom) // Load shaders manually
|
||||
|
||||
var shadersLoaded = 0;
|
||||
var numShaders = 2;
|
||||
if (raw.geometry_shader != null) numShaders++;
|
||||
if (raw.tesscontrol_shader != null) numShaders++;
|
||||
if (raw.tesseval_shader != null) numShaders++;
|
||||
|
||||
function loadShader(file: String, type: Int) {
|
||||
var path = ShaderData.shaderPath + file + ShaderData.shaderExt;
|
||||
Data.getBlob(path, function(b: kha.Blob) {
|
||||
if (type == 0) pipeState.vertexShader = new VertexShader([b], [file]);
|
||||
else if (type == 1) pipeState.fragmentShader = new FragmentShader([b], [file]);
|
||||
else if (type == 2) pipeState.geometryShader = new kha.graphics4.GeometryShader([b], [file]);
|
||||
else if (type == 3) pipeState.tessellationControlShader = new kha.graphics4.TessellationControlShader([b], [file]);
|
||||
else if (type == 4) pipeState.tessellationEvaluationShader = new kha.graphics4.TessellationEvaluationShader([b], [file]);
|
||||
shadersLoaded++;
|
||||
if (shadersLoaded >= numShaders) finishCompile(done);
|
||||
});
|
||||
}
|
||||
loadShader(raw.vertex_shader, 0);
|
||||
loadShader(raw.fragment_shader, 1);
|
||||
if (raw.geometry_shader != null) loadShader(raw.geometry_shader, 2);
|
||||
if (raw.tesscontrol_shader != null) loadShader(raw.tesscontrol_shader, 3);
|
||||
if (raw.tesseval_shader != null) loadShader(raw.tesseval_shader, 4);
|
||||
|
||||
#elseif lnx_shader_embed
|
||||
|
||||
pipeState.fragmentShader = kha.Shaders.getFragment(raw.fragment_shader);
|
||||
pipeState.vertexShader = kha.Shaders.getVertex(raw.vertex_shader);
|
||||
if (raw.geometry_shader != null) {
|
||||
pipeState.geometryShader = kha.Shaders.getGeometry(raw.geometry_shader);
|
||||
}
|
||||
finishCompile(done);
|
||||
|
||||
#else
|
||||
|
||||
pipeState.fragmentShader = Reflect.field(kha.Shaders, raw.fragment_shader.replace(".", "_"));
|
||||
pipeState.vertexShader = Reflect.field(kha.Shaders, raw.vertex_shader.replace(".", "_"));
|
||||
|
||||
if (raw.geometry_shader != null) {
|
||||
pipeState.geometryShader = Reflect.field(kha.Shaders, raw.geometry_shader.replace(".", "_"));
|
||||
}
|
||||
if (raw.tesscontrol_shader != null) {
|
||||
pipeState.tessellationControlShader = Reflect.field(kha.Shaders, raw.tesscontrol_shader.replace(".", "_"));
|
||||
}
|
||||
if (raw.tesseval_shader != null) {
|
||||
pipeState.tessellationEvaluationShader = Reflect.field(kha.Shaders, raw.tesseval_shader.replace(".", "_"));
|
||||
}
|
||||
finishCompile(done);
|
||||
|
||||
#end
|
||||
}
|
||||
}
|
||||
|
||||
function finishCompile(done: ShaderContext->Void) {
|
||||
// Override specified values
|
||||
if (overrideContext != null) {
|
||||
if (overrideContext.cull_mode != null) {
|
||||
pipeState.cullMode = getCullMode(overrideContext.cull_mode);
|
||||
}
|
||||
}
|
||||
|
||||
pipeState.compile();
|
||||
|
||||
if (raw.constants != null) {
|
||||
for (c in raw.constants) addConstant(c);
|
||||
}
|
||||
|
||||
if (raw.texture_units != null) {
|
||||
for (tu in raw.texture_units) addTexture(tu);
|
||||
}
|
||||
|
||||
done(this);
|
||||
}
|
||||
|
||||
public static function parseData(data: String): VertexData {
|
||||
if (data == "float1") return VertexData.Float1;
|
||||
else if (data == "float2") return VertexData.Float2;
|
||||
else if (data == "float3") return VertexData.Float3;
|
||||
else if (data == "float4") return VertexData.Float4;
|
||||
else if (data == "short2norm") return VertexData.Short2Norm;
|
||||
else if (data == "short4norm") return VertexData.Short4Norm;
|
||||
return VertexData.Float1;
|
||||
}
|
||||
|
||||
function parseVertexStructure() {
|
||||
structure = new VertexStructure();
|
||||
var ipos = false;
|
||||
var irot = false;
|
||||
var iscl = false;
|
||||
for (elem in raw.vertex_elements) {
|
||||
#if cpp
|
||||
if (Reflect.field(elem, "name") == "ipos") { ipos = true; continue; }
|
||||
if (Reflect.field(elem, "name") == "irot") { irot = true; continue; }
|
||||
if (Reflect.field(elem, "name") == "iscl") { iscl = true; continue; }
|
||||
#else
|
||||
if (elem.name == "ipos") { ipos = true; continue; }
|
||||
if (elem.name == "irot") { irot = true; continue; }
|
||||
if (elem.name == "iscl") { iscl = true; continue; }
|
||||
#end
|
||||
structure.add(elem.name, parseData(elem.data));
|
||||
}
|
||||
if (ipos && !irot && !iscl) instancingType = 1;
|
||||
else if (ipos && irot && !iscl) instancingType = 2;
|
||||
else if (ipos && !irot && iscl) instancingType = 3;
|
||||
else if (ipos && irot && iscl) instancingType = 4;
|
||||
}
|
||||
|
||||
public function delete() {
|
||||
if (pipeState.fragmentShader != null) pipeState.fragmentShader.delete();
|
||||
if (pipeState.vertexShader != null) pipeState.vertexShader.delete();
|
||||
if (pipeState.geometryShader != null) pipeState.geometryShader.delete();
|
||||
if (pipeState.tessellationControlShader != null) pipeState.tessellationControlShader.delete();
|
||||
if (pipeState.tessellationEvaluationShader != null) pipeState.tessellationEvaluationShader.delete();
|
||||
pipeState.delete();
|
||||
}
|
||||
|
||||
function getCompareMode(s: String): CompareMode {
|
||||
switch (s) {
|
||||
case "always": return CompareMode.Always;
|
||||
case "never": return CompareMode.Never;
|
||||
case "less": return CompareMode.Less;
|
||||
case "less_equal": return CompareMode.LessEqual;
|
||||
case "greater": return CompareMode.Greater;
|
||||
case "greater_equal": return CompareMode.GreaterEqual;
|
||||
case "equal": return CompareMode.Equal;
|
||||
case "not_equal": return CompareMode.NotEqual;
|
||||
default: return CompareMode.Less;
|
||||
}
|
||||
}
|
||||
|
||||
function getCullMode(s: String): CullMode {
|
||||
switch (s) {
|
||||
case "none": return CullMode.None;
|
||||
case "clockwise": return CullMode.Clockwise;
|
||||
default: return CullMode.CounterClockwise;
|
||||
}
|
||||
}
|
||||
|
||||
function getBlendingOperation(s: String): BlendingOperation {
|
||||
switch (s) {
|
||||
case "add": return BlendingOperation.Add;
|
||||
case "subtract": return BlendingOperation.Subtract;
|
||||
case "reverse_subtract": return BlendingOperation.ReverseSubtract;
|
||||
case "min": return BlendingOperation.Min;
|
||||
case "max": return BlendingOperation.Max;
|
||||
default: return BlendingOperation.Add;
|
||||
}
|
||||
}
|
||||
|
||||
function getBlendingFactor(s: String): BlendingFactor {
|
||||
switch (s) {
|
||||
case "blend_one": return BlendingFactor.BlendOne;
|
||||
case "blend_zero": return BlendingFactor.BlendZero;
|
||||
case "source_alpha": return BlendingFactor.SourceAlpha;
|
||||
case "destination_alpha": return BlendingFactor.DestinationAlpha;
|
||||
case "inverse_source_alpha": return BlendingFactor.InverseSourceAlpha;
|
||||
case "inverse_destination_alpha": return BlendingFactor.InverseDestinationAlpha;
|
||||
case "source_color": return BlendingFactor.SourceColor;
|
||||
case "destination_color": return BlendingFactor.DestinationColor;
|
||||
case "inverse_source_color": return BlendingFactor.InverseSourceColor;
|
||||
case "inverse_destination_color": return BlendingFactor.InverseDestinationColor;
|
||||
default: return BlendingFactor.Undefined;
|
||||
}
|
||||
}
|
||||
|
||||
function getTextureAddresing(s: String): TextureAddressing {
|
||||
switch (s) {
|
||||
case "repeat": return TextureAddressing.Repeat;
|
||||
case "mirror": return TextureAddressing.Mirror;
|
||||
default: return TextureAddressing.Clamp;
|
||||
}
|
||||
}
|
||||
|
||||
function getTextureFilter(s: String): TextureFilter {
|
||||
switch (s) {
|
||||
case "point": return TextureFilter.PointFilter;
|
||||
case "linear": return TextureFilter.LinearFilter;
|
||||
default: return TextureFilter.AnisotropicFilter;
|
||||
}
|
||||
}
|
||||
|
||||
function getMipmapFilter(s: String): MipMapFilter {
|
||||
switch (s) {
|
||||
case "no": return MipMapFilter.NoMipFilter;
|
||||
case "point": return MipMapFilter.PointMipFilter;
|
||||
default: return MipMapFilter.LinearMipFilter;
|
||||
}
|
||||
}
|
||||
|
||||
function getTextureFormat(s: String): TextureFormat {
|
||||
switch (s) {
|
||||
case "RGBA32": return TextureFormat.RGBA32;
|
||||
case "RGBA64": return TextureFormat.RGBA64;
|
||||
case "RGBA128": return TextureFormat.RGBA128;
|
||||
case "DEPTH16": return TextureFormat.DEPTH16;
|
||||
case "R32": return TextureFormat.A32;
|
||||
case "R16": return TextureFormat.A16;
|
||||
case "R8": return TextureFormat.L8;
|
||||
default: return TextureFormat.RGBA32;
|
||||
}
|
||||
}
|
||||
|
||||
function getDepthStencilFormat(s: String): DepthStencilFormat {
|
||||
switch (s) {
|
||||
case "DEPTH32": return DepthStencilFormat.DepthOnly;
|
||||
case "NONE": return DepthStencilFormat.NoDepthAndStencil;
|
||||
default: return DepthStencilFormat.DepthOnly;
|
||||
}
|
||||
}
|
||||
|
||||
function addConstant(c: TShaderConstant) {
|
||||
constants.push(pipeState.getConstantLocation(c.name));
|
||||
}
|
||||
|
||||
function addTexture(tu: TTextureUnit) {
|
||||
var unit = pipeState.getTextureUnit(tu.name);
|
||||
textureUnits.push(unit);
|
||||
}
|
||||
|
||||
public function setTextureParameters(g: kha.graphics4.Graphics, unitIndex: Int, tex: TBindTexture) {
|
||||
// This function is called for samplers set using material context
|
||||
var unit = textureUnits[unitIndex];
|
||||
g.setTextureParameters(unit,
|
||||
tex.u_addressing == null ? TextureAddressing.Repeat : getTextureAddresing(tex.u_addressing),
|
||||
tex.v_addressing == null ? TextureAddressing.Repeat : getTextureAddresing(tex.v_addressing),
|
||||
tex.min_filter == null ? TextureFilter.LinearFilter : getTextureFilter(tex.min_filter),
|
||||
tex.mag_filter == null ? TextureFilter.LinearFilter : getTextureFilter(tex.mag_filter),
|
||||
tex.mipmap_filter == null ? MipMapFilter.NoMipFilter : getMipmapFilter(tex.mipmap_filter));
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user