diff --git a/tetrislogic/tetrislogic.py b/tetrislogic/tetrislogic.py index 4626b16..c1357b1 100644 --- a/tetrislogic/tetrislogic.py +++ b/tetrislogic/tetrislogic.py @@ -262,7 +262,7 @@ class TetrisLogic: def rotate(self, spin): rotated_coords = tuple( - Coord(spin * mino.coord.y, -spin * mino.coord.x) + mino.coord.rotate(spin) for mino in self.matrix.piece ) for rotation_point, liberty_degree in enumerate( diff --git a/tetrislogic/utils.py b/tetrislogic/utils.py index 1a5309c..e532eea 100644 --- a/tetrislogic/utils.py +++ b/tetrislogic/utils.py @@ -7,6 +7,9 @@ class Coord: def __add__(self, other): return Coord(self.x + other.x, self.y + other.y) + def rotate(self, spin): + return Coord(spin * self.y, -spin * self.x) + class Movement: