armory/Sources/armory/logicnode/ApplyImpulseAtLocationNode.hx

package armory.logicnode;

import iron.object.Object;
import iron.math.Vec4;
import armory.trait.physics.RigidBody;

using armory.object.TransformExtension;

class ApplyImpulseAtLocationNode extends LogicNode {

	public function new(tree: LogicTree) {
		super(tree);
	}

	override function run(from: Int) {
		var object: Object = inputs[1].get();
		var impulse: Vec4 = inputs[2].get();
		var localImpulse: Bool = inputs.length > 3 ? inputs[3].get() : false;
        	var location: Vec4 = inputs[4].get();
		var localLoc: Bool = inputs.length > 5 ? inputs[5].get() : false;

		if (object == null || impulse == null || location == null) return;

#if arm_physics
		var rb: RigidBody = object.getTrait(RigidBody);

		if (localLoc) {
			location.applyQuat(object.transform.rot);
		}

		!localImpulse ? rb.applyImpulse(impulse, location) : rb.applyImpulse(object.transform.worldVecToOrientation(impulse), location);
#end

		runOutput(0);
	}

}