Othello.py

from bangtal import *
from enum import Enum

setGameOption(GameOption.ROOM_TITLE, False)
setGameOption(GameOption.INVENTORY_BUTTON, False)
setGameOption(GameOption.MESSAGE_BOX_BUTTON, False)

Main = Scene('20153182', 'Images/background.png')
Win = Scene('win', 'Images/win.png')
Lose = Scene('lose', 'Images/lose.png')

class State(Enum):
    BLANK = 0
    POSSIBLE = 1
    BLACK = 2
    WHITE = 3

class Turn(Enum):
    BLACK = 0
    WHITE = 1

stone = []
turn = Turn.BLACK

Num1 = Object('Images/L0.png')
Num1.locate(Main, 755, 220)
Num1.show()

Num2 = Object('Images/L2.png')
Num2.locate(Main, 825, 220)
Num2.show()

Num3 = Object('Images/L0.png')
Num3.locate(Main, 1075, 220)
Num3.show()

Num4 = Object('Images/L2.png')
Num4.locate(Main, 1145, 220)
Num4.show()



def score():
    global blackStone
    global whiteStone
    blackStone = 0
    whiteStone = 0

    for y in range(8):
        for x in range(8):
            if stone[y][x].state == State.BLACK:
                blackStone += 1
            elif stone[y][x].state == State.WHITE:
                whiteStone += 1
    Num1.setImage('Images/L' +str(blackStone//10)+ '.png')
    Num2.setImage('Images/L' +str(blackStone%10)+ '.png')
    Num3.setImage('Images/L' +str(whiteStone//10)+ '.png')       
    Num4.setImage('Images/L' +str(whiteStone%10)+ '.png')
    
def setState(x, y, state):
    object = stone[y][x]
    object.state = state

    if state == State.BLANK:
        object.setImage("Images/blank.png")
    elif state == State.BLACK:
        object.setImage("Images/black.png")
    elif state == State.WHITE:
        object.setImage("Images/white.png")
    elif turn == Turn.BLACK:
        object.setImage("Images/black possible.png")
    else:
        object.setImage("Images/white possible.png")
        
def stone_onMouseAction(x, y):
    global turn

    object = stone[y][x]

    if object.state == State.POSSIBLE:
        if turn == Turn.BLACK:
            setState(x, y, State.BLACK)
            reverse_xy(x, y)
            turn = Turn.WHITE
            
        else:
            setState(x, y, State.WHITE)
            reverse_xy(x, y)
            turn = Turn.BLACK

    if not setPossible():
        if turn == Turn.BLACK:
            turn = Turn.WHITE
        else:
            turn = Turn.BLACK

        if not setPossible():
            Result()
    
    score()
    computer()


def Result():
    if blackStone > whiteStone:
        Win.enter()
    else:
        Lose.enter()
        



def setPossible_xy_dir(x, y, dx, dy):
    if turn == Turn.BLACK:
        mine = State.BLACK
        other = State.WHITE
    
    else:
        mine = State.WHITE
        other = State.BLACK

    possible = False

    while True:
        x = x + dx
        y = y + dy

        if x < 0 or x > 7: return False
        if y < 0 or y > 7: return False

        object = stone[y][x]
        if object.state == other:
            possible = True
        elif object.state == mine:
            return possible # False?
        else:
            return False

def setPossible_xy(x, y):
    object = stone[y][x]
    if object.state == State.BLACK: return False
    if object.state == State.WHITE: return False
    setState(x, y, State.BLANK)

    if (setPossible_xy_dir(x, y, 0, 1)): return True
    if (setPossible_xy_dir(x, y, 1, 1)): return True
    if (setPossible_xy_dir(x, y, 1, 0)): return True
    if (setPossible_xy_dir(x, y, 1, -1)): return True
    if (setPossible_xy_dir(x, y, 0, -1)): return True
    if (setPossible_xy_dir(x, y, -1, -1)): return True
    if (setPossible_xy_dir(x, y, -1, 0)): return True
    if (setPossible_xy_dir(x, y, -1, 1)): return True
    return False

def setPossible():
    possible = False

    for y in range(8):
        for x in range(8):
            if setPossible_xy(x, y):
                setState(x, y, State.POSSIBLE)
                possible = True

    return possible

def reverse_xy_dir(x, y, dx, dy):
    if turn == Turn.BLACK:
        mine = State.BLACK
        other = State.WHITE
    
    else:
        mine = State.WHITE
        other = State.BLACK

    possible = False

    while True:
        x = x + dx
        y = y + dy

        if x < 0 or x > 7: return
        if y < 0 or y > 7: return

        object = stone[y][x]
        if object.state == other:
            possible = True
        elif object.state == mine:
            if possible:
                while True:
                    x = x - dx
                    y = y - dy

                    object = stone[y][x]
                    if object.state == other:
                        setState(x, y, mine)
                    else:
                        return
        else:
            return


def reverse_xy(x, y):
    reverse_xy_dir(x, y, 0, 1)
    reverse_xy_dir(x, y, 1, 1)
    reverse_xy_dir(x, y, 1, 0)
    reverse_xy_dir(x, y, 1, -1)
    reverse_xy_dir(x, y, 0, -1)
    reverse_xy_dir(x, y, -1, -1)
    reverse_xy_dir(x, y, -1, 0)
    reverse_xy_dir(x, y, -1, 1)
    return False


def count_xy_dir(x, y, dx, dy):
    count = 0
    if turn == Turn.BLACK:
        mine = State.BLACK
        other = State.WHITE
    
    else:
        mine = State.WHITE
        other = State.BLACK

    while True:
        x = x + dx
        y = y + dy

        if x < 0 or x > 7: return 0
        if y < 0 or y > 7: return 0

        object = stone[y][x]
        if object.state == other:
            count = count + 1
        elif object.state == mine:
            return count
        else:
            return 0

def count_xy(x, y):
    return 1 + count_xy_dir(x, y, 0, 1) 
    + count_xy_dir(x, y, 1, 1)
    + count_xy_dir(x, y, 1, 0)
    + count_xy_dir(x, y, 1, -1)
    + count_xy_dir(x, y, 0, -1)
    + count_xy_dir(x, y, -1, -1)
    + count_xy_dir(x, y, -1, 0)
    + count_xy_dir(x, y, -1, 1)


def computer():
    max = 0
    if turn == Turn.WHITE:
        for y in range(8):
            for x in range(8):
                if stone[y][x].state == State.POSSIBLE:
                    count = count_xy(x, y)
                    if count > max:
                        max = count
                        Mx = x
                        My = y

    stone_onMouseAction(Mx, My)

def GameStart():
    startGame(Main)


for y in range(8):
    stone.append([])

    for x in range(8):
        object = Object('Images/blank.png')
        object.locate(Main, 40 + x*80, 40 + y*80)
        object.state = State.BLANK
        object.show()
        
        object.onMouseAction = lambda mx, my, action, ix = x, iy = y: stone_onMouseAction(ix, iy)

        stone[y].append(object)



setState(3, 3, State.BLACK)
setState(3, 4, State.WHITE)
setState(4, 3, State.WHITE)
setState(4, 4, State.BLACK)


if __name__ == "__main__":
    startGame(Main)