92 lines
3.0 KiB
Python
92 lines
3.0 KiB
Python
|
from enum import Enum
|
|||
|
|
from typing import NamedTuple, Optional
|
|||
|
|
|
|||
|
|
|
|||
|
|
class Coords(NamedTuple):
|
|||
|
|
"""
|
|||
|
|
Вспомогательный класс для упрощения работы с координатами
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
x: int | float
|
|||
|
|
y: int | float
|
|||
|
|
|
|||
|
|
def __add__(self, other):
|
|||
|
|
if isinstance(other, self.__class__):
|
|||
|
|
return self.__class__(self.x + other.x, self.y + other.y)
|
|||
|
|
if isinstance(other, (int, float)):
|
|||
|
|
return self.__class__(self.x + other, self.y + other)
|
|||
|
|
return NotImplemented
|
|||
|
|
|
|||
|
|
def __sub__(self, other):
|
|||
|
|
if isinstance(other, self.__class__):
|
|||
|
|
return self.__class__(self.x - other.x, self.y - other.y)
|
|||
|
|
if isinstance(other, (int, float)):
|
|||
|
|
return self.__class__(self.x - other, self.y - other)
|
|||
|
|
return NotImplemented
|
|||
|
|
|
|||
|
|
def __floordiv__(self, other):
|
|||
|
|
if isinstance(other, self.__class__):
|
|||
|
|
return self.__class__(self.x // other.x, self.y // other.y)
|
|||
|
|
if isinstance(other, (int, float)):
|
|||
|
|
return self.__class__(self.x // other, self.y // other)
|
|||
|
|
return NotImplemented
|
|||
|
|
|
|||
|
|
def __mul__(self, other):
|
|||
|
|
if isinstance(other, self.__class__):
|
|||
|
|
return self.__class__(self.x * other.x, self.y * other.y)
|
|||
|
|
if isinstance(other, (int, float)):
|
|||
|
|
return self.__class__(self.x * other, self.y * other)
|
|||
|
|
return NotImplemented
|
|||
|
|
|
|||
|
|
def transform(self, ref: "Coords"):
|
|||
|
|
return self * ref
|
|||
|
|
|
|||
|
|
def dir_norm(self):
|
|||
|
|
"""нормализация вектора, но только для получения направления
|
|||
|
|
x, y могут быть только -1, 0, 1
|
|||
|
|
может быть только направление по горизонтали либо по вертикали
|
|||
|
|
либо без направления
|
|||
|
|
"""
|
|||
|
|
src = self
|
|||
|
|
if self.x and self.y:
|
|||
|
|
src = (
|
|||
|
|
self.__class__(self.x, 0)
|
|||
|
|
if abs(self.x) > abs(self.y)
|
|||
|
|
else self.__class__(0, self.y)
|
|||
|
|
)
|
|||
|
|
return self.__class__(*((n > 0) - (n < 0) for n in src))
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
def zero(cls):
|
|||
|
|
return cls(0, 0)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class Direction(Enum):
|
|||
|
|
LEFT = 1
|
|||
|
|
RIGHT = 2
|
|||
|
|
UP = 3
|
|||
|
|
DOWN = 4
|
|||
|
|
|
|||
|
|
def as_coords(self):
|
|||
|
|
match self:
|
|||
|
|
case Direction.LEFT:
|
|||
|
|
return Coords(-1, 0)
|
|||
|
|
case Direction.RIGHT:
|
|||
|
|
return Coords(1, 0)
|
|||
|
|
case Direction.UP:
|
|||
|
|
return Coords(0, -1)
|
|||
|
|
case Direction.DOWN:
|
|||
|
|
return Coords(0, 1)
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def from_coords(coords: Coords) -> Optional["Direction"]:
|
|||
|
|
match coords.dir_norm():
|
|||
|
|
case Coords(-1, 0):
|
|||
|
|
return Direction.LEFT
|
|||
|
|
case Coords(1, 0):
|
|||
|
|
return Direction.RIGHT
|
|||
|
|
case Coords(0, -1):
|
|||
|
|
return Direction.UP
|
|||
|
|
case Coords(0, 1):
|
|||
|
|
return Direction.DOWN
|