Info_edupage/TuGUI/Objects/nr_objects/controllable_objects/RopeInteractiveDot.py

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