79 lines
2.0 KiB
Rust
79 lines
2.0 KiB
Rust
use crate::{
|
|
core::{client::INTERNAL_CLIENT, typed_datamap::TypedDatamap},
|
|
nodes::{
|
|
input::{tip::Tip, InputMethod, InputType},
|
|
spatial::Spatial,
|
|
Node,
|
|
},
|
|
};
|
|
use color_eyre::eyre::Result;
|
|
use glam::{Mat4, Vec2, Vec3};
|
|
use serde::{Deserialize, Serialize};
|
|
use std::sync::Arc;
|
|
use stereokit::{
|
|
named_colors::WHITE, ButtonState, Handed, Model, RenderLayer, StereoKitDraw,
|
|
StereoKitMultiThread,
|
|
};
|
|
|
|
#[derive(Default, Deserialize, Serialize)]
|
|
struct ControllerDatamap {
|
|
select: f32,
|
|
grab: f32,
|
|
scroll: Vec2,
|
|
}
|
|
|
|
pub struct SkController {
|
|
_node: Arc<Node>,
|
|
input: Arc<InputMethod>,
|
|
model: Model,
|
|
handed: Handed,
|
|
datamap: TypedDatamap<ControllerDatamap>,
|
|
}
|
|
impl SkController {
|
|
pub fn new(sk: &impl StereoKitMultiThread, handed: Handed) -> Result<Self> {
|
|
let _node = Node::create(
|
|
&INTERNAL_CLIENT,
|
|
"",
|
|
if handed == Handed::Left {
|
|
"controller_left"
|
|
} else {
|
|
"controller_right"
|
|
},
|
|
false,
|
|
)
|
|
.add_to_scenegraph()?;
|
|
Spatial::add_to(&_node, None, Mat4::IDENTITY, false)?;
|
|
let model = sk.model_create_mem("cursor.glb", include_bytes!("cursor.glb"), None)?;
|
|
let tip = InputType::Tip(Tip::default());
|
|
let input = InputMethod::add_to(&_node, tip, None)?;
|
|
Ok(SkController {
|
|
_node,
|
|
input,
|
|
handed,
|
|
model,
|
|
datamap: Default::default(),
|
|
})
|
|
}
|
|
pub fn update(&mut self, sk: &impl StereoKitDraw) {
|
|
let controller = sk.input_controller(self.handed);
|
|
*self.input.enabled.lock() = controller.tracked.contains(ButtonState::ACTIVE);
|
|
if *self.input.enabled.lock() {
|
|
let world_transform = Mat4::from_rotation_translation(
|
|
controller.aim.orientation,
|
|
controller.aim.position,
|
|
);
|
|
sk.model_draw(
|
|
&self.model,
|
|
world_transform * Mat4::from_scale(Vec3::ONE * 0.02),
|
|
WHITE,
|
|
RenderLayer::LAYER0,
|
|
);
|
|
self.input.spatial.set_local_transform(world_transform);
|
|
}
|
|
self.datamap.select = controller.trigger;
|
|
self.datamap.grab = controller.grip;
|
|
self.datamap.scroll = controller.stick;
|
|
*self.input.datamap.lock() = self.datamap.to_datamap().ok();
|
|
}
|
|
}
|