feat(model): apply material replacements

This commit is contained in:
Nova
2022-08-22 14:42:28 -04:00
parent be8e8530a9
commit cd86b3a001

View File

@@ -18,6 +18,7 @@ use std::path::PathBuf;
use std::sync::Arc; use std::sync::Arc;
use stereokit::enums::RenderLayer; use stereokit::enums::RenderLayer;
use stereokit::lifecycle::DrawContext; use stereokit::lifecycle::DrawContext;
use stereokit::material::Material;
use stereokit::model::Model as SKModel; use stereokit::model::Model as SKModel;
use stereokit::texture::Texture; use stereokit::texture::Texture;
use stereokit::StereoKit; use stereokit::StereoKit;
@@ -36,6 +37,7 @@ pub struct Model {
resource_id: ResourceID, resource_id: ResourceID,
pending_model_path: OnceCell<PathBuf>, pending_model_path: OnceCell<PathBuf>,
pending_material_parameters: Mutex<FxHashMap<(u32, String), MaterialParameter>>, pending_material_parameters: Mutex<FxHashMap<(u32, String), MaterialParameter>>,
pub pending_material_replacements: Mutex<FxHashMap<u32, Arc<SendWrapper<Material>>>>,
sk_model: OnceCell<SendWrapper<SKModel>>, sk_model: OnceCell<SendWrapper<SKModel>>,
} }
@@ -54,6 +56,7 @@ impl Model {
resource_id, resource_id,
pending_model_path: OnceCell::new(), pending_model_path: OnceCell::new(),
pending_material_parameters: Mutex::new(FxHashMap::default()), pending_material_parameters: Mutex::new(FxHashMap::default()),
pending_material_replacements: Mutex::new(FxHashMap::default()),
sk_model: OnceCell::new(), sk_model: OnceCell::new(),
}; };
node.add_local_signal("setMaterialParameter", Model::set_material_parameter); node.add_local_signal("setMaterialParameter", Model::set_material_parameter);
@@ -81,8 +84,17 @@ impl Model {
.ok(); .ok();
if let Some(sk_model) = sk_model { if let Some(sk_model) = sk_model {
for ((material_idx, parameter_name), parameter_value) in {
self.pending_material_parameters.lock().iter() let mut material_replacements = self.pending_material_replacements.lock();
for (material_idx, replacement_material) in material_replacements.iter() {
sk_model.set_material(*material_idx as i32, replacement_material);
}
material_replacements.clear();
}
{
let mut material_parameters = self.pending_material_parameters.lock();
for ((material_idx, parameter_name), parameter_value) in material_parameters.iter()
{ {
if let Some(material) = sk_model.get_material(sk, *material_idx as i32) { if let Some(material) = sk_model.get_material(sk, *material_idx as i32) {
match parameter_value { match parameter_value {
@@ -94,7 +106,8 @@ impl Model {
} }
} }
} }
self.pending_material_parameters.lock().clear(); material_parameters.clear();
}
sk_model.draw( sk_model.draw(
draw_ctx, draw_ctx,