1pub mod core;
2pub mod policies;
3
4use std::{
5 cell::{Cell, RefCell},
6 fmt,
7 mem::size_of,
8 rc::Rc,
9 sync::OnceLock,
10};
11
12use cranpose_core::{
13 Applier, ApplierHost, Composer, ConcreteApplierHost, MemoryApplier, Node, NodeError, NodeId,
14 Phase, RuntimeHandle, SlotTable, SlotsHost, SnapshotStateObserver,
15};
16
17use self::core::Measurable;
18use self::core::Placeable;
19#[cfg(test)]
20use self::core::{HorizontalAlignment, VerticalAlignment};
21use crate::modifier::{
22 collect_semantics_from_modifier, DimensionConstraint, EdgeInsets, Modifier, ModifierNodeSlices,
23 ModifierNodeSlicesDebugStats, Point, Rect as GeometryRect, ResolvedModifiers, Size,
24};
25
26use crate::subcompose_layout::{CachedBatchMeasureInputs, SubcomposeLayoutNode};
27use crate::widgets::nodes::{IntrinsicKind, LayoutNode, LayoutNodeCacheHandles, LayoutState};
28use cranpose_foundation::{
29 text::TextRange, InvalidationKind, ModifierNodeContext, NodeCapabilities,
30 SemanticsConfiguration,
31};
32use cranpose_ui_layout::{Constraints, MeasurePolicy, Placement};
33use web_time::Instant;
34
35#[derive(Default)]
40pub(crate) struct LayoutNodeContext {
41 invalidations: Vec<InvalidationKind>,
42 update_requested: bool,
43 active_capabilities: Vec<NodeCapabilities>,
44}
45
46impl LayoutNodeContext {
47 pub(crate) fn new() -> Self {
48 Self::default()
49 }
50
51 pub(crate) fn take_invalidations(&mut self) -> Vec<InvalidationKind> {
52 std::mem::take(&mut self.invalidations)
53 }
54}
55
56impl ModifierNodeContext for LayoutNodeContext {
57 fn invalidate(&mut self, kind: InvalidationKind) {
58 if !self.invalidations.contains(&kind) {
59 self.invalidations.push(kind);
60 }
61 }
62
63 fn request_update(&mut self) {
64 self.update_requested = true;
65 }
66
67 fn push_active_capabilities(&mut self, capabilities: NodeCapabilities) {
68 self.active_capabilities.push(capabilities);
69 }
70
71 fn pop_active_capabilities(&mut self) {
72 self.active_capabilities.pop();
73 }
74}
75
76#[doc(hidden)]
99pub fn invalidate_all_layout_caches() {
100 crate::render_state::invalidate_layout_cache_epoch();
101}
102
103fn layout_measure_telemetry_threshold_ms() -> Option<f64> {
104 static THRESHOLD_MS: OnceLock<Option<f64>> = OnceLock::new();
105 *THRESHOLD_MS.get_or_init(|| {
106 std::env::var("CRANPOSE_LAYOUT_MEASURE_TELEMETRY_MS")
107 .ok()
108 .and_then(|value| value.parse::<f64>().ok())
109 .filter(|value| value.is_finite() && *value >= 0.0)
110 .or_else(|| {
111 std::env::var_os("CRANPOSE_LAYOUT_MEASURE_TELEMETRY")
112 .is_some()
113 .then_some(4.0)
114 })
115 })
116}
117
118struct LayoutMeasureTelemetry {
119 root: NodeId,
120 start: Instant,
121 after_repasses: Instant,
122 after_guard: Instant,
123 after_builder: Instant,
124 after_measure: Instant,
125 after_root_place: Instant,
126 after_aux: Instant,
127 after_builder_drop: Instant,
128 after_guard_drop: Instant,
129}
130
131fn log_layout_measure_telemetry(times: LayoutMeasureTelemetry) {
132 let Some(threshold_ms) = layout_measure_telemetry_threshold_ms() else {
133 return;
134 };
135
136 let total_ms = times
137 .after_guard_drop
138 .duration_since(times.start)
139 .as_secs_f64()
140 * 1000.0;
141 if total_ms < threshold_ms {
142 return;
143 }
144
145 let repass_ms = times
146 .after_repasses
147 .duration_since(times.start)
148 .as_secs_f64()
149 * 1000.0;
150 let guard_ms = times
151 .after_guard
152 .duration_since(times.after_repasses)
153 .as_secs_f64()
154 * 1000.0;
155 let builder_ms = times
156 .after_builder
157 .duration_since(times.after_guard)
158 .as_secs_f64()
159 * 1000.0;
160 let measure_ms = times
161 .after_measure
162 .duration_since(times.after_builder)
163 .as_secs_f64()
164 * 1000.0;
165 let root_place_ms = times
166 .after_root_place
167 .duration_since(times.after_measure)
168 .as_secs_f64()
169 * 1000.0;
170 let aux_ms = times
171 .after_aux
172 .duration_since(times.after_root_place)
173 .as_secs_f64()
174 * 1000.0;
175 let builder_drop_ms = times
176 .after_builder_drop
177 .duration_since(times.after_aux)
178 .as_secs_f64()
179 * 1000.0;
180 let guard_drop_ms = times
181 .after_guard_drop
182 .duration_since(times.after_builder_drop)
183 .as_secs_f64()
184 * 1000.0;
185 log::warn!(
186 "[layout-measure-telemetry] root={} total_ms={total_ms:.2} repass_ms={repass_ms:.2} guard_ms={guard_ms:.2} builder_ms={builder_ms:.2} measure_ms={measure_ms:.2} root_place_ms={root_place_ms:.2} aux_ms={aux_ms:.2} builder_drop_ms={builder_drop_ms:.2} guard_drop_ms={guard_drop_ms:.2}",
187 times.root
188 );
189}
190
191fn log_node_measure_telemetry(
192 kind: &'static str,
193 node_id: NodeId,
194 constraints: Constraints,
195 size: Size,
196 children: usize,
197 start: Instant,
198) {
199 let Some(threshold_ms) = layout_measure_telemetry_threshold_ms() else {
200 return;
201 };
202
203 let total_ms = start.elapsed().as_secs_f64() * 1000.0;
204 if total_ms < threshold_ms {
205 return;
206 }
207
208 log::warn!(
209 "[layout-node-telemetry] kind={kind} node={} total_ms={total_ms:.2} constraints=({:.1},{:.1},{:.1},{:.1}) size=({:.1},{:.1}) children={children}",
210 node_id,
211 constraints.min_width,
212 constraints.max_width,
213 constraints.min_height,
214 constraints.max_height,
215 size.width,
216 size.height,
217 );
218}
219
220struct ApplierSlotGuard<'a> {
231 target: &'a mut MemoryApplier,
233 host: Rc<ConcreteApplierHost<MemoryApplier>>,
235 slots: Rc<RefCell<SlotTable>>,
238}
239
240impl<'a> ApplierSlotGuard<'a> {
241 fn new(target: &'a mut MemoryApplier) -> Self {
245 let original_applier = std::mem::replace(target, MemoryApplier::new());
247 let host = Rc::new(ConcreteApplierHost::new(original_applier));
248
249 let slots = {
251 let mut applier_ref = host.borrow_typed();
252 std::mem::take(applier_ref.slots())
253 };
254 let slots = Rc::new(RefCell::new(slots));
255
256 Self {
257 target,
258 host,
259 slots,
260 }
261 }
262
263 fn host(&self) -> Rc<ConcreteApplierHost<MemoryApplier>> {
265 Rc::clone(&self.host)
266 }
267
268 fn slots_handle(&self) -> Rc<RefCell<SlotTable>> {
271 Rc::clone(&self.slots)
272 }
273}
274
275impl Drop for ApplierSlotGuard<'_> {
276 fn drop(&mut self) {
277 {
281 let mut applier_ref = self.host.borrow_typed();
282 *applier_ref.slots() = std::mem::take(&mut *self.slots.borrow_mut());
283 }
284
285 {
287 let mut applier_ref = self.host.borrow_typed();
288 let original_applier = std::mem::take(&mut *applier_ref);
289 let _ = std::mem::replace(self.target, original_applier);
290 }
291 }
293}
294
295struct ModifierChainMeasurement {
297 size: Size,
298 content_offset: Point,
300 offset: Point,
302}
303
304type LayoutModifierNodeData = (
305 usize,
306 Rc<RefCell<Box<dyn cranpose_foundation::ModifierNode>>>,
307);
308
309struct ScratchVecPool<T> {
310 available: Vec<Vec<T>>,
311}
312
313impl<T> ScratchVecPool<T> {
314 fn acquire(&mut self) -> Vec<T> {
315 self.available.pop().unwrap_or_default()
316 }
317
318 fn release(&mut self, mut values: Vec<T>) {
319 values.clear();
320 self.available.push(values);
321 }
322
323 #[cfg(test)]
324 fn available_count(&self) -> usize {
325 self.available.len()
326 }
327}
328
329impl<T> Default for ScratchVecPool<T> {
330 fn default() -> Self {
331 Self {
332 available: Vec::new(),
333 }
334 }
335}
336
337#[derive(Default)]
338pub(crate) struct FrameLayoutArena {
339 tmp_records: ScratchVecPool<(NodeId, ChildRecord)>,
340 tmp_child_ids: ScratchVecPool<NodeId>,
341 tmp_layout_node_data: ScratchVecPool<LayoutModifierNodeData>,
342 tmp_placements: ScratchVecPool<Placement>,
343}
344
345#[cfg(test)]
346impl FrameLayoutArena {
347 pub(crate) fn available_placement_scratch_count(&self) -> usize {
348 self.tmp_placements.available_count()
349 }
350
351 pub(crate) fn seed_placement_scratch_for_test(&mut self) {
352 self.tmp_placements.release(Vec::with_capacity(1));
353 }
354}
355
356#[derive(Clone, Debug, PartialEq, Eq)]
358pub struct SemanticsCallback {
359 node_id: NodeId,
360}
361
362impl SemanticsCallback {
363 pub fn new(node_id: NodeId) -> Self {
364 Self { node_id }
365 }
366
367 pub fn node_id(&self) -> NodeId {
368 self.node_id
369 }
370}
371
372#[derive(Clone, Debug, PartialEq, Eq)]
374pub enum SemanticsAction {
375 Click { handler: SemanticsCallback },
376}
377
378#[derive(Clone, Debug, PartialEq, Eq)]
381pub enum SemanticsRole {
382 Layout,
384 Subcompose,
386 Text { value: String },
388 Spacer,
390 Button,
392 Unknown,
394}
395
396#[derive(Clone, Debug, PartialEq, Eq)]
398pub struct SemanticsNode {
399 pub node_id: NodeId,
400 pub role: SemanticsRole,
401 pub actions: Vec<SemanticsAction>,
402 pub children: Vec<SemanticsNode>,
403 pub description: Option<String>,
404 pub editable_text: bool,
405 pub text_selection: Option<TextRange>,
406}
407
408impl SemanticsNode {
409 fn new(
410 node_id: NodeId,
411 role: SemanticsRole,
412 actions: Vec<SemanticsAction>,
413 children: Vec<SemanticsNode>,
414 description: Option<String>,
415 editable_text: bool,
416 text_selection: Option<TextRange>,
417 ) -> Self {
418 Self {
419 node_id,
420 role,
421 actions,
422 children,
423 description,
424 editable_text,
425 text_selection,
426 }
427 }
428}
429
430#[derive(Clone, Debug, PartialEq, Eq)]
432pub struct SemanticsTree {
433 root: SemanticsNode,
434}
435
436impl SemanticsTree {
437 fn new(root: SemanticsNode) -> Self {
438 Self { root }
439 }
440
441 pub fn root(&self) -> &SemanticsNode {
442 &self.root
443 }
444}
445
446#[derive(Clone, Copy, Debug, Default, PartialEq, Eq)]
447pub struct LayoutAllocationDebugStats {
448 pub layout_box_count: usize,
449 pub layout_box_child_count: usize,
450 pub layout_box_child_capacity: usize,
451 pub layout_box_heap_bytes: usize,
452 pub modifier_slice_count: usize,
453 pub modifier_slice_heap_bytes: usize,
454 pub modifier_draw_command_count: usize,
455 pub modifier_draw_command_capacity: usize,
456 pub modifier_pointer_input_count: usize,
457 pub modifier_pointer_input_capacity: usize,
458 pub modifier_click_handler_count: usize,
459 pub modifier_click_handler_capacity: usize,
460 pub modifier_text_content_count: usize,
461 pub modifier_text_style_count: usize,
462 pub modifier_text_layout_options_count: usize,
463 pub modifier_prepared_text_layout_count: usize,
464 pub modifier_graphics_layer_count: usize,
465 pub modifier_graphics_layer_resolver_count: usize,
466 pub semantics_node_count: usize,
467 pub semantics_action_count: usize,
468 pub semantics_action_capacity: usize,
469 pub semantics_child_count: usize,
470 pub semantics_child_capacity: usize,
471 pub semantics_description_count: usize,
472 pub semantics_description_bytes: usize,
473 pub semantics_text_role_bytes: usize,
474 pub semantics_heap_bytes: usize,
475}
476
477impl LayoutAllocationDebugStats {
478 fn add_modifier_slice(&mut self, stats: ModifierNodeSlicesDebugStats) {
479 self.modifier_slice_count += 1;
480 self.modifier_slice_heap_bytes += stats.heap_bytes;
481 self.modifier_draw_command_count += stats.draw_command_count;
482 self.modifier_draw_command_capacity += stats.draw_command_capacity;
483 self.modifier_pointer_input_count += stats.pointer_input_count;
484 self.modifier_pointer_input_capacity += stats.pointer_input_capacity;
485 self.modifier_click_handler_count += stats.click_handler_count;
486 self.modifier_click_handler_capacity += stats.click_handler_capacity;
487 self.modifier_text_content_count += usize::from(stats.has_text_content);
488 self.modifier_text_style_count += usize::from(stats.has_text_style);
489 self.modifier_text_layout_options_count += usize::from(stats.has_text_layout_options);
490 self.modifier_prepared_text_layout_count += usize::from(stats.has_prepared_text_layout);
491 self.modifier_graphics_layer_count += usize::from(stats.has_graphics_layer);
492 self.modifier_graphics_layer_resolver_count +=
493 usize::from(stats.has_graphics_layer_resolver);
494 }
495}
496
497#[derive(Debug, Clone)]
499pub struct LayoutTree {
500 root: LayoutBox,
501}
502
503impl LayoutTree {
504 pub fn new(root: LayoutBox) -> Self {
505 Self { root }
506 }
507
508 pub fn root(&self) -> &LayoutBox {
509 &self.root
510 }
511
512 pub fn root_mut(&mut self) -> &mut LayoutBox {
513 &mut self.root
514 }
515
516 pub fn into_root(self) -> LayoutBox {
517 self.root
518 }
519
520 pub fn debug_allocation_stats(&self) -> LayoutAllocationDebugStats {
521 let mut stats = LayoutAllocationDebugStats::default();
522 record_layout_box_allocation_stats(&self.root, &mut stats);
523 stats
524 }
525}
526
527#[derive(Debug, Clone)]
529pub struct LayoutBox {
530 pub node_id: NodeId,
531 pub rect: GeometryRect,
532 pub content_offset: Point,
534 pub node_data: LayoutNodeData,
535 pub children: Vec<LayoutBox>,
536}
537
538impl LayoutBox {
539 pub fn new(
540 node_id: NodeId,
541 rect: GeometryRect,
542 content_offset: Point,
543 node_data: LayoutNodeData,
544 children: Vec<LayoutBox>,
545 ) -> Self {
546 Self {
547 node_id,
548 rect,
549 content_offset,
550 node_data,
551 children,
552 }
553 }
554}
555
556#[derive(Debug, Clone)]
558pub struct LayoutNodeData {
559 pub modifier: Modifier,
560 pub resolved_modifiers: ResolvedModifiers,
561 pub modifier_slices: Rc<ModifierNodeSlices>,
562 pub kind: LayoutNodeKind,
563}
564
565impl LayoutNodeData {
566 pub fn new(
567 modifier: Modifier,
568 resolved_modifiers: ResolvedModifiers,
569 modifier_slices: Rc<ModifierNodeSlices>,
570 kind: LayoutNodeKind,
571 ) -> Self {
572 Self {
573 modifier,
574 resolved_modifiers,
575 modifier_slices,
576 kind,
577 }
578 }
579
580 pub fn resolved_modifiers(&self) -> ResolvedModifiers {
581 self.resolved_modifiers
582 }
583
584 pub fn modifier_slices(&self) -> &ModifierNodeSlices {
585 &self.modifier_slices
586 }
587}
588
589#[derive(Clone)]
596pub enum LayoutNodeKind {
597 Layout,
598 Subcompose,
599 Spacer,
600 Button { on_click: Rc<RefCell<dyn FnMut()>> },
601 Unknown,
602}
603
604impl fmt::Debug for LayoutNodeKind {
605 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
606 match self {
607 LayoutNodeKind::Layout => f.write_str("Layout"),
608 LayoutNodeKind::Subcompose => f.write_str("Subcompose"),
609 LayoutNodeKind::Spacer => f.write_str("Spacer"),
610 LayoutNodeKind::Button { .. } => f.write_str("Button"),
611 LayoutNodeKind::Unknown => f.write_str("Unknown"),
612 }
613 }
614}
615
616pub trait LayoutEngine {
618 fn compute_layout(&mut self, root: NodeId, max_size: Size) -> Result<LayoutTree, NodeError>;
619}
620
621impl LayoutEngine for MemoryApplier {
622 fn compute_layout(&mut self, root: NodeId, max_size: Size) -> Result<LayoutTree, NodeError> {
623 let measurements = measure_layout(self, root, max_size)?;
624 measurements
625 .into_layout_tree()
626 .ok_or(NodeError::MissingContext {
627 id: root,
628 reason: "layout tree was not requested",
629 })
630 }
631}
632
633#[derive(Debug, Clone)]
635pub struct LayoutMeasurements {
636 root: Rc<MeasuredNode>,
637 semantics: Option<SemanticsTree>,
638 layout_tree: Option<LayoutTree>,
639}
640
641impl LayoutMeasurements {
642 fn new(
643 root: Rc<MeasuredNode>,
644 semantics: Option<SemanticsTree>,
645 layout_tree: Option<LayoutTree>,
646 ) -> Self {
647 Self {
648 root,
649 semantics,
650 layout_tree,
651 }
652 }
653
654 pub fn root_size(&self) -> Size {
656 self.root.size
657 }
658
659 pub fn semantics_tree(&self) -> Option<&SemanticsTree> {
660 self.semantics.as_ref()
661 }
662
663 pub fn debug_allocation_stats(&self) -> LayoutAllocationDebugStats {
664 let mut stats = self
665 .layout_tree
666 .as_ref()
667 .map(LayoutTree::debug_allocation_stats)
668 .unwrap_or_default();
669 if let Some(semantics) = &self.semantics {
670 record_semantics_allocation_stats(semantics.root(), &mut stats);
671 }
672 stats
673 }
674
675 pub fn into_layout_tree(self) -> Option<LayoutTree> {
677 self.layout_tree
678 }
679
680 pub fn layout_tree(&self) -> Option<LayoutTree> {
682 self.layout_tree.clone()
683 }
684}
685
686pub fn build_semantics_tree_from_layout_tree(layout_tree: &LayoutTree) -> SemanticsTree {
691 SemanticsTree::new(build_semantics_node_from_layout_box(layout_tree.root()))
692}
693
694pub fn build_layout_tree_from_applier(
700 applier: &mut MemoryApplier,
701 root: NodeId,
702) -> Result<Option<LayoutTree>, NodeError> {
703 fn snapshot(
704 applier: &mut MemoryApplier,
705 node_id: NodeId,
706 ) -> Result<Option<(crate::widgets::nodes::layout_node::LayoutState, Vec<NodeId>)>, NodeError>
707 {
708 match applier.with_node::<LayoutNode, _>(node_id, |node| {
709 (node.layout_state(), node.children.clone())
710 }) {
711 Ok(snapshot) => return Ok(Some(snapshot)),
712 Err(NodeError::TypeMismatch { .. }) | Err(NodeError::Missing { .. }) => {}
713 Err(err) => return Err(err),
714 }
715
716 match applier.with_node::<SubcomposeLayoutNode, _>(node_id, |node| {
717 (node.layout_state(), node.active_children())
718 }) {
719 Ok(snapshot) => Ok(Some(snapshot)),
720 Err(NodeError::TypeMismatch { .. }) | Err(NodeError::Missing { .. }) => Ok(None),
721 Err(err) => Err(err),
722 }
723 }
724
725 fn place(
726 applier: &mut MemoryApplier,
727 node_id: NodeId,
728 parent_content_origin: Point,
729 parent_layer_translation: Point,
735 ) -> Result<Option<LayoutBox>, NodeError> {
736 let Some((state, child_ids)) = snapshot(applier, node_id)? else {
737 return Ok(None);
738 };
739 if !state.is_placed {
740 return Ok(None);
741 }
742
743 let top_left = Point {
744 x: parent_content_origin.x + state.position.x,
745 y: parent_content_origin.y + state.position.y,
746 };
747 let rect = GeometryRect {
748 x: top_left.x,
749 y: top_left.y,
750 width: state.size.width,
751 height: state.size.height,
752 };
753 let info = runtime_metadata_for(applier, node_id)?;
754 let kind = layout_kind_from_metadata(node_id, &info);
755 let RuntimeNodeMetadata {
756 modifier,
757 resolved_modifiers,
758 modifier_slices,
759 ..
760 } = info;
761
762 let layer_translation = match modifier_slices.graphics_layer() {
767 Some(layer) => Point {
768 x: parent_layer_translation.x + layer.translation_x,
769 y: parent_layer_translation.y + layer.translation_y,
770 },
771 None => parent_layer_translation,
772 };
773
774 if let Some(sink) = modifier_slices.text_field_window_origin() {
778 sink.set(Point {
779 x: top_left.x + layer_translation.x,
780 y: top_left.y + layer_translation.y,
781 });
782 }
783
784 if let Some(sink) = modifier_slices.viewport_window_rect() {
788 sink.set(GeometryRect {
789 x: top_left.x + layer_translation.x,
790 y: top_left.y + layer_translation.y,
791 width: state.size.width,
792 height: state.size.height,
793 });
794 }
795
796 let data = LayoutNodeData::new(modifier, resolved_modifiers, modifier_slices, kind);
797 let child_origin = Point {
798 x: top_left.x + state.content_offset.x,
799 y: top_left.y + state.content_offset.y,
800 };
801 let mut children = Vec::with_capacity(child_ids.len());
802 for child_id in child_ids {
803 if let Some(child) = place(applier, child_id, child_origin, layer_translation)? {
804 children.push(child);
805 }
806 }
807
808 Ok(Some(LayoutBox::new(
809 node_id,
810 rect,
811 state.content_offset,
812 data,
813 children,
814 )))
815 }
816
817 place(applier, root, Point::default(), Point::default()).map(|root| root.map(LayoutTree::new))
818}
819
820pub fn build_semantics_tree_from_applier(
826 applier: &mut MemoryApplier,
827 root: NodeId,
828) -> Result<Option<SemanticsTree>, NodeError> {
829 fn node(
830 applier: &mut MemoryApplier,
831 node_id: NodeId,
832 ) -> Result<Option<SemanticsNode>, NodeError> {
833 match applier.with_node::<LayoutNode, _>(node_id, |layout| {
834 let state = layout.layout_state();
835 if !state.is_placed {
836 return None;
837 }
838 let role = role_from_modifier_slices(&layout.modifier_slices_snapshot());
839 let config = layout.semantics_configuration();
840 let children = layout.children.clone();
841 layout.clear_needs_semantics();
842 Some((role, config, children))
843 }) {
844 Ok(Some((role, config, child_ids))) => {
845 let mut children = Vec::with_capacity(child_ids.len());
846 for child_id in child_ids {
847 if let Some(child) = node(applier, child_id)? {
848 children.push(child);
849 }
850 }
851 return Ok(Some(semantics_node_from_parts(
852 node_id, role, config, children,
853 )));
854 }
855 Ok(None) => return Ok(None),
856 Err(NodeError::TypeMismatch { .. }) | Err(NodeError::Missing { .. }) => {}
857 Err(err) => return Err(err),
858 }
859
860 match applier.with_node::<SubcomposeLayoutNode, _>(node_id, |subcompose| {
861 let state = subcompose.layout_state();
862 if !state.is_placed {
863 return None;
864 }
865 let config = collect_semantics_from_modifier(&subcompose.modifier());
866 let children = subcompose.active_children();
867 subcompose.clear_needs_semantics();
868 Some((config, children))
869 }) {
870 Ok(Some((config, child_ids))) => {
871 let mut children = Vec::with_capacity(child_ids.len());
872 for child_id in child_ids {
873 if let Some(child) = node(applier, child_id)? {
874 children.push(child);
875 }
876 }
877 Ok(Some(semantics_node_from_parts(
878 node_id,
879 SemanticsRole::Subcompose,
880 config,
881 children,
882 )))
883 }
884 Ok(None) | Err(NodeError::TypeMismatch { .. }) | Err(NodeError::Missing { .. }) => {
885 Ok(None)
886 }
887 Err(err) => Err(err),
888 }
889 }
890
891 node(applier, root).map(|root| root.map(SemanticsTree::new))
892}
893
894#[derive(Clone, Copy, Debug, PartialEq, Eq)]
895pub struct MeasureLayoutOptions {
896 pub collect_semantics: bool,
897 pub build_layout_tree: bool,
898}
899
900impl Default for MeasureLayoutOptions {
901 fn default() -> Self {
902 Self {
903 collect_semantics: true,
904 build_layout_tree: true,
905 }
906 }
907}
908
909pub fn tree_needs_layout(applier: &mut dyn Applier, root: NodeId) -> Result<bool, NodeError> {
917 Ok(applier.get_mut(root)?.needs_layout())
918}
919
920pub fn tree_needs_semantics(applier: &mut dyn Applier, root: NodeId) -> Result<bool, NodeError> {
926 Ok(applier.get_mut(root)?.needs_semantics())
927}
928
929#[cfg(test)]
931pub(crate) fn bubble_layout_dirty(applier: &mut MemoryApplier, node_id: NodeId) {
932 cranpose_core::bubble_layout_dirty(applier as &mut dyn Applier, node_id);
933}
934
935pub fn measure_layout(
937 applier: &mut MemoryApplier,
938 root: NodeId,
939 max_size: Size,
940) -> Result<LayoutMeasurements, NodeError> {
941 measure_layout_with_options(applier, root, max_size, MeasureLayoutOptions::default())
942}
943
944pub fn measure_layout_with_options(
945 applier: &mut MemoryApplier,
946 root: NodeId,
947 max_size: Size,
948 options: MeasureLayoutOptions,
949) -> Result<LayoutMeasurements, NodeError> {
950 let telemetry_start = Instant::now();
951 process_pending_layout_repasses(applier, root)?;
952 let after_repasses = Instant::now();
953
954 let constraints = Constraints {
955 min_width: 0.0,
956 max_width: max_size.width,
957 min_height: 0.0,
958 max_height: max_size.height,
959 };
960
961 let (needs_remeasure, _needs_semantics, cached_epoch) = match applier
972 .with_node::<LayoutNode, _>(root, |node| {
973 (
974 node.needs_measure(), node.needs_semantics(),
976 node.cache_handles().epoch(),
977 )
978 }) {
979 Ok(tuple) => tuple,
980 Err(NodeError::TypeMismatch { .. }) => {
981 let node = applier.get_mut(root)?;
982 let measure_dirty = node.needs_measure();
986 let semantics_dirty = node.needs_semantics();
987 (measure_dirty, semantics_dirty, 0)
988 }
989 Err(err) => return Err(err),
990 };
991
992 let epoch = if needs_remeasure {
993 crate::render_state::next_layout_cache_epoch()
994 } else if cached_epoch != 0 {
995 cached_epoch
996 } else {
997 crate::render_state::current_layout_cache_epoch()
999 };
1000
1001 let guard = ApplierSlotGuard::new(applier);
1009 let applier_host = guard.host();
1010 let slots_handle = guard.slots_handle();
1011 let after_guard = Instant::now();
1012
1013 let frame_arena = crate::render_state::take_layout_frame_arena();
1016 let mut builder = LayoutBuilder::new_with_epoch(
1017 Rc::clone(&applier_host),
1018 epoch,
1019 Rc::clone(&slots_handle),
1020 frame_arena,
1021 );
1022 let after_builder = Instant::now();
1023
1024 let measured = builder.measure_node(root, normalize_constraints(constraints))?;
1029 let after_measure = Instant::now();
1030
1031 if let Ok(mut applier) = applier_host.try_borrow_typed() {
1035 if applier
1036 .with_node::<LayoutNode, _>(root, |node| {
1037 node.set_position(Point::default());
1038 })
1039 .is_err()
1040 {
1041 let _ = applier.with_node::<SubcomposeLayoutNode, _>(root, |node| {
1042 node.set_position(Point::default());
1043 });
1044 }
1045 }
1046 let after_root_place = Instant::now();
1047
1048 let (layout_tree, semantics) = {
1049 let mut applier_ref = applier_host.borrow_typed();
1050 let layout_tree = if options.build_layout_tree {
1051 Some(build_layout_tree(&mut applier_ref, &measured)?)
1052 } else {
1053 None
1054 };
1055 let semantics = if options.collect_semantics {
1056 let semantics_tree = if let Some(layout_tree) = layout_tree.as_ref() {
1057 clear_semantics_dirty_flags(&mut applier_ref, &measured)?;
1058 build_semantics_tree_from_layout_tree(layout_tree)
1059 } else {
1060 build_semantics_tree_from_live_nodes(&mut applier_ref, &measured)?
1061 };
1062 Some(semantics_tree)
1063 } else {
1064 None
1065 };
1066 (layout_tree, semantics)
1067 };
1068 let after_aux = Instant::now();
1069
1070 drop(builder);
1073 let after_builder_drop = Instant::now();
1074
1075 drop(guard);
1078 let after_guard_drop = Instant::now();
1079
1080 log_layout_measure_telemetry(LayoutMeasureTelemetry {
1081 root,
1082 start: telemetry_start,
1083 after_repasses,
1084 after_guard,
1085 after_builder,
1086 after_measure,
1087 after_root_place,
1088 after_aux,
1089 after_builder_drop,
1090 after_guard_drop,
1091 });
1092
1093 Ok(LayoutMeasurements::new(measured, semantics, layout_tree))
1094}
1095
1096fn process_pending_layout_repasses(
1097 applier: &mut MemoryApplier,
1098 root: NodeId,
1099) -> Result<(), NodeError> {
1100 for node_id in crate::render_state::take_modifier_slice_repass_nodes() {
1101 if let Ok(node) = applier.get_mut(node_id) {
1102 let any = node.as_any_mut();
1103 if let Some(layout) = any.downcast_mut::<crate::widgets::nodes::LayoutNode>() {
1104 layout.mark_modifier_slices_dirty();
1105 } else if let Some(subcompose) =
1106 any.downcast_mut::<crate::subcompose_layout::SubcomposeLayoutNode>()
1107 {
1108 subcompose.mark_modifier_slices_dirty();
1109 }
1110 }
1111 }
1112 let measure_repass_nodes = crate::take_measure_repass_nodes();
1115 let repass_nodes = crate::take_layout_repass_nodes();
1116 if measure_repass_nodes.is_empty() && repass_nodes.is_empty() {
1117 return Ok(());
1118 }
1119 for node_id in measure_repass_nodes {
1120 cranpose_core::bubble_measure_dirty(applier as &mut dyn Applier, node_id);
1121 }
1122 for node_id in repass_nodes {
1123 cranpose_core::bubble_layout_dirty(applier as &mut dyn Applier, node_id);
1124 }
1125 applier.get_mut(root)?.mark_needs_layout();
1126 Ok(())
1127}
1128
1129struct LayoutBuilder {
1130 state: Rc<RefCell<LayoutBuilderState>>,
1131}
1132
1133impl LayoutBuilder {
1134 fn new_with_epoch(
1135 applier: Rc<ConcreteApplierHost<MemoryApplier>>,
1136 epoch: u64,
1137 slots: Rc<RefCell<SlotTable>>,
1138 frame_arena: FrameLayoutArena,
1139 ) -> Self {
1140 Self {
1141 state: Rc::new(RefCell::new(LayoutBuilderState::new_with_epoch(
1142 applier,
1143 epoch,
1144 slots,
1145 frame_arena,
1146 ))),
1147 }
1148 }
1149
1150 fn measure_node(
1151 &mut self,
1152 node_id: NodeId,
1153 constraints: Constraints,
1154 ) -> Result<Rc<MeasuredNode>, NodeError> {
1155 LayoutBuilderState::measure_node(Rc::clone(&self.state), node_id, constraints)
1156 }
1157
1158 fn set_runtime_handle(&mut self, handle: Option<RuntimeHandle>) {
1159 self.state.borrow_mut().runtime_handle = handle;
1160 }
1161}
1162
1163impl Drop for LayoutBuilder {
1164 fn drop(&mut self) {
1165 if Rc::strong_count(&self.state) != 1 {
1166 return;
1167 }
1168 let Ok(mut state) = self.state.try_borrow_mut() else {
1169 return;
1170 };
1171 crate::render_state::replace_layout_frame_arena(std::mem::take(&mut state.frame_arena));
1172 }
1173}
1174
1175struct LayoutBuilderState {
1176 applier: Rc<ConcreteApplierHost<MemoryApplier>>,
1177 runtime_handle: Option<RuntimeHandle>,
1178 slots: Rc<RefCell<SlotTable>>,
1181 cache_epoch: u64,
1182 frame_arena: FrameLayoutArena,
1183}
1184
1185struct LayoutRuntimeFrameBindingCleanup {
1186 state: Rc<RefCell<LayoutRuntimeState>>,
1187}
1188
1189impl LayoutRuntimeFrameBindingCleanup {
1190 fn new(state: Rc<RefCell<LayoutRuntimeState>>) -> Self {
1191 Self { state }
1192 }
1193}
1194
1195impl Drop for LayoutRuntimeFrameBindingCleanup {
1196 fn drop(&mut self) {
1197 self.state.borrow().clear_frame_bindings();
1198 }
1199}
1200
1201impl LayoutBuilderState {
1202 fn new_with_epoch(
1203 applier: Rc<ConcreteApplierHost<MemoryApplier>>,
1204 epoch: u64,
1205 slots: Rc<RefCell<SlotTable>>,
1206 frame_arena: FrameLayoutArena,
1207 ) -> Self {
1208 let runtime_handle = applier.borrow_typed().runtime_handle();
1209
1210 Self {
1211 applier,
1212 runtime_handle,
1213 slots,
1214 cache_epoch: epoch,
1215 frame_arena,
1216 }
1217 }
1218
1219 fn try_with_applier_result<R>(
1220 state_rc: &Rc<RefCell<Self>>,
1221 f: impl FnOnce(&mut MemoryApplier) -> Result<R, NodeError>,
1222 ) -> Option<Result<R, NodeError>> {
1223 let host = {
1224 let state = state_rc.borrow();
1225 Rc::clone(&state.applier)
1226 };
1227
1228 let Ok(mut applier) = host.try_borrow_typed() else {
1230 return None;
1231 };
1232
1233 Some(f(&mut applier))
1234 }
1235
1236 fn with_applier_result<R>(
1237 state_rc: &Rc<RefCell<Self>>,
1238 f: impl FnOnce(&mut MemoryApplier) -> Result<R, NodeError>,
1239 ) -> Result<R, NodeError> {
1240 Self::try_with_applier_result(state_rc, f).unwrap_or_else(|| {
1241 Err(NodeError::MissingContext {
1242 id: NodeId::default(),
1243 reason: "applier already borrowed",
1244 })
1245 })
1246 }
1247
1248 fn clear_node_placed(state_rc: &Rc<RefCell<Self>>, node_id: NodeId) {
1251 let host = {
1252 let state = state_rc.borrow();
1253 Rc::clone(&state.applier)
1254 };
1255 let Ok(mut applier) = host.try_borrow_typed() else {
1256 return;
1257 };
1258 if applier
1260 .with_node::<LayoutNode, _>(node_id, |node| {
1261 node.clear_placed();
1262 })
1263 .is_err()
1264 {
1265 let _ = applier.with_node::<SubcomposeLayoutNode, _>(node_id, |node| {
1266 node.clear_placed();
1267 });
1268 }
1269 }
1270
1271 fn measure_node(
1272 state_rc: Rc<RefCell<Self>>,
1273 node_id: NodeId,
1274 constraints: Constraints,
1275 ) -> Result<Rc<MeasuredNode>, NodeError> {
1276 let telemetry_start = Instant::now();
1277 Self::clear_node_placed(&state_rc, node_id);
1281
1282 if let Some(subcompose) =
1284 Self::try_measure_subcompose(Rc::clone(&state_rc), node_id, constraints)?
1285 {
1286 log_node_measure_telemetry(
1287 "subcompose",
1288 node_id,
1289 constraints,
1290 subcompose.size,
1291 subcompose.children.len(),
1292 telemetry_start,
1293 );
1294 return Ok(subcompose);
1295 }
1296
1297 if let Some(result) = Self::try_with_applier_result(&state_rc, |applier| {
1299 match applier.with_node::<LayoutNode, _>(node_id, |layout_node| {
1300 LayoutNodeSnapshot::from_layout_node(layout_node)
1301 }) {
1302 Ok(snapshot) => Ok(Some(snapshot)),
1303 Err(NodeError::TypeMismatch { .. }) | Err(NodeError::Missing { .. }) => Ok(None),
1304 Err(err) => Err(err),
1305 }
1306 }) {
1307 if let Some(snapshot) = result? {
1309 let measured = Self::measure_layout_node(
1310 Rc::clone(&state_rc),
1311 node_id,
1312 snapshot,
1313 constraints,
1314 )?;
1315 log_node_measure_telemetry(
1316 "layout",
1317 node_id,
1318 constraints,
1319 measured.size,
1320 measured.children.len(),
1321 telemetry_start,
1322 );
1323 return Ok(measured);
1324 }
1325 }
1326 let measured = Rc::new(MeasuredNode::new(
1331 node_id,
1332 Size::default(),
1333 Point { x: 0.0, y: 0.0 },
1334 Point::default(), Vec::new(),
1336 ));
1337 log_node_measure_telemetry(
1338 "fallback",
1339 node_id,
1340 constraints,
1341 measured.size,
1342 measured.children.len(),
1343 telemetry_start,
1344 );
1345 Ok(measured)
1346 }
1347
1348 fn cached_measure_node_with_applier(
1349 applier: &mut MemoryApplier,
1350 node_id: NodeId,
1351 constraints: Constraints,
1352 ) -> Result<Option<Rc<MeasuredNode>>, NodeError> {
1353 let Some(data) = Self::layout_child_measure_data(applier, node_id)? else {
1354 return Ok(None);
1355 };
1356 if data.needs_measure || data.cache.epoch() == 0 {
1357 return Ok(None);
1358 }
1359
1360 let Some(measured) = data.cache.get_measurement(constraints) else {
1361 return Ok(None);
1362 };
1363
1364 if let Some(layout_state) = data.layout_state {
1365 let mut layout_state = layout_state.borrow_mut();
1366 layout_state.size = measured.size;
1367 layout_state.measurement_constraints = constraints;
1368 drop(layout_state);
1369 let _ = applier.with_node::<LayoutNode, _>(node_id, |node| {
1370 if data.needs_layout {
1371 node.clear_needs_layout();
1372 }
1373 });
1374 } else {
1375 let _ = applier.with_node::<SubcomposeLayoutNode, _>(node_id, |node| {
1376 node.set_measured_size(measured.size);
1377 if data.needs_layout {
1378 node.clear_needs_layout();
1379 }
1380 });
1381 }
1382
1383 Ok(Some(measured))
1384 }
1385
1386 fn try_measure_subcompose(
1387 state_rc: Rc<RefCell<Self>>,
1388 node_id: NodeId,
1389 constraints: Constraints,
1390 ) -> Result<Option<Rc<MeasuredNode>>, NodeError> {
1391 let applier_host = {
1392 let state = state_rc.borrow();
1393 Rc::clone(&state.applier)
1394 };
1395
1396 let (node_handle, resolved_modifiers) = {
1397 let Ok(mut applier) = applier_host.try_borrow_typed() else {
1399 return Ok(None);
1400 };
1401 let node = match applier.get_mut(node_id) {
1402 Ok(node) => node,
1403 Err(NodeError::Missing { .. }) => return Ok(None),
1404 Err(err) => return Err(err),
1405 };
1406 let any = node.as_any_mut();
1407 if let Some(subcompose) =
1408 any.downcast_mut::<crate::subcompose_layout::SubcomposeLayoutNode>()
1409 {
1410 let handle = subcompose.handle();
1411 let resolved_modifiers = handle.resolved_modifiers();
1412 (handle, resolved_modifiers)
1413 } else {
1414 return Ok(None);
1415 }
1416 };
1417
1418 let runtime_handle = {
1419 let mut state = state_rc.borrow_mut();
1420 if state.runtime_handle.is_none() {
1421 if let Ok(applier) = applier_host.try_borrow_typed() {
1423 state.runtime_handle = applier.runtime_handle();
1424 }
1425 }
1426 state
1427 .runtime_handle
1428 .clone()
1429 .ok_or(NodeError::MissingContext {
1430 id: node_id,
1431 reason: "runtime handle required for subcomposition",
1432 })?
1433 };
1434
1435 let props = resolved_modifiers.layout_properties();
1436 let padding = resolved_modifiers.padding();
1437 let offset = resolved_modifiers.offset();
1438 let mut inner_constraints = normalize_constraints(subtract_padding(constraints, padding));
1439
1440 if let DimensionConstraint::Points(width) = props.width() {
1441 let constrained_width = width - padding.horizontal_sum();
1442 inner_constraints.max_width = inner_constraints.max_width.min(constrained_width);
1443 inner_constraints.min_width = inner_constraints.min_width.min(constrained_width);
1444 }
1445 if let DimensionConstraint::Points(height) = props.height() {
1446 let constrained_height = height - padding.vertical_sum();
1447 inner_constraints.max_height = inner_constraints.max_height.min(constrained_height);
1448 inner_constraints.min_height = inner_constraints.min_height.min(constrained_height);
1449 }
1450
1451 let mut slots_guard = SlotsGuard::take(Rc::clone(&state_rc));
1452 let slots_host = slots_guard.host();
1453 let applier_host_dyn: Rc<dyn ApplierHost> = applier_host.clone();
1454 let observer = SnapshotStateObserver::new(|callback| callback());
1455 let composer = Composer::new(
1456 Rc::clone(&slots_host),
1457 applier_host_dyn,
1458 runtime_handle.clone(),
1459 observer,
1460 Some(node_id),
1461 );
1462 composer.enter_phase(Phase::Measure);
1463
1464 let state_rc_clone = Rc::clone(&state_rc);
1465 let measure_error = RefCell::new(None);
1466 let state_rc_for_subcompose = Rc::clone(&state_rc_clone);
1467 let error_for_subcompose = &measure_error;
1468 let measured_children = node_handle.measured_children_scratch();
1469 let measured_children_for_subcompose = Rc::clone(&measured_children);
1470 let state_rc_for_cached = Rc::clone(&state_rc_clone);
1471 let error_for_cached = &measure_error;
1472 let measured_children_for_cached = Rc::clone(&measured_children);
1473 let measured_children_for_lookup = Rc::clone(&measured_children);
1474 let measured_children_for_retained = Rc::clone(&measured_children);
1475
1476 let measure_result = node_handle.measure_with_cached_batch(
1477 &composer,
1478 node_id,
1479 inner_constraints,
1480 CachedBatchMeasureInputs {
1481 measurer: Box::new(
1482 move |child_id: NodeId, child_constraints: Constraints| -> Size {
1483 match Self::measure_node(
1484 Rc::clone(&state_rc_for_subcompose),
1485 child_id,
1486 child_constraints,
1487 ) {
1488 Ok(measured) => {
1489 measured_children_for_subcompose
1490 .borrow_mut()
1491 .insert(child_id, Rc::clone(&measured));
1492 measured.size
1493 }
1494 Err(err) => {
1495 let mut slot = error_for_subcompose.borrow_mut();
1496 if slot.is_none() {
1497 *slot = Some(err);
1498 }
1499 Size::default()
1500 }
1501 }
1502 },
1503 ),
1504 cached_measure_batch_registrar: Box::new(
1505 move |child_ids: &[NodeId],
1506 child_constraints: Constraints,
1507 out: &mut Vec<Option<Size>>| {
1508 out.clear();
1509 out.resize(child_ids.len(), None);
1510
1511 let applier_host = {
1512 let state = state_rc_for_cached.borrow();
1513 Rc::clone(&state.applier)
1514 };
1515 let Ok(mut applier) = applier_host.try_borrow_typed() else {
1516 return;
1517 };
1518
1519 let mut measured_children = measured_children_for_cached.borrow_mut();
1520 for (index, &child_id) in child_ids.iter().enumerate() {
1521 match Self::cached_measure_node_with_applier(
1522 &mut applier,
1523 child_id,
1524 child_constraints,
1525 ) {
1526 Ok(Some(measured)) => {
1527 out[index] = Some(measured.size);
1528 measured_children.insert(child_id, Rc::clone(&measured));
1529 }
1530 Ok(None) => {}
1531 Err(err) => {
1532 let mut slot = error_for_cached.borrow_mut();
1533 if slot.is_none() {
1534 *slot = Some(err);
1535 }
1536 break;
1537 }
1538 }
1539 }
1540 },
1541 ),
1542 retained_measure_lookup: Box::new(move |child_id| {
1543 measured_children_for_lookup
1544 .borrow()
1545 .get(&child_id)
1546 .cloned()
1547 }),
1548 retained_measure_registrar: Box::new(move |measurements| {
1549 let mut measured_children = measured_children_for_retained.borrow_mut();
1550 for measured in measurements {
1551 measured_children.insert(measured.node_id(), Rc::clone(measured));
1552 }
1553 }),
1554 error: &measure_error,
1555 },
1556 )?;
1557 drop(composer);
1558 slots_guard.restore(slots_host.into_table()?);
1559
1560 if let Some(err) = measure_error.borrow_mut().take() {
1561 return Err(err);
1562 }
1563
1564 let cranpose_ui_layout::MeasureResult {
1568 size: measured_size,
1569 placements,
1570 } = measure_result;
1571
1572 let mut width = measured_size.width + padding.horizontal_sum();
1573 let mut height = measured_size.height + padding.vertical_sum();
1574
1575 width = resolve_dimension(
1576 width,
1577 props.width(),
1578 props.min_width(),
1579 props.max_width(),
1580 constraints.min_width,
1581 constraints.max_width,
1582 );
1583 height = resolve_dimension(
1584 height,
1585 props.height(),
1586 props.min_height(),
1587 props.max_height(),
1588 constraints.min_height,
1589 constraints.max_height,
1590 );
1591
1592 let mut children = Vec::with_capacity(placements.len());
1593 let mut measured_children_by_id = measured_children.borrow_mut();
1594
1595 if let Ok(mut applier) = applier_host.try_borrow_typed() {
1597 let _ = applier.with_node::<SubcomposeLayoutNode, _>(node_id, |parent_node| {
1598 parent_node.set_measured_size(Size { width, height });
1599 parent_node.clear_needs_measure();
1600 parent_node.clear_needs_layout();
1601 });
1602 }
1603
1604 for placement in &placements {
1605 let child = if let Some(measured) = measured_children_by_id.remove(&placement.node_id) {
1606 measured
1607 } else {
1608 Self::measure_node(Rc::clone(&state_rc), placement.node_id, inner_constraints)?
1614 };
1615 let position = Point {
1616 x: padding.left + placement.x,
1617 y: padding.top + placement.y,
1618 };
1619
1620 if let Ok(mut applier) = applier_host.try_borrow_typed() {
1628 if applier
1629 .with_node::<LayoutNode, _>(placement.node_id, |node| {
1630 node.set_position(position);
1631 })
1632 .is_err()
1633 {
1634 let _ =
1635 applier.with_node::<SubcomposeLayoutNode, _>(placement.node_id, |node| {
1636 node.set_position(position);
1637 });
1638 }
1639 }
1640
1641 children.push(MeasuredChild {
1642 node: child,
1643 offset: position,
1644 });
1645 }
1646
1647 node_handle.set_active_children(children.iter().map(|c| c.node.node_id));
1649 node_handle.recycle_placement_scratch(placements);
1650
1651 Ok(Some(Rc::new(MeasuredNode::new(
1652 node_id,
1653 Size { width, height },
1654 offset,
1655 Point::default(), children,
1657 ))))
1658 }
1659 fn measure_through_modifier_chain(
1666 state_rc: &Rc<RefCell<Self>>,
1667 node_id: NodeId,
1668 runtime_state: &mut LayoutRuntimeState,
1669 measure_policy: &Rc<dyn MeasurePolicy>,
1670 constraints: Constraints,
1671 layout_node_data: &mut Vec<LayoutModifierNodeData>,
1672 placements: &mut Vec<Placement>,
1673 ) -> ModifierChainMeasurement {
1674 use cranpose_foundation::NodeCapabilities;
1675
1676 layout_node_data.clear();
1678 let mut offset = Point::default();
1679
1680 {
1681 let state = state_rc.borrow();
1682 let mut applier = state.applier.borrow_typed();
1683
1684 let _ = applier.with_node::<LayoutNode, _>(node_id, |layout_node| {
1685 let chain_handle = layout_node.modifier_chain();
1686
1687 if !chain_handle.has_layout_nodes() {
1688 return;
1689 }
1690
1691 chain_handle.chain().for_each_forward_matching(
1693 NodeCapabilities::LAYOUT,
1694 |node_ref| {
1695 if let Some(index) = node_ref.entry_index() {
1696 if let Some(node_rc) = chain_handle.chain().get_node_rc(index) {
1698 layout_node_data.push((index, Rc::clone(&node_rc)));
1699 }
1700
1701 node_ref.with_node(|node| {
1705 if let Some(offset_node) =
1706 node.as_any()
1707 .downcast_ref::<crate::modifier_nodes::OffsetNode>()
1708 {
1709 let delta = offset_node.offset();
1710 offset.x += delta.x;
1711 offset.y += delta.y;
1712 }
1713 });
1714 }
1715 },
1716 );
1717 });
1718 }
1719
1720 if layout_node_data.is_empty() {
1723 let final_size = measure_policy.measure_into(
1724 runtime_state.child_measurables(),
1725 constraints,
1726 placements,
1727 );
1728
1729 return ModifierChainMeasurement {
1730 size: final_size,
1731 content_offset: Point::default(),
1732 offset,
1733 };
1734 }
1735
1736 runtime_state.reconcile_coordinator_chain(layout_node_data.as_slice());
1737 let frame = CoordinatorFrame::new(
1738 measure_policy,
1739 runtime_state.child_measurables(),
1740 placements,
1741 );
1742
1743 let placeable = runtime_state
1745 .coordinator_chain()
1746 .measure_from(0, &frame, constraints);
1747 let final_size = Size {
1748 width: placeable.width(),
1749 height: placeable.height(),
1750 };
1751
1752 let content_offset = placeable.content_offset();
1754 let all_placement_offset = Point {
1755 x: content_offset.0,
1756 y: content_offset.1,
1757 };
1758
1759 let content_offset = Point {
1763 x: all_placement_offset.x - offset.x,
1764 y: all_placement_offset.y - offset.y,
1765 };
1766
1767 let invalidations = frame.take_invalidations();
1771 if !invalidations.is_empty() {
1772 Self::with_applier_result(state_rc, |applier| {
1774 applier.with_node::<LayoutNode, _>(node_id, |layout_node| {
1775 for kind in invalidations {
1776 match kind {
1777 InvalidationKind::Layout => layout_node.mark_needs_measure(),
1778 InvalidationKind::Draw => layout_node.mark_needs_redraw(),
1779 InvalidationKind::Semantics => layout_node.mark_needs_semantics(),
1780 InvalidationKind::PointerInput => layout_node.mark_needs_pointer_pass(),
1781 InvalidationKind::Focus => layout_node.mark_needs_focus_sync(),
1782 }
1783 }
1784 })
1785 })
1786 .ok();
1787 }
1788
1789 ModifierChainMeasurement {
1790 size: final_size,
1791 content_offset,
1792 offset,
1793 }
1794 }
1795
1796 fn layout_child_measure_data(
1797 applier: &mut MemoryApplier,
1798 child_id: NodeId,
1799 ) -> Result<Option<LayoutChildMeasureData>, NodeError> {
1800 match applier.with_node::<LayoutNode, _>(child_id, |n| LayoutChildMeasureData {
1801 cache: n.cache_handles(),
1802 layout_state: Some(n.layout_state_handle()),
1803 needs_layout: n.needs_layout(),
1804 needs_measure: n.needs_measure(),
1805 }) {
1806 Ok(data) => Ok(Some(data)),
1807 Err(NodeError::TypeMismatch { .. }) => {
1808 match applier.with_node::<SubcomposeLayoutNode, _>(child_id, |n| {
1809 LayoutChildMeasureData {
1810 cache: n.cache_handles(),
1811 layout_state: None,
1812 needs_layout: n.needs_layout(),
1813 needs_measure: n.needs_measure(),
1814 }
1815 }) {
1816 Ok(data) => Ok(Some(data)),
1817 Err(NodeError::TypeMismatch { .. }) | Err(NodeError::Missing { .. }) => {
1818 Ok(None)
1819 }
1820 Err(err) => Err(err),
1821 }
1822 }
1823 Err(NodeError::Missing { .. }) => Ok(None),
1824 Err(err) => Err(err),
1825 }
1826 }
1827
1828 fn measure_layout_node(
1829 state_rc: Rc<RefCell<Self>>,
1830 node_id: NodeId,
1831 snapshot: LayoutNodeSnapshot,
1832 constraints: Constraints,
1833 ) -> Result<Rc<MeasuredNode>, NodeError> {
1834 let cache_epoch = {
1835 let state = state_rc.borrow();
1836 state.cache_epoch
1837 };
1838 let LayoutNodeSnapshot {
1839 measure_policy,
1840 cache,
1841 layout_runtime_state,
1842 needs_layout,
1843 needs_measure,
1844 } = snapshot;
1845 cache.activate(cache_epoch);
1846
1847 if needs_measure {
1848 }
1850
1851 if !needs_measure && !needs_layout {
1855 if let Some(cached) = cache.get_measurement(constraints) {
1857 Self::with_applier_result(&state_rc, |applier| {
1859 applier.with_node::<LayoutNode, _>(node_id, |node| {
1860 node.clear_needs_measure();
1861 node.clear_needs_layout();
1862 })
1863 })
1864 .ok();
1865 return Ok(cached);
1866 }
1867 }
1868
1869 let (runtime_handle, applier_host) = {
1870 let state = state_rc.borrow();
1871 (state.runtime_handle.clone(), Rc::clone(&state.applier))
1872 };
1873
1874 let measure_handle = LayoutMeasureHandle::new(Rc::clone(&state_rc));
1875 let error = Rc::new(RefCell::new(None));
1876 let mut pools = VecPools::acquire(Rc::clone(&state_rc));
1877 let (records, child_ids, layout_node_data, placements) = pools.parts();
1878
1879 applier_host
1880 .borrow_typed()
1881 .with_node::<LayoutNode, _>(node_id, |node| {
1882 child_ids.extend_from_slice(&node.children);
1883 })?;
1884
1885 let mut valid_child_count = 0;
1886 for index in 0..child_ids.len() {
1887 let child_id = child_ids[index];
1888 let child_exists = {
1889 let mut applier = applier_host.borrow_typed();
1890 Self::layout_child_measure_data(&mut applier, child_id)?.is_some()
1891 };
1892 if child_exists {
1893 child_ids[valid_child_count] = child_id;
1894 valid_child_count += 1;
1895 }
1896 }
1897 child_ids.truncate(valid_child_count);
1898
1899 let _frame_binding_cleanup =
1900 LayoutRuntimeFrameBindingCleanup::new(Rc::clone(&layout_runtime_state));
1901
1902 {
1903 let mut runtime_state = layout_runtime_state.borrow_mut();
1904 runtime_state.reconcile_child_measurables(child_ids.as_slice());
1905
1906 for (index, &child_id) in child_ids.iter().enumerate() {
1907 let data = {
1908 let mut applier = applier_host.borrow_typed();
1909 Self::layout_child_measure_data(&mut applier, child_id)?
1910 };
1911 let Some(data) = data else {
1912 continue;
1913 };
1914
1915 let child_is_dirty = data.needs_layout || data.needs_measure;
1916 let child_cache_epoch = if child_is_dirty {
1917 cache_epoch
1918 } else {
1919 data.cache.epoch()
1920 };
1921 let child_state = runtime_state.child_state(index);
1922 child_state.configure(LayoutChildMeasureConfig {
1923 applier: Rc::clone(&applier_host),
1924 node_id: child_id,
1925 error: Rc::clone(&error),
1926 runtime_handle: runtime_handle.clone(),
1927 cache: data.cache,
1928 cache_epoch: child_cache_epoch,
1929 force_remeasure: child_is_dirty,
1930 measure_handle: Some(measure_handle.clone()),
1931 layout_state: data.layout_state,
1932 });
1933 records.push((child_id, ChildRecord { state: child_state }));
1934 }
1935 }
1936
1937 let chain_constraints = constraints;
1938
1939 let modifier_chain_result = {
1940 let mut runtime_state = layout_runtime_state.borrow_mut();
1941 Self::measure_through_modifier_chain(
1942 &state_rc,
1943 node_id,
1944 &mut runtime_state,
1945 &measure_policy,
1946 chain_constraints,
1947 layout_node_data,
1948 placements,
1949 )
1950 };
1951
1952 let (width, height, content_offset, offset) = {
1954 let result = modifier_chain_result;
1955 if let Some(err) = error.borrow_mut().take() {
1958 return Err(err);
1959 }
1960
1961 (
1962 result.size.width,
1963 result.size.height,
1964 result.content_offset,
1965 result.offset,
1966 )
1967 };
1968
1969 let mut measured_children = Vec::with_capacity(records.len());
1970 for (child_id, record) in records.iter() {
1971 if let Some(measured) = record.state.take_measured() {
1972 let base_position = placements
1973 .iter()
1974 .find(|placement| placement.node_id == *child_id)
1975 .map(|placement| Point {
1976 x: placement.x,
1977 y: placement.y,
1978 })
1979 .or_else(|| record.state.last_position())
1980 .unwrap_or(Point { x: 0.0, y: 0.0 });
1981 let position = Point {
1983 x: content_offset.x + base_position.x,
1984 y: content_offset.y + base_position.y,
1985 };
1986 measured_children.push(MeasuredChild {
1987 node: measured,
1988 offset: position,
1989 });
1990 }
1991 }
1992
1993 let measured = Rc::new(MeasuredNode::new(
1994 node_id,
1995 Size { width, height },
1996 offset,
1997 content_offset,
1998 measured_children,
1999 ));
2000
2001 cache.store_measurement(constraints, Rc::clone(&measured));
2002
2003 Self::with_applier_result(&state_rc, |applier| {
2005 applier.with_node::<LayoutNode, _>(node_id, |node| {
2006 node.clear_needs_measure();
2007 node.clear_needs_layout();
2008 node.set_measured_size(Size { width, height });
2009 node.set_content_offset(content_offset);
2010 })
2011 })
2012 .ok();
2013
2014 Ok(measured)
2015 }
2016}
2017
2018struct LayoutChildMeasureData {
2019 cache: LayoutNodeCacheHandles,
2020 layout_state: Option<Rc<RefCell<LayoutState>>>,
2021 needs_layout: bool,
2022 needs_measure: bool,
2023}
2024
2025struct LayoutNodeSnapshot {
2032 measure_policy: Rc<dyn MeasurePolicy>,
2033 cache: LayoutNodeCacheHandles,
2034 layout_runtime_state: Rc<RefCell<LayoutRuntimeState>>,
2035 needs_layout: bool,
2036 needs_measure: bool,
2038}
2039
2040impl LayoutNodeSnapshot {
2041 fn from_layout_node(node: &LayoutNode) -> Self {
2042 Self {
2043 measure_policy: Rc::clone(&node.measure_policy),
2044 cache: node.cache_handles(),
2045 layout_runtime_state: node.layout_runtime_state_handle(),
2046 needs_layout: node.needs_layout(),
2047 needs_measure: node.needs_measure(),
2048 }
2049 }
2050}
2051
2052struct VecPools {
2054 state: Rc<RefCell<LayoutBuilderState>>,
2055 records: Vec<(NodeId, ChildRecord)>,
2056 child_ids: Vec<NodeId>,
2057 layout_node_data: Vec<LayoutModifierNodeData>,
2058 placements: Vec<Placement>,
2059}
2060
2061impl VecPools {
2062 fn acquire(state: Rc<RefCell<LayoutBuilderState>>) -> Self {
2063 let (records, child_ids, layout_node_data, placements) = {
2064 let mut state_mut = state.borrow_mut();
2065 (
2066 state_mut.frame_arena.tmp_records.acquire(),
2067 state_mut.frame_arena.tmp_child_ids.acquire(),
2068 state_mut.frame_arena.tmp_layout_node_data.acquire(),
2069 state_mut.frame_arena.tmp_placements.acquire(),
2070 )
2071 };
2072 Self {
2073 state,
2074 records,
2075 child_ids,
2076 layout_node_data,
2077 placements,
2078 }
2079 }
2080
2081 #[allow(clippy::type_complexity)] fn parts(
2083 &mut self,
2084 ) -> (
2085 &mut Vec<(NodeId, ChildRecord)>,
2086 &mut Vec<NodeId>,
2087 &mut Vec<LayoutModifierNodeData>,
2088 &mut Vec<Placement>,
2089 ) {
2090 (
2091 &mut self.records,
2092 &mut self.child_ids,
2093 &mut self.layout_node_data,
2094 &mut self.placements,
2095 )
2096 }
2097}
2098
2099impl Drop for VecPools {
2100 fn drop(&mut self) {
2101 let mut state = self.state.borrow_mut();
2102 state
2103 .frame_arena
2104 .tmp_records
2105 .release(std::mem::take(&mut self.records));
2106 state
2107 .frame_arena
2108 .tmp_child_ids
2109 .release(std::mem::take(&mut self.child_ids));
2110 state
2111 .frame_arena
2112 .tmp_layout_node_data
2113 .release(std::mem::take(&mut self.layout_node_data));
2114 state
2115 .frame_arena
2116 .tmp_placements
2117 .release(std::mem::take(&mut self.placements));
2118 }
2119}
2120
2121struct SlotsGuard {
2122 state: Rc<RefCell<LayoutBuilderState>>,
2123 slots: Option<SlotTable>,
2124}
2125
2126impl SlotsGuard {
2127 fn take(state: Rc<RefCell<LayoutBuilderState>>) -> Self {
2128 let slots = {
2129 let state_ref = state.borrow();
2130 let mut slots_ref = state_ref.slots.borrow_mut();
2131 std::mem::take(&mut *slots_ref)
2132 };
2133 Self {
2134 state,
2135 slots: Some(slots),
2136 }
2137 }
2138
2139 fn host(&mut self) -> Rc<SlotsHost> {
2140 let slots = self.slots.take().unwrap_or_default();
2141 Rc::new(SlotsHost::new(slots))
2142 }
2143
2144 fn restore(&mut self, slots: SlotTable) {
2145 debug_assert!(self.slots.is_none());
2146 self.slots = Some(slots);
2147 }
2148}
2149
2150impl Drop for SlotsGuard {
2151 fn drop(&mut self) {
2152 if let Some(slots) = self.slots.take() {
2153 let state_ref = self.state.borrow();
2154 *state_ref.slots.borrow_mut() = slots;
2155 }
2156 }
2157}
2158
2159#[derive(Clone)]
2160struct LayoutMeasureHandle {
2161 state: Rc<RefCell<LayoutBuilderState>>,
2162}
2163
2164impl LayoutMeasureHandle {
2165 fn new(state: Rc<RefCell<LayoutBuilderState>>) -> Self {
2166 Self { state }
2167 }
2168
2169 fn measure(
2170 &self,
2171 node_id: NodeId,
2172 constraints: Constraints,
2173 ) -> Result<Rc<MeasuredNode>, NodeError> {
2174 LayoutBuilderState::measure_node(Rc::clone(&self.state), node_id, constraints)
2175 }
2176}
2177
2178#[derive(Debug, Clone)]
2179pub(crate) struct MeasuredNode {
2180 node_id: NodeId,
2181 size: Size,
2182 offset: Point,
2184 content_offset: Point,
2186 children: Vec<MeasuredChild>,
2187}
2188
2189impl MeasuredNode {
2190 fn new(
2191 node_id: NodeId,
2192 size: Size,
2193 offset: Point,
2194 content_offset: Point,
2195 children: Vec<MeasuredChild>,
2196 ) -> Self {
2197 Self {
2198 node_id,
2199 size,
2200 offset,
2201 content_offset,
2202 children,
2203 }
2204 }
2205
2206 #[cfg(test)]
2207 pub(crate) fn leaf(node_id: NodeId, size: Size) -> Self {
2208 Self::new(
2209 node_id,
2210 size,
2211 Point::default(),
2212 Point::default(),
2213 Vec::new(),
2214 )
2215 }
2216
2217 pub(crate) fn node_id(&self) -> NodeId {
2218 self.node_id
2219 }
2220
2221 pub(crate) fn size(&self) -> Size {
2222 self.size
2223 }
2224}
2225
2226#[derive(Debug, Clone)]
2227struct MeasuredChild {
2228 node: Rc<MeasuredNode>,
2229 offset: Point,
2230}
2231
2232struct ChildRecord {
2233 state: Rc<LayoutChildMeasureState>,
2234}
2235
2236struct CoordinatorFrame<'a> {
2237 measure_policy: &'a Rc<dyn MeasurePolicy>,
2238 measurables: &'a [Box<dyn Measurable>],
2239 placements: RefCell<&'a mut Vec<Placement>>,
2240 context: RefCell<LayoutNodeContext>,
2241}
2242
2243impl<'a> CoordinatorFrame<'a> {
2244 fn new(
2245 measure_policy: &'a Rc<dyn MeasurePolicy>,
2246 measurables: &'a [Box<dyn Measurable>],
2247 placements: &'a mut Vec<Placement>,
2248 ) -> Self {
2249 Self {
2250 measure_policy,
2251 measurables,
2252 placements: RefCell::new(placements),
2253 context: RefCell::new(LayoutNodeContext::new()),
2254 }
2255 }
2256
2257 fn take_invalidations(&self) -> Vec<InvalidationKind> {
2258 self.context.borrow_mut().take_invalidations()
2259 }
2260}
2261
2262struct CoordinatorLink<'chain, 'frame_ref, 'frame_data> {
2263 chain: &'chain CoordinatorChain,
2264 frame: &'frame_ref CoordinatorFrame<'frame_data>,
2265 index: usize,
2266}
2267
2268impl Measurable for CoordinatorLink<'_, '_, '_> {
2269 fn measure(&self, constraints: Constraints) -> Placeable {
2270 self.chain.measure_from(self.index, self.frame, constraints)
2271 }
2272
2273 fn min_intrinsic_width(&self, height: f32) -> f32 {
2274 self.chain
2275 .min_intrinsic_width_from(self.index, self.frame, height)
2276 }
2277
2278 fn max_intrinsic_width(&self, height: f32) -> f32 {
2279 self.chain
2280 .max_intrinsic_width_from(self.index, self.frame, height)
2281 }
2282
2283 fn min_intrinsic_height(&self, width: f32) -> f32 {
2284 self.chain
2285 .min_intrinsic_height_from(self.index, self.frame, width)
2286 }
2287
2288 fn max_intrinsic_height(&self, width: f32) -> f32 {
2289 self.chain
2290 .max_intrinsic_height_from(self.index, self.frame, width)
2291 }
2292}
2293
2294struct CoordinatorNode {
2295 modifier_index: usize,
2296 node: Rc<RefCell<Box<dyn cranpose_foundation::ModifierNode>>>,
2297 measured_size: Cell<Size>,
2298 accumulated_offset: Cell<Point>,
2299}
2300
2301impl CoordinatorNode {
2302 fn new(
2303 modifier_index: usize,
2304 node: Rc<RefCell<Box<dyn cranpose_foundation::ModifierNode>>>,
2305 ) -> Self {
2306 Self {
2307 modifier_index,
2308 node,
2309 measured_size: Cell::new(Size::default()),
2310 accumulated_offset: Cell::new(Point::default()),
2311 }
2312 }
2313
2314 fn matches(
2315 &self,
2316 modifier_index: usize,
2317 node: &Rc<RefCell<Box<dyn cranpose_foundation::ModifierNode>>>,
2318 ) -> bool {
2319 self.modifier_index == modifier_index && Rc::ptr_eq(&self.node, node)
2320 }
2321
2322 #[cfg(test)]
2323 fn ptr(&self) -> usize {
2324 Rc::as_ptr(&self.node) as *const () as usize
2325 }
2326}
2327
2328#[derive(Default)]
2329struct CoordinatorChain {
2330 nodes: Vec<CoordinatorNode>,
2331}
2332
2333impl CoordinatorChain {
2334 fn reconcile(&mut self, layout_node_data: &[LayoutModifierNodeData]) {
2335 if self.matches(layout_node_data) {
2336 return;
2337 }
2338
2339 let mut previous_nodes = std::mem::take(&mut self.nodes);
2340 self.nodes.reserve(layout_node_data.len());
2341
2342 for (modifier_index, node) in layout_node_data.iter() {
2343 if let Some(position) = previous_nodes
2344 .iter()
2345 .position(|candidate| candidate.matches(*modifier_index, node))
2346 {
2347 self.nodes.push(previous_nodes.swap_remove(position));
2348 } else {
2349 self.nodes
2350 .push(CoordinatorNode::new(*modifier_index, Rc::clone(node)));
2351 }
2352 }
2353 }
2354
2355 fn matches(&self, layout_node_data: &[LayoutModifierNodeData]) -> bool {
2356 self.nodes.len() == layout_node_data.len()
2357 && self
2358 .nodes
2359 .iter()
2360 .zip(layout_node_data.iter())
2361 .all(|(node, (modifier_index, node_rc))| node.matches(*modifier_index, node_rc))
2362 }
2363
2364 fn measure_from(
2365 &self,
2366 index: usize,
2367 frame: &CoordinatorFrame<'_>,
2368 constraints: Constraints,
2369 ) -> Placeable {
2370 let Some(node) = self.nodes.get(index) else {
2371 let mut placements = frame.placements.borrow_mut();
2372 let size =
2373 frame
2374 .measure_policy
2375 .measure_into(frame.measurables, constraints, &mut placements);
2376 return Placeable::value(size.width, size.height, NodeId::default());
2377 };
2378
2379 let wrapped = CoordinatorLink {
2380 chain: self,
2381 frame,
2382 index: index + 1,
2383 };
2384 let node_borrow = node.node.borrow();
2385
2386 let Some(layout_node) = node_borrow.as_layout_node() else {
2387 let placeable = wrapped.measure(constraints);
2388 let child_accumulated = self.total_content_offset_from(index + 1);
2389 node.accumulated_offset.set(child_accumulated);
2390 return Placeable::value_with_offset(
2391 placeable.width(),
2392 placeable.height(),
2393 NodeId::default(),
2394 (child_accumulated.x, child_accumulated.y),
2395 );
2396 };
2397
2398 let result = match frame.context.try_borrow_mut() {
2399 Ok(mut context) => layout_node.measure(&mut *context, &wrapped, constraints),
2400 Err(_) => {
2401 let mut temp = LayoutNodeContext::new();
2402 let result = layout_node.measure(&mut temp, &wrapped, constraints);
2403 if let Ok(mut context) = frame.context.try_borrow_mut() {
2404 for kind in temp.take_invalidations() {
2405 context.invalidate(kind);
2406 }
2407 }
2408 result
2409 }
2410 };
2411
2412 node.measured_size.set(result.size);
2413 let local_offset = Point {
2414 x: result.placement_offset_x,
2415 y: result.placement_offset_y,
2416 };
2417 let child_accumulated = self.total_content_offset_from(index + 1);
2418 let accumulated = Point {
2419 x: local_offset.x + child_accumulated.x,
2420 y: local_offset.y + child_accumulated.y,
2421 };
2422 node.accumulated_offset.set(accumulated);
2423
2424 Placeable::value_with_offset(
2425 result.size.width,
2426 result.size.height,
2427 NodeId::default(),
2428 (accumulated.x, accumulated.y),
2429 )
2430 }
2431
2432 fn min_intrinsic_width_from(
2433 &self,
2434 index: usize,
2435 frame: &CoordinatorFrame<'_>,
2436 height: f32,
2437 ) -> f32 {
2438 let Some(node) = self.nodes.get(index) else {
2439 return frame
2440 .measure_policy
2441 .min_intrinsic_width(frame.measurables, height);
2442 };
2443 let wrapped = CoordinatorLink {
2444 chain: self,
2445 frame,
2446 index: index + 1,
2447 };
2448 let node_borrow = node.node.borrow();
2449 node_borrow
2450 .as_layout_node()
2451 .map(|layout_node| layout_node.min_intrinsic_width(&wrapped, height))
2452 .unwrap_or_else(|| wrapped.min_intrinsic_width(height))
2453 }
2454
2455 fn max_intrinsic_width_from(
2456 &self,
2457 index: usize,
2458 frame: &CoordinatorFrame<'_>,
2459 height: f32,
2460 ) -> f32 {
2461 let Some(node) = self.nodes.get(index) else {
2462 return frame
2463 .measure_policy
2464 .max_intrinsic_width(frame.measurables, height);
2465 };
2466 let wrapped = CoordinatorLink {
2467 chain: self,
2468 frame,
2469 index: index + 1,
2470 };
2471 let node_borrow = node.node.borrow();
2472 node_borrow
2473 .as_layout_node()
2474 .map(|layout_node| layout_node.max_intrinsic_width(&wrapped, height))
2475 .unwrap_or_else(|| wrapped.max_intrinsic_width(height))
2476 }
2477
2478 fn min_intrinsic_height_from(
2479 &self,
2480 index: usize,
2481 frame: &CoordinatorFrame<'_>,
2482 width: f32,
2483 ) -> f32 {
2484 let Some(node) = self.nodes.get(index) else {
2485 return frame
2486 .measure_policy
2487 .min_intrinsic_height(frame.measurables, width);
2488 };
2489 let wrapped = CoordinatorLink {
2490 chain: self,
2491 frame,
2492 index: index + 1,
2493 };
2494 let node_borrow = node.node.borrow();
2495 node_borrow
2496 .as_layout_node()
2497 .map(|layout_node| layout_node.min_intrinsic_height(&wrapped, width))
2498 .unwrap_or_else(|| wrapped.min_intrinsic_height(width))
2499 }
2500
2501 fn max_intrinsic_height_from(
2502 &self,
2503 index: usize,
2504 frame: &CoordinatorFrame<'_>,
2505 width: f32,
2506 ) -> f32 {
2507 let Some(node) = self.nodes.get(index) else {
2508 return frame
2509 .measure_policy
2510 .max_intrinsic_height(frame.measurables, width);
2511 };
2512 let wrapped = CoordinatorLink {
2513 chain: self,
2514 frame,
2515 index: index + 1,
2516 };
2517 let node_borrow = node.node.borrow();
2518 node_borrow
2519 .as_layout_node()
2520 .map(|layout_node| layout_node.max_intrinsic_height(&wrapped, width))
2521 .unwrap_or_else(|| wrapped.max_intrinsic_height(width))
2522 }
2523
2524 fn total_content_offset_from(&self, index: usize) -> Point {
2525 self.nodes
2526 .get(index)
2527 .map(|node| node.accumulated_offset.get())
2528 .unwrap_or_default()
2529 }
2530
2531 #[cfg(test)]
2532 fn debug_ptrs(&self) -> Vec<usize> {
2533 self.nodes.iter().map(CoordinatorNode::ptr).collect()
2534 }
2535}
2536
2537#[derive(Default)]
2538pub(crate) struct LayoutRuntimeState {
2539 child_ids: Vec<NodeId>,
2540 child_states: Vec<Rc<LayoutChildMeasureState>>,
2541 child_measurables: Vec<Box<dyn Measurable>>,
2542 coordinator_chain: CoordinatorChain,
2543}
2544
2545impl LayoutRuntimeState {
2546 fn reconcile_child_measurables(&mut self, child_ids: &[NodeId]) {
2547 if self.child_ids == child_ids {
2548 return;
2549 }
2550
2551 let mut previous_ids = std::mem::take(&mut self.child_ids);
2552 let mut previous_states = std::mem::take(&mut self.child_states);
2553 let mut previous_measurables = std::mem::take(&mut self.child_measurables);
2554
2555 self.child_ids.reserve(child_ids.len());
2556 self.child_states.reserve(child_ids.len());
2557 self.child_measurables.reserve(child_ids.len());
2558
2559 for &child_id in child_ids {
2560 if let Some(position) = previous_ids.iter().position(|&id| id == child_id) {
2561 self.child_ids.push(previous_ids.swap_remove(position));
2562 self.child_states
2563 .push(previous_states.swap_remove(position));
2564 self.child_measurables
2565 .push(previous_measurables.swap_remove(position));
2566 } else {
2567 let state = LayoutChildMeasureState::new(child_id);
2568 self.child_ids.push(child_id);
2569 self.child_states.push(Rc::clone(&state));
2570 self.child_measurables
2571 .push(Box::new(LayoutChildMeasurable::new(state)));
2572 }
2573 }
2574 }
2575
2576 fn child_state(&self, index: usize) -> Rc<LayoutChildMeasureState> {
2577 Rc::clone(&self.child_states[index])
2578 }
2579
2580 fn child_measurables(&self) -> &[Box<dyn Measurable>] {
2581 self.child_measurables.as_slice()
2582 }
2583
2584 fn reconcile_coordinator_chain(&mut self, layout_node_data: &[LayoutModifierNodeData]) {
2585 self.coordinator_chain.reconcile(layout_node_data);
2586 }
2587
2588 fn coordinator_chain(&self) -> &CoordinatorChain {
2589 &self.coordinator_chain
2590 }
2591
2592 fn clear_frame_bindings(&self) {
2593 for child_state in &self.child_states {
2594 child_state.clear_frame_bindings();
2595 }
2596 }
2597
2598 #[cfg(test)]
2599 pub(crate) fn debug_stats(&self) -> LayoutRuntimeDebugStats {
2600 LayoutRuntimeDebugStats {
2601 child_ids: self.child_ids.clone(),
2602 child_state_ptrs: self
2603 .child_states
2604 .iter()
2605 .map(|state| Rc::as_ptr(state) as *const () as usize)
2606 .collect(),
2607 child_measurable_ptrs: self
2608 .child_measurables
2609 .iter()
2610 .map(|measurable| {
2611 measurable.as_ref() as *const dyn Measurable as *const () as usize
2612 })
2613 .collect(),
2614 child_measurable_count: self.child_measurables.len(),
2615 coordinator_node_ptrs: self.coordinator_chain.debug_ptrs(),
2616 coordinator_node_count: self.coordinator_chain.nodes.len(),
2617 }
2618 }
2619}
2620
2621#[cfg(test)]
2622#[derive(Debug, Clone, PartialEq, Eq)]
2623pub(crate) struct LayoutRuntimeDebugStats {
2624 pub(crate) child_ids: Vec<NodeId>,
2625 pub(crate) child_state_ptrs: Vec<usize>,
2626 pub(crate) child_measurable_ptrs: Vec<usize>,
2627 pub(crate) child_measurable_count: usize,
2628 pub(crate) coordinator_node_ptrs: Vec<usize>,
2629 pub(crate) coordinator_node_count: usize,
2630}
2631
2632struct LayoutChildMeasureConfig {
2633 applier: Rc<ConcreteApplierHost<MemoryApplier>>,
2634 node_id: NodeId,
2635 error: Rc<RefCell<Option<NodeError>>>,
2636 runtime_handle: Option<RuntimeHandle>,
2637 cache: LayoutNodeCacheHandles,
2638 cache_epoch: u64,
2639 force_remeasure: bool,
2640 measure_handle: Option<LayoutMeasureHandle>,
2641 layout_state: Option<Rc<RefCell<LayoutState>>>,
2642}
2643
2644struct LayoutChildMeasureState {
2645 applier: RefCell<Option<Rc<ConcreteApplierHost<MemoryApplier>>>>,
2646 node_id: Cell<NodeId>,
2647 measured: RefCell<Option<Rc<MeasuredNode>>>,
2648 last_position: Cell<Option<Point>>,
2649 error: RefCell<Option<Rc<RefCell<Option<NodeError>>>>>,
2650 runtime_handle: RefCell<Option<RuntimeHandle>>,
2651 cache: RefCell<LayoutNodeCacheHandles>,
2652 cache_epoch: Cell<u64>,
2653 force_remeasure: Cell<bool>,
2654 measure_handle: RefCell<Option<LayoutMeasureHandle>>,
2655 layout_state: RefCell<Option<Rc<RefCell<LayoutState>>>>,
2656}
2657
2658impl LayoutChildMeasureState {
2659 fn new(node_id: NodeId) -> Rc<Self> {
2660 Rc::new(Self {
2661 applier: RefCell::new(None),
2662 node_id: Cell::new(node_id),
2663 measured: RefCell::new(None),
2664 last_position: Cell::new(None),
2665 error: RefCell::new(None),
2666 runtime_handle: RefCell::new(None),
2667 cache: RefCell::new(LayoutNodeCacheHandles::default()),
2668 cache_epoch: Cell::new(0),
2669 force_remeasure: Cell::new(true),
2670 measure_handle: RefCell::new(None),
2671 layout_state: RefCell::new(None),
2672 })
2673 }
2674
2675 fn configure(&self, config: LayoutChildMeasureConfig) {
2676 config.cache.activate(config.cache_epoch);
2677 *self.applier.borrow_mut() = Some(config.applier);
2678 self.node_id.set(config.node_id);
2679 self.measured.borrow_mut().take();
2680 self.last_position.set(None);
2681 *self.error.borrow_mut() = Some(config.error);
2682 *self.runtime_handle.borrow_mut() = config.runtime_handle;
2683 *self.cache.borrow_mut() = config.cache;
2684 self.cache_epoch.set(config.cache_epoch);
2685 self.force_remeasure.set(config.force_remeasure);
2686 *self.measure_handle.borrow_mut() = config.measure_handle;
2687 *self.layout_state.borrow_mut() = config.layout_state;
2688 }
2689
2690 fn clear_frame_bindings(&self) {
2691 self.measured.borrow_mut().take();
2692 *self.applier.borrow_mut() = None;
2693 *self.error.borrow_mut() = None;
2694 *self.runtime_handle.borrow_mut() = None;
2695 *self.measure_handle.borrow_mut() = None;
2696 *self.layout_state.borrow_mut() = None;
2697 }
2698
2699 fn node_id(&self) -> NodeId {
2700 self.node_id.get()
2701 }
2702
2703 fn cache(&self) -> LayoutNodeCacheHandles {
2704 self.cache.borrow().clone()
2705 }
2706
2707 fn applier(&self) -> Option<Rc<ConcreteApplierHost<MemoryApplier>>> {
2708 self.applier.borrow().clone()
2709 }
2710
2711 fn layout_state(&self) -> Option<Rc<RefCell<LayoutState>>> {
2712 self.layout_state.borrow().clone()
2713 }
2714
2715 fn take_measured(&self) -> Option<Rc<MeasuredNode>> {
2716 self.measured.borrow_mut().take()
2717 }
2718
2719 fn last_position(&self) -> Option<Point> {
2720 self.last_position.get()
2721 }
2722
2723 fn set_last_position(&self, position: Point) {
2724 self.last_position.set(Some(position));
2725 }
2726
2727 fn set_measured(&self, measured: Option<Rc<MeasuredNode>>) {
2728 *self.measured.borrow_mut() = measured;
2729 }
2730
2731 fn record_error(&self, err: NodeError) {
2732 let Some(error) = self.error.borrow().clone() else {
2733 return;
2734 };
2735 let mut slot = error.borrow_mut();
2736 if slot.is_none() {
2737 *slot = Some(err);
2738 }
2739 }
2740
2741 fn perform_measure(&self, constraints: Constraints) -> Result<Rc<MeasuredNode>, NodeError> {
2742 let node_id = self.node_id();
2743 if let Some(handle) = self.measure_handle.borrow().clone() {
2744 return handle.measure(node_id, constraints);
2745 }
2746 let applier = self.applier().ok_or(NodeError::MissingContext {
2747 id: node_id,
2748 reason: "layout child applier not configured",
2749 })?;
2750 measure_node_with_host(
2751 applier,
2752 self.runtime_handle.borrow().clone(),
2753 node_id,
2754 constraints,
2755 self.cache_epoch.get(),
2756 )
2757 }
2758
2759 fn intrinsic_measure(&self, constraints: Constraints) -> Option<Rc<MeasuredNode>> {
2760 let cache = self.cache();
2761 cache.activate(self.cache_epoch.get());
2762 if !self.force_remeasure.get() {
2763 if let Some(cached) = cache.get_measurement(constraints) {
2764 return Some(cached);
2765 }
2766 }
2767
2768 match self.perform_measure(constraints) {
2769 Ok(measured) => {
2770 self.force_remeasure.set(false);
2771 cache.store_measurement(constraints, Rc::clone(&measured));
2772 Some(measured)
2773 }
2774 Err(err) => {
2775 self.record_error(err);
2776 None
2777 }
2778 }
2779 }
2780}
2781
2782struct LayoutChildMeasurable {
2783 state: Rc<LayoutChildMeasureState>,
2784}
2785
2786impl LayoutChildMeasurable {
2787 fn new(state: Rc<LayoutChildMeasureState>) -> Self {
2788 Self { state }
2789 }
2790}
2791
2792impl Measurable for LayoutChildMeasurable {
2793 fn measure(&self, constraints: Constraints) -> Placeable {
2794 let state = &self.state;
2795 let cache = state.cache();
2796 cache.activate(state.cache_epoch.get());
2797 let measured_size;
2798 if !state.force_remeasure.get() {
2799 if let Some(cached) = cache.get_measurement(constraints) {
2800 measured_size = cached.size;
2801 state.set_measured(Some(Rc::clone(&cached)));
2802 } else {
2803 match state.perform_measure(constraints) {
2804 Ok(measured) => {
2805 state.force_remeasure.set(false);
2806 measured_size = measured.size;
2807 cache.store_measurement(constraints, Rc::clone(&measured));
2808 state.set_measured(Some(measured));
2809 }
2810 Err(err) => {
2811 state.record_error(err);
2812 state.set_measured(None);
2813 measured_size = Size {
2814 width: 0.0,
2815 height: 0.0,
2816 };
2817 }
2818 }
2819 }
2820 } else {
2821 match state.perform_measure(constraints) {
2822 Ok(measured) => {
2823 state.force_remeasure.set(false);
2824 measured_size = measured.size;
2825 cache.store_measurement(constraints, Rc::clone(&measured));
2826 state.set_measured(Some(measured));
2827 }
2828 Err(err) => {
2829 state.record_error(err);
2830 state.set_measured(None);
2831 measured_size = Size {
2832 width: 0.0,
2833 height: 0.0,
2834 };
2835 }
2836 }
2837 }
2838
2839 if let Some(layout_state) = state.layout_state() {
2840 let mut layout_state = layout_state.borrow_mut();
2841 layout_state.size = measured_size;
2842 layout_state.measurement_constraints = constraints;
2843 } else if let Some(applier) = state.applier() {
2844 let Ok(mut applier) = applier.try_borrow_typed() else {
2845 return Placeable::value(
2846 measured_size.width,
2847 measured_size.height,
2848 state.node_id(),
2849 );
2850 };
2851 let _ = applier.with_node::<LayoutNode, _>(state.node_id(), |node| {
2852 node.set_measured_size(measured_size);
2853 node.set_measurement_constraints(constraints);
2854 });
2855 }
2856
2857 let state = Rc::clone(&self.state);
2858 let applier = state.applier();
2859 let node_id = state.node_id();
2860 let layout_state = state.layout_state();
2861
2862 let place_fn = Rc::new(move |x: f32, y: f32| {
2863 let internal_offset = state
2864 .measured
2865 .borrow()
2866 .as_ref()
2867 .map(|m| m.offset)
2868 .unwrap_or_default();
2869
2870 let position = Point {
2871 x: x + internal_offset.x,
2872 y: y + internal_offset.y,
2873 };
2874 state.set_last_position(position);
2875
2876 if let Some(layout_state) = &layout_state {
2877 let mut layout_state = layout_state.borrow_mut();
2878 layout_state.position = position;
2879 layout_state.is_placed = true;
2880 } else if let Some(applier) = &applier {
2881 let Ok(mut applier) = applier.try_borrow_typed() else {
2882 return;
2883 };
2884 if applier
2885 .with_node::<LayoutNode, _>(node_id, |node| {
2886 node.set_position(position);
2887 })
2888 .is_err()
2889 {
2890 let _ = applier.with_node::<SubcomposeLayoutNode, _>(node_id, |node| {
2891 node.set_position(position);
2892 });
2893 }
2894 }
2895 });
2896
2897 Placeable::with_place_fn(measured_size.width, measured_size.height, node_id, place_fn)
2898 }
2899
2900 fn min_intrinsic_width(&self, height: f32) -> f32 {
2901 let kind = IntrinsicKind::MinWidth(height);
2902 let cache = self.state.cache();
2903 cache.activate(self.state.cache_epoch.get());
2904 if !self.state.force_remeasure.get() {
2905 if let Some(value) = cache.get_intrinsic(&kind) {
2906 return value;
2907 }
2908 }
2909 let constraints = Constraints {
2910 min_width: 0.0,
2911 max_width: f32::INFINITY,
2912 min_height: height,
2913 max_height: height,
2914 };
2915 if let Some(node) = self.state.intrinsic_measure(constraints) {
2916 let value = node.size.width;
2917 cache.store_intrinsic(kind, value);
2918 value
2919 } else {
2920 0.0
2921 }
2922 }
2923
2924 fn max_intrinsic_width(&self, height: f32) -> f32 {
2925 let kind = IntrinsicKind::MaxWidth(height);
2926 let cache = self.state.cache();
2927 cache.activate(self.state.cache_epoch.get());
2928 if !self.state.force_remeasure.get() {
2929 if let Some(value) = cache.get_intrinsic(&kind) {
2930 return value;
2931 }
2932 }
2933 let constraints = Constraints {
2934 min_width: 0.0,
2935 max_width: f32::INFINITY,
2936 min_height: 0.0,
2937 max_height: height,
2938 };
2939 if let Some(node) = self.state.intrinsic_measure(constraints) {
2940 let value = node.size.width;
2941 cache.store_intrinsic(kind, value);
2942 value
2943 } else {
2944 0.0
2945 }
2946 }
2947
2948 fn min_intrinsic_height(&self, width: f32) -> f32 {
2949 let kind = IntrinsicKind::MinHeight(width);
2950 let cache = self.state.cache();
2951 cache.activate(self.state.cache_epoch.get());
2952 if !self.state.force_remeasure.get() {
2953 if let Some(value) = cache.get_intrinsic(&kind) {
2954 return value;
2955 }
2956 }
2957 let constraints = Constraints {
2958 min_width: width,
2959 max_width: width,
2960 min_height: 0.0,
2961 max_height: f32::INFINITY,
2962 };
2963 if let Some(node) = self.state.intrinsic_measure(constraints) {
2964 let value = node.size.height;
2965 cache.store_intrinsic(kind, value);
2966 value
2967 } else {
2968 0.0
2969 }
2970 }
2971
2972 fn max_intrinsic_height(&self, width: f32) -> f32 {
2973 let kind = IntrinsicKind::MaxHeight(width);
2974 let cache = self.state.cache();
2975 cache.activate(self.state.cache_epoch.get());
2976 if !self.state.force_remeasure.get() {
2977 if let Some(value) = cache.get_intrinsic(&kind) {
2978 return value;
2979 }
2980 }
2981 let constraints = Constraints {
2982 min_width: 0.0,
2983 max_width: width,
2984 min_height: 0.0,
2985 max_height: f32::INFINITY,
2986 };
2987 if let Some(node) = self.state.intrinsic_measure(constraints) {
2988 let value = node.size.height;
2989 cache.store_intrinsic(kind, value);
2990 value
2991 } else {
2992 0.0
2993 }
2994 }
2995
2996 fn flex_parent_data(&self) -> Option<cranpose_ui_layout::FlexParentData> {
2997 let applier = self.state.applier()?;
2998 let node_id = self.state.node_id();
2999 let Ok(mut applier) = applier.try_borrow_typed() else {
3000 return None;
3001 };
3002
3003 applier
3004 .with_node::<LayoutNode, _>(node_id, |layout_node| {
3005 let props = layout_node.resolved_modifiers().layout_properties();
3006 props.weight().map(|weight_data| {
3007 cranpose_ui_layout::FlexParentData::new(weight_data.weight, weight_data.fill)
3008 })
3009 })
3010 .ok()
3011 .flatten()
3012 }
3013}
3014
3015fn measure_node_with_host(
3016 applier: Rc<ConcreteApplierHost<MemoryApplier>>,
3017 runtime_handle: Option<RuntimeHandle>,
3018 node_id: NodeId,
3019 constraints: Constraints,
3020 epoch: u64,
3021) -> Result<Rc<MeasuredNode>, NodeError> {
3022 let runtime_handle = match runtime_handle {
3023 Some(handle) => Some(handle),
3024 None => applier.borrow_typed().runtime_handle(),
3025 };
3026 let mut builder = LayoutBuilder::new_with_epoch(
3027 applier,
3028 epoch,
3029 Rc::new(RefCell::new(SlotTable::default())),
3030 FrameLayoutArena::default(),
3031 );
3032 builder.set_runtime_handle(runtime_handle);
3033 builder.measure_node(node_id, constraints)
3034}
3035
3036#[derive(Clone)]
3037struct RuntimeNodeMetadata {
3038 modifier: Modifier,
3039 resolved_modifiers: ResolvedModifiers,
3040 modifier_slices: Rc<ModifierNodeSlices>,
3041 role: SemanticsRole,
3042 button_handler: Option<Rc<RefCell<dyn FnMut()>>>,
3043}
3044
3045impl Default for RuntimeNodeMetadata {
3046 fn default() -> Self {
3047 Self {
3048 modifier: Modifier::empty(),
3049 resolved_modifiers: ResolvedModifiers::default(),
3050 modifier_slices: Rc::default(),
3051 role: SemanticsRole::Unknown,
3052 button_handler: None,
3053 }
3054 }
3055}
3056
3057fn role_from_modifier_slices(modifier_slices: &ModifierNodeSlices) -> SemanticsRole {
3058 modifier_slices
3059 .text_content()
3060 .map(|text| SemanticsRole::Text {
3061 value: text.to_string(),
3062 })
3063 .unwrap_or(SemanticsRole::Layout)
3064}
3065
3066fn runtime_metadata_for(
3067 applier: &mut MemoryApplier,
3068 node_id: NodeId,
3069) -> Result<RuntimeNodeMetadata, NodeError> {
3070 if let Ok(meta) = applier.with_node::<LayoutNode, _>(node_id, |layout| {
3075 let modifier = layout.modifier.clone();
3076 let resolved_modifiers = layout.resolved_modifiers();
3077 let modifier_slices = layout.modifier_slices_snapshot();
3078 let role = role_from_modifier_slices(&modifier_slices);
3079
3080 RuntimeNodeMetadata {
3081 modifier,
3082 resolved_modifiers,
3083 modifier_slices,
3084 role,
3085 button_handler: None,
3086 }
3087 }) {
3088 return Ok(meta);
3089 }
3090
3091 if let Ok((modifier, resolved_modifiers, modifier_slices)) = applier
3093 .with_node::<SubcomposeLayoutNode, _>(node_id, |node| {
3094 (
3095 node.modifier(),
3096 node.resolved_modifiers(),
3097 node.modifier_slices_snapshot(),
3098 )
3099 })
3100 {
3101 return Ok(RuntimeNodeMetadata {
3102 modifier,
3103 resolved_modifiers,
3104 modifier_slices,
3105 role: SemanticsRole::Subcompose,
3106 button_handler: None,
3107 });
3108 }
3109 Ok(RuntimeNodeMetadata::default())
3110}
3111
3112fn clear_semantics_dirty_flags(
3113 applier: &mut MemoryApplier,
3114 node: &MeasuredNode,
3115) -> Result<(), NodeError> {
3116 match applier.with_node::<LayoutNode, _>(node.node_id, |layout| {
3117 layout.clear_needs_semantics();
3118 }) {
3119 Ok(()) => {}
3120 Err(NodeError::Missing { .. }) => {}
3121 Err(NodeError::TypeMismatch { .. }) => {
3122 match applier.with_node::<SubcomposeLayoutNode, _>(node.node_id, |subcompose| {
3123 subcompose.clear_needs_semantics();
3124 }) {
3125 Ok(()) | Err(NodeError::Missing { .. }) | Err(NodeError::TypeMismatch { .. }) => {}
3126 Err(err) => return Err(err),
3127 }
3128 }
3129 Err(err) => return Err(err),
3130 }
3131
3132 for child in &node.children {
3133 clear_semantics_dirty_flags(applier, &child.node)?;
3134 }
3135
3136 Ok(())
3137}
3138
3139fn build_semantics_tree_from_live_nodes(
3140 applier: &mut MemoryApplier,
3141 node: &MeasuredNode,
3142) -> Result<SemanticsTree, NodeError> {
3143 Ok(SemanticsTree::new(build_semantics_node_from_live_nodes(
3144 applier, node,
3145 )?))
3146}
3147
3148fn semantics_node_from_parts(
3149 node_id: NodeId,
3150 mut role: SemanticsRole,
3151 config: Option<SemanticsConfiguration>,
3152 children: Vec<SemanticsNode>,
3153) -> SemanticsNode {
3154 let mut actions = Vec::new();
3155 let mut description = None;
3156 let mut editable_text = false;
3157 let mut text_selection = None;
3158
3159 if let Some(config) = config {
3160 if config.is_button {
3161 role = SemanticsRole::Button;
3162 }
3163 if config.is_clickable {
3164 actions.push(SemanticsAction::Click {
3165 handler: SemanticsCallback::new(node_id),
3166 });
3167 }
3168 description = config.content_description;
3169 editable_text = config.is_editable_text;
3170 text_selection = config.text_selection;
3171 }
3172
3173 SemanticsNode::new(
3174 node_id,
3175 role,
3176 actions,
3177 children,
3178 description,
3179 editable_text,
3180 text_selection,
3181 )
3182}
3183
3184fn build_semantics_node_from_live_nodes(
3185 applier: &mut MemoryApplier,
3186 node: &MeasuredNode,
3187) -> Result<SemanticsNode, NodeError> {
3188 let (role, config) = match applier.with_node::<LayoutNode, _>(node.node_id, |layout| {
3189 let role = role_from_modifier_slices(&layout.modifier_slices_snapshot());
3190 let config = layout.semantics_configuration();
3191 layout.clear_needs_semantics();
3192 (role, config)
3193 }) {
3194 Ok(data) => data,
3195 Err(NodeError::TypeMismatch { .. }) | Err(NodeError::Missing { .. }) => {
3196 match applier.with_node::<SubcomposeLayoutNode, _>(node.node_id, |subcompose| {
3197 subcompose.clear_needs_semantics();
3198 (
3199 SemanticsRole::Subcompose,
3200 collect_semantics_from_modifier(&subcompose.modifier()),
3201 )
3202 }) {
3203 Ok(data) => data,
3204 Err(NodeError::TypeMismatch { .. }) | Err(NodeError::Missing { .. }) => {
3205 (SemanticsRole::Unknown, None)
3206 }
3207 Err(err) => return Err(err),
3208 }
3209 }
3210 Err(err) => return Err(err),
3211 };
3212
3213 let mut children = Vec::with_capacity(node.children.len());
3214 for child in &node.children {
3215 children.push(build_semantics_node_from_live_nodes(applier, &child.node)?);
3216 }
3217
3218 Ok(semantics_node_from_parts(
3219 node.node_id,
3220 role,
3221 config,
3222 children,
3223 ))
3224}
3225
3226fn record_semantics_allocation_stats(node: &SemanticsNode, stats: &mut LayoutAllocationDebugStats) {
3227 stats.semantics_node_count += 1;
3228 stats.semantics_action_count += node.actions.len();
3229 stats.semantics_action_capacity += node.actions.capacity();
3230 stats.semantics_child_count += node.children.len();
3231 stats.semantics_child_capacity += node.children.capacity();
3232 stats.semantics_heap_bytes += node.actions.capacity() * size_of::<SemanticsAction>();
3233 stats.semantics_heap_bytes += node.children.capacity() * size_of::<SemanticsNode>();
3234
3235 if let Some(description) = &node.description {
3236 stats.semantics_description_count += 1;
3237 stats.semantics_description_bytes += description.capacity();
3238 stats.semantics_heap_bytes += description.capacity();
3239 }
3240 if let SemanticsRole::Text { value } = &node.role {
3241 stats.semantics_text_role_bytes += value.capacity();
3242 stats.semantics_heap_bytes += value.capacity();
3243 }
3244
3245 for child in &node.children {
3246 record_semantics_allocation_stats(child, stats);
3247 }
3248}
3249
3250fn record_layout_box_allocation_stats(
3251 layout_box: &LayoutBox,
3252 stats: &mut LayoutAllocationDebugStats,
3253) {
3254 stats.layout_box_count += 1;
3255 stats.layout_box_child_count += layout_box.children.len();
3256 stats.layout_box_child_capacity += layout_box.children.capacity();
3257 stats.layout_box_heap_bytes += layout_box.children.capacity() * size_of::<LayoutBox>();
3258 stats.add_modifier_slice(layout_box.node_data.modifier_slices().debug_stats());
3259
3260 for child in &layout_box.children {
3261 record_layout_box_allocation_stats(child, stats);
3262 }
3263}
3264
3265fn build_layout_tree(
3266 applier: &mut MemoryApplier,
3267 node: &MeasuredNode,
3268) -> Result<LayoutTree, NodeError> {
3269 fn place(
3270 applier: &mut MemoryApplier,
3271 node: &MeasuredNode,
3272 origin: Point,
3273 parent_layer_translation: Point,
3281 ) -> Result<LayoutBox, NodeError> {
3282 let top_left = Point {
3284 x: origin.x + node.offset.x,
3285 y: origin.y + node.offset.y,
3286 };
3287 let rect = GeometryRect {
3288 x: top_left.x,
3289 y: top_left.y,
3290 width: node.size.width,
3291 height: node.size.height,
3292 };
3293 let info = runtime_metadata_for(applier, node.node_id)?;
3294 let kind = layout_kind_from_metadata(node.node_id, &info);
3295 let RuntimeNodeMetadata {
3296 modifier,
3297 resolved_modifiers,
3298 modifier_slices,
3299 ..
3300 } = info;
3301
3302 let layer_translation = match modifier_slices.graphics_layer() {
3303 Some(layer) => Point {
3304 x: parent_layer_translation.x + layer.translation_x,
3305 y: parent_layer_translation.y + layer.translation_y,
3306 },
3307 None => parent_layer_translation,
3308 };
3309
3310 if let Some(sink) = modifier_slices.text_field_window_origin() {
3316 sink.set(Point {
3317 x: top_left.x + layer_translation.x,
3318 y: top_left.y + layer_translation.y,
3319 });
3320 }
3321
3322 if let Some(sink) = modifier_slices.viewport_window_rect() {
3325 sink.set(GeometryRect {
3326 x: top_left.x + layer_translation.x,
3327 y: top_left.y + layer_translation.y,
3328 width: node.size.width,
3329 height: node.size.height,
3330 });
3331 }
3332
3333 let data = LayoutNodeData::new(modifier, resolved_modifiers, modifier_slices, kind);
3334 let mut children = Vec::with_capacity(node.children.len());
3335 for child in &node.children {
3336 let child_origin = Point {
3337 x: top_left.x + child.offset.x,
3338 y: top_left.y + child.offset.y,
3339 };
3340 children.push(place(
3341 applier,
3342 &child.node,
3343 child_origin,
3344 layer_translation,
3345 )?);
3346 }
3347 Ok(LayoutBox::new(
3348 node.node_id,
3349 rect,
3350 node.content_offset,
3351 data,
3352 children,
3353 ))
3354 }
3355
3356 Ok(LayoutTree::new(place(
3357 applier,
3358 node,
3359 Point { x: 0.0, y: 0.0 },
3360 Point { x: 0.0, y: 0.0 },
3361 )?))
3362}
3363
3364fn semantics_role_from_layout_box(layout_box: &LayoutBox) -> SemanticsRole {
3365 match &layout_box.node_data.kind {
3366 LayoutNodeKind::Subcompose => SemanticsRole::Subcompose,
3367 LayoutNodeKind::Spacer => SemanticsRole::Spacer,
3368 LayoutNodeKind::Unknown => SemanticsRole::Unknown,
3369 LayoutNodeKind::Button { .. } => SemanticsRole::Button,
3370 LayoutNodeKind::Layout => layout_box
3371 .node_data
3372 .modifier_slices()
3373 .text_content()
3374 .map(|text| SemanticsRole::Text {
3375 value: text.to_string(),
3376 })
3377 .unwrap_or(SemanticsRole::Layout),
3378 }
3379}
3380
3381fn build_semantics_node_from_layout_box(layout_box: &LayoutBox) -> SemanticsNode {
3382 let children = layout_box
3383 .children
3384 .iter()
3385 .map(build_semantics_node_from_layout_box)
3386 .collect();
3387
3388 semantics_node_from_parts(
3389 layout_box.node_id,
3390 semantics_role_from_layout_box(layout_box),
3391 collect_semantics_from_modifier(&layout_box.node_data.modifier),
3392 children,
3393 )
3394}
3395
3396fn layout_kind_from_metadata(_node_id: NodeId, info: &RuntimeNodeMetadata) -> LayoutNodeKind {
3397 match &info.role {
3398 SemanticsRole::Layout => LayoutNodeKind::Layout,
3399 SemanticsRole::Subcompose => LayoutNodeKind::Subcompose,
3400 SemanticsRole::Text { .. } => {
3401 LayoutNodeKind::Layout
3405 }
3406 SemanticsRole::Spacer => LayoutNodeKind::Spacer,
3407 SemanticsRole::Button => {
3408 let handler = info
3409 .button_handler
3410 .as_ref()
3411 .cloned()
3412 .unwrap_or_else(|| Rc::new(RefCell::new(|| {})));
3413 LayoutNodeKind::Button { on_click: handler }
3414 }
3415 SemanticsRole::Unknown => LayoutNodeKind::Unknown,
3416 }
3417}
3418
3419fn subtract_padding(constraints: Constraints, padding: EdgeInsets) -> Constraints {
3420 let horizontal = padding.horizontal_sum();
3421 let vertical = padding.vertical_sum();
3422 let min_width = (constraints.min_width - horizontal).max(0.0);
3423 let mut max_width = constraints.max_width;
3424 if max_width.is_finite() {
3425 max_width = (max_width - horizontal).max(0.0);
3426 }
3427 let min_height = (constraints.min_height - vertical).max(0.0);
3428 let mut max_height = constraints.max_height;
3429 if max_height.is_finite() {
3430 max_height = (max_height - vertical).max(0.0);
3431 }
3432 normalize_constraints(Constraints {
3433 min_width,
3434 max_width,
3435 min_height,
3436 max_height,
3437 })
3438}
3439
3440#[cfg(test)]
3441pub(crate) fn align_horizontal(alignment: HorizontalAlignment, available: f32, child: f32) -> f32 {
3442 match alignment {
3443 HorizontalAlignment::Start => 0.0,
3444 HorizontalAlignment::CenterHorizontally => ((available - child) / 2.0).max(0.0),
3445 HorizontalAlignment::End => (available - child).max(0.0),
3446 }
3447}
3448
3449#[cfg(test)]
3450pub(crate) fn align_vertical(alignment: VerticalAlignment, available: f32, child: f32) -> f32 {
3451 match alignment {
3452 VerticalAlignment::Top => 0.0,
3453 VerticalAlignment::CenterVertically => ((available - child) / 2.0).max(0.0),
3454 VerticalAlignment::Bottom => (available - child).max(0.0),
3455 }
3456}
3457
3458fn resolve_dimension(
3459 base: f32,
3460 explicit: DimensionConstraint,
3461 min_override: Option<f32>,
3462 max_override: Option<f32>,
3463 min_limit: f32,
3464 max_limit: f32,
3465) -> f32 {
3466 let mut min_bound = min_limit;
3467 if let Some(min_value) = min_override {
3468 min_bound = min_bound.max(min_value);
3469 }
3470
3471 let mut max_bound = if max_limit.is_finite() {
3472 max_limit
3473 } else {
3474 max_override.unwrap_or(max_limit)
3475 };
3476 if let Some(max_value) = max_override {
3477 if max_bound.is_finite() {
3478 max_bound = max_bound.min(max_value);
3479 } else {
3480 max_bound = max_value;
3481 }
3482 }
3483 if max_bound < min_bound {
3484 max_bound = min_bound;
3485 }
3486
3487 let mut size = match explicit {
3488 DimensionConstraint::Points(points) => points,
3489 DimensionConstraint::Fraction(fraction) => {
3490 if max_limit.is_finite() {
3491 max_limit * fraction.clamp(0.0, 1.0)
3492 } else {
3493 base
3494 }
3495 }
3496 DimensionConstraint::Unspecified => base,
3497 DimensionConstraint::Intrinsic(_) => base,
3500 };
3501
3502 size = clamp_dimension(size, min_bound, max_bound);
3503 size = clamp_dimension(size, min_limit, max_limit);
3504 size.max(0.0)
3505}
3506
3507fn clamp_dimension(value: f32, min: f32, max: f32) -> f32 {
3508 let mut result = value.max(min);
3509 if max.is_finite() {
3510 result = result.min(max);
3511 }
3512 result
3513}
3514
3515fn normalize_constraints(mut constraints: Constraints) -> Constraints {
3516 if constraints.max_width < constraints.min_width {
3517 constraints.max_width = constraints.min_width;
3518 }
3519 if constraints.max_height < constraints.min_height {
3520 constraints.max_height = constraints.min_height;
3521 }
3522 constraints
3523}
3524
3525#[cfg(test)]
3526#[path = "tests/layout_tests.rs"]
3527mod tests;