Added debug_gizmos system

src/xr_input/debug_gizmos.rs

@@ -0,0 +1,258 @@
+use bevy::prelude::{Color, Gizmos, Plugin, Quat, Res, Transform, Update, Vec2, Vec3};
+
+use crate::{
+    input::XrInput,
+    resources::{XrFrameState, XrInstance, XrSession},
+};
+
+use crate::xr_input::{
+    oculus_touch::{OculusController, OculusControllerRef},
+    Hand, QuatConv, Vec3Conv,
+};
+
+/// add debug renderer for controllers
+#[derive(Default)]
+pub struct OpenXrDebugRenderer;
+
+impl Plugin for OpenXrDebugRenderer {
+    fn build(&self, app: &mut bevy::prelude::App) {
+        app.add_systems(Update, draw_gizmos);
+    }
+}
+
+pub fn draw_gizmos(
+    mut gizmos: Gizmos,
+    oculus_controller: Res<OculusController>,
+    frame_state: Res<XrFrameState>,
+    xr_input: Res<XrInput>,
+    instance: Res<XrInstance>,
+    session: Res<XrSession>,
+) {
+    //lock frame
+    let frame_state = *frame_state.lock().unwrap();
+    //get controller
+    let controller = oculus_controller.get_ref(&instance, &session, &frame_state, &xr_input);
+    //draw the hands
+    draw_hand_gizmo(&mut gizmos, &controller, Hand::Right);
+    draw_hand_gizmo(&mut gizmos, &controller, Hand::Left);
+}
+
+fn draw_hand_gizmo(gizmos: &mut Gizmos, controller: &OculusControllerRef<'_>, hand: Hand) {
+    match hand {
+        Hand::Left => {
+            //get left controller
+            let left_controller = controller.grip_space(Hand::Left);
+
+            let left_color = Color::YELLOW_GREEN;
+            let off_color = Color::BLUE;
+            let touch_color = Color::GREEN;
+            let pressed_color = Color::RED;
+
+            let grip_quat_offset = Quat::from_rotation_x(-1.4);
+            let face_quat_offset = Quat::from_rotation_x(1.05);
+
+            let controller_vec3 = left_controller.0.pose.position.to_vec3();
+            let controller_quat = left_controller.0.pose.orientation.to_quat();
+            let face_quat = controller_quat.mul_quat(face_quat_offset);
+            let face_quat_normal = face_quat.mul_vec3(Vec3::Z);
+
+            //draw grip
+            gizmos.rect(
+                controller_vec3,
+                controller_quat * grip_quat_offset,
+                Vec2::new(0.05, 0.1),
+                left_color,
+            );
+
+            let face_translation_offset = Quat::from_rotation_x(-1.7); //direction to move the face from the controller tracking point
+            let face_translation_vec3 = controller_vec3
+                + controller_quat
+                    .mul_quat(face_translation_offset)
+                    .mul_vec3(Vec3::Y * 0.075); //distance to move face by
+
+            //draw face
+            gizmos.circle(
+                face_translation_vec3,
+                face_quat_normal,
+                0.04,
+                Color::YELLOW_GREEN,
+            );
+
+            //button b
+            let mut b_color = off_color;
+            if controller.y_button_touched() {
+                b_color = touch_color;
+            }
+            if controller.y_button() {
+                b_color = pressed_color;
+            }
+
+            let b_offset_quat = face_quat;
+            let b_translation_vec3 =
+                face_translation_vec3 + b_offset_quat.mul_vec3(Vec3::new(0.025, -0.01, 0.0));
+            gizmos.circle(b_translation_vec3, face_quat_normal, 0.0075, b_color);
+
+            //button a
+            let mut a_color = off_color;
+            if controller.x_button_touched() {
+                a_color = touch_color;
+            }
+            if controller.x_button() {
+                a_color = pressed_color;
+            }
+
+            let a_offset_quat = face_quat;
+            let a_translation_vec3 =
+                face_translation_vec3 + a_offset_quat.mul_vec3(Vec3::new(0.025, 0.01, 0.0));
+            gizmos.circle(a_translation_vec3, face_quat_normal, 0.0075, a_color);
+
+            //joystick
+            let joystick_offset_quat = face_quat;
+            let joystick_base_vec =
+                face_translation_vec3 + joystick_offset_quat.mul_vec3(Vec3::new(-0.02, 0.0, 0.0));
+            let mut joystick_color = off_color;
+            if controller.thumbstick_touch(Hand::Left) {
+                joystick_color = touch_color;
+            }
+
+            //base
+            gizmos.circle(joystick_base_vec, face_quat_normal, 0.014, joystick_color);
+
+            let stick = controller.thumbstick(Hand::Left);
+            let input = Vec3::new(stick.x, -stick.y, 0.0);
+            let joystick_top_vec = face_translation_vec3
+                + joystick_offset_quat.mul_vec3(Vec3::new(-0.02, 0.0, -0.01))
+                + joystick_offset_quat.mul_vec3(input * 0.01);
+            //top
+            gizmos.circle(joystick_top_vec, face_quat_normal, 0.005, joystick_color);
+
+            //trigger
+            let trigger_state = controller.trigger(Hand::Left);
+            let trigger_rotation = Quat::from_rotation_x(-0.75 * trigger_state);
+            let mut trigger_color = off_color;
+            if controller.trigger_touched(Hand::Left) {
+                trigger_color = touch_color;
+            }
+            let trigger_transform = Transform {
+                translation: face_translation_vec3
+                    + face_quat
+                        .mul_quat(trigger_rotation)
+                        .mul_vec3(Vec3::new(0.0, 0.0, 0.02)),
+                rotation: face_quat.mul_quat(trigger_rotation),
+                scale: Vec3 {
+                    x: 0.01,
+                    y: 0.02,
+                    z: 0.03,
+                },
+            };
+            gizmos.cuboid(trigger_transform, trigger_color);
+        }
+        Hand::Right => {
+            //get right controller
+            let right_controller = controller.grip_space(Hand::Right);
+            let right_color = Color::YELLOW_GREEN;
+            let off_color = Color::BLUE;
+            let touch_color = Color::GREEN;
+            let pressed_color = Color::RED;
+
+            let grip_quat_offset = Quat::from_rotation_x(-1.4);
+            let face_quat_offset = Quat::from_rotation_x(1.05);
+
+            let controller_vec3 = right_controller.0.pose.position.to_vec3();
+            let controller_quat = right_controller.0.pose.orientation.to_quat();
+            let face_quat = controller_quat.mul_quat(face_quat_offset);
+            let face_quat_normal = face_quat.mul_vec3(Vec3::Z);
+
+            //grip
+            gizmos.rect(
+                controller_vec3,
+                controller_quat * grip_quat_offset,
+                Vec2::new(0.05, 0.1),
+                right_color,
+            );
+
+            let face_translation_offset = Quat::from_rotation_x(-1.7); //direction to move the face from the controller tracking point
+            let face_translation_vec3 = controller_vec3
+                + controller_quat
+                    .mul_quat(face_translation_offset)
+                    .mul_vec3(Vec3::Y * 0.075); //distance to move face by
+
+            //draw face
+            gizmos.circle(
+                face_translation_vec3,
+                face_quat_normal,
+                0.04,
+                Color::YELLOW_GREEN,
+            );
+
+            //button b
+            let mut b_color = off_color;
+            if controller.b_button_touched() {
+                b_color = touch_color;
+            }
+            if controller.b_button() {
+                b_color = pressed_color;
+            }
+
+            let b_offset_quat = face_quat;
+            let b_translation_vec3 =
+                face_translation_vec3 + b_offset_quat.mul_vec3(Vec3::new(-0.025, -0.01, 0.0));
+            gizmos.circle(b_translation_vec3, face_quat_normal, 0.0075, b_color);
+
+            //button a
+            let mut a_color = off_color;
+            if controller.a_button_touched() {
+                a_color = touch_color;
+            }
+            if controller.a_button() {
+                a_color = pressed_color;
+            }
+
+            let a_offset_quat = face_quat;
+            let a_translation_vec3 =
+                face_translation_vec3 + a_offset_quat.mul_vec3(Vec3::new(-0.025, 0.01, 0.0));
+            gizmos.circle(a_translation_vec3, face_quat_normal, 0.0075, a_color);
+
+            //joystick time
+            let joystick_offset_quat = face_quat;
+            let joystick_base_vec =
+                face_translation_vec3 + joystick_offset_quat.mul_vec3(Vec3::new(0.02, 0.0, 0.0));
+            let mut joystick_color = off_color;
+            if controller.thumbstick_touch(Hand::Right) {
+                joystick_color = touch_color;
+            }
+
+            //base
+            gizmos.circle(joystick_base_vec, face_quat_normal, 0.014, joystick_color);
+
+            let stick = controller.thumbstick(Hand::Right);
+            let input = Vec3::new(stick.x, -stick.y, 0.0);
+            let joystick_top_vec = face_translation_vec3
+                + joystick_offset_quat.mul_vec3(Vec3::new(0.02, 0.0, -0.01))
+                + joystick_offset_quat.mul_vec3(input * 0.01);
+            //top
+            gizmos.circle(joystick_top_vec, face_quat_normal, 0.005, joystick_color);
+
+            //trigger
+            let trigger_state = controller.trigger(Hand::Right);
+            let trigger_rotation = Quat::from_rotation_x(-0.75 * trigger_state);
+            let mut trigger_color = off_color;
+            if controller.trigger_touched(Hand::Right) {
+                trigger_color = touch_color;
+            }
+            let trigger_transform = Transform {
+                translation: face_translation_vec3
+                    + face_quat
+                        .mul_quat(trigger_rotation)
+                        .mul_vec3(Vec3::new(0.0, 0.0, 0.02)),
+                rotation: face_quat.mul_quat(trigger_rotation),
+                scale: Vec3 {
+                    x: 0.01,
+                    y: 0.02,
+                    z: 0.03,
+                },
+            };
+            gizmos.cuboid(trigger_transform, trigger_color);
+        }
+    }
+}

src/xr_input/mod.rs

@@ -1,4 +1,5 @@
 pub mod controllers;
+pub mod debug_gizmos;
 pub mod oculus_touch;
 pub mod xr_camera;