luboslenco
137 lines
3.7 KiB
Haxe
137 lines
3.7 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.util.MaterialUtil;
|
|
import arm.sys.Path;
|
|
import arm.sys.File;
|
|
import arm.node.NodesMaterial;
|
|
import arm.node.MaterialParser;
|
|
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 = "";
|
|
|
|
// 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);
|
|
}
|
|
|
|
// Create material
|
|
var isScene = UIHeader.inst.worktab.position == SpaceScene;
|
|
if (isScene) {
|
|
MaterialUtil.removeMaterialCache();
|
|
Data.getMaterial("Scene", "Material2", function(md: MaterialData) {
|
|
Context.materialScene = new MaterialSlot(md);
|
|
Project.materialsScene.push(Context.materialScene);
|
|
});
|
|
}
|
|
else {
|
|
Context.material = new MaterialSlot(Project.materials[0].data);
|
|
Project.materials.push(Context.material);
|
|
}
|
|
var nodes = isScene ? Context.materialScene.nodes : Context.material.nodes;
|
|
var canvas = isScene ? Context.materialScene.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++;
|
|
}
|
|
|
|
MaterialParser.parsePaintMaterial();
|
|
RenderUtil.makeMaterialPreview();
|
|
UISidebar.inst.hwnd1.redraws = 2;
|
|
}
|
|
|
|
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 = { id: nodes.getLinkId(canvas.links), from_id: n.id, from_socket: 0, to_id: to_id, to_socket: to_socket };
|
|
canvas.links.push(l);
|
|
}
|
|
}
|