pyxora.wrapper.functions

 1from typing import Tuple
 2import pygame
 3
 4__all__ = ["vector","rect"]
 5
 6def vector(x: int | float, y: int | float, z: int | float | None = None) -> pygame.math.Vector2 | pygame.math.Vector3:
 7    """
 8    Creates a 2D or 3D pygame vector based on the input arguments.
 9
10    Args:
11        x (int | float): X value of the vector.
12        y (int | float): Y value of the vector.
13        z (int | float | None): Optional Z value; if provided, returns a 3D vector.
14
15    Returns:
16        pygame.math.Vector2 or pygame.math.Vector3: A 2D or 3D vector.
17
18    Examples:
19        vector(1, 2) -> pygame.math.Vector2(1, 2)\n
20        vector(1, 2, 3) -> pygame.math.Vector3(1, 2, 3)
21    """
22    if z is not None:
23        return pygame.math.Vector3(x, y, z)
24    return pygame.math.Vector2(x, y)
25
26def rect(
27    pos: Tuple[int | float, int | float] | pygame.math.Vector2 | pygame.math.Vector3,
28    size: Tuple[int | float, int | float] | pygame.math.Vector2 | pygame.math.Vector3
29) -> pygame.Rect | pygame.FRect:
30    """
31    Returns the bounding rectangle (pygame.Rect or pygame.FRect) based on coordinate types.
32
33    Args:
34        pos (Tuple[int | float, int | float] | pygame.math.Vector2 | pygame.math.Vector3): Position of the rectangle.
35        size (Tuple[int | float, int | float] | pygame.math.Vector2 | pygame.math.Vector3): Size of the rectangle.
36
37    Returns:
38        pygame.Rect | pygame.FRect: The bounding rectangle of the shape.
39    """
40    if all(isinstance(i, int) for i in (pos[0], pos[1], size[0], size[1])):
41        return pygame.Rect(pos, size)  # Use pygame.Rect if all values are integers
42    return pygame.FRect(pos,size)  # Use pygame.FRect otherwise