Update state.py

state.py

@@ -47,14 +47,15 @@ def Minimax(state, depth, maximizingPlayer, alpha, beta):
         return beta, best_move
 
 class State:
-    def __init__(self, board = [], player = 'B', last_move = ()):
+    def __init__(self, board = [], player = 'B', last_move = (), playerMode = 1):
         self.nrows = 18
         self.ncols = 18
         self.board = board
         self.last_move = last_move
         if board == []:
             self.setup()
         self.gameOver = 0
+        self.playerMode = playerMode
 
     def setup(self):
         for row in range(self.nrows):
@@ -259,7 +260,16 @@ def eval(self, board, playerColor, opponentColor):
             return -1 * math.inf
         if opponent_moves == 0:
             return math.inf
-        return len(player_moves) - (len(opponent_moves)*3)
+        if self.playerMode == 1:
+            return len(player_moves) - (len(opponent_moves)*3)
+        if self.playerMode == 2:
+            return len(player_moves)/len(opponent_moves)
+        if self.playerMode == 3:
+            return len(player_moves)/(len(oppenent_moves)*2)
+        if self.playerMode == 4:
+            return len(player_moves)/(len(opponent_moves)*3)
+        if self.playerMode == 5:
+            return len(player_moves)/(len(opponent_moves)*4)
 
     def makeMove(self, color):
         if len(self.generate_moves(self.board, color)) > 0: