armorpaint/Sources/arm/LayerSlot.hx

package arm;

import kha.graphics4.TextureFormat;
import iron.RenderPath;
import arm.ui.UITrait;

class LayerSlot {
	public static var counter = 0;
	public var id = 0;
	public var visible = true;
	public var ext = "";

	public var name:String;

	public var texpaint:kha.Image;
	public var texpaint_nor:kha.Image;
	public var texpaint_pack:kha.Image;

	public var texpaint_preview:kha.Image; // Layer preview

	public var texpaint_mask:kha.Image = null; // Texture mask
	public var texpaint_mask_preview:kha.Image;
	public var maskOpacity = 1.0; // Opacity mask
	public var material_mask:MaterialSlot = null; // Fill layer

	// For undo layer
	public var targetLayer:LayerSlot = null;
	public var targetObject:iron.object.MeshObject = null;
	public var targetIsMask = false;

	static var first = true;

	public var objectMask = 0;
	public var paintBase = true;
	public var paintOpac = true;
	public var paintOcc = true;
	public var paintRough = true;
	public var paintMet = true;
	public var paintNor = true;
	public var paintHeight = true;
	public var paintEmis = true;
	public var paintSubs = true;

	var createMaskColor:Int;
	var createMaskImage:kha.Image;

	public function new(ext = "") {

		if (first) {
			first = false;
			{
				var t = new RenderTargetRaw();
				t.name = "texpaint_blend0";
				t.width = Config.getTextureRes();
				t.height = Config.getTextureRes();
				t.format = 'R8';
				RenderPath.active.createRenderTarget(t);
			}
			{
				var t = new RenderTargetRaw();
				t.name = "texpaint_blend1";
				t.width = Config.getTextureRes();
				t.height = Config.getTextureRes();
				t.format = 'R8';
				RenderPath.active.createRenderTarget(t);
			}
		}

		if (ext == "") {
			id = counter++;
			ext = id + "";
		}
		this.ext = ext;
		name = "Layer " + (id + 1);
		var format = UITrait.inst.bitsHandle.position == 0 ? 'RGBA32' : 'RGBA64';

		{
			var t = new RenderTargetRaw();
			t.name = "texpaint" + ext;
			t.width = Config.getTextureRes();
			t.height = Config.getTextureRes();
			t.format = format;
			texpaint = RenderPath.active.createRenderTarget(t).image;
		}
		{
			var t = new RenderTargetRaw();
			t.name = "texpaint_nor" + ext;
			t.width = Config.getTextureRes();
			t.height = Config.getTextureRes();
			t.format = format;
			texpaint_nor = RenderPath.active.createRenderTarget(t).image;
		}
		{
			var t = new RenderTargetRaw();
			t.name = "texpaint_pack" + ext;
			t.width = Config.getTextureRes();
			t.height = Config.getTextureRes();
			t.format = format;
			texpaint_pack = RenderPath.active.createRenderTarget(t).image;
		}

		texpaint_preview = kha.Image.createRenderTarget(200, 200, TextureFormat.RGBA32);
	}

	public function unload() {
		texpaint.unload();
		texpaint_nor.unload();
		texpaint_pack.unload();

		RenderPath.active.renderTargets.remove("texpaint" + ext);
		RenderPath.active.renderTargets.remove("texpaint_nor" + ext);
		RenderPath.active.renderTargets.remove("texpaint_pack" + ext);

		texpaint_preview.unload();

		deleteMask();
	}

	public function swap(other:LayerSlot) {
		RenderPath.active.renderTargets.get("texpaint" + ext).image = other.texpaint;
		RenderPath.active.renderTargets.get("texpaint_nor" + ext).image = other.texpaint_nor;
		RenderPath.active.renderTargets.get("texpaint_pack" + ext).image = other.texpaint_pack;
		
		RenderPath.active.renderTargets.get("texpaint" + other.ext).image = texpaint;
		RenderPath.active.renderTargets.get("texpaint_nor" + other.ext).image = texpaint_nor;
		RenderPath.active.renderTargets.get("texpaint_pack" + other.ext).image = texpaint_pack;

		var tp = texpaint;
		var tp_nor = texpaint_nor;
		var tp_pack = texpaint_pack;
		texpaint = other.texpaint;
		texpaint_nor = other.texpaint_nor;
		texpaint_pack = other.texpaint_pack;
		other.texpaint = tp;
		other.texpaint_nor = tp_nor;
		other.texpaint_pack = tp_pack;
	}

	public function swapMask(other:LayerSlot) {
		RenderPath.active.renderTargets.get("texpaint_mask" + ext).image = other.texpaint_mask;
		RenderPath.active.renderTargets.get("texpaint_mask" + other.ext).image = texpaint_mask;
		var tp_mask = texpaint_mask;
		texpaint_mask = other.texpaint_mask;
		other.texpaint_mask = tp_mask;
	}

	public function createMask(color:Int, clear = true, image:kha.Image = null) {
		if (texpaint_mask != null) return;

		{
			var t = new RenderTargetRaw();
			t.name = "texpaint_mask" + ext;
			t.width = Config.getTextureRes();
			t.height = Config.getTextureRes();
			t.format = 'R8';
			texpaint_mask = RenderPath.active.createRenderTarget(t).image;
		}

		texpaint_mask_preview = kha.Image.createRenderTarget(200, 200, TextureFormat.L8);

		if (clear) {
			createMaskColor = color;
			createMaskImage = image;
			iron.App.notifyOnRender(clearMask);
		}
	}

	function clearMask(g:kha.graphics4.Graphics) {
		g.end();

		texpaint_mask.g2.begin();
		
		if (createMaskImage != null) texpaint_mask.g2.drawScaledImage(createMaskImage, 0, 0, texpaint_mask.width, texpaint_mask.height);
		else texpaint_mask.g2.clear(createMaskColor);
		texpaint_mask.g2.end();

		g.begin();
		iron.App.removeRender(clearMask);

		UITrait.inst.layerPreviewDirty = true;
		createMaskColor = 0;
		createMaskImage = null;
	}

	public function deleteMask() {
		if (texpaint_mask == null) return;

		texpaint_mask.unload();
		RenderPath.active.renderTargets.remove("texpaint_mask" + ext);
		texpaint_mask = null;

		texpaint_mask_preview.unload();
	}

	public function applyMask() {
		if (texpaint_mask == null) return;

		if (Layers.pipe == null) Layers.makePipe();
		Layers.makeTempImg();

		// Copy layer to temp
		Layers.imga.g2.begin(false);
		Layers.imga.g2.pipeline = Layers.pipeCopy;
		Layers.imga.g2.drawImage(texpaint, 0, 0);
		Layers.imga.g2.end();

		// Merge mask
		if (iron.data.ConstData.screenAlignedVB == null) iron.data.ConstData.createScreenAlignedData();
		texpaint.g4.begin();
		texpaint.g4.setPipeline(Layers.pipeMask);
		texpaint.g4.setTexture(Layers.tex0Mask, Layers.imga);
		texpaint.g4.setTexture(Layers.texaMask, texpaint_mask);
		texpaint.g4.setVertexBuffer(iron.data.ConstData.screenAlignedVB);
		texpaint.g4.setIndexBuffer(iron.data.ConstData.screenAlignedIB);
		texpaint.g4.drawIndexedVertices();
		texpaint.g4.end();

		deleteMask();
	}

	public function duplicate() {
		var layers = UITrait.inst.layers;
		var i = 0;
		while (i++ < layers.length) if (layers[i] == this) break;
		i++;

		var l = new LayerSlot();
		layers.insert(i, l);

		if (Layers.pipe == null) Layers.makePipe();
		l.texpaint.g2.begin(false);
		l.texpaint.g2.pipeline = Layers.pipeCopy;
		l.texpaint.g2.drawImage(texpaint, 0, 0);
		l.texpaint.g2.end();
		l.texpaint_nor.g2.begin(false);
		l.texpaint_nor.g2.pipeline = Layers.pipeCopy;
		l.texpaint_nor.g2.drawImage(texpaint_nor, 0, 0);
		l.texpaint_nor.g2.end();
		l.texpaint_pack.g2.begin(false);
		l.texpaint_pack.g2.pipeline = Layers.pipeCopy;
		l.texpaint_pack.g2.drawImage(texpaint_pack, 0, 0);
		l.texpaint_pack.g2.end();
		
		l.texpaint_preview.g2.begin(true, 0xff000000);
		l.texpaint_preview.g2.drawScaledImage(texpaint_preview, 0, 0, texpaint_preview.width, texpaint_preview.height);
		l.texpaint_preview.g2.end();

		if (texpaint_mask != null) {
			l.createMask(0, false);
			l.texpaint_mask.g2.begin(false);
			l.texpaint_mask.g2.pipeline = Layers.pipeCopy;
			l.texpaint_mask.g2.drawImage(texpaint_mask, 0, 0);
			l.texpaint_mask.g2.end();

			l.texpaint_mask_preview.g2.begin(true, 0xff000000);
			l.texpaint_mask_preview.g2.drawScaledImage(texpaint_mask_preview, 0, 0, texpaint_mask_preview.width, texpaint_mask_preview.height);
			l.texpaint_mask_preview.g2.end();
		}

		l.visible = visible;
		l.maskOpacity = maskOpacity;
		l.material_mask = material_mask;
		l.objectMask = objectMask;
		l.paintBase = paintBase;
		l.paintOpac = paintOpac;
		l.paintOcc = paintOcc;
		l.paintRough = paintRough;
		l.paintMet = paintMet;
		l.paintNor = paintNor;
		l.paintHeight = paintHeight;
		l.paintEmis = paintEmis;
		l.paintSubs = paintSubs;
		
		return l;
	}

	public function resize() {
		var res = Config.getTextureRes();
		var rts = RenderPath.active.renderTargets;

		var texpaint = this.texpaint;
		var texpaint_nor = this.texpaint_nor;
		var texpaint_pack = this.texpaint_pack;

		this.texpaint = kha.Image.createRenderTarget(res, res, TextureFormat.RGBA32);
		this.texpaint_nor = kha.Image.createRenderTarget(res, res, TextureFormat.RGBA32);
		this.texpaint_pack = kha.Image.createRenderTarget(res, res, TextureFormat.RGBA32);

		this.texpaint.g2.begin(false);
		this.texpaint.g2.drawScaledImage(texpaint, 0, 0, res, res);
		this.texpaint.g2.end();

		this.texpaint_nor.g2.begin(false);
		this.texpaint_nor.g2.drawScaledImage(texpaint_nor, 0, 0, res, res);
		this.texpaint_nor.g2.end();

		this.texpaint_pack.g2.begin(false);
		this.texpaint_pack.g2.drawScaledImage(texpaint_pack, 0, 0, res, res);
		this.texpaint_pack.g2.end();

		texpaint.unload();
		texpaint_nor.unload();
		texpaint_pack.unload();

		rts.get("texpaint" + this.ext).image = this.texpaint;
		rts.get("texpaint_nor" + this.ext).image = this.texpaint_nor;
		rts.get("texpaint_pack" + this.ext).image = this.texpaint_pack;

		if (this.texpaint_mask != null) {
			var texpaint_mask = this.texpaint_mask;
			this.texpaint_mask = kha.Image.createRenderTarget(res, res, TextureFormat.L8);

			this.texpaint_mask.g2.begin(false);
			this.texpaint_mask.g2.drawScaledImage(texpaint_mask, 0, 0, res, res);
			this.texpaint_mask.g2.end();

			texpaint_mask.unload();

			rts.get("texpaint_mask" + this.ext).image = this.texpaint_mask;
		}
	}

	public function setBits(formatName:String) {
		var format = formatName == "RGBA32" ? TextureFormat.RGBA32 : TextureFormat.RGBA64;
		var res = this.texpaint.width;
		var rts = RenderPath.active.renderTargets;

		var texpaint = this.texpaint;
		var texpaint_nor = this.texpaint_nor;
		var texpaint_pack = this.texpaint_pack;

		this.texpaint = kha.Image.createRenderTarget(res, res, format);
		this.texpaint_nor = kha.Image.createRenderTarget(res, res, format);
		this.texpaint_pack = kha.Image.createRenderTarget(res, res, format);

		this.texpaint.g2.begin(false);
		this.texpaint.g2.drawScaledImage(texpaint, 0, 0, res, res);
		this.texpaint.g2.end();

		this.texpaint_nor.g2.begin(false);
		this.texpaint_nor.g2.drawScaledImage(texpaint_nor, 0, 0, res, res);
		this.texpaint_nor.g2.end();

		this.texpaint_pack.g2.begin(false);
		this.texpaint_pack.g2.drawScaledImage(texpaint_pack, 0, 0, res, res);
		this.texpaint_pack.g2.end();

		texpaint.unload();
		texpaint_nor.unload();
		texpaint_pack.unload();

		rts.get("texpaint" + this.ext).image = this.texpaint;
		rts.get("texpaint_nor" + this.ext).image = this.texpaint_nor;
		rts.get("texpaint_pack" + this.ext).image = this.texpaint_pack;
	}
}