re_context_menu 0.37.0

Support crate for context menu and actions.
Documentation 
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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106

use re_entity_db::InstancePath;
use re_viewer_context::{Item, ViewId};

use crate::{ContextMenuAction, ContextMenuContext};

/// Returns the resolved interactivity of `entity_path` in `view_id`, or `None` if
/// the view does not contain it.
fn entity_interactivity_in_view(
    ctx: &re_viewer_context::ViewerContext<'_>,
    view_id: ViewId,
    entity_path: &re_entity_db::EntityPath,
) -> Option<bool> {
    ctx.lookup_query_result(view_id)
        .result_for_entity(entity_path)
        .map(|dr| dr.is_interactive())
}

fn set_entity_interactivity_in_view(
    ctx: &re_viewer_context::ViewerContext<'_>,
    view_id: ViewId,
    entity_path: &re_entity_db::EntityPath,
    interactive: bool,
) {
    let query_result = ctx.lookup_query_result(view_id);
    if let Some(data_result) = query_result.result_for_entity(entity_path) {
        data_result.save_interactive(ctx, &query_result.tree, interactive);
    }
}

/// Make the selected items non-interactive
pub(crate) struct LockAction;

impl ContextMenuAction for LockAction {
    fn supports_selection(&self, ctx: &ContextMenuContext<'_>) -> bool {
        ctx.selection.iter().any(|(item, _)| {
            if let Item::DataResult(dr) = item {
                dr.instance_path.is_all()
                    && entity_interactivity_in_view(
                        ctx.viewer_context,
                        dr.view_id,
                        &dr.instance_path.entity_path,
                    )
                    .unwrap_or(false)
            } else {
                false
            }
        })
    }

    fn label(&self, _ctx: &ContextMenuContext<'_>) -> String {
        "Lock".to_owned()
    }

    fn process_data_result(
        &self,
        ctx: &ContextMenuContext<'_>,
        view_id: &ViewId,
        instance_path: &InstancePath,
    ) {
        set_entity_interactivity_in_view(
            ctx.viewer_context,
            *view_id,
            &instance_path.entity_path,
            false,
        );
    }
}

/// Make the selected items interactive again
pub(crate) struct UnlockAction;

impl ContextMenuAction for UnlockAction {
    fn supports_selection(&self, ctx: &ContextMenuContext<'_>) -> bool {
        ctx.selection.iter().any(|(item, _)| {
            if let Item::DataResult(dr) = item {
                dr.instance_path.is_all()
                    && entity_interactivity_in_view(
                        ctx.viewer_context,
                        dr.view_id,
                        &dr.instance_path.entity_path,
                    )
                    .is_some_and(|interactive| !interactive)
            } else {
                false
            }
        })
    }

    fn label(&self, _ctx: &ContextMenuContext<'_>) -> String {
        "Unlock".to_owned()
    }

    fn process_data_result(
        &self,
        ctx: &ContextMenuContext<'_>,
        view_id: &ViewId,
        instance_path: &InstancePath,
    ) {
        set_entity_interactivity_in_view(
            ctx.viewer_context,
            *view_id,
            &instance_path.entity_path,
            true,
        );
    }
}