tgmsbot/mscore.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import numpy as np
from random import shuffle, choice
from copy import deepcopy

# 0 - 8: means 0-8 mines, not opened
# opened block = the value of not opened block + 10
# 9 is mine, not opened
# 19 is flagged mine
# 20 is stepped mine

# default:
#  HEIGHT = 8
#  WIDTH = 8
#  MINES = 9

IS_MINE = 9
DEAD = 20

def check_params(height, width, mines):
    if height <= 0 or width <= 0:
        return (False, "地图太小!")
    elif mines > height * width:
        return (False, "放不下这么多雷嘛!")
    elif mines == height * width:
        return (False, "一点就爆炸，有意思吗？")
    elif mines < 0:
        return (False, "暂时还不会放负数颗雷呢。")
    elif mines <= height * width:
        return (True, "")
def get_row_col(width, index):
    row = index // width
    col = index - width * row
    return (row, col)
def get_index(width, row_col):
    (row, col) = row_col
    index = width * row + col
    return index
class Board():
    def __init__(self, height, width, mines):
        self.height = height
        self.width = width
        self.mines = mines
        self.map = None
        self.mmap = None
        self.moves = list()
        self.state = 0 # 0:not playing, 1:playing, 2:win, 3:dead
        # statistics
        self.__op = 0
        self.__is = 0
        self.__3bv = 0
    def __gen_map(self, first_move):
        height = self.height
        width = self.width
        mines = self.mines
        if mines >= height * width:
            return
        elif mines < 0:
            return
        # first_move should't be a mine, and if possible, it should be an open.
        self.map = np.zeros((height, width), dtype=np.int8)
        map_1d = [IS_MINE] * mines
        zero_blocks = list()
        fm_index = get_index(width, first_move)
        zero_blocks.append(fm_index)
        fm_nbrs = [rc for rc in self.__iter_neighbour(*first_move)]
        if height * width - mines - 1 >= len(fm_nbrs):
            fm_nbrs_index = [get_index(width, fm_nbr) for fm_nbr in fm_nbrs]
            zero_blocks += fm_nbrs_index
            map_1d += [0] * (height * width - mines - 1 - len(fm_nbrs))
        else:
            map_1d += [0] * (height * width - mines - 1)
        shuffle(map_1d)
        for mindex in sorted(zero_blocks):
            map_1d.insert(mindex, 0)
        for mindex in range(len(map_1d)):
            if map_1d[mindex] == IS_MINE:
                (row, col) = get_row_col(width, mindex)
                self.map[row][col] = IS_MINE
        for row in range(height):
            for col in range(width):
                if self.map[row][col] != IS_MINE:
                    mine_count = 0
                    for nbr_value in self.__iter_neighbour(row, col, return_rc=False):
                        if nbr_value == IS_MINE:
                            mine_count += 1
                    self.map[row][col] = mine_count
        self.mmap = deepcopy(self.map)
    def __iter_neighbour(self, row, col, return_rc=True):
        height = self.height
        width = self.width
        for i in [a - 1 for a in range(3)]:
            for j in [b - 1 for b in range(3)]:
                if (i != 0 or j != 0) and row + i >= 0 and row + i <= height - 1 and \
                col + j >= 0 and col + j <= width - 1:
                    if return_rc:
                        yield (row + i, col + j)
                    else:
                        yield self.map[row + i][col + j]
    def __do_i_win(self):
        unopened = 0
        mines_opened = 0
        for x in np.nditer(self.map):
            if x <= 8:
                unopened += 1
            elif x in (19, DEAD):
                mines_opened += 1
        if mines_opened == self.mines:
            return True
        elif unopened == 0:
            return True
        else:
            return False
    def __open(self, row, col, automatic=False):
        if self.state != 1:
            return
        if not automatic and self.map[row][col] == 9:
            self.map[row][col] = DEAD
            self.state = 3
            return
        elif self.map[row][col] == 0:
            self.map[row][col] += 10 # open this block
            # open other blocks
            for nbr in self.__iter_neighbour(row, col):
                self.__open(nbr[0], nbr[1], automatic=True)
        elif self.map[row][col] >= 10:
            # already opened
            if automatic:
                return
            neighbour_mine_opened = 0
            neighbour_unopened = 0
            for neighbour in self.__iter_neighbour(row, col, return_rc=False):
                if neighbour in (19, DEAD):
                    neighbour_mine_opened += 1
                if neighbour <= 9:
                    neighbour_unopened += 1
            if (neighbour_mine_opened == self.map[row][col] - 10) or \
            (neighbour_unopened == self.map[row][col] - 10 - neighbour_mine_opened):
                for nbr in self.__iter_neighbour(row, col):
                    self.__open(nbr[0], nbr[1], automatic=True)
        else:
            self.map[row][col] += 10
        if self.__do_i_win():
            self.state = 2

    def move(self, row_col):
        if self.state == 0:
            self.__gen_map(row_col)
            self.state = 1
        (row, col) = row_col
        self.__open(row, col)

    def gen_statistics(self):
        if self.__3bv != 0:
            return (self.__op, self.__is, self.__3bv)
        self.__visited = np.zeros((self.height, self.width), dtype=np.int8)
        def scan_open(row, col):
            self.__visited[row][col] = 1
            for nbr_rc in self.__iter_neighbour(row, col):
                (nrow, ncol) = nbr_rc
                if self.__visited[nrow][ncol] == 0:
                    nbr = self.mmap[nrow][ncol]
                    if nbr == 0:
                        scan_open(nrow, ncol)
                    elif nbr <= 8:
                        self.__visited[nrow][ncol] = 1
        def scan_island(row, col):
            self.__3bv += 1
            self.__visited[row, col] = 1
            for nbr_rc in self.__iter_neighbour(row, col):
                (nrow, ncol) = nbr_rc
                if self.__visited[nrow][ncol] == 0:
                    nbr = self.mmap[nrow][ncol]
                    if nbr >= 1 and nbr <= 8:
                        scan_island(nrow, ncol)

        for row in range(self.height):
            for col in range(self.width):
                if self.__visited[row][col] == 0 and self.mmap[row][col] == 0:
                    self.__op += 1
                    self.__3bv += 1
                    scan_open(row, col)
        for row in range(self.height):
            for col in range(self.width):
                cell = self.mmap[row][col]
                if self.__visited[row][col] == 0 and cell >= 1 and cell <= 8:
                    self.__is += 1
                    scan_island(row, col)
        return (self.__op, self.__is, self.__3bv)