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

use bevy::math::{Dir2, primitives::Plane2d};
use pyo3::prelude::*;

use crate::{dir2::PyDir2, vec2::PyVec2};

#[pyclass(name = "Plane2d", frozen, eq)]
#[derive(Clone, PartialEq)]
pub struct PyPlane2d {
    pub(crate) inner: Plane2d,
}

#[pymethods]
impl PyPlane2d {
    #[new]
    #[pyo3(signature = (normal = PyVec2::Y))]
    pub fn new(normal: PyVec2) -> PyResult<Self> {
        let dir = Dir2::new(normal.get())
            .map_err(|e| pyo3::exceptions::PyValueError::new_err(format!("{}", e)))?;
        Ok(Self {
            inner: Plane2d { normal: dir },
        })
    }

    #[staticmethod]
    pub fn from_dir(normal: PyDir2) -> Self {
        Self {
            inner: Plane2d {
                normal: normal.into_dir2(),
            },
        }
    }

    #[getter]
    pub fn normal(&self) -> PyDir2 {
        PyDir2::from_dir2(self.inner.normal)
    }

    fn __repr__(&self) -> String {
        format!("Plane2d(normal={})", self.inner.normal)
    }
}

impl From<Plane2d> for PyPlane2d {
    fn from(plane: Plane2d) -> Self {
        Self { inner: plane }
    }
}

impl From<PyPlane2d> for Plane2d {
    fn from(plane: PyPlane2d) -> Self {
        plane.inner
    }
}