From 798fdb828fc7d139b6f217bd9808ee587adcc025 Mon Sep 17 00:00:00 2001
From: Helen Li <helen.li@uconn.edu>
Date: Thu, 12 Dec 2019 20:02:47 -0500
Subject: [PATCH] Update state.py

---
 state.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/state.py b/state.py
index bcd0095..01e29ec 100644
--- a/state.py
+++ b/state.py
@@ -47,7 +47,7 @@ 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
@@ -55,6 +55,7 @@ def __init__(self, board = [], player = 'B', 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: