From c75798cbe9e54236c9a727b54e89ae96c07bb481 Mon Sep 17 00:00:00 2001
From: Nova <technobaboo@gmail.com>
Date: Sat, 1 Oct 2022 22:55:41 -0400
Subject: [PATCH] feat(input): tip

---
 src/nodes/input/mod.rs |  4 ++++
 src/nodes/input/tip.rs | 54 ++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 58 insertions(+)
 create mode 100644 src/nodes/input/tip.rs

diff --git a/src/nodes/input/mod.rs b/src/nodes/input/mod.rs
index 96c1fcc..7e07392 100644
--- a/src/nodes/input/mod.rs
+++ b/src/nodes/input/mod.rs
@@ -1,8 +1,10 @@
 pub mod hand;
 pub mod pointer;
+pub mod tip;
 
 use self::hand::Hand;
 use self::pointer::Pointer;
+use self::tip::Tip;
 
 use super::fields::Field;
 use super::spatial::{get_spatial_parent_flex, parse_transform, Spatial};
@@ -38,6 +40,7 @@ pub trait InputSpecialization: Send + Sync {
 pub enum InputType {
 	Pointer(Pointer),
 	Hand(Box<Hand>),
+	Tip(Tip),
 }
 impl Deref for InputType {
 	type Target = dyn InputSpecialization;
@@ -45,6 +48,7 @@ impl Deref for InputType {
 		match self {
 			InputType::Pointer(p) => p,
 			InputType::Hand(h) => h.as_ref(),
+			InputType::Tip(t) => t,
 		}
 	}
 }
diff --git a/src/nodes/input/tip.rs b/src/nodes/input/tip.rs
new file mode 100644
index 0000000..da37e36
--- /dev/null
+++ b/src/nodes/input/tip.rs
@@ -0,0 +1,54 @@
+use super::{DistanceLink, InputSpecialization};
+use crate::nodes::fields::Field;
+use crate::nodes::spatial::Spatial;
+use glam::{vec3a, Mat4};
+use portable_atomic::AtomicF32;
+use stardust_xr_schemas::input::InputDataRaw;
+use stardust_xr_schemas::input_tip;
+use std::sync::atomic::Ordering;
+use std::sync::Arc;
+
+#[derive(Default)]
+pub struct Tip {
+	pub radius: AtomicF32,
+	pub grab: AtomicF32,
+	pub select: AtomicF32,
+}
+
+impl InputSpecialization for Tip {
+	fn distance(&self, space: &Arc<Spatial>, field: &Field) -> f32 {
+		field.distance(space, vec3a(0.0, 0.0, 0.0))
+	}
+	fn serialize(
+		&self,
+		fbb: &mut flatbuffers::FlatBufferBuilder,
+		_distance_link: &DistanceLink,
+		local_to_handler_matrix: Mat4,
+	) -> (
+		InputDataRaw,
+		flatbuffers::WIPOffset<flatbuffers::UnionWIPOffset>,
+	) {
+		let (_, orientation, origin) = local_to_handler_matrix.to_scale_rotation_translation();
+
+		let origin: mint::Vector3<f32> = origin.into();
+		let orientation: mint::Quaternion<f32> = orientation.into();
+
+		let tip = input_tip::Tip::create(
+			fbb,
+			&input_tip::TipArgs {
+				origin: Some(&origin.into()),
+				orientation: Some(&orientation.into()),
+				radius: self.radius.load(Ordering::Relaxed),
+			},
+		);
+		(InputDataRaw::Tip, tip.as_union_value())
+	}
+	fn serialize_datamap(&self) -> Vec<u8> {
+		let mut fbb = flexbuffers::Builder::default();
+		let mut map = fbb.start_map();
+		map.push("grab", self.grab.load(Ordering::Relaxed));
+		map.push("select", self.select.load(Ordering::Relaxed));
+		map.end_map();
+		fbb.view().to_vec()
+	}
+}