armorpaint/Sources/arm/ui/TabTextures.hx

package arm.ui;

import zui.Zui;
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(1);

			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() - UISidebar.inst.selectTime < 0.25) UISidebar.inst.show2DView(1);
							UISidebar.inst.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);
									iron.system.Tween.timer(0.1, function() {
										arm.node.MaterialParser.parsePaintMaterial();
										arm.util.RenderUtil.makeMaterialPreview();
										UISidebar.inst.hwnd1.redraws = 2;
									});
								}
							}, 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"));
			}
		}
	}
}