_

TuGUI/Objects/nr_objects/controllable_objects/RopeInteractiveDot.py

@@ -0,0 +1,39 @@
+import math
+import pygame
+
+
+class RopeInteractiveDot:
+
+    object_type = "nr_c"
+
+    def __init__(self, position, r, color, force, forcefield_range, scene):
+        self.position = position
+        self.r = r
+        self.color = color
+        self.force = force
+        self.forcefield_range = forcefield_range
+        self.scene = scene
+
+    def blit(self):
+        pygame.draw.circle(self.scene.s, self.color, self.scene.matrix @ self.position, self.r * self.scene.pd_)
+
+    def input(self, mouse_pos, clicked):
+        self.position = mouse_pos
+
+        if clicked:
+            for object_batch in self.scene.get_i_objects():
+                rmp = object_batch[1]  # relative mouse position
+                for object_ in object_batch[0]:
+                    if object_.__class__.__name__ == "Rope":
+                        for node in object_.nodes:
+                            if node.locked is False:
+                                if math.sqrt(sum((node.position - rmp) ** 2)) < self.forcefield_range:
+                                    force_vector = node.position - rmp
+                                    full_force_vector = (force_vector / math.sqrt(sum(force_vector ** 2))) \
+                                                        * self.forcefield_range
+                                    force_vector = full_force_vector - force_vector
+                                    force_vector /= self.forcefield_range / self.force
+
+                                    node.position += force_vector
+
+