wayvr/src/backend/openvr/mod.rs

use std::{
    collections::VecDeque,
    sync::{
        atomic::{AtomicBool, Ordering},
        Arc,
    },
    time::{Duration, Instant},
};

use ovr_overlay::{
    sys::{ETrackedDeviceProperty, EVRApplicationType, EVREventType},
    TrackedDeviceIndex,
};
use vulkano::{
    device::{physical::PhysicalDevice, DeviceExtensions},
    instance::InstanceExtensions,
    Handle, VulkanObject,
};

use crate::{
    backend::{
        input::interact,
        openvr::{input::OpenVrInputSource, lines::LinePool},
    },
    graphics::WlxGraphics,
    state::AppState,
};

use self::{input::action_manifest_path, overlay::OpenVrOverlayData};

use super::common::{BackendError, OverlayContainer, TaskType};

pub mod helpers;
pub mod input;
pub mod lines;
pub mod overlay;
pub mod playspace;

pub fn openvr_run(running: Arc<AtomicBool>) -> Result<(), BackendError> {
    let app_type = EVRApplicationType::VRApplication_Overlay;
    let Ok(context) = ovr_overlay::Context::init(app_type) else {
        log::warn!("Will not use OpenVR: Context init failed");
        return Err(BackendError::NotSupported);
    };

    log::info!("Using OpenVR runtime");

    let mut overlay_mngr = context.overlay_mngr();
    //let mut settings_mngr = context.settings_mngr();
    let mut input_mngr = context.input_mngr();
    let mut system_mngr = context.system_mngr();
    let mut chaperone_mgr = context.chaperone_setup_mngr();
    let mut compositor_mngr = context.compositor_mngr();

    let device_extensions_fn = |device: &PhysicalDevice| {
        let names = compositor_mngr.get_vulkan_device_extensions_required(device.handle().as_raw());
        let ext = DeviceExtensions::from_iter(names.iter().map(|s| s.as_str()));
        ext
    };

    let mut compositor_mngr = context.compositor_mngr();
    let instance_extensions = {
        let names = compositor_mngr.get_vulkan_instance_extensions_required();
        InstanceExtensions::from_iter(names.iter().map(|s| s.as_str()))
    };

    let mut state = {
        let graphics = WlxGraphics::new_openvr(instance_extensions, device_extensions_fn);
        AppState::from_graphics(graphics)
    };

    let mut overlays = OverlayContainer::<OpenVrOverlayData>::new(&mut state);

    let mut space_mover = playspace::PlayspaceMover::new();

    state.hid_provider.set_desktop_extent(overlays.extent);

    if let Err(e) = input_mngr.set_action_manifest(action_manifest_path()) {
        log::error!("Failed to set action manifest: {}", e.description());
        return Err(BackendError::Fatal);
    };

    let Ok(mut input_source) = OpenVrInputSource::new(&mut input_mngr) else {
        log::error!("Failed to initialize input");
        return Err(BackendError::Fatal);
    };

    let Ok(refresh_rate) = system_mngr.get_tracked_device_property::<f32>(
        TrackedDeviceIndex::HMD,
        ETrackedDeviceProperty::Prop_DisplayFrequency_Float,
    ) else {
        log::error!("Failed to get display refresh rate");
        return Err(BackendError::Fatal);
    };

    log::info!("HMD running @ {} Hz", refresh_rate);

    let frame_time = (1000.0 / refresh_rate).floor() * 0.001;
    let mut next_device_update = Instant::now();
    let mut due_tasks = VecDeque::with_capacity(4);

    let mut lines = LinePool::new(state.graphics.clone());
    let pointer_lines = [
        lines.allocate(&mut overlay_mngr, &mut state),
        lines.allocate(&mut overlay_mngr, &mut state),
    ];

    'main_loop: loop {
        if !running.load(Ordering::Relaxed) {
            log::warn!("Received shutdown signal.");
            break 'main_loop;
        }

        while let Some(event) = system_mngr.poll_next_event() {
            match event.event_type {
                EVREventType::VREvent_Quit => {
                    log::warn!("Received quit event, shutting down.");
                    break 'main_loop;
                }
                EVREventType::VREvent_TrackedDeviceActivated
                | EVREventType::VREvent_TrackedDeviceDeactivated
                | EVREventType::VREvent_TrackedDeviceUpdated => {
                    next_device_update = Instant::now();
                }
                _ => {}
            }
        }

        if next_device_update <= Instant::now() {
            input_source.update_devices(&mut system_mngr, &mut state);
            next_device_update = Instant::now() + Duration::from_secs(30);
        }

        state.tasks.retrieve_due(&mut due_tasks);
        while let Some(task) = due_tasks.pop_front() {
            match task {
                TaskType::Global(f) => f(&mut state),
                TaskType::Overlay(sel, f) => {
                    if let Some(o) = overlays.mut_by_selector(&sel) {
                        f(&mut state, &mut o.state);
                    }
                }
                TaskType::Toast(t) => {
                    // TODO toasts
                    log::info!("Toast: {} {}", t.title, t.body);
                }
            }
        }

        state.input_state.pre_update();
        input_source.update(&mut input_mngr, &mut system_mngr, &mut state);
        state.input_state.post_update();

        if state
            .input_state
            .pointers
            .iter()
            .any(|p| p.now.show_hide && !p.before.show_hide)
        {
            overlays.show_hide(&mut state);
        }

        overlays
            .iter_mut()
            .for_each(|o| o.state.auto_movement(&mut state));

        space_mover.update(&mut chaperone_mgr, &mut overlays, &state);

        let lengths_haptics = interact(&mut overlays, &mut state);
        for (idx, (len, haptics)) in lengths_haptics.iter().enumerate() {
            lines.draw_from(
                pointer_lines[idx],
                state.input_state.pointers[idx].pose,
                *len,
                state.input_state.pointers[idx].interaction.mode as usize + 1,
                &state.input_state.hmd,
            );
            if let Some(haptics) = haptics {
                input_source.haptics(&mut input_mngr, idx, haptics)
            }
        }

        lines.update(&mut overlay_mngr, &mut state);

        overlays
            .iter_mut()
            .for_each(|o| o.after_input(&mut overlay_mngr, &mut state));

        log::debug!("Rendering frame");

        overlays
            .iter_mut()
            .filter(|o| o.state.want_visible)
            .for_each(|o| o.render(&mut state));

        log::debug!("Rendering overlays");

        overlays
            .iter_mut()
            .for_each(|o| o.after_render(&mut overlay_mngr, &state.graphics));

        // chaperone

        // close font handles?

        // playspace moved end frame

        state.hid_provider.on_new_frame();

        let mut seconds_since_vsync = 0f32;
        std::thread::sleep(Duration::from_secs_f32(
            if system_mngr.get_time_since_last_vsync(&mut seconds_since_vsync, &mut 0u64) {
                frame_time - (seconds_since_vsync % frame_time)
            } else {
                frame_time
            },
        ));
    }

    log::warn!("OpenVR shutdown");
    // context.shutdown() called by Drop

    Ok(())
}