1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81

use re_types::{
    blueprint::{
        archetypes,
        components::{Enabled, ForceDistance, ForceIterations, ForceStrength, VisualBounds2D},
    },
    components::Position2D,
    Archetype as _,
};
use re_ui::zoom_pan_area::fit_to_rect_in_scene;
use re_viewer_context::{TypedComponentFallbackProvider, ViewStateExt as _};

use crate::{ui::GraphViewState, GraphView};

fn valid_bound(rect: &egui::Rect) -> bool {
    rect.is_finite() && rect.is_positive()
}

impl TypedComponentFallbackProvider<VisualBounds2D> for GraphView {
    fn fallback_for(&self, ctx: &re_viewer_context::QueryContext<'_>) -> VisualBounds2D {
        let Ok(state) = ctx.view_state.downcast_ref::<GraphViewState>() else {
            return VisualBounds2D::default();
        };

        match state.layout_state.bounding_rect() {
            Some(rect) if valid_bound(&rect) => {
                if let Some(rect_in_ui) = state.rect_in_ui {
                    let ui_from_world = fit_to_rect_in_scene(rect_in_ui, rect);
                    (ui_from_world.inverse() * rect_in_ui).into()
                } else {
                    rect.into()
                }
            }
            _ => VisualBounds2D::default(),
        }
    }
}

impl TypedComponentFallbackProvider<Enabled> for GraphView {
    fn fallback_for(&self, ctx: &re_viewer_context::QueryContext<'_>) -> Enabled {
        match ctx.archetype_name {
            Some(name) if name == archetypes::ForceLink::name() => true.into(),
            Some(name) if name == archetypes::ForceManyBody::name() => true.into(),
            Some(name) if name == archetypes::ForcePosition::name() => true.into(),
            _ => false.into(),
        }
    }
}

impl TypedComponentFallbackProvider<ForceDistance> for GraphView {
    fn fallback_for(&self, _ctx: &re_viewer_context::QueryContext<'_>) -> ForceDistance {
        (60.).into()
    }
}

impl TypedComponentFallbackProvider<ForceStrength> for GraphView {
    fn fallback_for(&self, ctx: &re_viewer_context::QueryContext<'_>) -> ForceStrength {
        match ctx.archetype_name {
            Some(name) if name == archetypes::ForceManyBody::name() => (-60.0).into(),
            Some(name) if name == archetypes::ForcePosition::name() => (0.01).into(),
            _ => (1.0).into(),
        }
    }
}

impl TypedComponentFallbackProvider<ForceIterations> for GraphView {
    fn fallback_for(&self, ctx: &re_viewer_context::QueryContext<'_>) -> ForceIterations {
        match ctx.archetype_name {
            Some(name) if name == archetypes::ForceLink::name() => 3.into(),
            Some(name) if name == archetypes::ForceCollisionRadius::name() => 1.into(),
            _ => Default::default(),
        }
    }
}

impl TypedComponentFallbackProvider<Position2D> for GraphView {
    fn fallback_for(&self, _ctx: &re_viewer_context::QueryContext<'_>) -> Position2D {
        Default::default()
    }
}

re_viewer_context::impl_component_fallback_provider!(GraphView => [VisualBounds2D, Enabled, ForceDistance, ForceStrength, ForceIterations, Position2D]);