83 lines
2.4 KiB
Rust
83 lines
2.4 KiB
Rust
use super::{DistanceLink, InputSpecialization};
|
|
use crate::core::client::Client;
|
|
use crate::nodes::fields::Field;
|
|
use crate::nodes::input::{InputMethod, InputType};
|
|
use crate::nodes::spatial::{parse_transform, Spatial, Transform};
|
|
use crate::nodes::{Message, Node};
|
|
use color_eyre::eyre::Result;
|
|
use glam::{vec3a, Mat4};
|
|
use serde::Deserialize;
|
|
use stardust_xr::schemas::flat::{InputDataType, Tip as FlatTip};
|
|
use stardust_xr::schemas::flex::deserialize;
|
|
use stardust_xr::values::Datamap;
|
|
|
|
use std::sync::Arc;
|
|
|
|
#[derive(Default)]
|
|
pub struct Tip {
|
|
pub radius: f32,
|
|
}
|
|
impl Tip {
|
|
fn set_radius(node: Arc<Node>, _calling_client: Arc<Client>, message: Message) -> Result<()> {
|
|
let input_method = node.get_aspect::<InputMethod>()?;
|
|
if let InputType::Tip(tip) = &mut *input_method.specialization.lock() {
|
|
tip.radius = deserialize(message.as_ref())?;
|
|
}
|
|
Ok(())
|
|
}
|
|
}
|
|
impl InputSpecialization for Tip {
|
|
fn compare_distance(&self, space: &Arc<Spatial>, field: &Field) -> f32 {
|
|
field.distance(space, vec3a(0.0, 0.0, 0.0)).abs()
|
|
}
|
|
fn true_distance(&self, space: &Arc<Spatial>, field: &Field) -> f32 {
|
|
field.distance(space, vec3a(0.0, 0.0, 0.0))
|
|
}
|
|
fn serialize(
|
|
&self,
|
|
_distance_link: &DistanceLink,
|
|
local_to_handler_matrix: Mat4,
|
|
) -> InputDataType {
|
|
let (_, orientation, origin) = local_to_handler_matrix.to_scale_rotation_translation();
|
|
InputDataType::Tip(FlatTip {
|
|
origin: origin.into(),
|
|
orientation: orientation.into(),
|
|
radius: self.radius,
|
|
})
|
|
}
|
|
}
|
|
|
|
pub fn create_tip_flex(
|
|
_node: Arc<Node>,
|
|
calling_client: Arc<Client>,
|
|
message: Message,
|
|
) -> Result<()> {
|
|
#[derive(Deserialize)]
|
|
struct CreateTipInfo<'a> {
|
|
name: &'a str,
|
|
parent_path: &'a str,
|
|
transform: Transform,
|
|
radius: f32,
|
|
datamap: Option<Vec<u8>>,
|
|
}
|
|
let info: CreateTipInfo = deserialize(message.as_ref())?;
|
|
let node = Node::create_parent_name(&calling_client, "/input/method/tip", info.name, true);
|
|
let parent = calling_client
|
|
.get_node("Spatial parent", info.parent_path)?
|
|
.get_aspect::<Spatial>()?;
|
|
let transform = parse_transform(info.transform, true, true, false);
|
|
|
|
let node = node.add_to_scenegraph()?;
|
|
Spatial::add_to(&node, Some(parent.clone()), transform, false);
|
|
InputMethod::add_to(
|
|
&node,
|
|
InputType::Tip(Tip {
|
|
radius: info.radius,
|
|
}),
|
|
info.datamap
|
|
.and_then(|datamap| Datamap::from_raw(datamap).ok()),
|
|
)?;
|
|
node.add_local_signal("set_radius", Tip::set_radius);
|
|
Ok(())
|
|
}
|