use crate::{ app::{AppContext, MutableAppContext, WindowInvalidation}, elements::Element, font_cache::FontCache, geometry::rect::RectF, json::{self, ToJson}, keymap::Keystroke, platform::{CursorStyle, Event}, scene::{CursorRegion, MouseRegionEvent}, text_layout::TextLayoutCache, Action, AnyModelHandle, AnyViewHandle, AnyWeakModelHandle, AssetCache, ElementBox, Entity, FontSystem, ModelHandle, MouseButtonEvent, MouseMovedEvent, MouseRegion, MouseRegionId, ReadModel, ReadView, RenderContext, RenderParams, Scene, UpgradeModelHandle, UpgradeViewHandle, View, ViewHandle, WeakModelHandle, WeakViewHandle, }; use collections::{HashMap, HashSet}; use pathfinder_geometry::vector::{vec2f, Vector2F}; use serde_json::json; use smallvec::SmallVec; use std::{ marker::PhantomData, ops::{Deref, DerefMut, Range}, sync::Arc, }; pub struct Presenter { window_id: usize, pub(crate) rendered_views: HashMap, parents: HashMap, cursor_regions: Vec, mouse_regions: Vec<(MouseRegion, usize)>, font_cache: Arc, text_layout_cache: TextLayoutCache, asset_cache: Arc, last_mouse_moved_event: Option, hovered_region_ids: HashSet, clicked_region: Option, right_clicked_region: Option, prev_drag_position: Option, titlebar_height: f32, } impl Presenter { pub fn new( window_id: usize, titlebar_height: f32, font_cache: Arc, text_layout_cache: TextLayoutCache, asset_cache: Arc, cx: &mut MutableAppContext, ) -> Self { Self { window_id, rendered_views: cx.render_views(window_id, titlebar_height), parents: Default::default(), cursor_regions: Default::default(), mouse_regions: Default::default(), font_cache, text_layout_cache, asset_cache, last_mouse_moved_event: None, hovered_region_ids: Default::default(), clicked_region: None, right_clicked_region: None, prev_drag_position: None, titlebar_height, } } pub fn dispatch_path(&self, app: &AppContext) -> Vec { let mut path = Vec::new(); if let Some(view_id) = app.focused_view_id(self.window_id) { self.compute_dispatch_path_from(view_id, &mut path) } path } pub(crate) fn compute_dispatch_path_from(&self, mut view_id: usize, path: &mut Vec) { path.push(view_id); while let Some(parent_id) = self.parents.get(&view_id).copied() { path.push(parent_id); view_id = parent_id; } path.reverse(); } pub fn invalidate( &mut self, invalidation: &mut WindowInvalidation, cx: &mut MutableAppContext, ) { cx.start_frame(); for view_id in &invalidation.removed { invalidation.updated.remove(&view_id); self.rendered_views.remove(&view_id); self.parents.remove(&view_id); } for view_id in &invalidation.updated { self.rendered_views.insert( *view_id, cx.render_view(RenderParams { window_id: self.window_id, view_id: *view_id, titlebar_height: self.titlebar_height, hovered_region_ids: self.hovered_region_ids.clone(), clicked_region_id: self.clicked_region.as_ref().and_then(MouseRegion::id), right_clicked_region_id: self .right_clicked_region .as_ref() .and_then(MouseRegion::id), refreshing: false, }) .unwrap(), ); } } pub fn refresh(&mut self, invalidation: &mut WindowInvalidation, cx: &mut MutableAppContext) { self.invalidate(invalidation, cx); for (view_id, view) in &mut self.rendered_views { if !invalidation.updated.contains(view_id) { *view = cx .render_view(RenderParams { window_id: self.window_id, view_id: *view_id, titlebar_height: self.titlebar_height, hovered_region_ids: self.hovered_region_ids.clone(), clicked_region_id: self.clicked_region.as_ref().and_then(MouseRegion::id), right_clicked_region_id: self .right_clicked_region .as_ref() .and_then(MouseRegion::id), refreshing: true, }) .unwrap(); } } } pub fn build_scene( &mut self, window_size: Vector2F, scale_factor: f32, refreshing: bool, cx: &mut MutableAppContext, ) -> Scene { let mut scene = Scene::new(scale_factor); if let Some(root_view_id) = cx.root_view_id(self.window_id) { self.layout(window_size, refreshing, cx); let mut paint_cx = self.build_paint_context(&mut scene, window_size, cx); paint_cx.paint( root_view_id, Vector2F::zero(), RectF::new(Vector2F::zero(), window_size), ); self.text_layout_cache.finish_frame(); self.cursor_regions = scene.cursor_regions(); self.mouse_regions = scene.mouse_regions(); if cx.window_is_active(self.window_id) { if let Some(event) = self.last_mouse_moved_event.clone() { let mut invalidated_views = Vec::new(); self.handle_hover_events(&event, &mut invalidated_views, cx); for view_id in invalidated_views { cx.notify_view(self.window_id, view_id); } } } } else { log::error!("could not find root_view_id for window {}", self.window_id); } scene } fn layout(&mut self, window_size: Vector2F, refreshing: bool, cx: &mut MutableAppContext) { if let Some(root_view_id) = cx.root_view_id(self.window_id) { self.build_layout_context(window_size, refreshing, cx) .layout(root_view_id, SizeConstraint::strict(window_size)); } } pub fn build_layout_context<'a>( &'a mut self, window_size: Vector2F, refreshing: bool, cx: &'a mut MutableAppContext, ) -> LayoutContext<'a> { LayoutContext { window_id: self.window_id, rendered_views: &mut self.rendered_views, parents: &mut self.parents, font_cache: &self.font_cache, font_system: cx.platform().fonts(), text_layout_cache: &self.text_layout_cache, asset_cache: &self.asset_cache, view_stack: Vec::new(), refreshing, hovered_region_ids: self.hovered_region_ids.clone(), clicked_region_id: self.clicked_region.as_ref().and_then(MouseRegion::id), right_clicked_region_id: self.right_clicked_region.as_ref().and_then(MouseRegion::id), titlebar_height: self.titlebar_height, window_size, app: cx, } } pub fn build_paint_context<'a>( &'a mut self, scene: &'a mut Scene, window_size: Vector2F, cx: &'a mut MutableAppContext, ) -> PaintContext { PaintContext { scene, window_size, font_cache: &self.font_cache, text_layout_cache: &self.text_layout_cache, rendered_views: &mut self.rendered_views, view_stack: Vec::new(), app: cx, } } pub fn rect_for_text_range(&self, range_utf16: Range, cx: &AppContext) -> Option { cx.focused_view_id(self.window_id).and_then(|view_id| { let cx = MeasurementContext { app: cx, rendered_views: &self.rendered_views, window_id: self.window_id, }; cx.rect_for_text_range(view_id, range_utf16) }) } pub fn dispatch_event(&mut self, event: Event, cx: &mut MutableAppContext) -> bool { if let Some(root_view_id) = cx.root_view_id(self.window_id) { let mut invalidated_views = Vec::new(); let mut mouse_down_out_handlers = Vec::new(); let mut mouse_down_region = None; let mut clicked_region = None; let mut dragged_region = None; match &event { Event::MouseDown( e @ MouseButtonEvent { position, button, .. }, ) => { let mut hit = false; for (region, _) in self.mouse_regions.iter().rev() { if region.bounds.contains_point(*position) { if !hit { hit = true; invalidated_views.push(region.view_id); mouse_down_region = Some((region.clone(), MouseRegionEvent::Down(e.clone()))); self.clicked_region = Some(region.clone()); self.prev_drag_position = Some(*position); } } else if let Some(handler) = region .handlers .get(&(MouseRegionEvent::down_out_disc(), Some(*button))) { mouse_down_out_handlers.push(( handler, region.view_id, MouseRegionEvent::DownOut(e.clone()), )); } } } Event::MouseUp(e @ MouseButtonEvent { position, .. }) => { self.prev_drag_position.take(); if let Some(region) = self.clicked_region.take() { invalidated_views.push(region.view_id); if region.bounds.contains_point(*position) { clicked_region = Some((region, MouseRegionEvent::Click(e.clone()))); } } } Event::MouseMoved(e @ MouseMovedEvent { position, .. }) => { if let Some((clicked_region, prev_drag_position)) = self .clicked_region .as_ref() .zip(self.prev_drag_position.as_mut()) { dragged_region = Some(( clicked_region.clone(), MouseRegionEvent::Drag(*prev_drag_position, e.clone()), )); *prev_drag_position = *position; } self.last_mouse_moved_event = Some(event.clone()); } _ => {} } let (mut handled, mut event_cx) = self.handle_hover_events(&event, &mut invalidated_views, cx); for (handler, view_id, region_event) in mouse_down_out_handlers { event_cx.with_current_view(view_id, |event_cx| handler(region_event, event_cx)) } if let Some((mouse_down_region, region_event)) = mouse_down_region { handled = true; if let Some(mouse_down_callback) = mouse_down_region.handlers.get(®ion_event.handler_key()) { event_cx.with_current_view(mouse_down_region.view_id, |event_cx| { mouse_down_callback(region_event, event_cx); }) } } if let Some((clicked_region, region_event)) = clicked_region { handled = true; if let Some(click_callback) = clicked_region.handlers.get(®ion_event.handler_key()) { event_cx.with_current_view(clicked_region.view_id, |event_cx| { click_callback(region_event, event_cx); }) } } if let Some((dragged_region, region_event)) = dragged_region { handled = true; if let Some(drag_callback) = dragged_region.handlers.get(®ion_event.handler_key()) { event_cx.with_current_view(dragged_region.view_id, |event_cx| { drag_callback(region_event, event_cx); }) } } if !handled { handled = event_cx.dispatch_event(root_view_id, &event); } invalidated_views.extend(event_cx.invalidated_views); let dispatch_directives = event_cx.dispatched_actions; for view_id in invalidated_views { cx.notify_view(self.window_id, view_id); } let mut dispatch_path = Vec::new(); for directive in dispatch_directives { dispatch_path.clear(); if let Some(view_id) = directive.dispatcher_view_id { self.compute_dispatch_path_from(view_id, &mut dispatch_path); } cx.dispatch_action_any(self.window_id, &dispatch_path, directive.action.as_ref()); } handled } else { false } } fn handle_hover_events<'a>( &'a mut self, event: &Event, invalidated_views: &mut Vec, cx: &'a mut MutableAppContext, ) -> (bool, EventContext<'a>) { let mut hover_regions = Vec::new(); // let mut unhovered_regions = Vec::new(); // let mut hovered_regions = Vec::new(); if let Event::MouseMoved( e @ MouseMovedEvent { position, pressed_button, .. }, ) = event { if let None = pressed_button { let mut style_to_assign = CursorStyle::Arrow; for region in self.cursor_regions.iter().rev() { if region.bounds.contains_point(*position) { style_to_assign = region.style; break; } } cx.platform().set_cursor_style(style_to_assign); let mut hover_depth = None; for (region, depth) in self.mouse_regions.iter().rev() { if region.bounds.contains_point(*position) && hover_depth.map_or(true, |hover_depth| hover_depth == *depth) { hover_depth = Some(*depth); if let Some(region_id) = region.id() { if !self.hovered_region_ids.contains(®ion_id) { invalidated_views.push(region.view_id); hover_regions.push(( region.clone(), MouseRegionEvent::Hover(true, e.clone()), )); self.hovered_region_ids.insert(region_id); } } } else { if let Some(region_id) = region.id() { if self.hovered_region_ids.contains(®ion_id) { invalidated_views.push(region.view_id); hover_regions.push(( region.clone(), MouseRegionEvent::Hover(false, e.clone()), )); self.hovered_region_ids.remove(®ion_id); } } } } } } let mut event_cx = self.build_event_context(cx); let mut handled = false; for (hover_region, region_event) in hover_regions { handled = true; if let Some(hover_callback) = hover_region.handlers.get(®ion_event.handler_key()) { event_cx.with_current_view(hover_region.view_id, |event_cx| { hover_callback(region_event, event_cx); }) } } (handled, event_cx) } pub fn build_event_context<'a>( &'a mut self, cx: &'a mut MutableAppContext, ) -> EventContext<'a> { EventContext { rendered_views: &mut self.rendered_views, dispatched_actions: Default::default(), font_cache: &self.font_cache, text_layout_cache: &self.text_layout_cache, view_stack: Default::default(), invalidated_views: Default::default(), notify_count: 0, window_id: self.window_id, app: cx, } } pub fn debug_elements(&self, cx: &AppContext) -> Option { let view = cx.root_view(self.window_id)?; Some(json!({ "root_view": view.debug_json(cx), "root_element": self.rendered_views.get(&view.id()) .map(|root_element| { root_element.debug(&DebugContext { rendered_views: &self.rendered_views, font_cache: &self.font_cache, app: cx, }) }) })) } } pub struct DispatchDirective { pub dispatcher_view_id: Option, pub action: Box, } pub struct LayoutContext<'a> { window_id: usize, rendered_views: &'a mut HashMap, parents: &'a mut HashMap, view_stack: Vec, pub font_cache: &'a Arc, pub font_system: Arc, pub text_layout_cache: &'a TextLayoutCache, pub asset_cache: &'a AssetCache, pub app: &'a mut MutableAppContext, pub refreshing: bool, pub window_size: Vector2F, titlebar_height: f32, hovered_region_ids: HashSet, clicked_region_id: Option, right_clicked_region_id: Option, } impl<'a> LayoutContext<'a> { pub(crate) fn keystrokes_for_action( &self, action: &dyn Action, ) -> Option> { self.app .keystrokes_for_action(self.window_id, &self.view_stack, action) } fn layout(&mut self, view_id: usize, constraint: SizeConstraint) -> Vector2F { if let Some(parent_id) = self.view_stack.last() { self.parents.insert(view_id, *parent_id); } self.view_stack.push(view_id); let mut rendered_view = self.rendered_views.remove(&view_id).unwrap(); let size = rendered_view.layout(constraint, self); self.rendered_views.insert(view_id, rendered_view); self.view_stack.pop(); size } pub fn render(&mut self, handle: &ViewHandle, f: F) -> T where F: FnOnce(&mut V, &mut RenderContext) -> T, V: View, { handle.update(self.app, |view, cx| { let mut render_cx = RenderContext { app: cx, window_id: handle.window_id(), view_id: handle.id(), view_type: PhantomData, titlebar_height: self.titlebar_height, hovered_region_ids: self.hovered_region_ids.clone(), clicked_region_id: self.clicked_region_id, right_clicked_region_id: self.right_clicked_region_id, refreshing: self.refreshing, }; f(view, &mut render_cx) }) } } impl<'a> Deref for LayoutContext<'a> { type Target = MutableAppContext; fn deref(&self) -> &Self::Target { self.app } } impl<'a> DerefMut for LayoutContext<'a> { fn deref_mut(&mut self) -> &mut Self::Target { self.app } } impl<'a> ReadView for LayoutContext<'a> { fn read_view(&self, handle: &ViewHandle) -> &T { self.app.read_view(handle) } } impl<'a> ReadModel for LayoutContext<'a> { fn read_model(&self, handle: &ModelHandle) -> &T { self.app.read_model(handle) } } impl<'a> UpgradeModelHandle for LayoutContext<'a> { fn upgrade_model_handle( &self, handle: &WeakModelHandle, ) -> Option> { self.app.upgrade_model_handle(handle) } fn model_handle_is_upgradable(&self, handle: &WeakModelHandle) -> bool { self.app.model_handle_is_upgradable(handle) } fn upgrade_any_model_handle(&self, handle: &AnyWeakModelHandle) -> Option { self.app.upgrade_any_model_handle(handle) } } impl<'a> UpgradeViewHandle for LayoutContext<'a> { fn upgrade_view_handle(&self, handle: &WeakViewHandle) -> Option> { self.app.upgrade_view_handle(handle) } fn upgrade_any_view_handle(&self, handle: &crate::AnyWeakViewHandle) -> Option { self.app.upgrade_any_view_handle(handle) } } pub struct PaintContext<'a> { rendered_views: &'a mut HashMap, view_stack: Vec, pub window_size: Vector2F, pub scene: &'a mut Scene, pub font_cache: &'a FontCache, pub text_layout_cache: &'a TextLayoutCache, pub app: &'a AppContext, } impl<'a> PaintContext<'a> { fn paint(&mut self, view_id: usize, origin: Vector2F, visible_bounds: RectF) { if let Some(mut tree) = self.rendered_views.remove(&view_id) { self.view_stack.push(view_id); tree.paint(origin, visible_bounds, self); self.rendered_views.insert(view_id, tree); self.view_stack.pop(); } } #[inline] pub fn paint_layer(&mut self, clip_bounds: Option, f: F) where F: FnOnce(&mut Self) -> (), { self.scene.push_layer(clip_bounds); f(self); self.scene.pop_layer(); } pub fn current_view_id(&self) -> usize { *self.view_stack.last().unwrap() } } impl<'a> Deref for PaintContext<'a> { type Target = AppContext; fn deref(&self) -> &Self::Target { self.app } } pub struct EventContext<'a> { rendered_views: &'a mut HashMap, dispatched_actions: Vec, pub font_cache: &'a FontCache, pub text_layout_cache: &'a TextLayoutCache, pub app: &'a mut MutableAppContext, pub window_id: usize, pub notify_count: usize, view_stack: Vec, invalidated_views: HashSet, } impl<'a> EventContext<'a> { fn dispatch_event(&mut self, view_id: usize, event: &Event) -> bool { if let Some(mut element) = self.rendered_views.remove(&view_id) { let result = self.with_current_view(view_id, |this| element.dispatch_event(event, this)); self.rendered_views.insert(view_id, element); result } else { false } } fn with_current_view(&mut self, view_id: usize, f: F) -> T where F: FnOnce(&mut Self) -> T, { self.view_stack.push(view_id); let result = f(self); self.view_stack.pop(); result } pub fn window_id(&self) -> usize { self.window_id } pub fn view_id(&self) -> Option { self.view_stack.last().copied() } pub fn is_parent_view_focused(&self) -> bool { if let Some(parent_view_id) = self.view_stack.last() { self.app.focused_view_id(self.window_id) == Some(*parent_view_id) } else { false } } pub fn focus_parent_view(&mut self) { if let Some(parent_view_id) = self.view_stack.last() { self.app.focus(self.window_id, Some(*parent_view_id)) } } pub fn dispatch_any_action(&mut self, action: Box) { self.dispatched_actions.push(DispatchDirective { dispatcher_view_id: self.view_stack.last().copied(), action, }); } pub fn dispatch_action(&mut self, action: A) { self.dispatch_any_action(Box::new(action)); } pub fn notify(&mut self) { self.notify_count += 1; if let Some(view_id) = self.view_stack.last() { self.invalidated_views.insert(*view_id); } } pub fn notify_count(&self) -> usize { self.notify_count } } impl<'a> Deref for EventContext<'a> { type Target = MutableAppContext; fn deref(&self) -> &Self::Target { self.app } } impl<'a> DerefMut for EventContext<'a> { fn deref_mut(&mut self) -> &mut Self::Target { self.app } } pub struct MeasurementContext<'a> { app: &'a AppContext, rendered_views: &'a HashMap, pub window_id: usize, } impl<'a> Deref for MeasurementContext<'a> { type Target = AppContext; fn deref(&self) -> &Self::Target { self.app } } impl<'a> MeasurementContext<'a> { fn rect_for_text_range(&self, view_id: usize, range_utf16: Range) -> Option { let element = self.rendered_views.get(&view_id)?; element.rect_for_text_range(range_utf16, self) } } pub struct DebugContext<'a> { rendered_views: &'a HashMap, pub font_cache: &'a FontCache, pub app: &'a AppContext, } #[derive(Clone, Copy, Debug, Eq, PartialEq)] pub enum Axis { Horizontal, Vertical, } impl Axis { pub fn invert(self) -> Self { match self { Self::Horizontal => Self::Vertical, Self::Vertical => Self::Horizontal, } } } impl ToJson for Axis { fn to_json(&self) -> serde_json::Value { match self { Axis::Horizontal => json!("horizontal"), Axis::Vertical => json!("vertical"), } } } pub trait Vector2FExt { fn along(self, axis: Axis) -> f32; } impl Vector2FExt for Vector2F { fn along(self, axis: Axis) -> f32 { match axis { Axis::Horizontal => self.x(), Axis::Vertical => self.y(), } } } #[derive(Copy, Clone, Debug)] pub struct SizeConstraint { pub min: Vector2F, pub max: Vector2F, } impl SizeConstraint { pub fn new(min: Vector2F, max: Vector2F) -> Self { Self { min, max } } pub fn strict(size: Vector2F) -> Self { Self { min: size, max: size, } } pub fn strict_along(axis: Axis, max: f32) -> Self { match axis { Axis::Horizontal => Self { min: vec2f(max, 0.0), max: vec2f(max, f32::INFINITY), }, Axis::Vertical => Self { min: vec2f(0.0, max), max: vec2f(f32::INFINITY, max), }, } } pub fn max_along(&self, axis: Axis) -> f32 { match axis { Axis::Horizontal => self.max.x(), Axis::Vertical => self.max.y(), } } pub fn min_along(&self, axis: Axis) -> f32 { match axis { Axis::Horizontal => self.min.x(), Axis::Vertical => self.min.y(), } } pub fn constrain(&self, size: Vector2F) -> Vector2F { vec2f( size.x().min(self.max.x()).max(self.min.x()), size.y().min(self.max.y()).max(self.min.y()), ) } } impl Default for SizeConstraint { fn default() -> Self { SizeConstraint { min: Vector2F::zero(), max: Vector2F::splat(f32::INFINITY), } } } impl ToJson for SizeConstraint { fn to_json(&self) -> serde_json::Value { json!({ "min": self.min.to_json(), "max": self.max.to_json(), }) } } pub struct ChildView { view: AnyViewHandle, } impl ChildView { pub fn new(view: impl Into) -> Self { Self { view: view.into() } } } impl Element for ChildView { type LayoutState = (); type PaintState = (); fn layout( &mut self, constraint: SizeConstraint, cx: &mut LayoutContext, ) -> (Vector2F, Self::LayoutState) { let size = cx.layout(self.view.id(), constraint); (size, ()) } fn paint( &mut self, bounds: RectF, visible_bounds: RectF, _: &mut Self::LayoutState, cx: &mut PaintContext, ) -> Self::PaintState { cx.paint(self.view.id(), bounds.origin(), visible_bounds); } fn dispatch_event( &mut self, event: &Event, _: RectF, _: RectF, _: &mut Self::LayoutState, _: &mut Self::PaintState, cx: &mut EventContext, ) -> bool { cx.dispatch_event(self.view.id(), event) } fn rect_for_text_range( &self, range_utf16: Range, _: RectF, _: RectF, _: &Self::LayoutState, _: &Self::PaintState, cx: &MeasurementContext, ) -> Option { cx.rect_for_text_range(self.view.id(), range_utf16) } fn debug( &self, bounds: RectF, _: &Self::LayoutState, _: &Self::PaintState, cx: &DebugContext, ) -> serde_json::Value { json!({ "type": "ChildView", "view_id": self.view.id(), "bounds": bounds.to_json(), "view": self.view.debug_json(cx.app), "child": if let Some(view) = cx.rendered_views.get(&self.view.id()) { view.debug(cx) } else { json!(null) } }) } }