Add simple texture_breakdown plugin

2020-12-07 13:03:02 +01:00
parent 3f611090db
commit ca776be01d
1 changed files with 103 additions and 0 deletions
@@ -0,0 +1,103 @@
+
+let plugin = new arm.Plugin();
+let h1 = new zui.Handle();
+let h2 = new zui.Handle();
+
+let slots = ["base", "occ", "rough", "nor"];
+let breakdown = null;
+
+plugin.drawUI = function(ui) {
+	if (ui.panel(h1, "Texture Breakdown")) {
+
+		ui.g.end();
+		drawBreakdown();
+		ui.g.begin(false);
+
+		// ui.row([1 / 4]);
+		// ui.combo(h2, ["Material", "Viewport"], "Type");
+
+		ui.image(breakdown);
+		if (ui.isHovered && ui.inputReleasedR) {
+			let x = ui.inputX - ui._windowX;
+			let w = ui._windowW / slots.length;
+			let i = (x / w) | 0;
+			arm.UIMenu.draw(function(ui) {
+				ui.text(slots[i], 2, ui.t.HIGHLIGHT_COL);
+				if (ui.button("Delete", 0)) {
+					slots.splice(i, 1);
+				}
+			}, 2);
+		}
+
+		ui.row([1 / 4, 1 / 4]);
+
+		if (ui.button("Add")) {
+			arm.UIMenu.draw(function(ui) {
+				ui.text("Channel", 2, ui.t.HIGHLIGHT_COL);
+				if (ui.button("Base Color", 0)) { slots.push("base"); }
+				if (ui.button("Occlusion", 0)) { slots.push("occ"); }
+				if (ui.button("Roughness", 0)) { slots.push("rough"); }
+				if (ui.button("Metallic", 0)) { slots.push("metal"); }
+				if (ui.button("Normal Map", 0)) { slots.push("nor"); }
+			}, 6);
+		}
+
+		if (ui.button("Export")) {
+			arm.UIFiles.show("png", true, function(path) {
+				arm.App.notifyOnNextFrame(function() {
+					var f = arm.UIFiles.filename;
+					if (f === "") f = "untitled";
+					if (!f.endsWith(".png")) f += ".png";
+					var out = new core.BytesOutput();
+					var writer = new arm.PngWriter(out);
+					var data = arm.PngTools.build32RGBA(breakdown.get_width(), breakdown.get_height(), breakdown.getPixels());
+					writer.write(data);
+					Krom.fileSaveBytes(path + arm.Path.sep + f, out.b.buffer);
+				});
+			});
+		}
+	}
+}
+
+function drawBreakdown(type) {
+	if (breakdown === null) {
+		breakdown = core.Image.createRenderTarget(4096, 4096);
+	}
+	let g2 = breakdown.get_g2();
+	g2.begin(true, 0xff000000);
+	g2.disableScissor();
+
+	if (h2.position === 0) { // Material
+		var lay = arm.Context.layer;
+		for (let i = 0; i < slots.length; ++i) {
+			g2.set_pipeline(arm.UIView2D.pipe);
+			let image = lay.texpaint;
+			let channel = 0;
+			if (slots[i] === "occ") {
+				image = lay.texpaint_pack;
+				channel = 1;
+			}
+			else if (slots[i] === "rough") {
+				image = lay.texpaint_pack;
+				channel = 2;
+			}
+			else if (slots[i] === "metal") {
+				image = lay.texpaint_pack;
+				channel = 3;
+			}
+			else if (slots[i] === "nor") {
+				image = lay.texpaint_nor;
+				channel = 5;
+			}
+			breakdown.get_g4().setInt(arm.UIView2D.channelLocation, channel);
+			var step_source = image.get_width() / slots.length;
+			var step_dest = breakdown.get_width() / slots.length;
+			g2.drawScaledSubImage(image, step_source * i, 0, step_source, image.get_height(), step_dest * i, 0, step_dest, breakdown.get_height());
+			g2.flush();
+		}
+	}
+	else { // Viewport
+	}
+
+	g2.end();
+}