Initial public release

tools/nodes/arrangement/connections.py

@@ -0,0 +1,20 @@
+"""Socket endpoints for the node preview."""
+
+from .flow import socket_links
+
+
+def socket_position(socket, positions, sizes):
+    node = socket.node
+    x, y = positions[node]
+    if node.bl_idname == "NodeReroute":
+        return x + 8, y - 8
+    return x + (sizes[node].width if socket.is_output else 0), y - sizes[node].sockets[socket]
+
+
+def link_endpoints(link, positions, sizes):
+    start = socket_position(link.from_socket, positions, sizes)
+    x, y = socket_position(link.to_socket, positions, sizes)
+    if link.to_socket.is_multi_input:
+        siblings = socket_links(link.to_socket)
+        y -= (siblings.index(link) - (len(siblings) - 1) / 2) * 8
+    return start, (x, y)