Files
armorpaint/armorsculpt/Sources/MakeMaterial.ts
T
luboslenco ae67dc52ec Refactor
2024-02-05 20:57:20 +01:00

331 lines
9.9 KiB
TypeScript

class MakeMaterial {
static defaultScon: shader_context_t = null;
static defaultMcon: material_context_t = null;
static heightUsed = false;
static emisUsed = false;
static subsUsed = false;
static getMOut = (): bool => {
for (let n of UINodes.getCanvasMaterial().nodes) if (n.type == "OUTPUT_MATERIAL_PBR") return true;
return false;
}
static parseMeshMaterial = () => {
let m = Project.materials[0].data;
for (let c of m._shader._contexts) {
if (c.name == "mesh") {
array_remove(m._shader.contexts, c);
array_remove(m._shader._contexts, c);
MakeMaterial.deleteContext(c);
break;
}
}
if (MakeMesh.layerPassCount > 1) {
let i = 0;
while (i < m._shader._contexts.length) {
let c = m._shader._contexts[i];
for (let j = 1; j < MakeMesh.layerPassCount; ++j) {
if (c.name == "mesh" + j) {
array_remove(m._shader.contexts, c);
array_remove(m._shader._contexts, c);
MakeMaterial.deleteContext(c);
i--;
break;
}
}
i++;
}
i = 0;
while (i < m.contexts.length) {
let c = m.contexts[i];
for (let j = 1; j < MakeMesh.layerPassCount; ++j) {
if (c.name == "mesh" + j) {
array_remove(m.contexts, c);
array_remove(m._contexts, c);
i--;
break;
}
}
i++;
}
}
let con = MakeMesh.run({ name: "Material", canvas: null });
let scon: shader_context_t;
shader_context_create(con.data, (_scon: shader_context_t) => { scon = _scon; });
scon._override_context = {};
if (con.frag.sharedSamplers.length > 0) {
let sampler = con.frag.sharedSamplers[0];
scon._override_context.shared_sampler = sampler.substr(sampler.lastIndexOf(" ") + 1);
}
if (!Context.raw.textureFilter) {
scon._override_context.filter = "point";
}
m._shader.contexts.push(scon);
m._shader._contexts.push(scon);
for (let i = 1; i < MakeMesh.layerPassCount; ++i) {
let con = MakeMesh.run({ name: "Material", canvas: null }, i);
let scon: shader_context_t;
shader_context_create(con.data, (_scon: shader_context_t) => { scon = _scon; });
scon._override_context = {};
if (con.frag.sharedSamplers.length > 0) {
let sampler = con.frag.sharedSamplers[0];
scon._override_context.shared_sampler = sampler.substr(sampler.lastIndexOf(" ") + 1);
}
if (!Context.raw.textureFilter) {
scon._override_context.filter = "point";
}
m._shader.contexts.push(scon);
m._shader._contexts.push(scon);
let mcon: material_context_t;
MaterialContext.create({ name: "mesh" + i, bind_textures: [] }, (self: material_context_t) => { mcon = self; });
m.contexts.push(mcon);
m._contexts.push(mcon);
}
Context.raw.ddirty = 2;
///if arm_voxels
MakeMaterial.makeVoxel(m);
///end
}
static parseParticleMaterial = () => {
let m = Context.raw.particleMaterial;
let sc: shader_context_t = null;
for (let c of m._shader._contexts) {
if (c.name == "mesh") {
sc = c;
break;
}
}
if (sc != null) {
array_remove(m._shader.contexts, sc);
array_remove(m._shader._contexts, sc);
}
let con = MakeParticle.run({ name: "MaterialParticle", canvas: null });
if (sc != null) MakeMaterial.deleteContext(sc);
shader_context_create(con.data, (_sc: shader_context_t) => { sc = _sc; });
m._shader.contexts.push(sc);
m._shader._contexts.push(sc);
}
static parseMeshPreviewMaterial = () => {
if (!MakeMaterial.getMOut()) return;
let m = Project.materials[0].data;
let scon: shader_context_t = null;
for (let c of m._shader._contexts) {
if (c.name == "mesh") {
scon = c;
break;
}
}
array_remove(m._shader.contexts, scon);
array_remove(m._shader._contexts, scon);
let mcon: material_context_t = { name: "mesh", bind_textures: [] };
let sd: TMaterial = { name: "Material", canvas: null };
let con = MakeMeshPreview.run(sd, mcon);
for (let i = 0; i < m.contexts.length; ++i) {
if (m.contexts[i].name == "mesh") {
MaterialContext.create(mcon, (self: material_context_t) => { m.contexts[i] = self; });
break;
}
}
if (scon != null) MakeMaterial.deleteContext(scon);
let compileError = false;
shader_context_create(con.data, (_scon: shader_context_t) => {
if (_scon == null) compileError = true;
scon = _scon;
});
if (compileError) return;
m._shader.contexts.push(scon);
m._shader._contexts.push(scon);
}
///if arm_voxels
static makeVoxel = (m: material_data_t) => {
let rebuild = MakeMaterial.heightUsed;
if (Config.raw.rp_gi != false && rebuild) {
let scon: shader_context_t = null;
for (let c of m._shader._contexts) {
if (c.name == "voxel") {
scon = c;
break;
}
}
if (scon != null) MakeVoxel.run(scon);
}
}
///end
static parsePaintMaterial = (bakePreviews = true) => {
if (!MakeMaterial.getMOut()) return;
if (bakePreviews) {
let current = _g2_current;
if (current != null) g2_end(current);
MakeMaterial.bakeNodePreviews();
if (current != null) g2_begin(current, false);
}
let m = Project.materials[0].data;
let scon: shader_context_t = null;
let mcon: material_context_t = null;
for (let c of m._shader._contexts) {
if (c.name == "paint") {
array_remove(m._shader.contexts, c);
array_remove(m._shader._contexts, c);
if (c != MakeMaterial.defaultScon) MakeMaterial.deleteContext(c);
break;
}
}
for (let c of m.contexts) {
if (c.name == "paint") {
array_remove(m.contexts, c);
array_remove(m._contexts, c);
break;
}
}
let sdata: TMaterial = { name: "Material", canvas: UINodes.getCanvasMaterial() };
let mcon2: material_context_t = { name: "paint", bind_textures: [] };
let con = MakeSculpt.run(sdata, mcon2);
let compileError = false;
let scon2: shader_context_t;
shader_context_create(con.data, (_scon: shader_context_t) => {
if (_scon == null) compileError = true;
scon2 = _scon;
});
if (compileError) return;
scon2._override_context = {};
scon2._override_context.addressing = "repeat";
let mcon3: material_context_t;
MaterialContext.create(mcon2, (_mcon: material_context_t) => { mcon3 = _mcon; });
m._shader.contexts.push(scon2);
m._shader._contexts.push(scon2);
m.contexts.push(mcon3);
m._contexts.push(mcon3);
if (MakeMaterial.defaultScon == null) MakeMaterial.defaultScon = scon2;
if (MakeMaterial.defaultMcon == null) MakeMaterial.defaultMcon = mcon3;
}
static bakeNodePreviews = () => {
Context.raw.nodePreviewsUsed = [];
if (Context.raw.nodePreviews == null) Context.raw.nodePreviews = new Map();
MakeMaterial.traverseNodes(UINodes.getCanvasMaterial().nodes, null, []);
for (let key of Context.raw.nodePreviews.keys()) {
if (Context.raw.nodePreviewsUsed.indexOf(key) == -1) {
let image = Context.raw.nodePreviews.get(key);
Base.notifyOnNextFrame(function() { image_unload(image); });
Context.raw.nodePreviews.delete(key);
}
}
}
static traverseNodes = (nodes: TNode[], group: TNodeCanvas, parents: TNode[]) => {
for (let node of nodes) {
MakeMaterial.bakeNodePreview(node, group, parents);
if (node.type == "GROUP") {
for (let g of Project.materialGroups) {
if (g.canvas.name == node.name) {
parents.push(node);
MakeMaterial.traverseNodes(g.canvas.nodes, g.canvas, parents);
parents.pop();
break;
}
}
}
}
}
static bakeNodePreview = (node: TNode, group: TNodeCanvas, parents: TNode[]) => {
if (node.type == "BLUR") {
let id = ParserMaterial.node_name(node, parents);
let image = Context.raw.nodePreviews.get(id);
Context.raw.nodePreviewsUsed.push(id);
let resX = Math.floor(Config.getTextureResX() / 4);
let resY = Math.floor(Config.getTextureResY() / 4);
if (image == null || image.width != resX || image.height != resY) {
if (image != null) image_unload(image);
image = image_create_render_target(resX, resY);
Context.raw.nodePreviews.set(id, image);
}
ParserMaterial.blur_passthrough = true;
UtilRender.makeNodePreview(UINodes.getCanvasMaterial(), node, image, group, parents);
ParserMaterial.blur_passthrough = false;
}
else if (node.type == "DIRECT_WARP") {
let id = ParserMaterial.node_name(node, parents);
let image = Context.raw.nodePreviews.get(id);
Context.raw.nodePreviewsUsed.push(id);
let resX = Math.floor(Config.getTextureResX());
let resY = Math.floor(Config.getTextureResY());
if (image == null || image.width != resX || image.height != resY) {
if (image != null) image_unload(image);
image = image_create_render_target(resX, resY);
Context.raw.nodePreviews.set(id, image);
}
ParserMaterial.warp_passthrough = true;
UtilRender.makeNodePreview(UINodes.getCanvasMaterial(), node, image, group, parents);
ParserMaterial.warp_passthrough = false;
}
}
static parseNodePreviewMaterial = (node: TNode, group: TNodeCanvas = null, parents: TNode[] = null): { scon: shader_context_t, mcon: material_context_t } => {
if (node.outputs.length == 0) return null;
let sdata: TMaterial = { name: "Material", canvas: UINodes.getCanvasMaterial() };
let mcon_raw: material_context_t = { name: "mesh", bind_textures: [] };
let con = MakeNodePreview.run(sdata, mcon_raw, node, group, parents);
let compileError = false;
let scon: shader_context_t;
shader_context_create(con.data, (_scon: shader_context_t) => {
if (_scon == null) compileError = true;
scon = _scon;
});
if (compileError) return null;
let mcon: material_context_t;
MaterialContext.create(mcon_raw, (_mcon: material_context_t) => { mcon = _mcon; });
return { scon: scon, mcon: mcon };
}
static parseBrush = () => {
ParserLogic.parse(Context.raw.brush.canvas);
}
static getDisplaceStrength = (): f32 => {
let sc = Context.mainObject().base.transform.scale.x;
return Config.raw.displace_strength * 0.02 * sc;
}
static voxelgiHalfExtents = (): string => {
let ext = Context.raw.vxaoExt;
return `const vec3 voxelgiHalfExtents = vec3(${ext}, ${ext}, ${ext});`;
}
static deleteContext = (c: shader_context_t) => {
Base.notifyOnNextFrame(() => { // Ensure pipeline is no longer in use
shader_context_delete(c);
});
}
}