Add support for multiple AI-controlled players

maxman.py

@@ -1,10 +1,10 @@
 #!/bin/env python3
 
 import abc
+import argparse
 import dataclasses
 import math
 import random
-import sys
 from typing import List
 
 import pygame
@@ -157,10 +157,9 @@ dt = 0
 
 # set up board
 sw, sh = screen.get_width(), screen.get_height()
-ai = AI()
 maxman = Man(pygame.Vector2(sw / 4, sh / 4), name="Maxman")
 blackman = Man(
-    pygame.Vector2(sw * 0.75, sh * 0.75),
+    pygame.Vector2(sw * 0.75, sh * 0.25),
     name="Blackman",
     color="black",
     key_up=pygame.K_i,
@@ -168,11 +167,41 @@ blackman = Man(
     key_left=pygame.K_j,
     key_right=pygame.K_l,
 )
-players = [maxman, blackman]
+chrisman = Man(
+    pygame.Vector2(sw * 0.25, sh * 0.75),
+    name="Chrisman",
+    color="red",
+    key_up=pygame.K_UP,
+    key_down=pygame.K_DOWN,
+    key_left=pygame.K_LEFT,
+    key_right=pygame.K_RIGHT,
+)
+manman = Man(
+    pygame.Vector2(sw * 0.75, sh * 0.75),
+    name="Manman",
+    color="green",
+    key_up=pygame.K_KP_8,
+    key_down=pygame.K_KP_2,
+    key_left=pygame.K_KP_4,
+    key_right=pygame.K_KP_6,
+)
+players = [maxman, blackman, chrisman, manman]
 food = Food.new()
 
-if any([arg in ["--ai", "-a"] for arg in sys.argv]):
-    ai.hook(blackman)
+parser = argparse.ArgumentParser()
+parser.add_argument(
+    "--ai",
+    type=int,
+    choices=[0, 1, 2, 3, 4],
+    default=1,
+    help="number of AI players",
+)
+args = parser.parse_args()
+ais = []
+for i in range(args.ai):
+    ai = AI()
+    ai.hook(players[-i - 1])
+    ais.append(ai)
 
 while running:
     # check recent events
@@ -222,7 +251,9 @@ while running:
             )
 
         else:
-            ai.decide()
+            for ai in ais:
+                ai.decide()
+
             for p in players:
                 p.handle_input(keys, dt)