diff --git a/crates/bevy_openxr/src/openxr/features/overlay.rs b/crates/bevy_openxr/src/openxr/features/overlay.rs index 7b3d072..dfca5d3 100644 --- a/crates/bevy_openxr/src/openxr/features/overlay.rs +++ b/crates/bevy_openxr/src/openxr/features/overlay.rs @@ -1,12 +1,13 @@ use std::{mem, ptr}; use bevy::prelude::*; -use openxr::sys; +use openxr::{sys, Event}; use crate::{ next_chain::{OxrNextChainStructBase, OxrNextChainStructProvider}, openxr::exts::OxrEnabledExtensions, openxr_session_available, + poll_events::{OxrEvent, OxrEventHandlerExt}, session::{OxrSessionCreateNextChain, OxrSessionCreateNextProvider}, }; @@ -20,6 +21,17 @@ impl Plugin for OxrOverlayPlugin { First, add_overlay_info_to_chain.run_if(openxr_session_available), ); + app.add_oxr_event_handler(handle_overlay_event); + } +} + +fn handle_overlay_event(event: In, mut writer: EventWriter) { + // this unwrap will never panic since we are in a valid scope + if let Event::MainSessionVisibilityChangedEXTX(event) = unsafe { event.get() }.unwrap() { + writer.send(OxrOverlaySessionEvent::MainSessionVisibilityChanged { + visible: event.visible(), + flags: event.flags(), + }); } } diff --git a/crates/bevy_openxr/src/openxr/init.rs b/crates/bevy_openxr/src/openxr/init.rs index fa8d4fd..5567746 100644 --- a/crates/bevy_openxr/src/openxr/init.rs +++ b/crates/bevy_openxr/src/openxr/init.rs @@ -20,7 +20,6 @@ use bevy_mod_xr::session::*; use openxr::Event; use crate::error::OxrError; -use crate::features::overlay::OxrOverlaySessionEvent; use crate::graphics::*; use crate::resources::*; use crate::session::OxrSession; @@ -28,6 +27,8 @@ use crate::session::OxrSessionCreateNextChain; use crate::types::*; use super::exts::OxrEnabledExtensions; +use super::poll_events::OxrEvent; +use super::poll_events::OxrEventHandlerExt; pub fn session_started(started: Option>) -> bool { started.is_some_and(|started| started.0) @@ -106,12 +107,7 @@ impl Plugin for OxrInitPlugin { }, ExtractResourcePlugin::::default(), )) - .add_systems( - XrFirst, - poll_events - .in_set(XrHandleEvents::Poll) - .run_if(not(state_equals(XrState::Unavailable))), - ) + .add_oxr_event_handler(handle_events) .add_systems( XrFirst, ( @@ -282,65 +278,47 @@ impl OxrInitPlugin { #[derive(Event, Clone, Copy, Debug, Default)] pub struct OxrInteractionProfileChanged; -/// Polls any OpenXR events and handles them accordingly -pub fn poll_events( - instance: Res, +pub fn handle_events( + event: In, mut status: ResMut, mut changed_event: EventWriter, mut interaction_profile_changed_event: EventWriter, - mut overlay_writer: Option>>, ) { - let _span = info_span!("xr_poll_events"); - let mut buffer = Default::default(); - while let Some(event) = instance - .poll_event(&mut buffer) - .expect("Failed to poll event") - { - use openxr::Event::*; - match event { - SessionStateChanged(state) => { - use openxr::SessionState; + use openxr::Event::*; + // this unwrap will never panic since we are in a valid scope + match unsafe { event.get() }.unwrap() { + SessionStateChanged(state) => { + use openxr::SessionState; - let state = state.state(); + let state = state.state(); - info!("entered XR state {:?}", state); + info!("entered XR state {:?}", state); - let new_status = match state { - SessionState::IDLE => XrState::Idle, - SessionState::READY => XrState::Ready, - SessionState::SYNCHRONIZED | SessionState::VISIBLE | SessionState::FOCUSED => { - XrState::Running - } - SessionState::STOPPING => XrState::Stopping, - SessionState::EXITING => XrState::Exiting { - should_restart: false, - }, - SessionState::LOSS_PENDING => XrState::Exiting { - should_restart: true, - }, - _ => unreachable!(), - }; - changed_event.send(XrStateChanged(new_status)); - *status = new_status; - } - InstanceLossPending(_) => {} - EventsLost(e) => warn!("lost {} XR events", e.lost_event_count()), - MainSessionVisibilityChangedEXTX(d) => { - if let Some(writer) = overlay_writer.as_mut() { - writer.send(OxrOverlaySessionEvent::MainSessionVisibilityChanged { - visible: d.visible(), - flags: d.flags(), - }); - } else { - warn!("Overlay Event Recieved without the OverlayPlugin being added!"); + let new_status = match state { + SessionState::IDLE => XrState::Idle, + SessionState::READY => XrState::Ready, + SessionState::SYNCHRONIZED | SessionState::VISIBLE | SessionState::FOCUSED => { + XrState::Running } - } - // we might want to check if this is the correct session? - Event::InteractionProfileChanged(_) => { - interaction_profile_changed_event.send_default(); - } - _ => {} + SessionState::STOPPING => XrState::Stopping, + SessionState::EXITING => XrState::Exiting { + should_restart: false, + }, + SessionState::LOSS_PENDING => XrState::Exiting { + should_restart: true, + }, + _ => unreachable!(), + }; + changed_event.send(XrStateChanged(new_status)); + *status = new_status; } + InstanceLossPending(_) => {} + EventsLost(e) => warn!("lost {} XR events", e.lost_event_count()), + // we might want to check if this is the correct session? + Event::InteractionProfileChanged(_) => { + interaction_profile_changed_event.send_default(); + } + _ => {} } } diff --git a/crates/bevy_openxr/src/openxr/mod.rs b/crates/bevy_openxr/src/openxr/mod.rs index d01c0a1..85c5c81 100644 --- a/crates/bevy_openxr/src/openxr/mod.rs +++ b/crates/bevy_openxr/src/openxr/mod.rs @@ -9,6 +9,7 @@ use bevy::{ use bevy_mod_xr::session::XrSessionPlugin; use bevy_mod_xr::{camera::XrCameraPlugin, session::XrState}; use init::OxrInitPlugin; +use poll_events::OxrEventsPlugin; use render::OxrRenderPlugin; use resources::OxrInstance; use session::OxrSession; @@ -29,6 +30,7 @@ pub mod helper_traits; pub mod init; pub mod layer_builder; pub mod next_chain; +pub mod poll_events; pub mod reference_space; pub mod render; pub mod resources; @@ -60,6 +62,7 @@ pub fn add_xr_plugins(plugins: G) -> PluginGroupBuilder { // .disable::() .add_before::(XrSessionPlugin { auto_handle: true }) .add_before::(OxrInitPlugin::default()) + .add(OxrEventsPlugin) .add(OxrReferenceSpacePlugin::default()) .add(OxrRenderPlugin) .add(OxrPassthroughPlugin) diff --git a/crates/bevy_openxr/src/openxr/poll_events.rs b/crates/bevy_openxr/src/openxr/poll_events.rs new file mode 100644 index 0000000..4b4d696 --- /dev/null +++ b/crates/bevy_openxr/src/openxr/poll_events.rs @@ -0,0 +1,88 @@ +use std::{cell::RefCell, mem, ops::Deref, rc::Rc}; + +use bevy::{ecs::system::SystemId, prelude::*}; +use bevy_mod_xr::session::{XrFirst, XrHandleEvents}; +use openxr::{Event, EventDataBuffer}; + +pub struct OxrEventsPlugin; + +impl Plugin for OxrEventsPlugin { + fn build(&self, app: &mut App) { + app.init_resource::(); + app.add_systems( + XrFirst, + poll_events + .in_set(XrHandleEvents::Poll) + .run_if(openxr_session_available), + ); + } +} +/// Polls any OpenXR events and handles them accordingly +pub fn poll_events(world: &mut World) { + let _span = info_span!("xr_poll_events"); + let instance = world.resource::().clone(); + let handlers = world.remove_resource::().unwrap(); + let mut buffer = EventDataBuffer::default(); + while let Some(event) = instance + .poll_event(&mut buffer) + .expect("Failed to poll event") + { + let event = Rc::new(RefCell::new(Some(event))); + for handler in handlers.handlers.iter() { + if let Err(err) = + world.run_system_with_input::<_, ()>(*handler, OxrEvent::new(event.clone())) + { + error!("error when running oxr event handler: {err}"); + }; + } + event.deref().take(); + } + world.insert_resource(handlers); +} + +use super::{openxr_session_available, resources::OxrInstance}; +#[derive(Resource, Debug, Default)] +pub struct OxrEventHandlers { + pub handlers: Vec, +} +pub type OxrEventHandler = SystemId; + +pub struct OxrEvent { + event: Rc>>>, +} + +impl OxrEvent { + pub(crate) fn new<'a>(event: Rc>>>) -> Self { + Self { + event: unsafe { mem::transmute(event) }, + } + } + /// always returns [Some] if called in a valid scope + /// # Safety + /// The event is only valid for the duration of the poll event callback, + /// don't Store the [Event] anywhere!! + #[allow(clippy::needless_lifetimes)] + pub unsafe fn get<'a>(&'a self) -> Option> { + self.event.borrow().clone() + } +} +pub trait OxrEventHandlerExt { + fn add_oxr_event_handler( + &mut self, + system: impl IntoSystem + 'static, + ) -> &mut Self; +} +impl OxrEventHandlerExt for App { + fn add_oxr_event_handler( + &mut self, + system: impl IntoSystem + 'static, + ) -> &mut Self { + self.init_resource::(); + let id = self.register_system(system); + self.world_mut() + .resource_mut::() + .handlers + .push(id); + self + } +}