MathemanFlo
146 lines
3.5 KiB
Haxe
146 lines
3.5 KiB
Haxe
package arm.io;
|
|
|
|
import zui.Nodes;
|
|
import iron.data.Data;
|
|
import iron.data.MaterialData;
|
|
import arm.ui.UIHeader;
|
|
import arm.ui.UISidebar;
|
|
import arm.util.RenderUtil;
|
|
import arm.sys.Path;
|
|
import arm.sys.File;
|
|
import arm.shader.NodesMaterial;
|
|
import arm.node.MakeMaterial;
|
|
import arm.data.MaterialSlot;
|
|
import arm.Enums;
|
|
|
|
class ImportFolder {
|
|
|
|
public static function run(path: String) {
|
|
var files = File.readDirectory(path);
|
|
var mapbase = "";
|
|
var mapopac = "";
|
|
var mapnor = "";
|
|
var mapocc = "";
|
|
var maprough = "";
|
|
var mapmet = "";
|
|
var mapheight = "";
|
|
|
|
var foundTexture = false;
|
|
// Import maps
|
|
for (f in files) {
|
|
if (!Path.isTexture(f)) continue;
|
|
|
|
// TODO: handle -albedo
|
|
|
|
var base = f.substr(0, f.lastIndexOf(".")).toLowerCase();
|
|
var valid = false;
|
|
if (mapbase == "" && Path.isBaseColorTex(base)) {
|
|
mapbase = f;
|
|
valid = true;
|
|
}
|
|
if (mapopac == "" && Path.isOpacityTex(base)) {
|
|
mapopac = f;
|
|
valid = true;
|
|
}
|
|
if (mapnor == "" && Path.isNormalMapTex(base)) {
|
|
mapnor = f;
|
|
valid = true;
|
|
}
|
|
if (mapocc == "" && Path.isOcclusionTex(base)) {
|
|
mapocc = f;
|
|
valid = true;
|
|
}
|
|
if (maprough == "" && Path.isRoughnessTex(base)) {
|
|
maprough = f;
|
|
valid = true;
|
|
}
|
|
if (mapmet == "" && Path.isMetallicTex(base)) {
|
|
mapmet = f;
|
|
valid = true;
|
|
}
|
|
if (mapheight == "" && Path.isDisplacementTex(base)) {
|
|
mapheight = f;
|
|
valid = true;
|
|
}
|
|
|
|
if (valid) {
|
|
ImportTexture.run(path + Path.sep + f, false);
|
|
foundTexture = true;
|
|
}
|
|
}
|
|
|
|
if (!foundTexture) {
|
|
Console.info(tr("Folder does not contain textures"));
|
|
return;
|
|
}
|
|
|
|
// Create material
|
|
Context.material = new MaterialSlot(Project.materials[0].data);
|
|
Project.materials.push(Context.material);
|
|
var nodes = Context.material.nodes;
|
|
var canvas = Context.material.canvas;
|
|
var dirs = path.split(Path.sep);
|
|
canvas.name = dirs[dirs.length - 1];
|
|
var nout: TNode = null;
|
|
for (n in canvas.nodes) {
|
|
if (n.type == "OUTPUT_MATERIAL_PBR") {
|
|
nout = n;
|
|
break;
|
|
}
|
|
}
|
|
for (n in canvas.nodes) {
|
|
if (n.name == "RGB") {
|
|
nodes.removeNode(n, canvas);
|
|
break;
|
|
}
|
|
}
|
|
|
|
// Place nodes
|
|
var pos = 0;
|
|
var startY = 100;
|
|
var nodeH = 164;
|
|
if (mapbase != "") {
|
|
placeImageNode(nodes, canvas, mapbase, startY + nodeH * pos, nout.id, 0);
|
|
pos++;
|
|
}
|
|
if (mapopac != "") {
|
|
placeImageNode(nodes, canvas, mapopac, startY + nodeH * pos, nout.id, 1);
|
|
pos++;
|
|
}
|
|
if (mapocc != "") {
|
|
placeImageNode(nodes, canvas, mapocc, startY + nodeH * pos, nout.id, 2);
|
|
pos++;
|
|
}
|
|
if (maprough != "") {
|
|
placeImageNode(nodes, canvas, maprough, startY + nodeH * pos, nout.id, 3);
|
|
pos++;
|
|
}
|
|
if (mapmet != "") {
|
|
placeImageNode(nodes, canvas, mapmet, startY + nodeH * pos, nout.id, 4);
|
|
pos++;
|
|
}
|
|
if (mapnor != "") {
|
|
placeImageNode(nodes, canvas, mapnor, startY + nodeH * pos, nout.id, 5);
|
|
pos++;
|
|
}
|
|
if (mapheight != "") {
|
|
placeImageNode(nodes, canvas, mapheight, startY + nodeH * pos, nout.id, 7);
|
|
pos++;
|
|
}
|
|
|
|
MakeMaterial.parsePaintMaterial();
|
|
RenderUtil.makeMaterialPreview();
|
|
UISidebar.inst.hwnd1.redraws = 2;
|
|
History.newMaterial();
|
|
}
|
|
|
|
static function placeImageNode(nodes: Nodes, canvas: TNodeCanvas, asset: String, ny: Int, to_id: Int, to_socket: Int) {
|
|
var n = NodesMaterial.createNode("TEX_IMAGE");
|
|
n.buttons[0].default_value = App.getAssetIndex(asset);
|
|
n.x = 72;
|
|
n.y = ny;
|
|
var l: TNodeLink = { id: nodes.getLinkId(canvas.links), from_id: n.id, from_socket: 0, to_id: to_id, to_socket: to_socket };
|
|
canvas.links.push(l);
|
|
}
|
|
}
|