diff --git a/tetrislogic/tetrislogic.py b/tetrislogic/tetrislogic.py index c1357b1..d1c4c64 100644 --- a/tetrislogic/tetrislogic.py +++ b/tetrislogic/tetrislogic.py @@ -262,7 +262,7 @@ class TetrisLogic: def rotate(self, spin): rotated_coords = tuple( - mino.coord.rotate(spin) + mino.coord @ 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 e532eea..84d0c30 100644 --- a/tetrislogic/utils.py +++ b/tetrislogic/utils.py @@ -7,7 +7,7 @@ class Coord: def __add__(self, other): return Coord(self.x + other.x, self.y + other.y) - def rotate(self, spin): + def __matmul__(self, spin): return Coord(spin * self.y, -spin * self.x)