armorpaint/Sources/arm/ui/TabTextures.hx

package arm.ui;

import zui.Zui;
import zui.Nodes;
import iron.data.Data;
import iron.system.Time;
import iron.system.Input;
import arm.io.ImportAsset;
import arm.sys.Path;

class TabTextures {

	@:access(zui.Zui)
	public static function draw() {
		var ui = UISidebar.inst.ui;
		if (ui.tab(UISidebar.inst.htab2, tr("Textures"))) {
			ui.row([1 / 4, 1 / 4]);

			if (ui.button(tr("Import"))) {
				UIFiles.show(Path.textureFormats.join(","), false, function(path: String) {
					ImportAsset.run(path);
				});
			}
			if (ui.isHovered) ui.tooltip(tr("Import texture file") + ' (${Config.keymap.file_import_assets})');

			if (ui.button(tr("2D View"))) UISidebar.inst.show2DView(View2DAsset);

			if (Project.assets.length > 0) {

				var slotw = Std.int(51 * ui.SCALE());
				var num = Std.int(UISidebar.inst.windowW / slotw);

				for (row in 0...Std.int(Math.ceil(Project.assets.length / num))) {
					ui.row([for (i in 0...num) 1 / num]);

					ui._x += 2;
					if (row > 0) ui._y += 6;

					for (j in 0...num) {
						var imgw = Std.int(50 * ui.SCALE());
						var i = j + row * num;
						if (i >= Project.assets.length) {
							@:privateAccess ui.endElement(imgw);
							continue;
						}

						var asset = Project.assets[i];
						var img = UISidebar.inst.getImage(asset);
						var uix = ui._x;
						var uiy = ui._y;
						var sw = img.height < img.width ? img.height : 0;
						if (ui.image(img, 0xffffffff, slotw, 0, 0, sw, sw) == State.Started && ui.inputY > ui._windowY) {
							var mouse = Input.getMouse();
							App.dragOffX = -(mouse.x - uix - ui._windowX - 3);
							App.dragOffY = -(mouse.y - uiy - ui._windowY + 1);
							App.dragAsset = asset;
							Context.texture = asset;

							if (Time.time() - Context.selectTime < 0.25) UISidebar.inst.show2DView(View2DAsset);
							Context.selectTime = Time.time();
							UIView2D.inst.hwnd.redraws = 2;
						}

						var _uix = ui._x;
						var _uiy = ui._y;
						ui._x = uix;
						ui._y = uiy;
						if (asset == Context.texture) {
							var off = i % 2 == 1 ? 1 : 0;
							var w = 50;
							ui.fill(0,               0, w + 3,       2, ui.t.HIGHLIGHT_COL);
							ui.fill(0,     w - off + 2, w + 3, 2 + off, ui.t.HIGHLIGHT_COL);
							ui.fill(0,               0,     2,   w + 3, ui.t.HIGHLIGHT_COL);
							ui.fill(w + 2,           0,     2,   w + 4, ui.t.HIGHLIGHT_COL);
						}
						ui._x = _uix;
						ui._y = _uiy;

						// End of row align
						if (i % num == num - 1) ui._y = uiy + slotw;

						if (ui.isHovered) ui.tooltipImage(img, 256);

						if (ui.isHovered && ui.inputReleasedR) {
							UIMenu.draw(function(ui: Zui) {
								ui.text(asset.name, Right, ui.t.HIGHLIGHT_COL);
								if (ui.button(tr("Export"), Left)) {
									UIFiles.show("png", true, function(path: String) {
										var target = kha.Image.createRenderTarget(img.width, img.height);
										function exportTexture(g: kha.graphics4.Graphics) {
											if (Layers.pipeMerge == null) Layers.makePipe();
											target.g2.begin(false);
											target.g2.pipeline = Layers.pipeCopy;
											target.g2.drawImage(img, 0, 0);
											target.g2.pipeline = null;
											target.g2.end();
											var f = UIFiles.filename;
											if (f == "") f = tr("untitled");
											if (!f.endsWith(".png")) f += ".png";
											var out = new haxe.io.BytesOutput();
											var writer = new arm.format.PngWriter(out);
											var data = arm.format.PngTools.build32RGB1(target.width, target.height, target.getPixels());
											writer.write(data);
											Krom.fileSaveBytes(path + Path.sep + f, out.getBytes().getData());
											iron.App.removeRender(exportTexture);
										};
										iron.App.notifyOnRender(exportTexture);
									});
								}
								if (ui.button(tr("To Mask"), Left)) {
									Layers.createImageMask(asset);
								}
								if (ui.button(tr("Delete"), Left)) {
									UISidebar.inst.hwnd2.redraws = 2;
									Data.deleteImage(asset.file);
									Project.assetMap.remove(asset.id);
									Project.assets.splice(i, 1);
									Project.assetNames.splice(i, 1);
									function _parse(g: kha.graphics4.Graphics) {
										arm.node.MaterialParser.parsePaintMaterial();
										arm.util.RenderUtil.makeMaterialPreview();
										UISidebar.inst.hwnd1.redraws = 2;
										iron.App.removeRender(_parse);
									}
									iron.App.notifyOnRender(_parse);

									for (m in Project.materials) updateTexturePointers(m.canvas.nodes, i);
									for (b in Project.brushes) updateTexturePointers(b.canvas.nodes, i);
								}
							}, 4);
						}
					}
				}
			}
			else {
				var img = Res.get("icons.k");
				var r = Res.tile50(img, 0, 1);
				ui.image(img, ui.t.BUTTON_COL, r.h, r.x, r.y, r.w, r.h);
				if (ui.isHovered) ui.tooltip(tr("Drag and drop files here"));
			}
		}
	}

	static function updateTexturePointers(nodes: Array<TNode>, i: Int) {
		for (n in nodes) {
			if (n.type == "TEX_IMAGE") {
				if (n.buttons[0].default_value == i) {
					n.buttons[0].default_value = 9999; // Texture deleted, use pink now
				}
				else if (n.buttons[0].default_value > i) {
					n.buttons[0].default_value--; // Offset by deleted texture
				}
			}
		}
	}
}