armorpaint/Sources/arm/plugin/Gizmo.hx

package arm.plugin;

import iron.object.MeshObject;
import iron.object.LightObject;
import iron.system.Input;
import iron.math.RayCaster;
import iron.math.Vec4;
import iron.Scene;
import arm.ui.UITrait;

class Gizmo {

	public static function update() {
		var gizmo = UITrait.inst.gizmo;
		if (!gizmo.visible) return;

		if (Context.object != null) {
			var cam = Scene.active.camera;
			gizmo.transform.loc.setFrom(Context.object.transform.loc);
			var dist = Vec4.distance(cam.transform.loc, gizmo.transform.loc) / 10;
			gizmo.transform.scale.set(dist, dist, dist);
			UITrait.inst.gizmoX.transform.scale.set(dist, dist, dist);
			UITrait.inst.gizmoY.transform.scale.set(dist, dist, dist);
			UITrait.inst.gizmoZ.transform.scale.set(dist, dist, dist);
			gizmo.transform.buildMatrix();
		}

		var mouse = Input.getMouse();
		var kb = Input.getKeyboard();

		if (mouse.viewX < App.w()) {
			if (kb.started("delete") || kb.started("backspace")) {
				if (Context.object != null) {
					Context.object.remove();
					Context.selectObject(Scene.active.getChild("Scene"));
				}
			}
			if (kb.started("c") && Context.object != null) {
				if (Std.is(Context.object, MeshObject)) {
					var mo = cast(Context.object, MeshObject);
					var object = Scene.active.addMeshObject(mo.data, mo.materials, Scene.active.getChild("Scene"));
					object.name = mo.name + '.1';

					object.transform.loc.setFrom(mo.transform.loc);
					object.transform.rot.setFrom(mo.transform.rot);
					object.transform.scale.setFrom(mo.transform.scale);

					var hit = RayCaster.planeIntersect(Vec4.zAxis(), new Vec4(), mouse.viewX, mouse.viewY, Scene.active.camera);
					if (hit != null) {
						object.transform.loc.x = hit.x;
						object.transform.loc.y = hit.y;
						object.transform.setRotation(0, 0, Math.random() * 3.1516 * 2);
					}

					object.transform.buildMatrix();

					#if arm_physics
					object.addTrait(new armory.trait.physics.RigidBody(0, 0));
					#end

					Context.selectObject(object);
				}
				else if (Std.is(Context.object, LightObject)) {
					var lo = cast(Context.object, LightObject);
					var object = Scene.active.addLightObject(lo.data, Scene.active.getChild("Scene"));
					object.name = lo.name + '.1';

					object.transform.loc.setFrom(lo.transform.loc);
					object.transform.rot.setFrom(lo.transform.rot);
					object.transform.scale.setFrom(lo.transform.scale);
					object.transform.buildMatrix();

					#if arm_physics
					object.addTrait(new armory.trait.physics.RigidBody(0, 0));
					#end

					Context.selectObject(object);
				}
				Context.rdirty = 3;
				Context.ddirty = 3;
			}
			if (kb.started("m")) { // skip voxel
				var raw = Context.materialScene.data.raw;
				raw.skip_context = raw.skip_context == '' ? 'voxel' : '';
			}
		}

		if (mouse.started("middle")) {
			#if arm_physics
			var physics = armory.trait.physics.PhysicsWorld.active;
			var rb = physics.pickClosest(mouse.viewX, mouse.viewY);
			if (rb != null) Context.selectObject(rb.object);
			#end
		}

		if (mouse.started("left") && Context.object.name != "Scene") {
			gizmo.transform.buildMatrix();
			var trs = [UITrait.inst.gizmoX.transform, UITrait.inst.gizmoY.transform, UITrait.inst.gizmoZ.transform];
			var hit = RayCaster.closestBoxIntersect(trs, mouse.viewX, mouse.viewY, Scene.active.camera);
			if (hit != null) {
				if (hit.object == UITrait.inst.gizmoX) UITrait.inst.axisX = true;
				else if (hit.object == UITrait.inst.gizmoY) UITrait.inst.axisY = true;
				else if (hit.object == UITrait.inst.gizmoZ) UITrait.inst.axisZ = true;
				if (UITrait.inst.axisX || UITrait.inst.axisY || UITrait.inst.axisZ) UITrait.inst.axisStart = 0.0;
			}
		}
		else if (mouse.released("left")) {
			UITrait.inst.axisX = UITrait.inst.axisY = UITrait.inst.axisZ = false;
		}

		if (UITrait.inst.axisX || UITrait.inst.axisY || UITrait.inst.axisZ) {
			var t = Context.object.transform;
			var v = new Vec4();
			v.set(t.worldx(), t.worldy(), t.worldz());

			if (UITrait.inst.axisX) {
				var hit = RayCaster.planeIntersect(Vec4.yAxis(), v, mouse.viewX, mouse.viewY, Scene.active.camera);
				if (hit != null) {
					if (UITrait.inst.axisStart == 0) UITrait.inst.axisStart = hit.x - Context.object.transform.loc.x;
					Context.object.transform.loc.x = hit.x - UITrait.inst.axisStart;
					Context.object.transform.buildMatrix();
					#if arm_physics
					var rb = Context.object.getTrait(armory.trait.physics.RigidBody);
					if (rb != null) rb.syncTransform();
					#end
				}
			}
			else if (UITrait.inst.axisY) {
				var hit = RayCaster.planeIntersect(Vec4.xAxis(), v, mouse.viewX, mouse.viewY, Scene.active.camera);
				if (hit != null) {
					if (UITrait.inst.axisStart == 0) UITrait.inst.axisStart = hit.y - Context.object.transform.loc.y;
					Context.object.transform.loc.y = hit.y - UITrait.inst.axisStart;
					Context.object.transform.buildMatrix();
					#if arm_physics
					var rb = Context.object.getTrait(armory.trait.physics.RigidBody);
					if (rb != null) rb.syncTransform();
					#end
				}
			}
			else if (UITrait.inst.axisZ) {
				var hit = RayCaster.planeIntersect(Vec4.xAxis(), v, mouse.viewX, mouse.viewY, Scene.active.camera);
				if (hit != null) {
					if (UITrait.inst.axisStart == 0) UITrait.inst.axisStart = hit.z - Context.object.transform.loc.z;
					Context.object.transform.loc.z = hit.z - UITrait.inst.axisStart;
					Context.object.transform.buildMatrix();
					#if arm_physics
					var rb = Context.object.getTrait(armory.trait.physics.RigidBody);
					if (rb != null) rb.syncTransform();
					#end
				}
			}
		}

		Input.occupied = (UITrait.inst.axisX || UITrait.inst.axisY || UITrait.inst.axisZ) && mouse.viewX < App.w();
	}
}