player.py

import pieces, constants, board, numpy as np

class Player:
    
    players = [constants.PLAYER1_VALUE, constants.PLAYER2_VALUE]
    
    def __init__(self, player_number: int, color: list, is_ai: bool, ai_name = None, ai_class = None):
        self.number = player_number
        self.remaining_pieces = pieces.get_pieces()
        self.discarded_pieces = []
        self.current_piece = {"piece": "", "arr": [], "rotated": 0, "flipped": 0, "rects": [], "place_on_board_at": []}
        self.color = color
        self.score = board.scoring_fn(self.remaining_pieces)
        self.turn_number = 1
        
        #tl = top left, bl = bottom left, tr = top right, br = bottom right
        self.board_corners = {"bl":[], "br":[], "tl":[], "tr":[]}
        self.is_1st_move = True
        #Used for AIs
        self.is_ai = is_ai
        self.ai_name = ai_name
        #To initialise things like a class containing a TF model
        self.ai_class = ai_class

    def update_score(self):
        self.score = board.scoring_fn(self.remaining_pieces)
    
    """def set_current_piece(self, piece_name):
        self.current_piece = {"piece": piece_name, "arr": pieces.get_pieces()[piece_name]}"""
    
    def empty_current_piece(self):
        self.current_piece = {"piece": "", "arr": [], "rotated": 0, "flipped": 0, "rects": [], "place_on_board_at": []}
    
    #We only rotate whatever is the current piece. We keep track of which rotated state its in
    def rotate_current_piece(self, clockwise = True):
        max_rots = pieces.get_pieces()[self.current_piece["piece"]]["rots"]
        current_state = self.current_piece["rotated"]
    
        if clockwise:
            if current_state == max_rots - 1:
                current_state = 0
            else:
                current_state += 1
            self.current_piece["rotated"] = current_state
            self.current_piece["arr"] = np.rot90(self.current_piece["arr"], k = 1)
        else:
            if current_state == 0:
                current_state = max_rots - 1
            else:
                current_state -= 1
            self.current_piece["rotated"] = current_state
            self.current_piece["arr"] = np.rot90(self.current_piece["arr"], k = -1)
        if constants.VERBOSITY > 1:
            print("New piece array for %s is %s" % (self.current_piece["piece"], self.current_piece["arr"]))

    #We only flip whatever is the current piece. We keep track of which flipped state its in
    def flip_current_piece(self):
        if not pieces.get_pieces()[self.current_piece["piece"]]["flips"] == 1:
            if self.current_piece["flipped"] == 1:
                self.current_piece["flipped"] = 0
            else:
                self.current_piece["flipped"] += 1
            self.current_piece["arr"] = np.flipud(self.current_piece["arr"])
    
    def discard_piece(self, piece):
        #Delete from dictionary of current pieces
        del self.remaining_pieces[piece["piece"]]
        #Append to dictionary containing discarded pieces
        self.discarded_pieces.append(piece)
    
    def retrieve_last_piece(self):
        piece = self.discarded_pieces[-1]
        self.remaining_pieces[piece["piece"]] = pieces.get_pieces()[piece["piece"]]
        return piece

def switch_active_player(active_player, opponent):
    if constants.VERBOSITY > 0:
        print("Player number %d is now active" % opponent.number)
    return opponent, active_player