détail

env.py

@@ -91,17 +91,31 @@ def reward(self, state, action):
         return r
 
 # test the environment and visualize the results
-'''env = Env()
+env = Env()
 print(env.reset())
 indices = [[0, 10000, 0]]
+for i in range(70):
+    if i%20 >= 10:
+        a = 1
+    else:
+        a = -1
+    state, r, ok = env.step([0.1, -0.2, 0.01, 0.1])
+    indices.append([state[0], state[1], state[2]]) 
+for i in range(30):
+    if i%20 >= 10:
+        a = 1
+    else:
+        a = -1
+    state, r, ok = env.step([0, 0.1, 0., -0.1])
+    indices.append([state[0], state[1], state[2]]) 
 for i in range(100):
     if i%20 >= 10:
         a = 1
     else:
         a = -1
-    state, r, ok = env.step([0.2, 0.2, a*0.1, 0.1])
+    state, r, ok = env.step([-0.2, 0.4, -0.1, 0.])
     indices.append([state[0], state[1], state[2]])
 
-vis.visualize(indices)'''
+vis.visualize(indices)
 
 

visualize.py

@@ -20,17 +20,17 @@
 EARTH_GREEN = (0, 128, 0)
 
 # Chargement de l'image du vaisseau
-vaisseau_original = pygame.image.load("spacecraft.png")
+vaisseau_original = pygame.image.load("vaisseau.png")
 vaisseau_original_rect = vaisseau_original.get_rect()
 # Redimensionnement de l'image du vaisseau
-vaisseau_width = 30
-vaisseau_height = 20
+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 = 300
+point_y = 315
 
 def draw_circle(screen):
     pygame.draw.circle(screen, BROWN, (650, point_y), 15)