armorpaint/Sources/arm/RenderUtil.hx

package arm;

import iron.object.MeshObject;
import iron.math.Mat4;

class RenderUtil {
	public static function makeStickerPreview() {
		if (UITrait.inst.stickerImage == null) {
			UITrait.inst.stickerImage = kha.Image.createRenderTarget(512, 512);
		}
		UITrait.inst.stickerPreview = true;

		var painto = UITrait.inst.paintObject;
		for (p in UITrait.inst.paintObjects) p.visible = false;

		var plane:MeshObject = cast iron.Scene.active.getChild("Plane");
		plane.visible = true;
		UITrait.inst.paintObject = plane;
		
		UITrait.inst.savedCamera.setFrom(iron.Scene.active.camera.transform.local);
		var m = Mat4.identity();
		m.translate(0, 0, 1);
		iron.Scene.active.camera.transform.setMatrix(m);
		var savedFov = iron.Scene.active.camera.data.raw.fov;
		iron.Scene.active.camera.data.raw.fov = 0.92;
		var light = iron.Scene.active.lights[0];
		light.data.raw.cast_shadow = false;

		// No jitter
		// @:privateAccess iron.Scene.active.camera.frame = 0;
		// No resize
		@:privateAccess iron.RenderPath.active.lastW = 512;
		@:privateAccess iron.RenderPath.active.lastH = 512;
		iron.Scene.active.camera.buildProjection();
		iron.Scene.active.camera.buildMatrix();

		UINodes.inst.parseMeshPreviewMaterial();
		iron.RenderPath.active.commands = arm.renderpath.RenderPathDeferred.commandsSticker;
		iron.RenderPath.active.renderFrame(iron.RenderPath.active.frameG);
		iron.RenderPath.active.commands = arm.renderpath.RenderPathDeferred.commands;

		UITrait.inst.stickerPreview = false;
		@:privateAccess iron.RenderPath.active.lastW = iron.App.w();
		@:privateAccess iron.RenderPath.active.lastH = iron.App.h();

		// Restore
		plane.visible = false;
		for (p in UITrait.inst.paintObjects) p.visible = true;
		UITrait.inst.paintObject = painto;

		iron.Scene.active.camera.transform.setMatrix(UITrait.inst.savedCamera);
		iron.Scene.active.camera.data.raw.fov = savedFov;
		iron.Scene.active.camera.buildProjection();
		iron.Scene.active.camera.buildMatrix();
		var light = iron.Scene.active.lights[0];
		light.data.raw.cast_shadow = true;
		UINodes.inst.parseMeshMaterial();
		UITrait.inst.ddirty = 2;
	}

	public static function makeMaterialPreview() {
		UITrait.inst.materialPreview = true;

		var painto = UITrait.inst.paintObject;
		for (p in UITrait.inst.paintObjects) p.visible = false;

		var sphere:MeshObject = cast iron.Scene.active.getChild("Sphere");
		sphere.visible = true;
		UITrait.inst.paintObject = sphere;

		#if arm_editor
		sphere.materials[0] = UITrait.inst.htab.position == 0 ? UITrait.inst.selectedMaterial2.data : UITrait.inst.materials[0].data;
		var gizmo_vis = UITrait.inst.gizmo.visible;
		var grid_vis = UITrait.inst.grid.visible;
		UITrait.inst.gizmo.visible = false;
		UITrait.inst.grid.visible = false;
		#end
		
		UITrait.inst.savedCamera.setFrom(iron.Scene.active.camera.transform.local);
		var m = new Mat4(0.9146286343879498, -0.0032648027153306235, 0.404281837254303, 0.4659988049397712, 0.404295023959927, 0.007367569133732468, -0.9145989516155143, -1.0687517188018691, 0.000007410128652369705, 0.9999675337275382, 0.008058532943908717, 0.015935682577325486, 0, 0, 0, 1);
		iron.Scene.active.camera.transform.setMatrix(m);
		var savedFov = iron.Scene.active.camera.data.raw.fov;
		iron.Scene.active.camera.data.raw.fov = 0.92;
		var light = iron.Scene.active.lights[0];
		light.data.raw.cast_shadow = false;
		iron.Scene.active.world.envmap = UITrait.inst.previewEnvmap;

		// No jitter
		// @:privateAccess iron.Scene.active.camera.frame = 0;
		// No resize
		@:privateAccess iron.RenderPath.active.lastW = 100;
		@:privateAccess iron.RenderPath.active.lastH = 100;
		iron.Scene.active.camera.buildProjection();
		iron.Scene.active.camera.buildMatrix();

		UINodes.inst.parseMeshPreviewMaterial();
		iron.RenderPath.active.commands = arm.renderpath.RenderPathDeferred.commandsPreview;
		iron.RenderPath.active.renderFrame(iron.RenderPath.active.frameG);
		iron.RenderPath.active.commands = arm.renderpath.RenderPathDeferred.commands;

		UITrait.inst.materialPreview = false;
		@:privateAccess iron.RenderPath.active.lastW = iron.App.w();
		@:privateAccess iron.RenderPath.active.lastH = iron.App.h();

		// Restore
		sphere.visible = false;
		for (p in UITrait.inst.paintObjects) p.visible = true;
		UITrait.inst.paintObject = painto;

		#if arm_editor
		UITrait.inst.gizmo.visible = gizmo_vis;
		UITrait.inst.grid.visible = grid_vis;
		#end

		iron.Scene.active.camera.transform.setMatrix(UITrait.inst.savedCamera);
		iron.Scene.active.camera.data.raw.fov = savedFov;
		iron.Scene.active.camera.buildProjection();
		iron.Scene.active.camera.buildMatrix();
		var light = iron.Scene.active.lights[0];
		light.data.raw.cast_shadow = true;
		iron.Scene.active.world.envmap = UITrait.inst.showEnvmap ? UITrait.inst.savedEnvmap : UITrait.inst.emptyEnvmap;
		UINodes.inst.parseMeshMaterial();
	}
}