luboslenco
104 lines
2.8 KiB
JavaScript
104 lines
2.8 KiB
JavaScript
|
|
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, out.getBytes().length);
|
|
});
|
|
});
|
|
}
|
|
}
|
|
}
|
|
|
|
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();
|
|
}
|