feat(drawable): set sky

src/nodes/drawable/mod.rs

@@ -1,19 +1,60 @@
 pub mod model;
 
 use super::Node;
-use crate::core::{client::Client, registry::Registry};
-use std::sync::Arc;
-use stereokit::{lifecycle::DrawContext, StereoKit};
-
-pub trait Drawable: Send + Sync {
-	fn draw(&self, sk: &StereoKit, draw_ctx: &DrawContext);
-}
-
-pub static DRAWABLE_REGISTRY: Registry<dyn Drawable> = Registry::new();
+use crate::core::client::Client;
+use anyhow::Result;
+use parking_lot::Mutex;
+use std::{path::PathBuf, str::FromStr, sync::Arc};
+use stereokit::{lifecycle::DrawContext, texture::Texture, StereoKit};
 
 pub fn create_interface(client: &Arc<Client>) {
 	let node = Node::create(client, "", "drawable", false);
 	node.add_local_signal("createModelFromFile", model::create_from_file);
 	node.add_local_signal("createModelFromResource", model::create_from_resource);
+	node.add_local_signal("setSkyFile", set_sky_file_flex);
 	node.add_to_scenegraph();
 }
+
+pub fn draw(sk: &mut StereoKit, draw_ctx: &DrawContext) {
+	model::draw_all(sk, draw_ctx);
+
+	let new_skytex = QUEUED_SKYTEX.lock().take();
+	let mut new_skylight = QUEUED_SKYLIGHT.lock().take();
+	let same_file = new_skytex == new_skylight;
+
+	if let Some(skytex) = new_skytex {
+		if let Some((skytex, skylight)) =
+			Texture::from_cubemap_equirectangular(sk, &skytex, true, i32::MAX)
+		{
+			sk.set_skytex(&skytex);
+			if same_file {
+				sk.set_skylight(&skylight);
+				new_skylight = None;
+			}
+		}
+	}
+	if let Some(skylight) = new_skylight {
+		if let Some((_, skylight)) =
+			Texture::from_cubemap_equirectangular(sk, &skylight, true, i32::MAX)
+		{
+			sk.set_skylight(&skylight);
+		}
+	}
+}
+
+static QUEUED_SKYLIGHT: Mutex<Option<PathBuf>> = Mutex::new(None);
+static QUEUED_SKYTEX: Mutex<Option<PathBuf>> = Mutex::new(None);
+
+fn set_sky_file_flex(_node: &Node, _calling_client: Arc<Client>, data: &[u8]) -> Result<()> {
+	let flex_vec = flexbuffers::Reader::get_root(data)?.get_vector()?;
+	let path = PathBuf::from_str(flex_vec.index(0)?.get_str()?)?;
+	path.metadata()?;
+	if flex_vec.idx(1).as_bool() {
+		QUEUED_SKYTEX.lock().replace(path.clone());
+	}
+	if flex_vec.idx(2).as_bool() {
+		QUEUED_SKYLIGHT.lock().replace(path);
+	}
+
+	Ok(())
+}