added first draft litris

2023-07-22 09:44:48 +02:00
parent c2b5416542
commit 3176bf5269
4 changed files with 507 additions and 19 deletions
--- a/tetromino.py
+++ b/tetromino.py
@@ -0,0 +1,202 @@
+#!/usr/bin/python
+
+class Tetromino():
+
+    TYPES = ['i','a', 'b', 'c', 'o', 'd', 't', 's', 'z', 'j', 'l']
+
+    def __init__(self, state, letter):
+        # assert that there are rows
+        assert len(state) > 0
+        # assert rows and columns form a rectangle
+        assert len({len(row) for row in state}) == 1
+        self.state = state
+        self.letter = letter
+
+    @staticmethod
+    def ITetromino(): #ok
+        return Tetromino(
+            [
+                ['I', 'I', 'I', 'I']
+            ],
+            'i'
+        )
+
+    @staticmethod
+    def ATetromino(): #ok
+        return Tetromino(
+            [
+                ['I', 'I', 'I']
+            ],
+            'a'
+        )
+
+    @staticmethod
+    def BTetromino(): #ok
+        return Tetromino(
+            [
+                ['I', 'I',]
+            ],
+            'b'
+        )
+
+    @staticmethod
+    def CTetromino(): #ok
+        return Tetromino(
+            [
+                ['I']
+            ],
+            'c'
+        )
+
+    @staticmethod
+    def OTetromino(): #ok
+        return Tetromino(
+            [
+                ['O', 'O'],
+                ['O', 'O']
+            ],
+            'o'
+        )
+
+
+    @staticmethod
+    def DTetromino(): #ok
+        return Tetromino(
+            [
+                ['O', 'O'],
+                [' ', 'O']
+            ],
+            'd'
+        )
+
+    @staticmethod
+    def TTetromino(): #ok
+        return Tetromino(
+            [
+                ['T', 'T', 'T'],
+                [' ', 'T', ' ']
+            ],
+            't'
+        )
+
+    @staticmethod
+    def STetromino(): #ok
+        return Tetromino(
+            [
+                [' ', 'S', 'S'],
+                ['S', 'S', ' ']
+            ],
+            's'
+        )
+
+    @staticmethod
+    def ZTetromino(): #ok
+        return Tetromino(
+            [
+                ['Z', 'Z', ' '],
+                [' ', 'Z', 'Z']
+            ],
+            'z'
+        )
+
+    @staticmethod
+    def JTetromino(): #ok
+        return Tetromino(
+            [
+                ['J', 'J', 'J'],
+                [' ', ' ', 'J']
+            ],
+            'j'
+        )
+
+    @staticmethod
+    def LTetromino():
+        return Tetromino(
+            [
+                [' ', ' ', 'L'],
+                ['L', 'L', 'L']
+            ],
+            'l'
+        )
+
+    @staticmethod
+    def create(letter):
+        assert letter.lower() in Tetromino.TYPES
+        return getattr(Tetromino, '{}Tetromino'.format(letter.upper()))()
+
+    def __str__(self):
+        return "\n".join(["".join(x) for x in self.state])
+
+    def __getitem__(self, key):
+        return self.state[key]
+
+    def copy(self):
+        return Tetromino([row[:] for row in self.state], self.letter)
+
+    def width(self):
+        return len(self.state[0])
+
+    def height(self):
+        return len(self.state)
+
+    def rotate(self, change):
+        while change < 0:
+            change += 4
+        change = (change % 4)
+        assert 0 <= change and change <= 3
+        if change == 0:
+            return None
+        elif change == 1:
+            self.rotate_right()
+        elif change == 2:
+            self.flip()
+        elif change == 3:
+            self.rotate_left()
+        else:
+            raise Exception('This should never happen!')
+
+    def rotate_right(self):
+        self.state = list(zip(*self.state[::-1]))
+        return self
+
+    def rotate_left(self):
+        self.state = list(reversed(list(zip(*self.state))))
+        return self
+
+    def flip(self):
+        self.state = [row[::-1] for row in self.state[::-1]]
+        return self
+
+    def get_offset_column(self, rotation):
+        offset_map= {
+            'i': {0: 10, 1: 9, 2: 10, 3: 9},
+            'a': {0: 9, 1: 9, 2: 9, 3: 9},
+            'b': {0: 9, 1: 8, 2: 10, 3: 8},
+            'c': {0: 8, 1: 8, 2: 8, 3: 8},
+            'o': {0: 9, 1: 9, 2: 9, 3: 9},
+            'd': {0: 9, 1: 9, 2: 9, 3: 9},
+            't': {0: 9, 1: 9, 2: 10, 3: 9},
+            's': {0: 9, 1: 9, 2: 9, 3: 9},
+            'z': {0: 9, 1: 9, 2: 9, 3: 9},
+            'j': {0: 9, 1: 8, 2: 8, 3: 8},
+            'l': {0: 9, 1: 8, 2: 8, 3: 8}
+         }
+        return offset_map.get(self.letter)[rotation]
+
+
+if __name__ == '__main__':
+    t = Tetromino.LTetromino()
+    print(t)
+    print()
+    t.rotate_right()
+    print(t)
+    print()
+    t.rotate_right()
+    print(t)
+    print()
+    t.rotate_left()
+    print(t)
+    print(t.height())
+    print(t.width())
+    t.flip()
+    print(t)