""" A vehicle to represent roaches on foot. """
-import math
-from .base import Vehicle, Seat
-from ..roaches import default_roaches, roaches_quartet, roaches_nonet, WorldRoach
+from .base import Vehicle, circle_of_seats
+from ..roaches import (
+ default_roaches, roaches_quartet, roaches_nonet, WorldRoach)
+
class Walking(Vehicle):
vehicle_type = "walking"
def init_seats(self):
- n_seats = 6
- d_theta = 2 * math.pi / n_seats
- return [
- Seat(
- vehicle=self,
- pos=(math.sin(i * d_theta), math.cos(i * d_theta)))
- for i in range(n_seats)
- ]
+ return circle_of_seats(6, vehicle=self)
def get_avatar(self, world):
num_roaches = len(world.roaches)
avatar.anchor = (0, 0)
elif num_roaches < 6:
avatar = roaches_quartet.assemble(roach)
- avatar.anchor = (-16, 0)
+ avatar.anchor = (0, 0)
else:
avatar = roaches_nonet.assemble(roach)
+ avatar.anchor = (0, 0)
return avatar
-
- def changed(self):
- return False