pybevy_pbr 0.2.1

use bevy::pbr::Lightmap;
use pybevy_core::{ComponentStorage, PyComponent, PyHandle, extract_handle_from_any};
use pybevy_macros::component_storage;
use pybevy_math::rect::PyRect;
use pyo3::prelude::*;

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

#[pymethods]
impl PyLightmap {
    #[new]
    #[pyo3(signature = (image, uv_rect = None, bicubic_sampling = false))]
    pub fn new(
        image: &Bound<'_, PyAny>,
        uv_rect: Option<PyRect>,
        bicubic_sampling: bool,
    ) -> PyResult<(Self, PyComponent)> {
        let handle = extract_handle_from_any(image)?;
        Ok(Self::from_owned(Lightmap {
            image: handle.try_into()?,
            uv_rect: uv_rect
                .map(Into::into)
                .unwrap_or(bevy::math::Rect::new(0.0, 0.0, 1.0, 1.0)),
            bicubic_sampling,
        }))
    }

    #[getter]
    pub fn image(&self) -> PyResult<PyHandle> {
        Ok(PyHandle::from(&self.as_ref()?.image))
    }

    #[setter]
    pub fn set_image(&mut self, image: &Bound<'_, PyAny>) -> PyResult<()> {
        let handle = extract_handle_from_any(image)?;
        self.as_mut()?.image = handle.try_into()?;
        Ok(())
    }

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

    #[setter]
    pub fn set_uv_rect(&mut self, rect: PyRect) -> PyResult<()> {
        self.as_mut()?.uv_rect = rect.into();
        Ok(())
    }

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

    #[setter]
    pub fn set_bicubic_sampling(&mut self, value: bool) -> PyResult<()> {
        self.as_mut()?.bicubic_sampling = value;
        Ok(())
    }
}