1#![deny(unsafe_code)]
2#![allow(clippy::type_complexity)]
3
4mod fps_monitor;
5mod hit_path_tracker;
6mod shell_debug;
7mod shell_frame;
8mod shell_input;
9#[cfg(test)]
10use shell_frame::build_draw_refresh_scope;
11
12pub use fps_monitor::FpsStats;
13
14use std::fmt::{Debug, Write};
15use std::rc::Rc;
16use std::sync::{
17 atomic::{AtomicBool, Ordering},
18 Mutex, MutexGuard,
19};
20use web_time::Instant;
22
23use cranpose_core::{
24 enter_event_handler_scope, location_key, run_in_mutable_snapshot, Applier, Composition, Key,
25 MemoryApplier, NodeError, NodeId,
26};
27use cranpose_foundation::{PointerButton, PointerButtons, PointerEvent, PointerEventKind};
28use cranpose_render_common::{HitTestTarget, RenderScene, Renderer};
29use cranpose_runtime_std::StdRuntime;
30use cranpose_ui::{
31 clear_transient_scroll_motion_contexts, format_layout_tree, format_render_scene,
32 format_screen_summary, has_pending_focus_invalidations, has_pending_pointer_repasses,
33 peek_focus_invalidation, peek_layout_invalidation, peek_pointer_invalidation,
34 peek_render_invalidation, process_focus_invalidations, process_pointer_repasses,
35 request_render_invalidation, take_draw_repass_nodes, take_focus_invalidation,
36 take_layout_invalidation, take_pointer_invalidation, take_render_invalidation,
37 HeadlessRenderer, LayoutBox, LayoutNode, LayoutTree, MeasureLayoutOptions, SemanticsTree,
38 SubcomposeLayoutNode,
39};
40use cranpose_ui_graphics::{Point, Rect, Size};
41use hit_path_tracker::{HitPathTracker, PointerId};
42use std::collections::HashSet;
43
44pub use cranpose_ui::{KeyCode, KeyEvent, KeyEventType, Modifiers};
46pub use cranpose_foundation::PointerSource;
48
49#[cfg(all(
54 feature = "clipboard-native",
55 not(target_arch = "wasm32"),
56 not(target_os = "android"),
57 not(target_os = "ios")
58))]
59struct ShellClipboard {
60 inner: std::rc::Rc<std::cell::RefCell<Option<arboard::Clipboard>>>,
61}
62
63#[cfg(all(
64 feature = "clipboard-native",
65 not(target_arch = "wasm32"),
66 not(target_os = "android"),
67 not(target_os = "ios")
68))]
69impl cranpose_ui::clipboard_session::PlatformClipboard for ShellClipboard {
70 fn write_text(&self, text: &str) {
71 if let Some(clipboard) = self.inner.borrow_mut().as_mut() {
72 let _ = clipboard.set_text(text);
73 }
74 }
75
76 fn read_text(&self) -> Option<String> {
77 self.inner
78 .borrow_mut()
79 .as_mut()
80 .and_then(|clipboard| clipboard.get_text().ok())
81 }
82}
83pub use cranpose_ui::PlatformTextInputHandler;
85pub use cranpose_ui::ImeEditorState;
87
88#[cfg(any(test, feature = "test-support"))]
89use cranpose_core::{
90 debug_recompose_scope_registry_stats, MemoryApplierDebugStats,
91 RecomposeScopeRegistryDebugStats, SlotTableDebugStats,
92};
93#[cfg(any(test, feature = "test-support"))]
94use cranpose_core::{
95 runtime::{RuntimeDebugStats, StateArenaDebugStats},
96 snapshot_pinning::{debug_snapshot_pinning_stats, SnapshotPinningDebugStats},
97 snapshot_state_observer::SnapshotStateObserverDebugStats,
98 snapshot_v2::{debug_snapshot_v2_stats, SnapshotV2DebugStats},
99 CompositionPassDebugStats, SlotId,
100};
101
102pub struct AppShell<R>
103where
104 R: Renderer,
105{
106 app_context: Rc<cranpose_ui::AppContext>,
107 runtime: StdRuntime,
108 composition: Composition<MemoryApplier>,
109 content: Box<dyn FnMut()>,
110 renderer: R,
111 cursor: (f32, f32),
112 viewport: (f32, f32),
113 buffer_size: (u32, u32),
114 start_time: Instant,
115 last_frame_time_nanos: u64,
116 layout_tree: Option<LayoutTree>,
117 semantics_tree: Option<SemanticsTree>,
118 semantics_enabled: bool,
119 layout_requested: bool,
120 force_layout_pass: bool,
121 scene_dirty: bool,
122 scoped_layout_scene_nodes: Vec<NodeId>,
123 is_dirty: bool,
124 buttons_pressed: PointerButtons,
126 pointer_source: PointerSource,
132 hit_path_tracker: HitPathTracker,
139 hovered_nodes: Vec<NodeId>,
142 #[cfg(all(
144 feature = "clipboard-native",
145 not(target_arch = "wasm32"),
146 not(target_os = "android"),
147 not(target_os = "ios")
148 ))]
149 clipboard: Option<arboard::Clipboard>,
150 dev_options: DevOptions,
152 dev_overlay_controls: Vec<DevOverlayControl>,
153 dev_overlay_text: String,
154 dev_overlay_last_refresh: Option<Instant>,
155 dev_overlay_viewport: Option<Size>,
156 fps_monitor: fps_monitor::FpsMonitor,
157 frame_scheduler: FrameScheduler,
158}
159
160fn update_stage_telemetry_threshold_ms() -> Option<f64> {
161 static THRESHOLD_MS: std::sync::OnceLock<Option<f64>> = std::sync::OnceLock::new();
162 *THRESHOLD_MS.get_or_init(|| {
163 std::env::var("CRANPOSE_UPDATE_STAGE_TELEMETRY_MS")
164 .ok()
165 .and_then(|value| value.parse::<f64>().ok())
166 .filter(|value| value.is_finite() && *value >= 0.0)
167 })
168}
169
170#[derive(Clone, Copy, Debug)]
171struct UpdateStageTelemetry {
172 started_at: Instant,
173 after_frame_callbacks: Instant,
174 after_ui_drain: Instant,
175 after_reconcile: Instant,
176 after_process_frame: Instant,
177 should_render: bool,
178 reconcile_attempted: bool,
179 reconcile_changed: bool,
180}
181
182fn log_update_stage_telemetry(telemetry: UpdateStageTelemetry) {
183 let Some(threshold_ms) = update_stage_telemetry_threshold_ms() else {
184 return;
185 };
186 let total_ms = telemetry
187 .after_process_frame
188 .duration_since(telemetry.started_at)
189 .as_secs_f64()
190 * 1000.0;
191 if total_ms < threshold_ms {
192 return;
193 }
194
195 let frame_callbacks_ms = telemetry
196 .after_frame_callbacks
197 .duration_since(telemetry.started_at)
198 .as_secs_f64()
199 * 1000.0;
200 let ui_drain_ms = telemetry
201 .after_ui_drain
202 .duration_since(telemetry.after_frame_callbacks)
203 .as_secs_f64()
204 * 1000.0;
205 let reconcile_ms = telemetry
206 .after_reconcile
207 .duration_since(telemetry.after_ui_drain)
208 .as_secs_f64()
209 * 1000.0;
210 let process_frame_ms = telemetry
211 .after_process_frame
212 .duration_since(telemetry.after_reconcile)
213 .as_secs_f64()
214 * 1000.0;
215 eprintln!(
216 "[update-stage-telemetry] total_ms={total_ms:.2} frame_callbacks_ms={frame_callbacks_ms:.2} ui_drain_ms={ui_drain_ms:.2} reconcile_ms={reconcile_ms:.2} process_frame_ms={process_frame_ms:.2} should_render={} reconcile_attempted={} reconcile_changed={}",
217 telemetry.should_render,
218 telemetry.reconcile_attempted,
219 telemetry.reconcile_changed
220 );
221}
222
223#[derive(Clone, Copy, Debug, Default, Eq, PartialEq)]
224pub enum FramePacingMode {
225 #[default]
228 Vsync,
229 Hard60,
230 Hard120,
231 NoVsync,
234}
235
236impl FramePacingMode {
237 pub const ALL: [Self; 4] = [Self::Vsync, Self::Hard60, Self::Hard120, Self::NoVsync];
238
239 pub fn label(self) -> &'static str {
240 match self {
241 Self::Vsync => "VSync",
242 Self::Hard60 => "60fps",
243 Self::Hard120 => "120fps",
244 Self::NoVsync => "NoVSync",
245 }
246 }
247
248 pub fn target_fps(self) -> Option<u32> {
249 match self {
250 Self::Hard60 => Some(60),
251 Self::Hard120 => Some(120),
252 Self::Vsync | Self::NoVsync => None,
253 }
254 }
255}
256
257#[derive(Clone, Copy, Debug, PartialEq)]
258pub struct FrameSchedule {
259 pub needs_update: bool,
260 pub needs_frame: bool,
261 pub next_deadline: Option<web_time::Instant>,
262}
263
264#[derive(Clone, Copy, Debug, Default, Eq, PartialEq)]
265pub struct FrameUpdateResult {
266 pub visual_changed: bool,
267 pub structure_changed: bool,
268}
269
270pub trait PlatformFrameDriver {
271 fn request_frame(&self);
272 fn request_wake_at(&self, deadline: web_time::Instant);
273 fn clear_wake(&self);
274}
275
276#[derive(Debug)]
277pub struct FrameScheduler {
278 update_pending: AtomicBool,
279 frame_pending: AtomicBool,
280 next_deadline: Mutex<Option<web_time::Instant>>,
281}
282
283impl Default for FrameScheduler {
284 fn default() -> Self {
285 Self {
286 update_pending: AtomicBool::new(false),
287 frame_pending: AtomicBool::new(false),
288 next_deadline: Mutex::new(None),
289 }
290 }
291}
292
293impl FrameScheduler {
294 fn lock_deadline(&self) -> MutexGuard<'_, Option<web_time::Instant>> {
295 self.next_deadline
296 .lock()
297 .unwrap_or_else(|poisoned| poisoned.into_inner())
298 }
299
300 pub fn record(&self, schedule: FrameSchedule) {
301 self.update_pending
302 .store(schedule.needs_update, Ordering::SeqCst);
303 self.frame_pending
304 .store(schedule.needs_frame, Ordering::SeqCst);
305 let mut next_deadline = self.lock_deadline();
306 *next_deadline = if schedule.needs_update {
307 None
308 } else {
309 schedule.next_deadline
310 };
311 }
312
313 pub fn schedule<D>(&self, schedule: FrameSchedule, driver: &D)
314 where
315 D: PlatformFrameDriver + ?Sized,
316 {
317 self.record(schedule);
318 schedule.apply_to(driver);
319 }
320
321 pub fn snapshot(&self) -> FrameSchedule {
322 FrameSchedule {
323 needs_update: self.update_pending.load(Ordering::SeqCst),
324 needs_frame: self.frame_pending.load(Ordering::SeqCst),
325 next_deadline: *self.lock_deadline(),
326 }
327 }
328}
329
330impl FrameSchedule {
331 pub fn apply_to<D>(self, driver: &D)
332 where
333 D: PlatformFrameDriver + ?Sized,
334 {
335 if self.needs_frame {
336 driver.clear_wake();
337 driver.request_frame();
338 } else if self.needs_update {
339 driver.request_wake_at(web_time::Instant::now());
340 } else if let Some(deadline) = self.next_deadline {
341 driver.request_wake_at(deadline);
342 } else {
343 driver.clear_wake();
344 }
345 }
346}
347
348#[derive(Clone, Copy, Debug)]
349struct DevOverlayControl {
350 bounds: Rect,
351 mode: FramePacingMode,
352}
353
354#[derive(Clone, Debug, Default)]
359pub struct DevOptions {
360 pub fps_counter: bool,
362 pub recomposition_counter: bool,
364 pub layout_timing: bool,
366 pub frame_pacing_controls: bool,
367 pub frame_pacing_mode: FramePacingMode,
368}
369
370#[cfg(any(test, feature = "test-support"))]
371#[doc(hidden)]
372#[derive(Clone, Copy, Debug)]
373pub struct RuntimeLeakDebugStats {
374 pub applier_stats: MemoryApplierDebugStats,
375 pub live_node_heap_bytes: usize,
376 pub recycled_node_heap_bytes: usize,
377 pub slot_table_heap_bytes: usize,
378 pub pass_stats: CompositionPassDebugStats,
379 pub slot_stats: SlotTableDebugStats,
380 pub observer_stats: SnapshotStateObserverDebugStats,
381 pub runtime_stats: RuntimeDebugStats,
382 pub state_arena_stats: StateArenaDebugStats,
383 pub recompose_scope_stats: RecomposeScopeRegistryDebugStats,
384 pub snapshot_v2_stats: SnapshotV2DebugStats,
385 pub snapshot_pinning_stats: SnapshotPinningDebugStats,
386}
387
388impl<R> AppShell<R>
389where
390 R: Renderer,
391 R::Error: Debug,
392{
393 pub fn new(renderer: R, root_key: Key, content: impl FnMut() + 'static) -> Self {
394 Self::new_with_size(renderer, root_key, content, (800, 600), (800.0, 600.0))
395 }
396
397 pub fn new_with_size(
398 renderer: R,
399 root_key: Key,
400 content: impl FnMut() + 'static,
401 buffer_size: (u32, u32),
402 viewport: (f32, f32),
403 ) -> Self {
404 Self::new_with_size_and_density(renderer, root_key, content, buffer_size, viewport, 1.0)
405 }
406
407 pub fn new_with_size_and_density(
408 mut renderer: R,
409 root_key: Key,
410 content: impl FnMut() + 'static,
411 buffer_size: (u32, u32),
412 viewport: (f32, f32),
413 density: f32,
414 ) -> Self {
415 let app_context = cranpose_ui::AppContext::new_with_density(density);
416 let runtime = StdRuntime::new();
417 let mut composition = Composition::with_runtime(MemoryApplier::new(), runtime.runtime());
418 let app_content = Rc::new(std::cell::RefCell::new(content));
424 let mut build: Box<dyn FnMut()> = Box::new(move || {
425 let app_content = Rc::clone(&app_content);
426 cranpose_ui::widgets::PopupHost(move || {
427 (app_content.borrow_mut())();
428 });
429 });
430 renderer.attach_app_context_services(&app_context);
431 app_context.enter(|| {
432 #[cfg(all(
435 feature = "clipboard-native",
436 not(target_arch = "wasm32"),
437 not(target_os = "android"),
438 not(target_os = "ios")
439 ))]
440 {
441 let clipboard =
442 std::rc::Rc::new(std::cell::RefCell::new(arboard::Clipboard::new().ok()));
443 cranpose_ui::clipboard_session::set_platform_clipboard(std::rc::Rc::new(
444 ShellClipboard { inner: clipboard },
445 ));
446 }
447 if let Err(err) = composition.render_stable(root_key, &mut *build) {
448 log::error!("initial render failed: {err}");
449 }
450 });
451 renderer.scene_mut().clear();
452 let mut shell = Self {
453 app_context,
454 runtime,
455 composition,
456 content: build,
457 renderer,
458 cursor: (0.0, 0.0),
459 viewport,
460 buffer_size,
461 start_time: Instant::now(),
462 last_frame_time_nanos: 0,
463 layout_tree: None,
464 semantics_tree: None,
465 semantics_enabled: false,
466 layout_requested: true,
467 force_layout_pass: true,
468 scene_dirty: true,
469 scoped_layout_scene_nodes: Vec::new(),
470 is_dirty: true,
471 buttons_pressed: PointerButtons::NONE,
472 pointer_source: PointerSource::Unknown,
473 hit_path_tracker: HitPathTracker::new(),
474 hovered_nodes: Vec::new(),
475 #[cfg(all(
476 feature = "clipboard-native",
477 not(target_arch = "wasm32"),
478 not(target_os = "android"),
479 not(target_os = "ios")
480 ))]
481 clipboard: arboard::Clipboard::new().ok(),
482 dev_options: DevOptions::default(),
483 dev_overlay_controls: Vec::new(),
484 dev_overlay_text: String::new(),
485 dev_overlay_last_refresh: None,
486 dev_overlay_viewport: None,
487 fps_monitor: fps_monitor::FpsMonitor::new(),
488 frame_scheduler: FrameScheduler::default(),
489 };
490 shell.process_frame();
491 shell
492 }
493
494 pub fn app_context(&self) -> &Rc<cranpose_ui::AppContext> {
499 &self.app_context
500 }
501
502 pub fn set_dev_options(&mut self, options: DevOptions) {
507 self.dev_options = options;
508 self.invalidate_dev_overlay_text();
509 let app_context = Rc::clone(&self.app_context);
510 app_context.enter(request_render_invalidation);
511 self.mark_dirty();
512 }
513
514 pub fn dev_options(&self) -> &DevOptions {
516 &self.dev_options
517 }
518
519 pub fn frame_pacing_mode(&self) -> FramePacingMode {
520 self.dev_options.frame_pacing_mode
521 }
522
523 pub fn current_fps(&self) -> f32 {
524 self.fps_monitor.current_fps()
525 }
526
527 pub fn fps_stats(&self) -> FpsStats {
528 self.fps_monitor.stats()
529 }
530
531 pub fn reset_fps_stats(&mut self) {
532 self.fps_monitor.reset_stats();
533 self.invalidate_dev_overlay_text();
534 }
535
536 pub fn record_presented_frame(
537 &mut self,
538 frame_started_at: Instant,
539 frame_finished_at: Instant,
540 ) {
541 self.fps_monitor
542 .record_frame_work(frame_started_at, frame_finished_at);
543 }
544
545 #[cfg(any(test, feature = "test-support"))]
546 #[doc(hidden)]
547 pub fn record_presented_frame_for_test(
548 &mut self,
549 frame_started_nanos: u64,
550 frame_finished_nanos: u64,
551 ) {
552 let started = self.start_time + std::time::Duration::from_nanos(frame_started_nanos);
553 let finished = self.start_time + std::time::Duration::from_nanos(frame_finished_nanos);
554 self.record_presented_frame(started, finished);
555 }
556
557 pub fn set_frame_pacing_mode(&mut self, mode: FramePacingMode) {
558 if self.dev_options.frame_pacing_mode == mode {
559 return;
560 }
561 self.dev_options.frame_pacing_mode = mode;
562 self.invalidate_dev_overlay_text();
563 let app_context = Rc::clone(&self.app_context);
564 app_context.enter(request_render_invalidation);
565 self.mark_dirty();
566 }
567
568 pub fn handle_dev_overlay_click(&mut self, x: f32, y: f32) -> Option<FramePacingMode> {
569 if !self.dev_options.frame_pacing_controls {
570 return None;
571 }
572 let mode = self
573 .dev_overlay_controls
574 .iter()
575 .find(|control| control.bounds.contains(x, y))
576 .map(|control| control.mode)?;
577 self.set_frame_pacing_mode(mode);
578 Some(mode)
579 }
580
581 fn invalidate_dev_overlay_text(&mut self) {
582 self.dev_overlay_text.clear();
583 self.dev_overlay_last_refresh = None;
584 self.dev_overlay_viewport = None;
585 }
586
587 pub fn set_viewport(&mut self, width: f32, height: f32) {
588 self.viewport = (width, height);
589 self.request_forced_layout_pass();
590 self.mark_dirty();
591 self.process_frame();
592 }
593
594 pub fn viewport_size(&self) -> (f32, f32) {
595 self.viewport
596 }
597
598 pub fn set_buffer_size(&mut self, width: u32, height: u32) {
599 self.buffer_size = (width, height);
600 }
601
602 pub fn buffer_size(&self) -> (u32, u32) {
603 self.buffer_size
604 }
605
606 pub fn scene(&self) -> &R::Scene {
607 self.renderer.scene()
608 }
609
610 pub fn renderer(&mut self) -> &mut R {
611 &mut self.renderer
612 }
613
614 #[cfg(not(target_arch = "wasm32"))]
615 pub fn set_frame_waker(&mut self, waker: impl Fn() + Send + Sync + 'static) {
616 self.runtime.set_frame_waker(waker);
617 }
618
619 #[cfg(target_arch = "wasm32")]
620 pub fn set_frame_waker(&mut self, waker: impl Fn() + 'static) {
621 self.runtime.set_frame_waker(waker);
622 }
623
624 pub fn clear_frame_waker(&mut self) {
625 self.runtime.clear_frame_waker();
626 }
627
628 pub fn should_render(&self) -> bool {
629 let app_context = Rc::clone(&self.app_context);
630 app_context.enter(|| {
631 if self.layout_requested
632 || self.scene_dirty
633 || peek_render_invalidation()
634 || peek_pointer_invalidation()
635 || peek_focus_invalidation()
636 || peek_layout_invalidation()
637 {
638 return true;
639 }
640 self.composition.should_render()
641 })
642 }
643
644 fn needs_ui_update_in_context(&self) -> bool {
645 if self.is_dirty
646 || self.layout_requested
647 || self.scene_dirty
648 || peek_render_invalidation()
649 || peek_pointer_invalidation()
650 || peek_focus_invalidation()
651 || peek_layout_invalidation()
652 || cranpose_ui::has_pending_layout_repasses()
653 || cranpose_ui::has_pending_measure_repasses()
654 || cranpose_ui::has_pending_draw_repasses()
655 || has_pending_pointer_repasses()
656 || has_pending_focus_invalidations()
657 {
658 return true;
659 }
660
661 self.composition.should_render()
662 }
663
664 pub fn needs_update(&self) -> bool {
665 let app_context = Rc::clone(&self.app_context);
666 app_context.enter(|| self.needs_ui_update_in_context())
667 }
668
669 pub fn needs_redraw(&self) -> bool {
672 let app_context = Rc::clone(&self.app_context);
673 app_context
674 .enter(|| self.needs_ui_update_in_context() || self.renderer.needs_frame_warmup())
675 }
676
677 pub fn mark_dirty(&mut self) {
679 self.is_dirty = true;
680 }
681
682 pub fn request_root_render(&mut self) {
683 self.composition.request_root_render();
684 self.request_forced_layout_pass();
685 let app_context = Rc::clone(&self.app_context);
686 app_context.enter(request_render_invalidation);
687 self.mark_dirty();
688 }
689
690 pub fn set_density(&mut self, density: f32) {
691 let app_context = Rc::clone(&self.app_context);
692 let changed = app_context.enter(|| {
693 let previous = cranpose_ui::current_density().to_bits();
694 cranpose_ui::set_density(density);
695 previous != cranpose_ui::current_density().to_bits()
696 });
697 if changed {
698 self.request_forced_layout_pass();
699 self.mark_dirty();
700 }
701 }
702
703 #[cfg(any(test, feature = "test-support"))]
704 #[doc(hidden)]
705 pub fn debug_current_density(&self) -> f32 {
706 let app_context = Rc::clone(&self.app_context);
707 app_context.enter(cranpose_ui::current_density)
708 }
709
710 #[cfg(any(test, feature = "test-support"))]
711 #[doc(hidden)]
712 pub fn debug_enter_app_context<T>(&self, block: impl FnOnce() -> T) -> T {
713 let app_context = Rc::clone(&self.app_context);
714 app_context.enter(block)
715 }
716
717 fn request_layout_pass(&mut self) {
718 self.layout_requested = true;
719 }
720
721 fn request_forced_layout_pass(&mut self) {
722 self.layout_requested = true;
723 self.force_layout_pass = true;
724 }
725
726 fn composition_tree_needs_layout(&mut self) -> bool {
727 let Some(root) = self.composition.root() else {
728 return true;
729 };
730 let mut applier = self.composition.applier_mut();
731 cranpose_ui::tree_needs_layout(&mut *applier, root).unwrap_or_else(|err| {
732 log::warn!(
733 "Cannot check layout dirty status for root #{}: {}",
734 root,
735 err
736 );
737 true
738 })
739 }
740
741 pub fn has_active_animations(&self) -> bool {
743 self.composition.should_render()
744 }
745
746 pub fn has_active_pointer_gesture(&self) -> bool {
747 self.buttons_pressed != PointerButtons::NONE
748 && self.hit_path_tracker.has_path(PointerId::PRIMARY)
749 }
750
751 pub fn next_event_time(&self) -> Option<web_time::Instant> {
754 let app_context = Rc::clone(&self.app_context);
755 app_context.enter(cranpose_ui::next_cursor_blink_time)
756 }
757
758 fn compute_frame_schedule(&self) -> FrameSchedule {
759 let needs_update = self.needs_update();
760 let needs_frame = needs_update
761 || self.has_active_animations()
762 || self.has_active_pointer_gesture()
763 || self.renderer.needs_frame_warmup();
764 FrameSchedule {
765 needs_update,
766 needs_frame,
767 next_deadline: self.next_event_time(),
768 }
769 }
770
771 pub fn frame_schedule(&self) -> FrameSchedule {
772 let schedule = self.compute_frame_schedule();
773 self.frame_scheduler.record(schedule);
774 schedule
775 }
776
777 pub fn schedule_platform_frame<D>(&self, driver: &D) -> FrameSchedule
778 where
779 D: PlatformFrameDriver + ?Sized,
780 {
781 let schedule = self.compute_frame_schedule();
782 self.frame_scheduler.schedule(schedule, driver);
783 schedule
784 }
785
786 pub fn frame_scheduler_snapshot(&self) -> FrameSchedule {
787 self.frame_scheduler.snapshot()
788 }
789
790 fn frame_time_nanos_at(&self, now: Instant) -> u64 {
791 now.checked_duration_since(self.start_time)
792 .unwrap_or_default()
793 .as_nanos()
794 .min(u128::from(u64::MAX)) as u64
795 }
796
797 pub fn update_after_frame_interval(
798 &mut self,
799 frame_interval: std::time::Duration,
800 ) -> FrameUpdateResult {
801 let wall_frame_time = self.frame_time_nanos_at(Instant::now());
802 let base_frame_time = self.last_frame_time_nanos.max(wall_frame_time);
803 let frame_time = base_frame_time
804 .saturating_add(frame_interval.as_nanos().min(u128::from(u64::MAX)) as u64);
805 self.update_at_frame_time_nanos(frame_time)
806 }
807
808 pub fn update_at_frame_time_nanos(&mut self, frame_time: u64) -> FrameUpdateResult {
809 let app_context = Rc::clone(&self.app_context);
810 app_context.enter(|| {
811 let update_started_at = Instant::now();
812 let frame_time = frame_time.max(self.last_frame_time_nanos);
813 self.last_frame_time_nanos = frame_time;
814 let runtime_handle = self.runtime.runtime_handle();
815 runtime_handle.with_deferred_state_releases(|| {
816 self.runtime.drain_frame_callbacks(frame_time);
817 let after_frame_callbacks = Instant::now();
818 runtime_handle.drain_ui();
819 let after_ui_drain = Instant::now();
820 let should_render = self.composition.should_render();
821 let mut reconcile_attempted = false;
822 let mut reconcile_changed = false;
823 if should_render {
824 log::trace!(
825 target: "cranpose::input",
826 "update begin: should_render=true layout_requested={} scene_dirty={} is_dirty={}",
827 self.layout_requested,
828 self.scene_dirty,
829 self.is_dirty
830 );
831 }
832 if should_render {
833 let Some(root_key) = self.composition.root_key() else {
834 let result = self.process_frame_in_context(reconcile_changed);
835 let after_process_frame = Instant::now();
836 log_update_stage_telemetry(UpdateStageTelemetry {
837 started_at: update_started_at,
838 after_frame_callbacks,
839 after_ui_drain,
840 after_reconcile: after_ui_drain,
841 after_process_frame,
842 should_render,
843 reconcile_attempted,
844 reconcile_changed,
845 });
846 self.is_dirty = false;
847 return result;
848 };
849 reconcile_attempted = true;
850 match self.composition.reconcile(root_key, &mut *self.content) {
851 Ok(changed) => {
852 reconcile_changed = changed;
853 log::trace!(
854 target: "cranpose::input",
855 "reconcile changed={changed}"
856 );
857 if changed {
858 self.fps_monitor.record_recomposition();
859 if self.composition_tree_needs_layout() {
860 self.request_layout_pass();
861 }
862 request_render_invalidation();
863 }
864 }
865 Err(NodeError::Missing { id }) => {
866 log::debug!("Recomposition skipped: node {} no longer exists", id);
867 self.request_layout_pass();
868 request_render_invalidation();
869 }
870 Err(err) => {
871 log::error!("recomposition failed: {err}");
872 self.request_layout_pass();
873 request_render_invalidation();
874 }
875 }
876 }
877 let after_reconcile = Instant::now();
878 let result = self.process_frame_in_context(reconcile_changed);
879 let after_process_frame = Instant::now();
880 log_update_stage_telemetry(UpdateStageTelemetry {
881 started_at: update_started_at,
882 after_frame_callbacks,
883 after_ui_drain,
884 after_reconcile,
885 after_process_frame,
886 should_render,
887 reconcile_attempted,
888 reconcile_changed,
889 });
890 self.is_dirty = false;
891 result
892 })
893 })
894 }
895
896 pub fn update(&mut self) -> FrameUpdateResult {
897 let frame_time = self.frame_time_nanos_at(Instant::now());
898 self.update_at_frame_time_nanos(frame_time)
899 }
900}
901
902impl<R> Drop for AppShell<R>
903where
904 R: Renderer,
905{
906 fn drop(&mut self) {
907 self.runtime.clear_frame_waker();
908 }
909}
910
911pub fn default_root_key() -> Key {
912 location_key(file!(), line!(), column!())
913}
914
915#[cfg(test)]
916mod frame_pacing_tests {
917 use super::{FramePacingMode, FrameSchedule, FrameScheduler, PlatformFrameDriver};
918 use std::cell::RefCell;
919 use std::panic::{catch_unwind, AssertUnwindSafe};
920 use std::time::Duration;
921 use web_time::Instant;
922
923 #[derive(Clone, Copy, Debug, PartialEq)]
924 enum DriverCall {
925 RequestFrame,
926 RequestWakeAt(Instant),
927 ClearWake,
928 }
929
930 #[derive(Default)]
931 struct RecordingFrameDriver {
932 calls: RefCell<Vec<DriverCall>>,
933 }
934
935 impl RecordingFrameDriver {
936 fn calls(&self) -> Vec<DriverCall> {
937 self.calls.borrow().clone()
938 }
939 }
940
941 impl PlatformFrameDriver for RecordingFrameDriver {
942 fn request_frame(&self) {
943 self.calls.borrow_mut().push(DriverCall::RequestFrame);
944 }
945
946 fn request_wake_at(&self, deadline: Instant) {
947 self.calls
948 .borrow_mut()
949 .push(DriverCall::RequestWakeAt(deadline));
950 }
951
952 fn clear_wake(&self) {
953 self.calls.borrow_mut().push(DriverCall::ClearWake);
954 }
955 }
956
957 #[test]
958 fn frame_pacing_labels_match_overlay_modes() {
959 assert_eq!(FramePacingMode::Vsync.label(), "VSync");
960 assert_eq!(FramePacingMode::Hard60.label(), "60fps");
961 assert_eq!(FramePacingMode::Hard120.label(), "120fps");
962 assert_eq!(FramePacingMode::NoVsync.label(), "NoVSync");
963 }
964
965 #[test]
966 fn only_hard_modes_have_fixed_targets() {
967 assert_eq!(FramePacingMode::Vsync.target_fps(), None);
968 assert_eq!(FramePacingMode::Hard60.target_fps(), Some(60));
969 assert_eq!(FramePacingMode::Hard120.target_fps(), Some(120));
970 assert_eq!(FramePacingMode::NoVsync.target_fps(), None);
971 }
972
973 #[test]
974 fn frame_schedule_requests_immediate_frame_and_clears_deadline() {
975 let driver = RecordingFrameDriver::default();
976 let deadline = Instant::now() + Duration::from_millis(25);
977
978 FrameSchedule {
979 needs_update: true,
980 needs_frame: true,
981 next_deadline: Some(deadline),
982 }
983 .apply_to(&driver);
984
985 assert_eq!(
986 driver.calls(),
987 vec![DriverCall::ClearWake, DriverCall::RequestFrame]
988 );
989 }
990
991 #[test]
992 fn frame_schedule_requests_deadline_when_idle_until_timer() {
993 let driver = RecordingFrameDriver::default();
994 let deadline = Instant::now() + Duration::from_millis(25);
995
996 FrameSchedule {
997 needs_update: false,
998 needs_frame: false,
999 next_deadline: Some(deadline),
1000 }
1001 .apply_to(&driver);
1002
1003 assert_eq!(driver.calls(), vec![DriverCall::RequestWakeAt(deadline)]);
1004 }
1005
1006 #[test]
1007 fn frame_schedule_wakes_without_requesting_frame_for_update_only_work() {
1008 let driver = RecordingFrameDriver::default();
1009 let before = Instant::now();
1010
1011 FrameSchedule {
1012 needs_update: true,
1013 needs_frame: false,
1014 next_deadline: None,
1015 }
1016 .apply_to(&driver);
1017
1018 let calls = driver.calls();
1019 assert_eq!(calls.len(), 1);
1020 match calls[0] {
1021 DriverCall::RequestWakeAt(deadline) => {
1022 assert!(deadline >= before);
1023 }
1024 other => panic!("update-only work must wake without requesting a frame: {other:?}"),
1025 }
1026 }
1027
1028 #[test]
1029 fn frame_schedule_clears_wake_when_fully_idle() {
1030 let driver = RecordingFrameDriver::default();
1031
1032 FrameSchedule {
1033 needs_update: false,
1034 needs_frame: false,
1035 next_deadline: None,
1036 }
1037 .apply_to(&driver);
1038
1039 assert_eq!(driver.calls(), vec![DriverCall::ClearWake]);
1040 }
1041
1042 #[test]
1043 fn frame_scheduler_records_latest_schedule_and_applies_driver() {
1044 let scheduler = FrameScheduler::default();
1045 let driver = RecordingFrameDriver::default();
1046 let deadline = Instant::now() + Duration::from_millis(25);
1047
1048 scheduler.schedule(
1049 FrameSchedule {
1050 needs_update: false,
1051 needs_frame: false,
1052 next_deadline: Some(deadline),
1053 },
1054 &driver,
1055 );
1056
1057 assert_eq!(
1058 scheduler.snapshot(),
1059 FrameSchedule {
1060 needs_update: false,
1061 needs_frame: false,
1062 next_deadline: Some(deadline),
1063 }
1064 );
1065 assert_eq!(driver.calls(), vec![DriverCall::RequestWakeAt(deadline)]);
1066 }
1067
1068 #[test]
1069 fn frame_scheduler_clears_deadline_for_immediate_frame() {
1070 let scheduler = FrameScheduler::default();
1071 let driver = RecordingFrameDriver::default();
1072 let deadline = Instant::now() + Duration::from_millis(25);
1073
1074 scheduler.schedule(
1075 FrameSchedule {
1076 needs_update: true,
1077 needs_frame: true,
1078 next_deadline: Some(deadline),
1079 },
1080 &driver,
1081 );
1082
1083 assert_eq!(
1084 scheduler.snapshot(),
1085 FrameSchedule {
1086 needs_update: true,
1087 needs_frame: true,
1088 next_deadline: None,
1089 }
1090 );
1091 assert_eq!(
1092 driver.calls(),
1093 vec![DriverCall::ClearWake, DriverCall::RequestFrame]
1094 );
1095 }
1096
1097 #[test]
1098 fn frame_scheduler_recovers_poisoned_deadline_lock() {
1099 let scheduler = FrameScheduler::default();
1100 let deadline = Instant::now() + Duration::from_millis(25);
1101
1102 let _ = catch_unwind(AssertUnwindSafe(|| {
1103 let _guard = scheduler.lock_deadline();
1104 panic!("poison frame scheduler deadline lock");
1105 }));
1106
1107 scheduler.record(FrameSchedule {
1108 needs_update: false,
1109 needs_frame: false,
1110 next_deadline: Some(deadline),
1111 });
1112
1113 assert_eq!(
1114 scheduler.snapshot(),
1115 FrameSchedule {
1116 needs_update: false,
1117 needs_frame: false,
1118 next_deadline: Some(deadline),
1119 }
1120 );
1121 }
1122}
1123
1124#[cfg(test)]
1125#[path = "tests/app_shell_tests.rs"]
1126mod tests;