pybevy_pbr 0.2.1

PBR components 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

use bevy::{color::Color, pbr::DistanceFog};
use pybevy_color::PyColor;
use pybevy_core::{ComponentStorage, PyComponent};
use pybevy_macros::component_storage;
use pyo3::prelude::*;

use crate::PyFogFalloff;

#[component_storage(DistanceFog)]
#[pyclass(name = "DistanceFog", extends = PyComponent)]
#[derive(Debug, Clone)]
pub struct PyDistanceFog {
    pub(crate) storage: ComponentStorage<DistanceFog>,
}

#[pymethods]
impl PyDistanceFog {
    #[new]
    #[pyo3(signature = (
        color = PyColor(Color::WHITE),
        falloff = PyFogFalloff::Linear { start: 0.0, end: 100.0 },
        directional_light_color = PyColor(Color::WHITE),
        directional_light_exponent = 8.0
    ))]
    pub fn new(
        color: PyColor,
        falloff: PyFogFalloff,
        directional_light_color: PyColor,
        directional_light_exponent: f32,
    ) -> (Self, PyComponent) {
        Self::from_owned(DistanceFog {
            color: color.0,
            falloff: falloff.into(),
            directional_light_color: directional_light_color.0,
            directional_light_exponent,
        })
    }

    #[getter]
    pub fn color(&self, py: Python) -> PyResult<Py<PyColor>> {
        PyColor::from_color(self.as_ref()?.color, py)
    }

    #[setter]
    pub fn set_color(&mut self, color: PyColor) -> PyResult<()> {
        self.as_mut()?.color = color.0;
        Ok(())
    }

    #[getter]
    pub fn falloff(&self) -> PyResult<PyFogFalloff> {
        Ok(self.as_ref()?.falloff.clone().into())
    }

    #[setter]
    pub fn set_falloff(&mut self, falloff: PyFogFalloff) -> PyResult<()> {
        self.as_mut()?.falloff = falloff.into();
        Ok(())
    }

    #[getter]
    pub fn directional_light_color(&self, py: Python) -> PyResult<Py<PyColor>> {
        PyColor::from_color(self.as_ref()?.directional_light_color, py)
    }

    #[setter]
    pub fn set_directional_light_color(&mut self, color: PyColor) -> PyResult<()> {
        self.as_mut()?.directional_light_color = color.0;
        Ok(())
    }

    #[getter]
    pub fn directional_light_exponent(&self) -> PyResult<f32> {
        Ok(self.as_ref()?.directional_light_exponent)
    }

    #[setter]
    pub fn set_directional_light_exponent(&mut self, exponent: f32) -> PyResult<()> {
        self.as_mut()?.directional_light_exponent = exponent;
        Ok(())
    }
}