# An OO wrapper around the Vector2 struct, by @Emtyloc

from pyray import *

class Vector2Ex(list):
    def __init__(self, x, y):
        super(Vector2Ex, self).__init__([x, y])

    @property
    def x(self):
        return self[0]

    @x.setter
    def x(self, value):
        self[0]= value

    @property
    def y(self):
        return self[1]

    @y.setter
    def y(self, value):
        self[1]= value

    @staticmethod
    def to_Vec2(v: Vector2):
        """
        Cast Vector2 to Vec2.
        """
        return Vector2Ex(v.x, v.y)

    def __repr__(self) -> str:
        return f"{self.x}, {self.y}"

    def __eq__(self, other):
        if isinstance(other, Vector2Ex):
            return self.x == other.x and self.y == other.y
        return False

    def __add__(self, other):
        if isinstance(other, Vector2Ex):
            return Vector2Ex(self.x + other.x, self.y + other.y)
        return Vector2Ex(self.x + other, self.y + other)

    def __iadd__(self, other):
        if isinstance(other, Vector2Ex):
            self.x += other.x
            self.y += other.y
        else:
            res = vector2_add_value(self, other)
            self.x = res.x
            self.y = res.y
        return self

    def __radd__(self, other):
        return self + other

    def __sub__(self, other):
        if isinstance(other, Vector2Ex):
            return Vector2Ex(self.x - other.x, self.y - other.y)
        return Vector2Ex(self.x - other, self.y - other)

    def __isub__(self, other):
        if isinstance(other, Vector2Ex):
            self.x -= other.x
            self.y -= other.y
        else:
            self.x -= other
            self.y -= other
        return self

    def __rsub__(self, other):
        return Vector2Ex(other - self.x, other - self.y)

    def __mul__(self, other):
        if isinstance(other, Vector2Ex):
            res = vector2_multiply(self, other)
            return self.to_Vec2(res)
        return Vector2Ex(self.x * other, self.y * other)

    def __imul__(self, other):
        if isinstance(other, Vector2Ex):
            res = vector2_multiply(self, other)
        else:
            res = vector2_scale(self, other)
        self.x = res.x
        self.y = res.y
        return self

    def __truediv__(self, other):
        if isinstance(other, Vector2Ex):
            res = vector_2divide(self, other)
            return self.to_Vec2(res)
        return Vector2Ex(self.x / other, self.y / other)

    def __itruediv__(self, other):
        if isinstance(other, Vector2Ex):
            res = vector_2divide(self, other)
        else:
            res = vector2_scale(self, 1/other)
        self.x = res.x
        self.y = res.y
        return self

    def __neg__(self):
        return Vector2Ex(-self.x, -self.y)

    def __pos__(self):
        return Vector2Ex(self.x, self.y)

    def __pow__(self, exponent):
        return Vector2Ex(self.x ** exponent, self.y ** exponent)

    # PyRay mapped vector2 functions

    def angle(self, vec2):
        return vector2_angle(self, vec2)

    def clamp(self, min_vec2, max_vec2):
        res = vector2_clamp(self, min_vec2, max_vec2)
        return self.to_Vec2(res)

    def clamp_value(self, min_val: float, max_val: float):
        res = vector2_clamp_value(self, min_val, max_val)
        return self.to_Vec2(res)

    def distance(self, vec2):
        return vector_2distance(self, vec2)

    def distance_sqr(self, vec2) -> float:
        return vector_2distance_sqr(self, vec2)

    def dot_product(self, vec2) -> float:
        return vector_2dot_product(self, vec2)

    def invert(self):
        res = vector2_invert(self)
        return self.to_Vec2(res)

    def length(self):
        return vector2_length(self)

    def length_sqr(self) -> float:
        return vector2_length_sqr(self)

    def lerp(self, vec2, amount: float):
        res = vector2_lerp(self, vec2, amount)
        return self.to_Vec2(res)

    def move_towards(self, target_vec2, max_distance: float):
        res = vector2_move_towards(self, target_vec2, max_distance)
        return self.to_Vec2(res)

    def negate(self):
        res = vector2_negate(self)
        return self.to_Vec2(res)

    def normalize(self):
        res =  vector2_normalize(self)
        return self.to_Vec2(res)

    def reflect(self, normal_vec2):
        res = vector2_reflect(self, normal_vec2)
        return self.to_Vec2(res)

    def rotate(self, angle: float):
        res = vector2_rotate(self, angle)
        return self.to_Vec2(res)

    def transform(self, mat: Matrix):
        res = vector2_transform(self, mat)
        return self.to_Vec2(res)

    @staticmethod
    def line_angle(start_vec2, end_vec2) -> float:
        return vector2_line_angle(start_vec2, end_vec2)

    @staticmethod
    def one():
        return Vector2Ex(1, 1)

    @staticmethod
    def zero():
        return Vector2Ex(0, 0)


if __name__ == "__main__":
    # Arithmetic ops
    v1 = Vector2Ex(5, 5)
    v2 = Vector2Ex(10, 10)

    print(v1 + v2) # 15, 15
    print(v1 - v2) # -5, -5

    print(v1 * v2) # 50.0, 50.0
    print(v1 / v2) # 0.5, 0.5

    print(v1 * 2) # 10, 10
    print(v2 / 2) # 5.0, 5.0

    v1+=v2
    print(v1) #15, 15
    v2-=v1
    print(v2) #-5, -5

    v1/=-v2
    print(v1) #3.0, 3.0
    v2*=v1
    print(v2) #-15.0, -15.0

    v3 = Vector2Ex(3, 5)
    print(v3 ** 2) #9, 25

    v1 = Vector2Ex.one()
    print(v1)

    v0 = Vector2Ex.zero()
    print(v0)

    # Vector2 pyray methods
    v1 = Vector2Ex(3, 4)
    v2 = Vector2Ex(1, 2)
    v_min = Vector2Ex(0, 0)
    v_max = Vector2Ex(5, 5)

    print("Angle:", v1.angle(v2))

    print("Clamp:", v1.clamp(v_min, v_max))

    print("Clamp value:", v1.clamp_value(1.5, 3.5))

    print("Distance:", v1.distance(v2))

    print("Distance Sqr:", v1.distance_sqr(v2))

    print("Dot Product:", v1.dot_product(v2))

    print("Invert:", v1.invert())

    print("Length:", v1.length())

    print("Length Sqr:", v1.length_sqr())

    print("Lerp:", v1.lerp(v2, 0.5))

    print("Line Angle:", Vector2Ex.line_angle(v1, v2))

    print("Move Towards:", v1.move_towards(v2, 0.5))

    print("Negate:", v1.negate())

    print("Normalize:", v1.normalize())

    print("Reflect:", v1.reflect(v2))

    print("Rotate:", v1.rotate(45))

    # I don't know why this not work
    # mat = Matrix2x2(1, 0, 0, 1)
    # print("Transform:", v1.transform(mat))