# A simple vector 2 class to track unit and city positions.
#
# NOTE: You can make use of the distance_line and distance_man
# functions in your AI code.

import math

class Vec2:
    def __init__(self, x, y):
        self.x = x
        self.y = y

    def __eq__(self, o):
        return o.x == self.x and o.y == self.y

    def __ne__(self, o):
        return not (self == o)

    def __hash__(self):
        return hash((self.x, self.y))

    def __str__(self):
        return f"({self.x},{self.y})"

    def __add__(self, o):
        return Vec2(self.x + o.x, self.y + o.y)
    
    # Straight-line distance
    def distance_line(self, o):
        return math.sqrt(
            (o.x - self.x)**2 +
            (o.y - self.y)**2)

    # Manhattan distance
    def distance_man(self, o):
        return (o.x - self.x) + (o.y - self.y)

    def mod(self, maxX, maxY):
        self.x = (self.x + maxX) % maxX
        self.y = (self.y + maxY) % maxY



MOVES = {
    'N': Vec2(0,-1),
    'E': Vec2(1,0),
    'S': Vec2(0,1),
    'W': Vec2(-1,0)
    }