rocket_engine_plot.py

import pygame
import sys
import rocket_engine

# 1. Inicjalizacja Pygame i większe okno
pygame.init()
WIDTH, HEIGHT = 1000, 750
screen = pygame.display.set_mode((WIDTH, HEIGHT))
pygame.display.set_caption("Rocket Flight Simulator - C++ Engine")
clock = pygame.time.Clock()

# 2. Utworzenie rakiety z C++
rocket = rocket_engine.Rocket()

# Parametry: fuel_mass, rocket_mass, burn_rate, exhaust_velocity, diameter, drag_coeff
rocket.set_parameters(15.0, 10.0, 0.5, 2000.0, 0.1, 0.3)

# Kolory i czcionki
BLACK = (15, 15, 20)
WHITE = (240, 240, 240)
RED = (255, 70, 70)
ORANGE = (255, 165, 0)
GREEN = (70, 255, 100)
GRAY = (100, 100, 100)
BLUE = (50, 100, 200)
font = pygame.font.SysFont("Consolas", 18)

running = True
while running:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False

    # Krok fizyki co 0.05s dla płynności
    rocket.update_step(0.05)

    # Pobranie stanu bezpośrednio z C++
    alt = rocket.get_height()
    vel = rocket.get_velocity()
    acc = rocket.get_acceleration()
    fuel = rocket.get_fuel_mass()
    burning = rocket.is_burning()

    screen.fill(BLACK)

    #Dynamic camera followng the rocket
    scale = 0.05
    center_x = WIDTH // 2
    center_y = HEIGHT // 2 + 100  # Rakieta stoi nieco poniżej środka ekranu

    # Obliczamy gdzie znajduje się Ziemia na ekranie względem rakiety
    ground_y = center_y + (alt * scale)

    # Rysujemy ziemię
    if ground_y < HEIGHT:
        pygame.draw.rect(screen, GRAY, (0, ground_y, WIDTH, HEIGHT - ground_y + 100))

    # Rysowanie płomienia (gdy silnik pracuje)
    if burning and alt > 0:
        pygame.draw.polygon(screen, ORANGE, [
            (center_x - 6, center_y + 20),
            (center_x + 6, center_y + 20),
            (center_x, center_y + 45)
        ])
        pygame.draw.polygon(screen, RED, [
            (center_x - 3, center_y + 20),
            (center_x + 3, center_y + 20),
            (center_x, center_y + 35)
        ])

    # Korpus rakiety (zawsze w punkcie center_x, center_y)
    pygame.draw.rect(screen, WHITE, (center_x - 8, center_y - 20, 16, 40))
    pygame.draw.polygon(screen, RED, [
        (center_x - 8, center_y - 20),
        (center_x + 8, center_y - 20),
        (center_x, center_y - 35)
    ])

    # TELEMETRIA (Panel po lewej)
    telemetry = [
        f"ALTITUDE:     {alt:.1f} m",
        f"VELOCITY:     {vel:.1f} m/s",
        f"ACCELERATION: {acc:.1f} m/s^2",
        f"FUEL MASS:    {fuel:.1f} kg",
        f"ENGINE STATE: {'BURNING' if burning else 'COASTING'}"
    ]

    for i, line in enumerate(telemetry):
        color = GREEN if "BURNING" in line else (RED if "COASTING" in line else WHITE)
        text_surface = font.render(line, True, color)
        screen.blit(text_surface, (20, 20 + i * 25))

    pygame.display.flip()
    clock.tick(60)

pygame.quit()
sys.exit()