-
Notifications
You must be signed in to change notification settings - Fork 1
/
visualization.py
58 lines (50 loc) · 1.56 KB
/
visualization.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
import pygame
from pygame.locals import *
from OpenGL.GL import *
from OpenGL.GLUT import *
from OpenGL.GLU import *
from simulator import RocketSimulator
def draw_rocket(rocket, altitude):
# Implement your rocket drawing code here using OpenGL commands
glPushMatrix()
glTranslatef(0, altitude, 0)
glColor3f(1, 0, 0)
glBegin(GL_TRIANGLES)
glVertex2f(-0.5, -0.5)
glVertex2f(0.5, -0.5)
glVertex2f(0, 0.5)
glEnd()
glPopMatrix()
pass
def draw_environment():
# Implement your environment drawing code here using OpenGL commands
glBegin(GL_QUADS)
glColor3f(0.5, 0.5, 0.5)
glVertex3f(-10, -0.5, -10)
glVertex3f(-10, -0.5, 10)
glVertex3f(10, -0.5, 10)
glVertex3f(10, -0.5, -10)
glEnd()
pass
pass
def visualize_rocket(rocket, config):
pygame.init()
display = (800, 600)
pygame.display.set_mode(display, DOUBLEBUF | OPENGL)
gluPerspective(45, (display[0] / display[1]), 0.1, 50.0)
glTranslatef(0.0, 0.0, -5)
while True:
for event in pygame.event.get():
if event.type == pygame.QUIT:
pygame.quit()
quit()
glRotatef(1, 3, 1, 1) # Rotate the view
glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT)
draw_environment()
altitude = rocket_simulate(rocket, config) # Simulate rocket and get altitude
draw_rocket(rocket, altitude)
pygame.display.flip()
pygame.time.wait(10)
def rocket_simulate(rocket, config):
simulator = RocketSimulator(rocket, config)
return simulator.simulate()