luboslenco
294 lines
8.4 KiB
Haxe
294 lines
8.4 KiB
Haxe
package arm;
|
|
|
|
import kha.graphics4.DepthStencilFormat;
|
|
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;
|
|
|
|
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 t = new RenderTargetRaw();
|
|
t.name = "texpaint" + ext;
|
|
t.width = Config.getTextureRes();
|
|
t.height = Config.getTextureRes();
|
|
t.format = 'RGBA32';
|
|
t.depth_buffer = "paintdb";
|
|
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 = 'RGBA32';
|
|
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 = 'RGBA32';
|
|
texpaint_pack = RenderPath.active.createRenderTarget(t).image;
|
|
}
|
|
|
|
texpaint_preview = kha.Image.createRenderTarget(200, 200, TextureFormat.RGBA32);
|
|
}
|
|
|
|
public function unload() {
|
|
// Set null depth so paintdb stays alive
|
|
if (UITrait.inst.layers.length > 0 && UITrait.inst.layers[0] != this) {
|
|
texpaint.setDepthStencilFrom(texpaint_nor);
|
|
}
|
|
|
|
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.setDepthStencilFrom(other.texpaint);
|
|
|
|
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) {
|
|
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;
|
|
iron.App.notifyOnRender(clearMask);
|
|
}
|
|
}
|
|
|
|
function clearMask(g:kha.graphics4.Graphics) {
|
|
g.end();
|
|
|
|
texpaint_mask.g4.begin();
|
|
texpaint_mask.g4.clear(createMaskColor);
|
|
texpaint_mask.g4.end();
|
|
|
|
g.begin();
|
|
iron.App.removeRender(clearMask);
|
|
}
|
|
|
|
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();
|
|
|
|
return l;
|
|
}
|
|
|
|
public function resize(hasDepth:Bool) {
|
|
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;
|
|
|
|
var depthFormat = hasDepth ? DepthStencilFormat.Depth16 : DepthStencilFormat.NoDepthAndStencil;
|
|
this.texpaint = kha.Image.createRenderTarget(res, res, TextureFormat.RGBA32, depthFormat);
|
|
this.texpaint_nor = kha.Image.createRenderTarget(res, res, TextureFormat.RGBA32, DepthStencilFormat.NoDepthAndStencil);
|
|
this.texpaint_pack = kha.Image.createRenderTarget(res, res, TextureFormat.RGBA32, DepthStencilFormat.NoDepthAndStencil);
|
|
|
|
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;
|
|
}
|
|
}
|
|
}
|