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

import math
import pygame


class RopeInteractiveDot:
    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_ in self.scene.get_i_objects():
                if object_.__class__.__name__ == "Rope":
                    for node in object_.nodes:
                        if node.locked is False:
                            if math.sqrt(sum((node.position - self.position) ** 2)) < self.forcefield_range:
                                force_vector = node.position - self.position
                                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