Litcraft_Python_B/field.py

#!/usr/bin/python
from copy import copy

from tetromino import Tetromino

class Field():

    WIDTH = 20
    HEIGHT = 20

    def __init__(self, state=None):
        if state:
            self.state = state
        else:
            self.state = [[' ' for cols in range(Field.WIDTH)]
                          for rows in range(Field.HEIGHT)]
        #self.rotate_90_degree_clckwise(self.state)
        self.cleared_rows = 1

    def __str__(self):
        BAR = '   ' + '-' * (Field.WIDTH * 2 + 1) + '\n    ' + \
            ' '.join(map(str, range(Field.WIDTH))) + '\n'
        return BAR + '\n'.join([
            '{:2d} |'.format(i) + ' '.join(row) + '|'
                for i, row in enumerate(self.state)]) + '\n' + BAR

    def matrixflip(self, m, d):
        tempm = m.copy()
        if d == 'h':
            for i in range(0, len(tempm), 1):
                tempm[i].reverse()
        elif d == 'v':
            tempm.reverse()
        return (tempm)

    def rotate_90_degree_anticlckwise(self, matrix):
        new_matrix = []
        for i in range(len(matrix[0]), 0, -1):
            new_matrix.append(list(map(lambda x: x[i - 1], matrix)))

        return new_matrix

    def rotate_90_degree_clckwise(self, matrix):
        new_matrix = []
        for i in range(len(matrix[0])):
            li = list(map(lambda x: x[i], matrix))
            li.reverse()
            new_matrix.append(li)

        return new_matrix

    def reset_field(self):
        self.state = [[' ' for cols in range(Field.WIDTH)]
                      for rows in range(Field.HEIGHT)]

    def reset_half_field(self):
        for row in range(int(self.HEIGHT/2)):
            self.state[row] = [' ' for cols in range(Field.WIDTH)]

    def rotate_state(self):
        self.state = self.rotate_90_degree_anticlckwise(self.state)
        #self.state = self.matrixflip(self.state, 'v')
        #print(self.state)

    def get_line_count(self):
        return self.cleared_rows

    def _test_tetromino(self, tetromino, row, column):
        """
        Tests to see if a tetromino can be placed at the specified row and
        column. It performs the test with the bottom left corner of the
        tetromino at the specified row and column.
        """
        assert column >= 0
        assert column + tetromino.width() <= Field.WIDTH
        assert row - tetromino.height() + 1 >= 0
        assert row < Field.HEIGHT
        for ti, si in list(enumerate(range(row - tetromino.height() + 1,
                                           row + 1)))[::-1]:
            for tj, sj in enumerate(range(column, column + tetromino.width())):
                if tetromino[ti][tj] != ' ' and self.state[si][sj] != ' ':
                    return False
        return True

    def _place_tetromino(self, tetromino, row, column):
        """
        Place a tetromino at the specified row and column.
        The bottom left corner of the tetromino will be placed at the specified
        row and column. This function does not perform checks and will overwrite
        filled spaces in the field.
        """
        assert column >= 0
        assert column + tetromino.width() <= Field.WIDTH
        assert row - tetromino.height() + 1 >= 0
        assert row < Field.HEIGHT
        for ti, si in list(enumerate(range(row - tetromino.height() + 1,
                                           row + 1)))[::-1]:
            for tj, sj in enumerate(range(column, column + tetromino.width())):
                if tetromino[ti][tj] != ' ':
                    self.state[si][sj] = tetromino[ti][tj]

    def _get_tetromino_drop_row(self, tetromino, column):
        """
        Given a tetromino and a column, return the row that the tetromino
        would end up in if it were dropped in that column.
        Assumes the leftmost column of the tetromino will be aligned with the
        specified column.
        """
        assert isinstance(tetromino, Tetromino)
        assert column >= 0
        assert column + tetromino.width() <= Field.WIDTH
        last_fit = -1
        for row in range(tetromino.height(), Field.HEIGHT):
            if self._test_tetromino(tetromino, row, column):
                last_fit = row
            else:
                return last_fit
        return last_fit

    def _line_clear(self):
        """
        Checks and removes all filled lines.
        """
        self.state = list(filter(lambda row: row.count(' ') != 0, self.state))
        while len(self.state) < Field.HEIGHT:
            self.state.insert(0, [' ' for col in range(Field.WIDTH)])
            self.cleared_rows = self.cleared_rows + 1
            print("cleared rows: ", self.cleared_rows)

    def copy(self):
        """
        Returns a shallow copy of the field.
        """
        return Field([row[:] for row in self.state])

    def drop(self, tetromino, column):
        """
        Drops a tetromino in the specified column.
        The leftmost column of the tetromino will be aligned with the specified
        column.
        Returns the row it was dropped in for computations.
        """
        assert isinstance(tetromino, Tetromino)
        assert column >= 0
        assert column + tetromino.width() <= Field.WIDTH
        row = self._get_tetromino_drop_row(tetromino, column)
        assert row != -1
        self._place_tetromino(tetromino, row, column)
        self._line_clear()
        return row



    def predict_gaps_in_next_rotation(self):
        tmp_state = copy(self.state)
        if tmp_state[10][0] is not ' ' and tmp_state[10][1] is not ' ':
            return 10
        tmp_state = self.rotate_90_degree_anticlckwise(tmp_state)
        for row in range(int(self.HEIGHT/2)):
            tmp_state[row] = [' ' for cols in range(Field.WIDTH)]
        return sum(
            ["".join([row[col] for row in tmp_state]).lstrip().count(' ')
             for col in range(Field.WIDTH)])


    def count_gaps(self):
        """
        Check each column one by one to make sure there are no gaps in the
        column.
        """
        return sum(
            ["".join([row[col] for row in self.state]).lstrip().count(' ')
             for col in range(Field.WIDTH)])

    def height(self):
        """
        Returns the height on the field of the highest placed tetromino on the
        field.
        """
        for i, row in enumerate(self.state):
            if ''.join(row).strip():
                return Field.HEIGHT - i

if __name__ == '__main__':
    import sys
    f = Field()
    if len(sys.argv) > 1 and sys.argv[1] == 'sim':
        from optimizer import Optimizer
        i = input()
        while i != 'q':
            t = Tetromino.create(i)
            opt = Optimizer.get_optimal_drop(f, t)
            t.rotate(opt['orientation'])
            f.drop(t, opt['column'])
            print(f)
            i = input()
    t = Tetromino.JTetromino().rotate_right()
    print(t)
    f.drop(t, 0)
    print(f)
    # f.drop(Tetromino.LTetromino(), 2)
    # print(f)
    # f.drop(Tetromino.JTetromino().rotate_left(), 5)
    # print(f)
    # t = Tetromino.LTetromino().flip()
    # f.drop(t, 0)
    # f.drop(Tetromino.TTetromino().flip(), 0)
    # f.drop(Tetromino.JTetromino(), 4)
    # print(f)