add 4 robot random walk, but return objects todo

Author: superPershing

wedge_animation.py

@@ -30,7 +30,7 @@ def talk(self):
         pass
 
     def update_direction(self): #TODO stuck in corner and border.
-        if (self.x + self.dx) < 10 and (self.x + self.dx) > -10 and (self.y + self.dy) < 10 and (self.y + self.dx) > -10:
+        if (self.x + self.dx) < 10 and (self.x + self.dx) > -10 and (self.y + self.dy) < 10 and (self.y + self.dy) > -10:
             self.x += self.dx    
             self.y += self.dy
         else:
@@ -49,34 +49,53 @@ def update_direction(self): #TODO stuck in corner and border.
         elif self.angle >= 270 and self.angle < 360:
             self.dx = 1 / math.sqrt(1 + self.k ** 2)
             self.dy = - math.sqrt(1 - self.dx ** 2)
-robot0 =  Robot(random.randint(-10, 10), random.randint(-10, 10), 0, 1)
+
+# robot0 =  Robot(random.randint(-10, 10), random.randint(-10, 10), 0, 1)
+robots = [Robot(random.randint(-10, 10), random.randint(-10, 10), 0, 1) for i in range(4)]
+
 
 # First set up the figure, the axis, and the plot element we want to animate
 fig = plt.figure()
 ax = plt.axes(xlim=(-12, 12), ylim=(-12, 12))
 plt.grid(True)
 # patch_one = matplotlib.patches.Circle((0, 0), 1)
-patch_one = matplotlib.patches.Wedge((robot0.x,robot0.y), 1, robot0.angle - 25, robot0.angle + 25)
+# patch_one = matplotlib.patches.Wedge((robot0.x,robot0.y), 1, robot0.angle - 25, robot0.angle + 25)
+patchs = [matplotlib.patches.Wedge((robot_i.x,robot_i.y), 1, robot_i.angle - 25, robot_i.angle + 25) for robot_i in robots]
 
 # initialization function: plot the background of each frame
 def init():
-    patch_one.radius = 1
-    ax.add_patch(patch_one)
-    return patch_one,
+    # patch_one.radius = 1
+    # ax.add_patch(patch_one)
+    # return patch_one,
+    for i in range(4):
+        patchs[i].radius = 1
+        ax.add_patch(patchs[i])
+    return patchs[0],patchs[1],patchs[2],patchs[3],
+
 
 # animation function.  This is called sequentially
 def animate(frame):
-    robot0.update_direction()
-    # robot0.random_walk()
-    patch_one.set_center((robot0.x, robot0.y))
-    patch_one.set_theta1(robot0.angle - 25)
-    patch_one.set_theta2(robot0.angle + 25)
-    patch_one._recompute_path()
-    print(robot0.angle)
-    return patch_one,
+    # robot0.update_direction()
+    # # robot0.random_walk()
+    # patch_one.set_center((robot0.x, robot0.y))
+    # patch_one.set_theta1(robot0.angle - 25)
+    # patch_one.set_theta2(robot0.angle + 25)
+    # patch_one._recompute_path()
+    # print(robot0.angle)
+    # return patch_one,
+    for i in range(4):
+        robots[i].update_direction()
+        # robot0.random_walk()
+        patchs[i].set_center((robots[i].x, robots[i].y))
+        patchs[i].set_theta1(robots[i].angle - 25)
+        patchs[i].set_theta2(robots[i].angle + 25)
+        patchs[i]._recompute_path()
+        # print(robots[i].angle)
+    return patchs[0],patchs[1],patchs[2],patchs[3],
+
 
 # call the animator.  blit=True means only re-draw the parts that have changed.
 anim = animation.FuncAnimation(fig, animate, init_func=init,
-                               frames=200, interval=1000, blit=True)
+                               frames=200, interval=10, blit=True)
 
 plt.show()