import random

board = [[0, 0, 0, 0],
         [0, 0, 0, 0],
         [0, 0, 0, 0],
         [0, 0, 0, 0]]


# 打印游戲界面
def display(board, score):
    print('{0:4} {1:4} {2:4} {3:4}'.format(board[0][0], board[0][1], board[0][2], board[0][3]))
    print('{0:4} {1:4} {2:4} {3:4}'.format(board[1][0], board[1][1], board[1][2], board[1][3]))
    print('{0:4} {1:4} {2:4} {3:4}'.format(board[2][0], board[2][1], board[2][2], board[2][3]))
    print('{0:4} {1:4} {2:4} {3:4}'.format(board[3][0], board[3][1], board[3][2], board[3][3]), '      分數:', score)


# 初始化游戲,在4*4里面隨機生成兩個2
def init(board):
    # 游戲先都重置為0
    for i in range(4):
        for j in range(4):
            board[i][j] = 0
    # 隨機生成兩個2保存的位置
    randomposition = random.sample(range(0, 15), 2)
    board[int(randomposition[0] / 4)][randomposition[0] % 4] = 2
    board[int(randomposition[1] / 4)][randomposition[1] % 4] = 2


def addSameNumber(boardList, direction):
    '''需要在列表中查找相鄰相同的數字相加，返回增加的分數

    :param boardList: 經過對齊非零的數字處理過后的二維數組
    :param direction: direction == 'left'從右向左查找，找到相同且相鄰的兩個數字，左側數字翻倍，右側數字置0
                      direction == 'right'從左向右查找，找到相同且相鄰的兩個數字，右側數字翻倍，左側數字置0
    :return:
    '''
    addNumber = 0
    # 向左以及向上的操作
    if direction == 'left':
        for i in [0, 1, 2]:
            if boardList[i] == boardList[i+1] != 0:
                boardList[i] *= 2
                boardList[i + 1] = 0
                addNumber += boardList[i]
                return {'continueRun': True, 'addNumber': addNumber}
        return {'continueRun': False, 'addNumber': addNumber}
    # 向右以及向下的操作
    else:
        for i in [3, 2, 1]:
            if boardList[i] == boardList[i-1] != 0:
                boardList[i] *= 2
                boardList[i - 1] = 0
                addNumber += boardList[i]
                return {'continueRun': True, 'addNumber': addNumber}
        return {'continueRun': False, 'addNumber': addNumber}


def align(boardList, direction):
    '''對齊非零的數字

    direction == 'left'：向左對齊，例如[8,0,0,2]左對齊后[8,2,0,0]
    direction == 'right'：向右對齊，例如[8,0,0,2]右對齊后[0,0,8,2]
    '''
    # 先移除列表里面的0，如[8,0,0,2]->[8,2],1.先找0的個數，然后依照個數進行清理
    # boardList.remove(0)：移除列表中的某個值的第一個匹配項，所以[8,0,0,2]會移除兩次0
    for x in range(boardList.count(0)):
        boardList.remove(0)
    # 移除的0重新補充回去,[8,2]->[8,2,0,0]
    if direction == 'left':
        boardList.extend([0 for x in range(4 - len(boardList))])
    else:
        boardList[:0] = [0 for x in range(4 - len(boardList))]


def handle(boardList, direction):
    '''
    處理一行（列）中的數據，得到最終的該行（列）的數字狀態值, 返回得分
    :param boardList: 列表結構，存儲了一行（列）中的數據
    :param direction: 移動方向,向上和向左都使用方向'left'，向右和向下都使用'right'
    :return: 返回一行（列）處理后加的分數
    '''
    addscore = 0
    # 先處理數據，把數據都往指定方向進行運動
    align(boardList, direction)
    result = addSameNumber(boardList, direction)
    # 當result['continueRun'] 為True，代表需要再次執行
    while result['continueRun']:
        # 重新對其，然后重新執行合并，直到再也無法合并為止
        addscore += result['addNumber']
        align(boardList, direction)
        result = addSameNumber(boardList, direction)
    # 直到執行完畢，及一行的數據都不存在相同的
    return {'addscore': addscore}


# 游戲操作函數，根據移動方向重新計算矩陣狀態值，并記錄得分
def operator(board):
    # 每一次的操作所加的分數，以及操作后游戲是否觸發結束狀態（即數據占滿位置）
    addScore = 0
    gameOver = False
    # 默認向左
    direction = 'left'
    op = input("請輸入您的操作:")
    if op in ['a', 'A']:
        # 方向向左
        direction = 'left'
        # 一行一行進行處理
        for row in range(4):
            addScore += handle(board[row], direction)['addscore']

    elif op in ['d', 'D']:
        direction = 'right'
        for row in range(4):
            addScore += handle(board[row], direction)['addscore']

    elif op in ['w', 'W']:
        # 向上相當于向左的轉置處理
        direction = 'left'
        board = list(map(list, zip(*board)))
        # 一行一行進行處理
        for row in range(4):
            addScore += handle(board[row], direction)['addscore']
        board = list(map(list, zip(*board)))

    elif op in ['s', 'S']:
        # 向下相當于向右的轉置處理
        direction = 'right'
        board = list(map(list, zip(*board)))
        # 一行一行進行處理
        for row in range(4):
            addScore += handle(board[row], direction)['addscore']
        board = list(map(list, zip(*board)))
    else:
        print("錯誤輸入！請輸入[W, S, A, D]或者對應小寫")
        return {'gameOver': gameOver, 'addScore': addScore, 'board': board}

    # 每一次操作后都需要判斷0的數量，如果滿了，則游戲結束
    number_0 = 0
    for q in board:
        # count(0)是指0出現的個數,是掃描每一行的
        number_0 += q.count(0)
    # 如果number_0為0，說明滿了
    if number_0 == 0:
        gameOver = True
        return {'gameOver': gameOver, 'addScore': addScore, 'board': board}
    # 說明還沒有滿，則在空的位置上加上一個2或者4，概率為3：1
    else:
        addnum = random.choice([2,2,2,4])
        position_0_list = []
        # 找出0的位置，并保存起來
        for i in range(4):
            for j in range(4):
                if board[i][j] == 0:
                    position_0_list.append(i*4 + j)
    # 在剛才記錄的0的位置里面隨便找一個，然后替換成生成的2或者4
    randomposition = random.sample(position_0_list, 1)
    board[int(randomposition[0] / 4)][randomposition[0] % 4] = addnum
    return {'gameOver': gameOver, 'addScore': addScore, 'board': board}


if __name__ == '__main__':
    print('輸入：W(上) S(下) A(左) D(右).')
    # 初始化游戲界面，游戲分數
    gameOver = False
    init(board)
    score = 0
    # 游戲未結束，則一直運行
    while gameOver != True:
        display(board, score)
        operator_result = operator(board)
        board = operator_result['board']
        if operator_result['gameOver'] == True:
            print("游戲結束，你輸了！")
            print("你的最終得分:", score)
            gameOver = operator_result['gameOver']
            break
        else:
            # 加上這一步的分
            score += operator_result['addScore']
            if score >= 2048:
                print("牛啊牛啊，你吊竟然贏了！")
                print("你的最終得分:", score)
                # 結束游戲
                gameOver = True
                break