forked from LeenkxTeam/LNXSDK
add new Set Look At Rotation Node for making logic nodes great again
This commit is contained in:
190
leenkx/Sources/leenkx/logicnode/SetLookAtRotationNode.hx
Normal file
190
leenkx/Sources/leenkx/logicnode/SetLookAtRotationNode.hx
Normal file
@ -0,0 +1,190 @@
|
||||
package leenkx.logicnode;
|
||||
|
||||
import iron.math.Vec4;
|
||||
import iron.math.Quat;
|
||||
import iron.math.Mat4;
|
||||
import iron.object.Object;
|
||||
|
||||
class SetLookAtRotationNode extends LogicNode {
|
||||
|
||||
public var property0: String; // Axis to align
|
||||
public var property1: String; // Use vector for target (true/false)
|
||||
public var property2: String; // Use vector for source (true/false)
|
||||
public var property3: String; // Damping value (backward compatibility, now input socket)
|
||||
public var property4: String; // Disable rotation on aligning axis (true/false)
|
||||
|
||||
// Store the calculated rotation for output
|
||||
var calculatedRotation: Quat = null;
|
||||
// Store the previous rotation for smooth interpolation
|
||||
var previousRotation: Quat = null;
|
||||
|
||||
public function new(tree: LogicTree) {
|
||||
super(tree);
|
||||
previousRotation = new Quat();
|
||||
}
|
||||
|
||||
override function run(from: Int): Void {
|
||||
// Determine if we're using a vector or an object as source
|
||||
var useSourceVector: Bool = property2 == "true";
|
||||
var objectToUse: Object = null;
|
||||
var objectLoc: Vec4 = null;
|
||||
|
||||
if (useSourceVector) {
|
||||
// Use tree.object as the object to rotate
|
||||
objectToUse = tree.object;
|
||||
if (objectToUse == null) {
|
||||
runOutput(0);
|
||||
return;
|
||||
}
|
||||
|
||||
// Get the source location directly
|
||||
objectLoc = inputs[1].get();
|
||||
if (objectLoc == null) {
|
||||
runOutput(0);
|
||||
return;
|
||||
}
|
||||
} else {
|
||||
// Get the source object (or fallback to tree.object)
|
||||
objectToUse = (inputs.length > 1 && inputs[1] != null) ? inputs[1].get() : tree.object;
|
||||
if (objectToUse == null) {
|
||||
runOutput(0);
|
||||
return;
|
||||
}
|
||||
|
||||
// Get source object's position
|
||||
objectLoc = objectToUse.transform.loc;
|
||||
}
|
||||
|
||||
// Determine if we're using a vector or an object as target
|
||||
var useTargetVector: Bool = property1 == "true";
|
||||
var targetLoc: Vec4 = null;
|
||||
|
||||
if (useTargetVector) {
|
||||
// Get the target location directly
|
||||
targetLoc = inputs[2].get();
|
||||
if (targetLoc == null) {
|
||||
runOutput(0);
|
||||
return;
|
||||
}
|
||||
} else {
|
||||
// Get the target object
|
||||
var targetObject: Object = inputs[2].get();
|
||||
if (targetObject == null) {
|
||||
runOutput(0);
|
||||
return;
|
||||
}
|
||||
|
||||
// Get target object's position
|
||||
targetLoc = targetObject.transform.loc;
|
||||
}
|
||||
|
||||
// Calculate direction to target
|
||||
var direction = new Vec4(
|
||||
targetLoc.x - objectLoc.x,
|
||||
targetLoc.y - objectLoc.y,
|
||||
targetLoc.z - objectLoc.z
|
||||
);
|
||||
direction.normalize();
|
||||
|
||||
// Calculate target rotation based on selected axis
|
||||
calculatedRotation = new Quat();
|
||||
switch (property0) {
|
||||
case "X":
|
||||
calculatedRotation.fromTo(new Vec4(1, 0, 0), direction);
|
||||
case "-X":
|
||||
calculatedRotation.fromTo(new Vec4(-1, 0, 0), direction);
|
||||
case "Y":
|
||||
calculatedRotation.fromTo(new Vec4(0, 1, 0), direction);
|
||||
case "-Y":
|
||||
calculatedRotation.fromTo(new Vec4(0, -1, 0), direction);
|
||||
case "Z":
|
||||
calculatedRotation.fromTo(new Vec4(0, 0, 1), direction);
|
||||
case "-Z":
|
||||
calculatedRotation.fromTo(new Vec4(0, 0, -1), direction);
|
||||
}
|
||||
|
||||
// If disable rotation on aligning axis is enabled, constrain the target rotation
|
||||
if (property4 == "true") {
|
||||
// Apply constraint to the target rotation BEFORE damping to avoid jiggling
|
||||
var eulerAngles = calculatedRotation.toEulerOrdered("XYZ");
|
||||
|
||||
// Set the rotation around the selected axis to 0
|
||||
switch (property0) {
|
||||
case "X", "-X":
|
||||
eulerAngles.x = 0.0;
|
||||
case "Y", "-Y":
|
||||
eulerAngles.y = 0.0;
|
||||
case "Z", "-Z":
|
||||
eulerAngles.z = 0.0;
|
||||
}
|
||||
|
||||
// Convert back to quaternion
|
||||
calculatedRotation.fromEulerOrdered(eulerAngles, "XYZ");
|
||||
}
|
||||
|
||||
// Apply rotation with damping
|
||||
var dampingValue: Float = 0.0;
|
||||
|
||||
// Try to get damping from input socket first (index 3), fallback to property
|
||||
if (inputs.length > 3 && inputs[3] != null) {
|
||||
var dampingInput: Dynamic = inputs[3].get();
|
||||
if (dampingInput != null) {
|
||||
dampingValue = dampingInput;
|
||||
}
|
||||
} else {
|
||||
// Fallback to property for backward compatibility
|
||||
dampingValue = Std.parseFloat(property3);
|
||||
}
|
||||
|
||||
if (dampingValue > 0.0) {
|
||||
// Create a fixed interpolation rate that never reaches exactly 1.0
|
||||
// Higher damping = slower rotation (smaller step)
|
||||
var step = Math.max(0.001, (1.0 - dampingValue) * 0.2); // 0.001 to 0.2 range
|
||||
|
||||
// Get current rotation as quaternion
|
||||
var currentRot = new Quat().setFrom(objectToUse.transform.rot);
|
||||
|
||||
// Calculate the difference between current and target rotation
|
||||
var diffQuat = new Quat();
|
||||
// q1 * inverse(q2) gives the rotation from q2 to q1
|
||||
var invCurrent = new Quat().setFrom(currentRot);
|
||||
invCurrent.x = -invCurrent.x;
|
||||
invCurrent.y = -invCurrent.y;
|
||||
invCurrent.z = -invCurrent.z;
|
||||
diffQuat.multquats(calculatedRotation, invCurrent);
|
||||
|
||||
// Convert to axis-angle representation
|
||||
var axis = new Vec4();
|
||||
var angle = diffQuat.toAxisAngle(axis);
|
||||
|
||||
// Apply only a portion of this rotation (step)
|
||||
var partialAngle = angle * step;
|
||||
|
||||
// Create partial rotation quaternion
|
||||
var partialRot = new Quat().fromAxisAngle(axis, partialAngle);
|
||||
|
||||
// Apply this partial rotation to current
|
||||
var newRot = new Quat();
|
||||
newRot.multquats(partialRot, currentRot);
|
||||
|
||||
// Apply the new rotation
|
||||
objectToUse.transform.rot.setFrom(newRot);
|
||||
} else {
|
||||
// No damping, apply instant rotation
|
||||
objectToUse.transform.rot.setFrom(calculatedRotation);
|
||||
}
|
||||
|
||||
objectToUse.transform.buildMatrix();
|
||||
|
||||
runOutput(0);
|
||||
}
|
||||
|
||||
// Getter method for output sockets
|
||||
override function get(from: Int): Dynamic {
|
||||
// Output index 1 is the rotation socket (global rotation)
|
||||
if (from == 1) {
|
||||
return calculatedRotation;
|
||||
}
|
||||
return null;
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,367 @@
|
||||
from lnx.logicnode.lnx_nodes import *
|
||||
import bpy
|
||||
|
||||
class SetLookAtRotationNode(LnxLogicTreeNode):
|
||||
"""Returns a rotation that makes an object look at a target object or location"""
|
||||
bl_idname = 'LNSetLookAtRotationNode'
|
||||
bl_label = 'Set Look At Rotation'
|
||||
lnx_section = 'rotation'
|
||||
lnx_version = 1
|
||||
|
||||
use_vector: bpy.props.BoolProperty(
|
||||
name='Use Vector for Target Location',
|
||||
description='Use a vector location instead of a target object',
|
||||
default=False,
|
||||
update=lambda self, context: self.update_sockets(context)
|
||||
)
|
||||
|
||||
use_source_vector: bpy.props.BoolProperty(
|
||||
name='Use Vector for Source',
|
||||
description='Use a vector location instead of a source object',
|
||||
default=False,
|
||||
update=lambda self, context: self.update_sockets(context)
|
||||
)
|
||||
|
||||
disable_rotation_on_align_axis: bpy.props.BoolProperty(
|
||||
name='Disable Rotation on Aligning Axis',
|
||||
description='Zero out the rotation on the aligning axis after look at is applied',
|
||||
default=False,
|
||||
update=lambda self, context: self.update_sockets(context)
|
||||
)
|
||||
|
||||
damping: bpy.props.FloatProperty(
|
||||
name='Damping',
|
||||
description='Amount of damping for rotation (0.0 = instant, 1.0 = no movement)',
|
||||
default=0.0,
|
||||
min=0.0,
|
||||
max=1.0,
|
||||
step=10,
|
||||
precision=2,
|
||||
update=lambda self, context: self.update_damping_socket(context)
|
||||
)
|
||||
|
||||
# Store object references as custom properties
|
||||
source_object_name: bpy.props.StringProperty(default="")
|
||||
target_object_name: bpy.props.StringProperty(default="")
|
||||
|
||||
property0: HaxeEnumProperty(
|
||||
'property0',
|
||||
items = [('X', ' X', 'X'),
|
||||
('-X', '-X', '-X'),
|
||||
('Y', ' Y', 'Y'),
|
||||
('-Y', '-Y', '-Y'),
|
||||
('Z', ' Z', 'Z'),
|
||||
('-Z', '-Z', '-Z')],
|
||||
name='With', default='Z')
|
||||
|
||||
property1: HaxeEnumProperty(
|
||||
'property1',
|
||||
items = [('true', 'True', 'True'),
|
||||
('false', 'False', 'False')],
|
||||
name='Use Vector for Target Location', default='false')
|
||||
|
||||
property2: HaxeEnumProperty(
|
||||
'property2',
|
||||
items = [('true', 'True', 'True'),
|
||||
('false', 'False', 'False')],
|
||||
name='Use Vector for Source', default='false')
|
||||
|
||||
property3: bpy.props.StringProperty(name='Damping', default='0.0')
|
||||
|
||||
property4: HaxeEnumProperty(
|
||||
'property4',
|
||||
items = [('true', 'True', 'True'),
|
||||
('false', 'False', 'False')],
|
||||
name='Disable Rotation on Aligning Axis', default='false')
|
||||
|
||||
def lnx_init(self, context):
|
||||
# Add inputs in standard order
|
||||
self.inputs.new('LnxNodeSocketAction', 'In')
|
||||
|
||||
# Add the initial source input
|
||||
self.inputs.new('LnxNodeSocketObject', 'Source Object')
|
||||
|
||||
# Add the initial target input
|
||||
self.inputs.new('LnxNodeSocketObject', 'Target Object')
|
||||
|
||||
# Add damping input socket with default value
|
||||
damping_socket = self.inputs.new('LnxFloatSocket', 'Damping')
|
||||
damping_socket.default_value_raw = 0.0
|
||||
|
||||
# Add outputs
|
||||
self.add_output('LnxNodeSocketAction', 'Out')
|
||||
# Add rotation output socket
|
||||
self.add_output('LnxRotationSocket', 'Rotation')
|
||||
|
||||
def draw_buttons(self, context, layout):
|
||||
# 1. Axis Selector
|
||||
layout.prop(self, 'property0')
|
||||
|
||||
# 2. 'Use Vector for Source' checkbox
|
||||
layout.prop(self, 'use_source_vector')
|
||||
|
||||
# 3. 'Use Vector for Target Location' checkbox
|
||||
layout.prop(self, 'use_vector')
|
||||
|
||||
# 4. 'Disable Rotation on Aligning Axis' checkbox
|
||||
layout.prop(self, 'disable_rotation_on_align_axis')
|
||||
|
||||
# Note: Damping is now handled by the input socket
|
||||
|
||||
def update_sockets(self, context):
|
||||
# Update the Haxe properties to match the Python properties
|
||||
self.property1 = 'true' if self.use_vector else 'false'
|
||||
self.property2 = 'true' if self.use_source_vector else 'false'
|
||||
self.property3 = str(self.damping) # Keep for backward compatibility
|
||||
self.property4 = 'true' if self.disable_rotation_on_align_axis else 'false'
|
||||
|
||||
# Store current object references before changing sockets
|
||||
self.save_object_references()
|
||||
|
||||
# Helper to find a socket by name
|
||||
def find_input_socket(name):
|
||||
for i, s in enumerate(self.inputs):
|
||||
if s.name == name:
|
||||
return i, s
|
||||
return -1, None
|
||||
|
||||
# Ensure we have the 'In' socket at index 0
|
||||
in_idx, in_socket = find_input_socket('In')
|
||||
if in_idx == -1:
|
||||
# If 'In' socket is missing, add it at index 0
|
||||
self.inputs.new('LnxNodeSocketAction', 'In')
|
||||
|
||||
# Fixed indices for our sockets
|
||||
SOURCE_INDEX = 1
|
||||
TARGET_INDEX = 2
|
||||
DAMPING_INDEX = 3
|
||||
|
||||
# Get current socket information
|
||||
source_names = ['Source Object', 'Source Location']
|
||||
target_names = ['Target Object', 'Target Location']
|
||||
|
||||
# Find current source and target sockets
|
||||
source_idx = -1
|
||||
source_socket = None
|
||||
for name in source_names:
|
||||
idx, socket = find_input_socket(name)
|
||||
if idx != -1:
|
||||
source_idx = idx
|
||||
source_socket = socket
|
||||
break
|
||||
|
||||
target_idx = -1
|
||||
target_socket = None
|
||||
for name in target_names:
|
||||
idx, socket = find_input_socket(name)
|
||||
if idx != -1:
|
||||
target_idx = idx
|
||||
target_socket = socket
|
||||
break
|
||||
|
||||
# Expected types based on current settings
|
||||
expected_source_name = 'Source Location' if self.use_source_vector else 'Source Object'
|
||||
expected_source_type = 'LnxVectorSocket' if self.use_source_vector else 'LnxNodeSocketObject'
|
||||
|
||||
expected_target_name = 'Target Location' if self.use_vector else 'Target Object'
|
||||
expected_target_type = 'LnxVectorSocket' if self.use_vector else 'LnxNodeSocketObject'
|
||||
|
||||
# Ensure we have exactly 4 sockets (In, Source, Target, Damping) in the correct order
|
||||
while len(self.inputs) > 4:
|
||||
# Remove any extra sockets
|
||||
self.inputs.remove(self.inputs[-1])
|
||||
|
||||
# Make sure we have exactly 4 sockets
|
||||
while len(self.inputs) < 4:
|
||||
if len(self.inputs) == 0:
|
||||
self.inputs.new('LnxNodeSocketAction', 'In')
|
||||
elif len(self.inputs) == 1:
|
||||
self.inputs.new(expected_source_type, expected_source_name)
|
||||
elif len(self.inputs) == 2:
|
||||
self.inputs.new(expected_target_type, expected_target_name)
|
||||
elif len(self.inputs) == 3:
|
||||
damping_socket = self.inputs.new('LnxFloatSocket', 'Damping')
|
||||
damping_socket.default_value_raw = self.damping
|
||||
|
||||
# Now update the source socket if needed
|
||||
if source_socket and source_socket.name != expected_source_name:
|
||||
# Store links before removing
|
||||
links = []
|
||||
for link in source_socket.links:
|
||||
links.append(link.from_socket)
|
||||
|
||||
# Get the index where this socket should be
|
||||
correct_idx = SOURCE_INDEX
|
||||
|
||||
# Create the new socket at the correct position
|
||||
self.inputs.remove(source_socket)
|
||||
new_socket = self.inputs.new(expected_source_type, expected_source_name)
|
||||
|
||||
# Move the new socket to the correct position
|
||||
if not new_socket.is_linked: # Only move if not linked
|
||||
# Move the socket to the correct index
|
||||
if correct_idx < len(self.inputs) - 1:
|
||||
self.inputs.move(len(self.inputs) - 1, correct_idx)
|
||||
|
||||
# Restore links
|
||||
if links and hasattr(context, 'space_data') and context.space_data and hasattr(context.space_data, 'edit_tree'):
|
||||
for link_socket in links:
|
||||
context.space_data.edit_tree.links.new(link_socket, new_socket)
|
||||
|
||||
# Update the target socket if needed
|
||||
if target_socket and target_socket.name != expected_target_name:
|
||||
# Store links before removing
|
||||
links = []
|
||||
for link in target_socket.links:
|
||||
links.append(link.from_socket)
|
||||
|
||||
# Get the index where this socket should be
|
||||
correct_idx = TARGET_INDEX
|
||||
|
||||
# Create the new socket at the correct position
|
||||
self.inputs.remove(target_socket)
|
||||
new_socket = self.inputs.new(expected_target_type, expected_target_name)
|
||||
|
||||
# Move the new socket to the correct position
|
||||
if not new_socket.is_linked: # Only move if not linked
|
||||
# Move the socket to the correct index
|
||||
if correct_idx < len(self.inputs) - 1:
|
||||
self.inputs.move(len(self.inputs) - 1, correct_idx)
|
||||
|
||||
# Restore links
|
||||
if links and hasattr(context, 'space_data') and context.space_data and hasattr(context.space_data, 'edit_tree'):
|
||||
for link_socket in links:
|
||||
context.space_data.edit_tree.links.new(link_socket, new_socket)
|
||||
|
||||
# Make a final check to ensure the sockets are in the correct order
|
||||
# This is a safety measure to ensure consistent UI
|
||||
in_idx, in_socket = find_input_socket('In')
|
||||
source_idx = -1
|
||||
for name in source_names:
|
||||
idx, _ = find_input_socket(name)
|
||||
if idx != -1:
|
||||
source_idx = idx
|
||||
break
|
||||
|
||||
target_idx = -1
|
||||
for name in target_names:
|
||||
idx, _ = find_input_socket(name)
|
||||
if idx != -1:
|
||||
target_idx = idx
|
||||
break
|
||||
|
||||
damping_idx, damping_socket = find_input_socket('Damping')
|
||||
|
||||
# If the order is wrong, fix it by recreating the sockets in the correct order
|
||||
if not (in_idx == 0 and source_idx == 1 and target_idx == 2 and damping_idx == 3):
|
||||
# Store all links
|
||||
all_links = {}
|
||||
|
||||
for i, socket in enumerate(self.inputs):
|
||||
# Store links
|
||||
links = []
|
||||
for link in socket.links:
|
||||
links.append(link.from_socket)
|
||||
all_links[socket.name] = links
|
||||
|
||||
# Clear all inputs
|
||||
while len(self.inputs) > 0:
|
||||
self.inputs.remove(self.inputs[0])
|
||||
|
||||
# Recreate in the correct order
|
||||
in_socket = self.inputs.new('LnxNodeSocketAction', 'In')
|
||||
source_socket = self.inputs.new(expected_source_type, expected_source_name)
|
||||
target_socket = self.inputs.new(expected_target_type, expected_target_name)
|
||||
damping_socket = self.inputs.new('LnxFloatSocket', 'Damping')
|
||||
damping_socket.default_value_raw = self.damping
|
||||
|
||||
# Restore links
|
||||
if hasattr(context, 'space_data') and context.space_data and hasattr(context.space_data, 'edit_tree'):
|
||||
# Restore In links
|
||||
if 'In' in all_links:
|
||||
for link_socket in all_links['In']:
|
||||
context.space_data.edit_tree.links.new(link_socket, in_socket)
|
||||
|
||||
# Restore Source links
|
||||
source_links = []
|
||||
for name in source_names:
|
||||
if name in all_links:
|
||||
source_links.extend(all_links[name])
|
||||
for link_socket in source_links:
|
||||
context.space_data.edit_tree.links.new(link_socket, source_socket)
|
||||
|
||||
# Restore Target links
|
||||
target_links = []
|
||||
for name in target_names:
|
||||
if name in all_links:
|
||||
target_links.extend(all_links[name])
|
||||
for link_socket in target_links:
|
||||
context.space_data.edit_tree.links.new(link_socket, target_socket)
|
||||
|
||||
# Restore Damping links
|
||||
if 'Damping' in all_links:
|
||||
for link_socket in all_links['Damping']:
|
||||
context.space_data.edit_tree.links.new(link_socket, damping_socket)
|
||||
|
||||
# Restore object references after socket changes
|
||||
self.restore_object_references()
|
||||
|
||||
def update_damping_socket(self, context):
|
||||
"""Update the damping socket default value when the slider changes"""
|
||||
for socket in self.inputs:
|
||||
if socket.name == 'Damping' and hasattr(socket, 'default_value_raw'):
|
||||
socket.default_value_raw = self.damping
|
||||
break
|
||||
|
||||
def save_object_references(self):
|
||||
"""Save object references to custom properties"""
|
||||
# Find source and target object sockets
|
||||
for socket in self.inputs:
|
||||
if socket.name == 'Source Object' and socket.is_linked:
|
||||
for link in socket.links:
|
||||
if hasattr(link.from_node, 'item') and link.from_node.item:
|
||||
self.source_object_name = link.from_node.item.name
|
||||
|
||||
if socket.name == 'Target Object' and socket.is_linked:
|
||||
for link in socket.links:
|
||||
if hasattr(link.from_node, 'item') and link.from_node.item:
|
||||
self.target_object_name = link.from_node.item.name
|
||||
|
||||
def restore_object_references(self):
|
||||
"""Restore object references from custom properties"""
|
||||
# Only restore if we're not using vector inputs
|
||||
if not self.use_source_vector:
|
||||
# Find source object socket
|
||||
for socket in self.inputs:
|
||||
if socket.name == 'Source Object' and not socket.is_linked:
|
||||
# Try to find the object in the scene
|
||||
if self.source_object_name and self.source_object_name in bpy.context.scene.objects:
|
||||
# Find the appropriate object node in the node tree
|
||||
if hasattr(self, 'id_data') and hasattr(self.id_data, 'nodes'):
|
||||
for node in self.id_data.nodes:
|
||||
if (node.bl_idname == 'LNObjectNode' and
|
||||
hasattr(node, 'item') and
|
||||
node.item and
|
||||
node.item.name == self.source_object_name):
|
||||
# Create a link between the nodes
|
||||
if hasattr(self.id_data, 'links'):
|
||||
self.id_data.links.new(node.outputs[0], socket)
|
||||
break
|
||||
|
||||
if not self.use_vector:
|
||||
# Find target object socket
|
||||
for socket in self.inputs:
|
||||
if socket.name == 'Target Object' and not socket.is_linked:
|
||||
# Try to find the object in the scene
|
||||
if self.target_object_name and self.target_object_name in bpy.context.scene.objects:
|
||||
# Find the appropriate object node in the node tree
|
||||
if hasattr(self, 'id_data') and hasattr(self.id_data, 'nodes'):
|
||||
for node in self.id_data.nodes:
|
||||
if (node.bl_idname == 'LNObjectNode' and
|
||||
hasattr(node, 'item') and
|
||||
node.item and
|
||||
node.item.name == self.target_object_name):
|
||||
# Create a link between the nodes
|
||||
if hasattr(self.id_data, 'links'):
|
||||
self.id_data.links.new(node.outputs[0], socket)
|
||||
break
|
||||
Reference in New Issue
Block a user