visualize.py

import pygame
import sys

# flake8: noqa

# Initialisation de Pygame
pygame.init()

# Définition de la taille de la fenêtre
screen_width = 800
screen_height = 600

# Couleurs
BROWN =(165, 80, 80)
BLACK = (0, 0, 0)
WHITE = (255, 255, 255)
RED = (255, 0, 0)
BLUE = (0, 0, 255)
EARTH_BLUE = (0, 69, 240)
EARTH_GREEN = (0, 128, 0)

# Chargement de l'image du vaisseau
vaisseau_original = pygame.image.load("vaisseau.png")
vaisseau_original_rect = vaisseau_original.get_rect()
# Redimensionnement de l'image du vaisseau
vaisseau_width = 40
vaisseau_height = 30
vaisseau_image = pygame.transform.scale(vaisseau_original, (vaisseau_width, vaisseau_height))
vaisseau_rect = vaisseau_image.get_rect()

# Coordonnées du point sur la droite
point_x = 650
point_y = 315

def draw_circle(screen):
    pygame.draw.circle(screen, BROWN, (650, point_y), 15)

def draw_earth(screen):
    pygame.draw.rect(screen, EARTH_BLUE, (0, 0, screen_width, 30))
    pygame.draw.rect(screen, EARTH_GREEN, (0, 0, screen_width, 15))

# Fonction pour dessiner la droite
def draw_line(screen):
    pygame.draw.line(screen, WHITE, (0, point_y), (screen_width, point_y))

# Fonction pour mettre à jour la position du vaisseau
def update_vaisseau(screen, x, y):
    vaisseau_rect.center = (point_x - y, point_y + x)

def dot(screen, x, y, z):
    pygame.draw.circle(screen, (255, -min(0, max(-255, z*255/1000)), max(0, min(255, z*255/1000))), (point_x - y, point_y + x), 2)

# Boucle principale
def visualize(indices):
    screen = pygame.display.set_mode((screen_width, screen_height))
    pygame.display.set_caption("Chaser trajectory visualization")

    running = True
    cruise = True   
    while running:
        while cruise:
            for i in range(len(indices)-1):
                if i == len(indices) - 2:
                    cruise = False

                # Effacer l'écran
                screen.fill(BLACK)

                # Dessiner la droite
                draw_earth(screen)
                draw_line(screen)
                draw_circle(screen)

                # Mettre à jour la position du vaisseau
                update_vaisseau(screen, indices[i][0]*6/10000, indices[i][1]*6/100)  # Mettez vos coordonnées x et y ici

                # Afficher le vaisseau
                screen.blit(vaisseau_image, vaisseau_rect)

                if i > 0:
                    for j in range(i-1):
                        dot(screen, indices[j][0]*6/10000, indices[j][1]*6/100, indices[j][2])

                # Actualiser l'affichage
                pygame.display.flip()

                # Limiter le nombre d'images par seconde
                pygame.time.Clock().tick(50)
            
        for event in pygame.event.get():
            if event.type == pygame.MOUSEBUTTONDOWN and event.button == 1:
            
                mouse_x, mouse_y = pygame.mouse.get_pos()
                if 0 <= mouse_x <= screen_width and 0 <= mouse_y <= screen_height:
                    # Si le clic est à l'intérieur de la fenêtre, fermer la fenêtre
                    running = False

    # Quitter Pygame
    pygame.quit()


'''indices = [[0, 10000 - i*100] for i in range(50)]
visualize(indices)'''