armorpaint/Sources/arm/UIView2D.hx

package arm;

import zui.*;

@:access(zui.Zui)
class UIView2D extends iron.Trait {

	public static var inst:UIView2D;
	public var show = false;
	public var wx:Int;
	public var wy:Int;
	public var ww:Int;
	public var uvmapCached = false;
	public var ui:Zui;
	public var hwnd = Id.handle();
	
	var panX = 0.0;
	var panY = 0.0;
	var panScale = 1.0;
	var pipe:kha.graphics4.PipelineState;
	var texType = 0;
	var uvmap:kha.Image = null;
	var uvmapShow = false;

	public function new() {
		super();
		inst = this;

		pipe = new kha.graphics4.PipelineState();
		pipe.vertexShader = kha.graphics4.VertexShader.fromSource(ConstData.painterVert);
		pipe.fragmentShader = kha.graphics4.FragmentShader.fromSource(ConstData.painterFrag);
		var vs = new kha.graphics4.VertexStructure();
		vs.add("vertexPosition", kha.graphics4.VertexData.Float3);
		vs.add("texPosition", kha.graphics4.VertexData.Float2);
		vs.add("vertexColor", kha.graphics4.VertexData.Float4);
		pipe.inputLayout = [vs];
		pipe.compile();

		var t = Reflect.copy(arm.App.theme);
		t.ELEMENT_H = 18;
		t.BUTTON_H = 16;
		var scale = armory.data.Config.raw.window_scale;
		ui = new Zui({font: arm.App.font, theme: t, color_wheel: arm.App.color_wheel, scaleFactor: scale});
		ui.scrollEnabled = false;

		notifyOnRender2D(render2D);
		notifyOnUpdate(update);
	}

	function render2D(g:kha.graphics2.Graphics) {
		ww = Std.int(iron.App.w());
		var lay = UITrait.inst.C.ui_layout;
		wx = lay == 0 ? Std.int(iron.App.w()) : UITrait.inst.windowW;
		wy = 0;

		if (!show ||
			arm.App.realw() == 0 ||
			arm.App.realh() == 0) return;

		if (UITrait.inst.pdirty >= 0) hwnd.redraws = 2; // Paint was active

		var tw = iron.App.w() * 0.95;
		var tx = iron.App.w() / 2 - tw / 2;
		var ty = iron.App.h() / 2 - tw / 2;

		tx += panX;
		ty += panY;
		tw *= panScale;

		g.end();
		
		// Cache grid
		if (UINodes.inst.grid == null) UINodes.inst.drawGrid();

		// Cache UV map
		if (uvmap == null && uvmapShow) {
			uvmap = kha.Image.createRenderTarget(2048, 2048);
		}
		if (!uvmapCached && uvmapShow) {
			uvmapCached = true;
			var mesh = UITrait.inst.paintObject.data.raw;
			var texa = mesh.vertex_arrays[2].values;
			var inda = mesh.index_arrays[0].values;
			uvmap.g2.begin(true, 0x00000000);
			uvmap.g2.color = 0xffffffff;
			for (i in 0...Std.int(inda.length / 3)) {
				var x1 = (texa[inda[i * 3 + 0] * 2 + 0]) / 32767 * uvmap.width;
				var x2 = (texa[inda[i * 3 + 1] * 2 + 0]) / 32767 * uvmap.width;
				var x3 = (texa[inda[i * 3 + 2] * 2 + 0]) / 32767 * uvmap.width;
				var y1 = (texa[inda[i * 3 + 0] * 2 + 1]) / 32767 * uvmap.width;
				var y2 = (texa[inda[i * 3 + 1] * 2 + 1]) / 32767 * uvmap.width;
				var y3 = (texa[inda[i * 3 + 2] * 2 + 1]) / 32767 * uvmap.width;
				uvmap.g2.drawLine(x1, y1, x2, y2);
				uvmap.g2.drawLine(x2, y2, x3, y3);
				uvmap.g2.drawLine(x3, y3, x1, y1);
			}
			uvmap.g2.end();
		}
		
		ui.begin(g);
		if (ui.window(hwnd, wx, wy, ww, iron.App.h())) {

			// Grid
			ui.g.color = 0xffffffff;
			ui.g.drawImage(UINodes.inst.grid, (panX * panScale) % 40 - 40, (panY * panScale) % 40 - 40);

			// Texture
			ui.g.pipeline = pipe;
			// var l = UITrait.inst.selectedLayer;
			var l = UITrait.inst.layers[0];
			var tex = texType == 0 ? l.texpaint : texType == 1 ? l.texpaint_nor : l.texpaint_pack;
	 		ui.g.drawScaledImage(tex, tx, ty, tw, tw);
			ui.g.pipeline = null;

			// UV map
			if (uvmapShow) {
				ui.g.drawScaledImage(uvmap, tx, ty, tw, tw);
			}

			// Controls
			var ew = Std.int(ui.ELEMENT_W());
			ui.g.color = ui.t.WINDOW_BG_COL;
			ui.g.fillRect(0, 0, ww, 24);
			ui.g.color = 0xffffffff;
			ui._x = 3;
			ui._y = 3;
			ui._w = ew;
			texType = ui.combo(Id.handle({position: texType}), ["Base", "Normal", "ORM"], "Texture");
			ui._x += ew + 3;
			ui._y = 3;
			uvmapShow = ui.check(Id.handle({selected: uvmapShow}), "UV Map");
			ui._x += ew + 3;
			ui._y = 3;
		}
		ui.end();
		g.begin(false);
	}

	function update() {
		var m = iron.system.Input.getMouse();

		if (!arm.App.uienabled ||
			!show ||
			m.x + App.x() < wx ||
			m.x + App.x() > wx + ww) return;
		
		if (m.down("right")) {
			panX += m.movementX;
			panY += m.movementY;
		}
		if (m.wheelDelta != 0) {
			panScale -= m.wheelDelta / 10;
			if (panScale < 0.1) panScale = 0.1;
			if (panScale > 3.0) panScale = 3.0;
		}
	}
}