biy2/piecepositionhelper.py at main · jautung/biy2 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
from movedirection import MoveDirection
from pieceposition import PiecePosition


def get_position_after_move(
    position: PiecePosition, direction: MoveDirection
) -> PiecePosition:
    if direction == MoveDirection.UP:
        return PiecePosition(
            x=position.x,
            y=position.y + 1,
        )
    elif direction == MoveDirection.DOWN:
        return PiecePosition(
            x=position.x,
            y=position.y - 1,
        )
    elif direction == MoveDirection.LEFT:
        return PiecePosition(
            x=position.x - 1,
            y=position.y,
        )
    elif direction == MoveDirection.RIGHT:
        return PiecePosition(
            x=position.x + 1,
            y=position.y,
        )
    elif direction == MoveDirection.WAIT:
        return PiecePosition(
            x=position.x,
            y=position.y,
        )