added transform utils for snapping to position and rotation

This commit is contained in:
Jay Christy
2024-06-11 13:17:16 -04:00
parent 8807b5f927
commit 8ecebb7d2c
4 changed files with 127 additions and 41 deletions

View File

@@ -1,9 +1,12 @@
//! A simple 3D scene with light shining over a cube sitting on a plane. //! A simple example of how to use the transform utils to set the players position and orientation
use bevy::prelude::*; use bevy::prelude::*;
use bevy_openxr::add_xr_plugins; use bevy_openxr::add_xr_plugins;
use bevy_xr_utils::xr_utils_actions::{ActiveSet, XRUtilsAction, XRUtilsActionSet, XRUtilsActionState, XRUtilsActionSystemSet, XRUtilsActionsPlugin, XRUtilsBinding}; use bevy_xr_utils::transform_utils::{self, SnapToPosition, SnapToRotation};
use bevy_xr_utils::transform_utils; use bevy_xr_utils::xr_utils_actions::{
ActiveSet, XRUtilsAction, XRUtilsActionSet, XRUtilsActionState, XRUtilsActionSystemSet,
XRUtilsActionsPlugin, XRUtilsBinding,
};
fn main() { fn main() {
App::new() App::new()
@@ -18,8 +21,16 @@ fn main() {
) )
.add_systems( .add_systems(
Update, Update,
read_action_with_marker_component.after(XRUtilsActionSystemSet::SyncActionStates), send_look_at_red_cube_event.after(XRUtilsActionSystemSet::SyncActionStates),
) )
.add_systems(
Update,
send_recenter.after(XRUtilsActionSystemSet::SyncActionStates),
)
.insert_resource(AmbientLight {
color: Default::default(),
brightness: 500.0,
})
.run(); .run();
} }
@@ -43,7 +54,7 @@ fn setup(
transform: Transform::from_xyz(4.0, 0.5, 0.0).with_scale(Vec3::splat(0.5)), transform: Transform::from_xyz(4.0, 0.5, 0.0).with_scale(Vec3::splat(0.5)),
..default() ..default()
}); });
// red cube // blue cube
commands.spawn(PbrBundle { commands.spawn(PbrBundle {
mesh: meshes.add(Cuboid::new(1.0, 1.0, 1.0)), mesh: meshes.add(Cuboid::new(1.0, 1.0, 1.0)),
material: materials.add(Color::rgb_u8(3, 28, 252)), material: materials.add(Color::rgb_u8(3, 28, 252)),
@@ -64,15 +75,14 @@ fn setup(
transform: Transform::from_xyz(0.0, 0.5, -4.0).with_scale(Vec3::splat(0.5)), transform: Transform::from_xyz(0.0, 0.5, -4.0).with_scale(Vec3::splat(0.5)),
..default() ..default()
}); });
// light // black cube
commands.spawn(PointLightBundle { commands.spawn(PbrBundle {
point_light: PointLight { mesh: meshes.add(Cuboid::new(1.0, 1.0, 1.0)),
shadows_enabled: true, material: materials.add(Color::rgb_u8(0, 0, 0)),
..default() transform: Transform::from_xyz(0.0, 0.1, 0.0).with_scale(Vec3::splat(0.2)),
},
transform: Transform::from_xyz(4.0, 8.0, 4.0),
..default() ..default()
}); });
commands.spawn(Camera3dBundle { commands.spawn(Camera3dBundle {
transform: Transform::from_xyz(-2.5, 4.5, 9.0).looking_at(Vec3::ZERO, Vec3::Y), transform: Transform::from_xyz(-2.5, 4.5, 9.0).looking_at(Vec3::ZERO, Vec3::Y),
..default() ..default()
@@ -110,7 +120,31 @@ fn create_action_entities(mut commands: Commands) {
let binding = commands let binding = commands
.spawn(XRUtilsBinding { .spawn(XRUtilsBinding {
profile: "/interaction_profiles/valve/index_controller".into(), profile: "/interaction_profiles/valve/index_controller".into(),
binding: "/user/hand/left//input/a/click".into(), binding: "/user/hand/left/input/a/click".into(),
})
.id();
//add action to set, this isnt the best
//TODO look into a better system
commands.entity(action).add_child(binding);
commands.entity(set).add_child(action);
let action = commands
.spawn((
XRUtilsAction {
action_name: "center".into(),
localized_name: "center_localized".into(),
action_type: bevy_xr::actions::ActionType::Bool,
},
Center, //lets try a marker component
))
.id();
//create a binding
let binding = commands
.spawn(XRUtilsBinding {
profile: "/interaction_profiles/valve/index_controller".into(),
binding: "/user/hand/right/input/a/click".into(),
}) })
.id(); .id();
@@ -120,11 +154,48 @@ fn create_action_entities(mut commands: Commands) {
commands.entity(set).add_child(action); commands.entity(set).add_child(action);
} }
fn read_action_with_marker_component( fn send_look_at_red_cube_event(
mut action_query: Query<&XRUtilsActionState, With<FaceRedAction>>, mut action_query: Query<&XRUtilsActionState, With<FaceRedAction>>,
mut event_writer: EventWriter<SnapToRotation>,
) { ) {
//now for the actual checking //now for the actual checking
for state in action_query.iter_mut() { for state in action_query.iter_mut() {
info!("action state is: {:?}", state); match state {
XRUtilsActionState::Bool(state) => {
let send = state.current_state && state.changed_since_last_sync;
if send {
info!("send facing");
let quat = Transform::default()
.looking_at(Transform::from_xyz(4.0, 0.0, 0.0).translation, Vec3::Y); //this is a transform facing the red cube from the center of the scene, you should use the HMD posision but I was lazy.
event_writer.send(SnapToRotation(quat.rotation));
}
}
XRUtilsActionState::Float(_) => (),
XRUtilsActionState::Vector(_) => (),
}
}
}
#[derive(Component)]
pub struct Center;
fn send_recenter(
mut action_query: Query<&XRUtilsActionState, With<Center>>,
mut event_writer: EventWriter<SnapToPosition>,
) {
//now for the actual checking
for state in action_query.iter_mut() {
match state {
XRUtilsActionState::Bool(state) => {
let send = state.current_state && state.changed_since_last_sync;
if send {
let center = Transform::default().translation;
event_writer.send(SnapToPosition(center));
}
}
XRUtilsActionState::Float(_) => (),
XRUtilsActionState::Vector(_) => (),
}
} }
} }

View File

@@ -14,7 +14,7 @@ impl Plugin for OxrActionBindingPlugin {
app.add_schedule(Schedule::new(OxrSendActionBindings)); app.add_schedule(Schedule::new(OxrSendActionBindings));
app.add_event::<OxrSuggestActionBinding>(); app.add_event::<OxrSuggestActionBinding>();
app.add_systems( app.add_systems(
PostUpdate, Update,
run_action_binding_sugestion.run_if( run_action_binding_sugestion.run_if(
|mut session_state: EventReader<OxrSessionStatusEvent>| { |mut session_state: EventReader<OxrSessionStatusEvent>| {
session_state session_state

View File

@@ -1,5 +1,10 @@
use bevy::prelude::*; use bevy::prelude::*;
use bevy_openxr::{helper_traits::{ToQuat, ToVec3}, init::OxrTrackingRoot, resources::OxrViews}; use bevy_openxr::{
helper_traits::{ToQuat, ToVec3},
init::OxrTrackingRoot,
resources::OxrViews,
};
pub struct TransformUtilitiesPlugin; pub struct TransformUtilitiesPlugin;
@@ -27,30 +32,37 @@ pub fn handle_transform_events(
let result = root_query.get_single_mut(); let result = root_query.get_single_mut();
match result { match result {
Ok(mut root_transform) => { Ok(mut root_transform) => {
//rotation first
let view = views.first(); let view = views.first();
match view { match view {
Some(view) => { Some(view) => {
//get our parameters together //we want the view translation with a height of zero for a few calculations
let mut view_translation = view.pose.position.to_vec3(); let mut view_translation = view.pose.position.to_vec3();
let view_quat = view.pose.orientation.to_quat(); view_translation.y = 0.0;
let view_yaw = view_quat.to_euler(EulerRot::XYZ).1;
let science = Quat::from_axis_angle(*root_transform.up(), view_yaw);
let invert_science = science.inverse();
view_translation.y = 0.0; //we want to do rotations around the same height as the root
let root_local = root_transform.translation;
let view_global = root_transform.rotation.mul_vec3(view_translation) + root_local;
//now set our rotation for every event, this does mean only the last event matters
for snap in rotation_reader.read() {
let rotation_quaternion = snap.0.mul_quat(invert_science);
root_transform.rotate_around(view_global, rotation_quaternion);
}
//position second
for position in position_reader.read() {
root_transform.translation = position.0;
}
}, //position
for position in position_reader.read() {
root_transform.translation = position.0 - root_transform.rotation.mul_vec3(view_translation);
}
//rotation
let root_local = root_transform.translation.clone();
let hmd_global = root_transform.rotation.mul_vec3(view_translation) + root_local;
let view_rot = view.pose.orientation.to_quat();
let root_rot = root_transform.rotation;
let view_global_rotation = root_rot.mul_quat(view_rot).normalize();
let (global_view_yaw, _pitch, _roll) = view_global_rotation.to_euler(bevy::math::EulerRot::YXZ);
let up = Vec3::Y;
for rotation in rotation_reader.read() {
let (target_yaw, _pitch, _roll) =
rotation.0.normalize().to_euler(bevy::math::EulerRot::YXZ);
let diff_yaw = target_yaw - global_view_yaw;
//build a rotation quat?
let rotation_quat = Quat::from_axis_angle(up, diff_yaw);
//apply rotation this works
root_transform.rotate_around(hmd_global, rotation_quat);
}
}
None => debug!("error getting first view"), None => debug!("error getting first view"),
} }
} }

View File

@@ -68,22 +68,28 @@ impl Plugin for XRUtilsActionsPlugin {
Startup, Startup,
create_openxr_events.in_set(XRUtilsActionSystemSet::CreateEvents), create_openxr_events.in_set(XRUtilsActionSystemSet::CreateEvents),
); );
app.add_systems(Update, sync_active_action_sets); app.add_systems(
Update,
sync_active_action_sets.run_if(resource_exists::<OxrSession>),
);
app.add_systems( app.add_systems(
Update, Update,
sync_and_update_action_states_f32 sync_and_update_action_states_f32
.run_if(resource_exists::<OxrSession>)
.in_set(XRUtilsActionSystemSet::SyncActionStates) .in_set(XRUtilsActionSystemSet::SyncActionStates)
.after(sync_active_action_sets), .after(sync_active_action_sets),
); );
app.add_systems( app.add_systems(
Update, Update,
sync_and_update_action_states_bool sync_and_update_action_states_bool
.run_if(resource_exists::<OxrSession>)
.in_set(XRUtilsActionSystemSet::SyncActionStates) .in_set(XRUtilsActionSystemSet::SyncActionStates)
.after(sync_active_action_sets), .after(sync_active_action_sets),
); );
app.add_systems( app.add_systems(
Update, Update,
sync_and_update_action_states_vector sync_and_update_action_states_vector
.run_if(resource_exists::<OxrSession>)
.in_set(XRUtilsActionSystemSet::SyncActionStates) .in_set(XRUtilsActionSystemSet::SyncActionStates)
.after(sync_active_action_sets), .after(sync_active_action_sets),
); );
@@ -248,7 +254,7 @@ fn sync_active_action_sets(
.collect::<Vec<_>>(); .collect::<Vec<_>>();
let sync = session.sync_actions(&active_sets[..]); let sync = session.sync_actions(&active_sets[..]);
match sync { match sync {
Ok(_) => info!("sync ok"), Ok(_) => (),
Err(_) => error!("sync error"), Err(_) => error!("sync error"),
} }
} }
@@ -262,7 +268,6 @@ fn sync_and_update_action_states_f32(
let state = reference.action.state(&session, Path::NULL); let state = reference.action.state(&session, Path::NULL);
match state { match state {
Ok(s) => { Ok(s) => {
info!("we found a state");
let new_state = XRUtilsActionState::Float(ActionStateFloat { let new_state = XRUtilsActionState::Float(ActionStateFloat {
current_state: s.current_state, current_state: s.current_state,
changed_since_last_sync: s.changed_since_last_sync, changed_since_last_sync: s.changed_since_last_sync,
@@ -288,7 +293,6 @@ fn sync_and_update_action_states_bool(
let state = reference.action.state(&session, Path::NULL); let state = reference.action.state(&session, Path::NULL);
match state { match state {
Ok(s) => { Ok(s) => {
info!("we found a state");
let new_state = XRUtilsActionState::Bool(ActionStateBool { let new_state = XRUtilsActionState::Bool(ActionStateBool {
current_state: s.current_state, current_state: s.current_state,
changed_since_last_sync: s.changed_since_last_sync, changed_since_last_sync: s.changed_since_last_sync,
@@ -314,7 +318,6 @@ fn sync_and_update_action_states_vector(
let state = reference.action.state(&session, Path::NULL); let state = reference.action.state(&session, Path::NULL);
match state { match state {
Ok(s) => { Ok(s) => {
info!("we found a state");
let new_state = XRUtilsActionState::Vector(ActionStateVector { let new_state = XRUtilsActionState::Vector(ActionStateVector {
current_state: [s.current_state.x, s.current_state.y], current_state: [s.current_state.x, s.current_state.y],
changed_since_last_sync: s.changed_since_last_sync, changed_since_last_sync: s.changed_since_last_sync,