refactor(fields): submodules for types

src/nodes/fields/cylinder.rs

@@ -0,0 +1,93 @@
+use super::{Field, FieldTrait, Node};
+use crate::core::client::Client;
+use crate::nodes::spatial::{get_spatial_parent_flex, Spatial};
+use anyhow::{anyhow, ensure, Result};
+use glam::{swizzles::*, vec2, Mat4, Vec3A};
+use libstardustxr::{flex_to_quat, flex_to_vec3};
+use portable_atomic::AtomicF32;
+use std::sync::atomic::Ordering;
+use std::sync::Arc;
+
+pub struct CylinderField {
+	space: Arc<Spatial>,
+	length: AtomicF32,
+	radius: AtomicF32,
+}
+
+impl CylinderField {
+	pub fn add_to(node: &Arc<Node>, length: f32, radius: f32) -> Result<()> {
+		ensure!(
+			node.spatial.get().is_some(),
+			"Internal: Node does not have a spatial attached!"
+		);
+		ensure!(
+			node.field.get().is_none(),
+			"Internal: Node already has a field attached!"
+		);
+		let cylinder_field = CylinderField {
+			space: node.spatial.get().unwrap().clone(),
+			length: AtomicF32::new(length),
+			radius: AtomicF32::new(radius),
+		};
+		cylinder_field.add_field_methods(node);
+		node.add_local_signal("setSize", CylinderField::set_size_flex);
+		let _ = node.field.set(Arc::new(Field::Cylinder(cylinder_field)));
+		Ok(())
+	}
+
+	pub fn set_size(&self, length: f32, radius: f32) {
+		self.length.store(length, Ordering::Relaxed);
+		self.radius.store(radius, Ordering::Relaxed);
+	}
+
+	pub fn set_size_flex(node: &Node, _calling_client: Arc<Client>, data: &[u8]) -> Result<()> {
+		let flex_vec = flexbuffers::Reader::get_root(data)?.get_vector()?;
+		let length = flex_vec.idx(0).as_f32();
+		let radius = flex_vec.idx(1).as_f32();
+		if let Field::Cylinder(cylinder_field) = node.field.get().unwrap().as_ref() {
+			cylinder_field.set_size(length, radius);
+		}
+		Ok(())
+	}
+}
+
+impl FieldTrait for CylinderField {
+	fn local_distance(&self, p: Vec3A) -> f32 {
+		let radius = self.length.load(Ordering::Relaxed);
+		let d = vec2(p.xy().length().abs() - radius, p.z.abs() - (radius * 0.5));
+
+		d.x.max(d.y).min(0_f32)
+			+ (if d.x >= 0_f32 && d.y >= 0_f32 {
+				d.length()
+			} else {
+				0_f32
+			})
+	}
+	fn spatial_ref(&self) -> &Spatial {
+		self.space.as_ref()
+	}
+}
+
+pub fn create_cylinder_field_flex(
+	_node: &Node,
+	calling_client: Arc<Client>,
+	data: &[u8],
+) -> Result<()> {
+	let flex_vec = flexbuffers::Reader::get_root(data)?.get_vector()?;
+	let node = Node::create(&calling_client, "/field", flex_vec.idx(0).get_str()?, true);
+	let parent = get_spatial_parent_flex(&calling_client, flex_vec.idx(1).get_str()?)?;
+	let transform = Mat4::from_rotation_translation(
+		flex_to_quat!(flex_vec.idx(3))
+			.ok_or_else(|| anyhow!("Rotation not found"))?
+			.into(),
+		flex_to_vec3!(flex_vec.idx(2))
+			.ok_or_else(|| anyhow!("Position not found"))?
+			.into(),
+	);
+	let length = flex_vec.idx(0).as_f32();
+	let radius = flex_vec.idx(1).as_f32();
+	let node = node.add_to_scenegraph();
+	Spatial::add_to(&node, Some(parent), transform)?;
+	CylinderField::add_to(&node, length, radius)?;
+	Ok(())
+}