re_viewer 0.3.0-alpha.1

The Rerun viewer
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

use std::collections::BTreeMap;

use re_arrow_store::LatestAtQuery;
use re_data_store::EntityPath;
use re_log::warn_once;
use re_log_types::component_types::{self, InstanceKey, Tensor, TensorTrait as _};
use re_query::query_entity_with_primary;

use crate::{misc::ViewerContext, ui::scene::SceneQuery};

/// A bar chart scene, with everything needed to render it.
#[derive(Default)]
pub struct SceneBarChart {
    pub charts: BTreeMap<(EntityPath, InstanceKey), Tensor>,
}

impl SceneBarChart {
    pub(crate) fn load(&mut self, ctx: &mut ViewerContext<'_>, query: &SceneQuery<'_>) {
        crate::profile_function!();

        self.load_tensors(ctx, query);
    }

    fn load_tensors(&mut self, ctx: &mut ViewerContext<'_>, query: &SceneQuery<'_>) {
        crate::profile_function!();

        let store = &ctx.log_db.entity_db.data_store;

        for (ent_path, props) in query.iter_entities() {
            if !props.visible {
                continue;
            }

            let query = LatestAtQuery::new(query.timeline, query.latest_at);
            let ent_view =
                query_entity_with_primary::<component_types::Tensor>(store, &query, ent_path, &[]);
            let Ok(ent_view) = ent_view else {
                warn_once!("bar chart query failed for {ent_path:?}");
                continue;
            };
            let Ok(instance_keys) = ent_view.iter_instance_keys() else {
                warn_once!("bar chart query failed for {ent_path:?}");
                continue;
            };
            let Ok(tensors) = ent_view.iter_primary() else {
                warn_once!("bar chart query failed for {ent_path:?}");
                continue;
            };

            for (instance_key, tensor) in instance_keys.zip(tensors) {
                let tensor = tensor.unwrap(); // primary
                if tensor.is_vector() {
                    self.charts.insert(
                        (ent_path.clone(), instance_key),
                        tensor.clone(), /* shallow */
                    );
                }
            }
        }
    }
}