maplibre-rs/maplibre/src/map_schedule.rs

use std::{cell::RefCell, marker::PhantomData, mem, rc::Rc};

use crate::{
    context::{MapContext, ViewState},
    coords::{LatLon, WorldCoords, Zoom, TILE_SIZE},
    error::Error,
    io::{
        scheduler::Scheduler,
        source_client::{HttpClient, HttpSourceClient},
        tile_repository::TileRepository,
    },
    render::{create_default_render_graph, register_default_render_stages},
    schedule::{Schedule, Stage},
    stages::register_stages,
    style::Style,
    Environment, HeadedMapWindow, MapWindowConfig, Renderer, RendererSettings, WgpuSettings,
    WindowSize,
};

/// Stores the state of the map, dispatches tile fetching and caching, tessellation and drawing.
pub struct InteractiveMapSchedule<E: Environment> {
    map_window_config: E::MapWindowConfig,

    // FIXME (wasm-executor): avoid RefCell, change ownership model
    pub apc: Rc<RefCell<E::AsyncProcedureCall>>,

    map_context: EventuallyMapContext,

    schedule: Schedule,

    suspended: bool,
}

impl<E: Environment> InteractiveMapSchedule<E> {
    pub fn new(
        map_window_config: E::MapWindowConfig,
        window_size: WindowSize,
        renderer: Option<Renderer>,
        scheduler: E::Scheduler, // TODO: unused
        apc: E::AsyncProcedureCall,
        http_client: E::HttpClient,
        style: Style,
        wgpu_settings: WgpuSettings,
        renderer_settings: RendererSettings,
    ) -> Self {
        let zoom = style.zoom.map(|zoom| Zoom::new(zoom)).unwrap_or_default();
        let position = style
            .center
            .map(|center| WorldCoords::from_lat_lon(LatLon::new(center[0], center[1]), zoom))
            .unwrap_or_default();
        let pitch = style.pitch.unwrap_or_default();
        let view_state = ViewState::new(&window_size, position, zoom, pitch, cgmath::Deg(110.0));

        let tile_repository = TileRepository::new();
        let mut schedule = Schedule::default();

        let apc = Rc::new(RefCell::new(apc));

        let http_source_client: HttpSourceClient<E::HttpClient> =
            HttpSourceClient::new(http_client);
        register_stages::<E>(&mut schedule, http_source_client, apc.clone());

        let graph = create_default_render_graph().unwrap(); // TODO: Remove unwrap
        register_default_render_stages(graph, &mut schedule);

        Self {
            apc,
            map_window_config,
            map_context: match renderer {
                None => EventuallyMapContext::Premature(PrematureMapContext {
                    view_state,
                    style,
                    tile_repository,
                    wgpu_settings,
                    renderer_settings,
                }),
                Some(renderer) => EventuallyMapContext::Full(MapContext {
                    view_state,
                    style,
                    tile_repository,
                    renderer,
                }),
            },
            schedule,
            suspended: false,
        }
    }

    #[tracing::instrument(name = "update_and_redraw", skip_all)]
    pub fn update_and_redraw(&mut self) -> Result<(), Error> {
        if self.suspended {
            return Ok(());
        }

        if let EventuallyMapContext::Full(map_context) = &mut self.map_context {
            self.schedule.run(map_context)
        }

        Ok(())
    }

    pub fn resize(&mut self, width: u32, height: u32) {
        if let EventuallyMapContext::Full(map_context) = &mut self.map_context {
            let view_state = &mut map_context.view_state;
            view_state.perspective.resize(width, height);
            view_state.camera.resize(width, height);

            map_context.renderer.resize(width, height)
        }
    }

    pub fn is_initialized(&self) -> bool {
        match &self.map_context {
            EventuallyMapContext::Full(_) => true,
            _ => false,
        }
    }

    pub fn view_state_mut(&mut self) -> &mut ViewState {
        match &mut self.map_context {
            EventuallyMapContext::Full(MapContext { view_state, .. }) => view_state,
            EventuallyMapContext::Premature(PrematureMapContext { view_state, .. }) => view_state,
            _ => panic!("should not happen"),
        }
    }

    pub fn apc(&self) -> &Rc<RefCell<E::AsyncProcedureCall>> {
        &self.apc
    }
}

impl<E: Environment> InteractiveMapSchedule<E>
where
    <E::MapWindowConfig as MapWindowConfig>::MapWindow: HeadedMapWindow,
{
    pub fn suspend(&mut self) {
        self.suspended = true;
    }

    pub fn resume(&mut self, window: &<E::MapWindowConfig as MapWindowConfig>::MapWindow) {
        if let EventuallyMapContext::Full(map_context) = &mut self.map_context {
            let renderer = &mut map_context.renderer;
            renderer.state.recreate_surface(window, &renderer.instance);
            self.suspended = false;
        }
    }

    pub async fn late_init(&mut self) -> bool {
        match &self.map_context {
            EventuallyMapContext::Full(_) => false,
            EventuallyMapContext::Premature(PrematureMapContext {
                wgpu_settings,
                renderer_settings,
                ..
            }) => {
                let window = self.map_window_config.create();
                let renderer =
                    Renderer::initialize(&window, wgpu_settings.clone(), renderer_settings.clone())
                        .await
                        .unwrap(); // TODO: Remove unwrap
                self.map_context.make_full(renderer);
                true
            }
            EventuallyMapContext::_Uninitialized => false,
        }
    }
}

pub struct PrematureMapContext {
    view_state: ViewState,
    style: Style,

    tile_repository: TileRepository,

    wgpu_settings: WgpuSettings,
    renderer_settings: RendererSettings,
}

pub enum EventuallyMapContext {
    Full(MapContext),
    Premature(PrematureMapContext),
    _Uninitialized,
}

impl EventuallyMapContext {
    pub fn make_full(&mut self, renderer: Renderer) {
        let context = mem::replace(self, EventuallyMapContext::_Uninitialized);

        match context {
            EventuallyMapContext::Full(_) => {}
            EventuallyMapContext::Premature(PrematureMapContext {
                view_state,
                style,
                tile_repository,
                ..
            }) => {
                *self = EventuallyMapContext::Full(MapContext {
                    view_state,
                    style,
                    tile_repository,
                    renderer,
                });
            }
            EventuallyMapContext::_Uninitialized => {}
        }
    }
}