pybevy_math 0.2.1

Math types (Vec2, Vec3, Vec4, Quat, Mat3, Mat4) for PyBevy
Documentation 
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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102

use bevy::math::{CompassOctant, CompassQuadrant};
use pybevy_macros::bevy_enum;
use pyo3::prelude::*;

use crate::PyVec2;

#[bevy_enum(CompassOctant, from_only)]
#[pyclass(name = "CompassOctant", eq)]
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum PyCompassOctant {
    North,
    NorthEast,
    East,
    SouthEast,
    South,
    SouthWest,
    West,
    NorthWest,
}

#[pymethods]
impl PyCompassOctant {
    #[staticmethod]
    pub fn from_index(index: usize) -> Option<Self> {
        CompassOctant::from_index(index).map(|c| c.into())
    }

    pub fn to_index(&self) -> usize {
        let octant: CompassOctant = (*self).into();
        octant.to_index()
    }

    pub fn opposite(&self) -> Self {
        let octant: CompassOctant = (*self).into();
        octant.opposite().into()
    }

    pub fn is_in_direction(&self, origin: PyVec2, candidate: PyVec2) -> bool {
        let octant: CompassOctant = (*self).into();
        octant.is_in_direction(origin.into(), candidate.into())
    }

    pub fn __neg__(&self) -> Self {
        self.opposite()
    }

    fn __repr__(&self) -> String {
        let name = match self {
            Self::North => "North",
            Self::NorthEast => "NorthEast",
            Self::East => "East",
            Self::SouthEast => "SouthEast",
            Self::South => "South",
            Self::SouthWest => "SouthWest",
            Self::West => "West",
            Self::NorthWest => "NorthWest",
        };
        format!("CompassOctant.{name}")
    }
}

#[bevy_enum(CompassQuadrant, from_only)]
#[pyclass(name = "CompassQuadrant", eq)]
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum PyCompassQuadrant {
    North,
    East,
    South,
    West,
}

#[pymethods]
impl PyCompassQuadrant {
    #[staticmethod]
    pub fn from_index(index: usize) -> Option<Self> {
        CompassQuadrant::from_index(index).map(|c| c.into())
    }

    pub fn to_index(&self) -> usize {
        let quadrant: CompassQuadrant = (*self).into();
        quadrant.to_index()
    }

    pub fn opposite(&self) -> Self {
        let quadrant: CompassQuadrant = (*self).into();
        quadrant.opposite().into()
    }

    pub fn __neg__(&self) -> Self {
        self.opposite()
    }

    fn __repr__(&self) -> String {
        let name = match self {
            Self::North => "North",
            Self::East => "East",
            Self::South => "South",
            Self::West => "West",
        };
        format!("CompassQuadrant.{name}")
    }
}