use crate::builtin::ObjectMaterial2d;
use crate::resource::Material2d;
use std::cell::RefCell;
use std::collections::HashMap;
use std::rc::Rc;
thread_local!(static KEY_MATERIAL_MANAGER: RefCell<MaterialManager2d> = RefCell::new(MaterialManager2d::new()));
pub struct MaterialManager2d {
default_material: Rc<RefCell<Box<dyn Material2d + 'static>>>,
materials: HashMap<String, Rc<RefCell<Box<dyn Material2d + 'static>>>>,
}
impl Default for MaterialManager2d {
fn default() -> Self {
Self::new()
}
}
impl MaterialManager2d {
pub fn new() -> MaterialManager2d {
let mut materials = HashMap::new();
let om = Rc::new(RefCell::new(
Box::new(ObjectMaterial2d::new()) as Box<dyn Material2d + 'static>
));
let _ = materials.insert("object".to_string(), om.clone());
MaterialManager2d {
default_material: om,
materials,
}
}
pub fn get_global_manager<T, F: FnMut(&mut MaterialManager2d) -> T>(mut f: F) -> T {
KEY_MATERIAL_MANAGER.with(|manager| f(&mut manager.borrow_mut()))
}
pub fn get_default(&self) -> Rc<RefCell<Box<dyn Material2d + 'static>>> {
self.default_material.clone()
}
pub fn get(&mut self, name: &str) -> Option<Rc<RefCell<Box<dyn Material2d + 'static>>>> {
self.materials.get(name).cloned()
}
pub fn add(&mut self, material: Rc<RefCell<Box<dyn Material2d + 'static>>>, name: &str) {
let _ = self.materials.insert(name.to_string(), material);
}
pub fn remove(&mut self, name: &str) {
let _ = self.materials.remove(name);
}
pub fn begin_frame(&mut self) {
for material in self.materials.values() {
material.borrow_mut().begin_frame();
}
}
pub fn flush(&mut self) {
for material in self.materials.values() {
material.borrow_mut().flush();
}
}
pub fn reset_global_manager() {
KEY_MATERIAL_MANAGER.with(|manager| {
let mut manager = manager.borrow_mut();
manager.materials.clear();
});
}
}