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.io.ImportTexture;
import arm.sys.Path;
import arm.sys.File;
import arm.Enums;

class TabTextures {

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

			ui.beginSticky();
			ui.row([1 / 4, 1 / 4]);

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

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

			ui.endSticky();

			if (Project.assets.length > 0) {

				var slotw = Std.int(51 * ui.SCALE());
				var num = Std.int(Config.raw.layout[LayoutSidebarW] / slotw);

				for (row in 0...Std.int(Math.ceil(Project.assets.length / num))) {
					var mult = Config.raw.show_asset_names ? 2 : 1;
					ui.row([for (i in 0...num * mult) 1 / num]);

					ui._x += 2;
					var off = Config.raw.show_asset_names ? ui.ELEMENT_OFFSET() * 10.0 : 6;
					if (row > 0) ui._y += off;

					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);
							if (Config.raw.show_asset_names) @:privateAccess ui.endElement(0);
							continue;
						}

						var asset = Project.assets[i];
						var img = Project.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;
						}


						if (asset == Context.texture) {
							var _uix = ui._x;
							var _uiy = ui._y;
							ui._x = uix;
							ui._y = uiy;
							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;
						}

						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) {
										App.notifyOnNextFrame(function () {
											if (Layers.pipeMerge == null) Layers.makePipe();
											var target = kha.Image.createRenderTarget(to_pow2(img.width), to_pow2(img.height));
											target.g2.begin(false);
											target.g2.pipeline = Layers.pipeCopy;
											target.g2.drawScaledImage(img, 0, 0, target.width, target.height);
											target.g2.pipeline = null;
											target.g2.end();
											App.notifyOnNextFrame(function () {
												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.build32RGBA(target.width, target.height, target.getPixels());
												writer.write(data);
												Krom.fileSaveBytes(path + Path.sep + f, out.getBytes().getData(), out.getBytes().length);
												target.unload();
											});
										});
									});
								}
								if (ui.button(tr("Reimport"), Left)) {
									Data.deleteImage(asset.file);
									Project.assetMap.remove(asset.id);
									Project.assets.splice(i, 1);
									Project.assetNames.splice(i, 1);
									ImportTexture.run(asset.file);
									function _next() {
										arm.node.MakeMaterial.parsePaintMaterial();
										arm.util.RenderUtil.makeMaterialPreview();
										UISidebar.inst.hwnd1.redraws = 2;
									}
									App.notifyOnNextFrame(_next);
								}
								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 _next() {
										arm.node.MakeMaterial.parsePaintMaterial();
										arm.util.RenderUtil.makeMaterialPreview();
										UISidebar.inst.hwnd1.redraws = 2;
									}
									App.notifyOnNextFrame(_next);

									for (m in Project.materials) updateTexturePointers(m.canvas.nodes, i);
									for (b in Project.brushes) updateTexturePointers(b.canvas.nodes, i);
								}
								if (ui.button(tr("Open Containing Directory..."), Left)) {
									File.explorer(asset.file.substr(0, asset.file.lastIndexOf(Path.sep)));
								}
							}, 6);
						}

						if (Config.raw.show_asset_names) {
							ui._x = uix;
							ui._y += slotw * 0.9;
							ui.text(Project.assets[i].name, Center);
							if (ui.isHovered) ui.tooltip(Project.assets[i].name);
							ui._y -= slotw * 0.9;
							if (i == Project.assets.length - 1) {
								ui._y += j == num - 1 ? imgw : imgw + ui.ELEMENT_H() + ui.ELEMENT_OFFSET();
							}
						}
					}
				}
			}
			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 to_pow2(i: Int): Int {
		i--;
		i |= i >> 1;
		i |= i >> 2;
		i |= i >> 4;
		i |= i >> 8;
		i |= i >> 16;
		i++;
		return i;
	}

	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
				}
			}
		}
	}
}