pybevy_pbr 0.2.1

use bevy::pbr::{DefaultOpaqueRendererMethod, OpaqueRendererMethod};
use pybevy_core::{PyResource, ResourceStorage};
use pybevy_macros::resource_storage;
use pybevy_render::PyOpaqueRenderMethod;
use pyo3::prelude::*;

#[resource_storage(DefaultOpaqueRendererMethod)]
#[pyclass(name = "DefaultOpaqueRendererMethod", extends = PyResource)]
#[derive(Debug)]
pub struct PyDefaultOpaqueRendererMethod {
    pub storage: ResourceStorage<DefaultOpaqueRendererMethod>,
}

#[pymethods]
impl PyDefaultOpaqueRendererMethod {
    #[new]
    #[pyo3(signature = (method = PyOpaqueRenderMethod::Forward))]
    pub fn new(method: PyOpaqueRenderMethod) -> (Self, PyResource) {
        let bevy_method: OpaqueRendererMethod = method.into();
        let resource = match bevy_method {
            OpaqueRendererMethod::Forward => DefaultOpaqueRendererMethod::forward(),
            OpaqueRendererMethod::Deferred => DefaultOpaqueRendererMethod::deferred(),
            OpaqueRendererMethod::Auto => DefaultOpaqueRendererMethod::forward(),
        };
        Self::from_owned(resource)
    }

    #[staticmethod]
    pub fn forward(py: Python) -> PyResult<Py<Self>> {
        let (obj, base) = Self::from_owned(DefaultOpaqueRendererMethod::forward());
        Py::new(py, (obj, base))
    }

    #[staticmethod]
    pub fn deferred(py: Python) -> PyResult<Py<Self>> {
        let (obj, base) = Self::from_owned(DefaultOpaqueRendererMethod::deferred());
        Py::new(py, (obj, base))
    }

    pub fn set_to_forward(&mut self) -> PyResult<()> {
        self.as_mut()?.set_to_forward();
        Ok(())
    }

    pub fn set_to_deferred(&mut self) -> PyResult<()> {
        self.as_mut()?.set_to_deferred();
        Ok(())
    }
}