Fix line rotate

[itzpr3d4t0r]

from geometry import Line
import pygame


def main():
    pygame.init()

    line = Line((100, 100), (700, 700))

    screen = pygame.display.set_mode((800, 800))
    clock = pygame.time.Clock()

    keep = True
    clicked = False

    rot_point = line.center

    while keep:
        dt = clock.tick_busy_loop(60)

        if clicked:
            line.rotate_ip(dt, rot_point)

        screen.fill((0, 0, 0))

        pygame.draw.line(screen, (255, 255, 255), line.a, line.b)
        pygame.draw.circle(screen, (255, 0, 0), rot_point, 5)

        pygame.display.flip()

        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                keep = False
            elif event.type == pygame.MOUSEBUTTONDOWN:
                clicked = True
            elif event.type == pygame.MOUSEBUTTONUP:
                clicked = False
                if event.button == 3:
                    rot_point = event.pos



if __name__ == "__main__":
    main()