csh/utils.py

from math import atan2, hypot, pi


class Vec:
    def __init__(self, x: int | float | None = None, y: int | float | None = None, z: int | float | None = None) -> None:
        if x != None:
            self.new(x, y, z)
        else:
            self.x = 0
            self.y = 0
            self.z = 0

    def new(self, x: int | float, y: int | float | None, z: int | float | None) -> None:
        """Change values of Vector"""
        if y == None:
            raise TypeError

        # Determine type of class
        self.type = int
        if type(x) == float:
            self.type = float

        # Verify other type
        if type(y) != self.type:
            raise TypeError

        self.x = self.type(x)
        self.y = self.type(y)

        if z == None:
            self.z = None
        else:
            self.z = self.type(z)

    def __str__(self):
        max_precision = 3

        x = round(self.x, max_precision)
        y = round(self.y, max_precision)
        z = None if self.z == None else round(self.z, max_precision)

        return f"{self.__class__.__name__}({x}, {y}{'' if z == None else f', {z}'})"

    def plus(self, other: 'Vec') -> 'Vec':
        x = self.x + other.x
        y = self.y + other.y

        if self.z == None or other.z == None:
            return Vec(x, y)

        return Vec(x, y, self.z + other.z)

    def minus(self, other: 'Vec') -> 'Vec':
        x = self.x - other.x
        y = self.y - other.y

        if self.z == None or other.z == None:
            return Vec(x, y)

        return Vec(x, y, self.z - other.z)

    def times(self, factor: float) -> 'Vec':
        x = self.x * factor
        y = self.y * factor

        if self.z == None:
            return Vec(x, y)

        return Vec(x, y, self.z * factor)

    def div(self, factor: float) -> 'Vec':
        x = self.x / factor
        y = self.y / factor

        if self.z == None:
            return Vec(x, y)

        return Vec(x, y, self.z / factor)

    def to_angle(self):
        if self.z == None:
            raise TypeError

        deg_to_rad = (180. / pi)

        return Vec(
            atan2(-self.z, hypot(self.x, self.y)) * deg_to_rad,
            atan2(self.y, self.x) * deg_to_rad
        )

    def is_zero(self):
        xy = (float(self.x) == 0.) and (float(self.y) == 0.)
        if self.z == None:
            return xy

        return xy and float(self.z) == 0.


def angle_fixer(angle: Vec) -> Vec:
    """Force the angle to respect game limitation"""
    # Limit of pitch in game is ]-89; 180[
    if angle.x > 89.:
        angle.x = 89.
    if angle.x < -89.:
        angle.x = -89

    # Limit of yaw in game is ]-180; 360[
    while (angle.y > 180.):
        angle.y -= 360.
    while (angle.y < -180.):
        angle.y += 360.

    return angle