Add support for Visible History to time series space views

crates/re_arrow_store/src/store_read.rs

@@ -168,6 +168,34 @@ impl DataStore {
             .map_or(false, |table| table.all_components.contains(component))
     }
 
+    /// Find the earliest time at which something was logged for a given entity on the specified
+    /// timeline.
+    ///
+    /// # Temporal semantics
+    ///
+    /// Only considers temporal results—timeless data is ignored.
+    pub fn entity_min_time(&self, timeline: &Timeline, ent_path: &EntityPath) -> Option<TimeInt> {
+        let ent_path_hash = ent_path.hash();
+
+        let min_time = self
+            .tables
+            .get(&(*timeline, ent_path_hash))?
+            .buckets
+            .first_key_value()?
+            .1
+            .inner
+            .read()
+            .time_range
+            .min;
+
+        // handle case where no data was logged
+        if min_time == TimeInt::MIN {
+            None
+        } else {
+            Some(min_time)
+        }
+    }
+
     /// Queries the datastore for the cells of the specified `components`, as seen from the point
     /// of view of the so-called `primary` component.
     ///

crates/re_arrow_store/tests/correctness.rs

@@ -12,7 +12,7 @@ use re_arrow_store::{
 };
 use re_log_types::example_components::MyPoint;
 use re_log_types::{
-    build_frame_nr, build_log_time, DataCell, DataRow, Duration, EntityPath, RowId, Time,
+    build_frame_nr, build_log_time, DataCell, DataRow, Duration, EntityPath, RowId, Time, TimeInt,
     TimePoint, TimeType, Timeline,
 };
 use re_types::components::InstanceKey;
@@ -493,3 +493,95 @@ pub fn init_logs() {
         re_log::setup_native_logging();
     }
 }
+
+// ---
+
+#[test]
+fn entity_min_time_correct() {
+    init_logs();
+
+    for config in re_arrow_store::test_util::all_configs() {
+        let mut store = DataStore::new(InstanceKey::name(), config.clone());
+        entity_min_time_correct_impl(&mut store);
+    }
+}
+
+fn entity_min_time_correct_impl(store: &mut DataStore) {
+    let ent_path = EntityPath::from("this/that");
+    let wrong_ent_path = EntityPath::from("this/that/other");
+
+    let positions = build_some_positions2d(3);
+    let colors = build_some_colors(3);
+    let now = Time::now();
+    let now_plus_one = now + Duration::from_secs(1.0);
+    let now_minus_one = now - Duration::from_secs(1.0);
+    let row = test_row!(ent_path @ [build_log_time(now), build_frame_nr(42.into())] => 3; [positions.clone(), colors.clone()]);
+    store.insert_row(&row).unwrap();
+
+    let timeline_wrong_name = Timeline::new("lag_time", TimeType::Time);
+    let timeline_wrong_kind = Timeline::new("log_time", TimeType::Sequence);
+    let timeline_frame_nr = Timeline::new("frame_nr", TimeType::Sequence);
+    let timeline_log_time = Timeline::log_time();
+
+    assert!(store
+        .entity_min_time(&timeline_wrong_name, &ent_path)
+        .is_none());
+    assert!(store
+        .entity_min_time(&timeline_wrong_kind, &ent_path)
+        .is_none());
+    assert_eq!(
+        store.entity_min_time(&timeline_frame_nr, &ent_path),
+        Some(TimeInt::from(42))
+    );
+    assert_eq!(
+        store.entity_min_time(&timeline_log_time, &ent_path),
+        Some(TimeInt::from(now))
+    );
+    assert!(store
+        .entity_min_time(&timeline_frame_nr, &wrong_ent_path)
+        .is_none());
+
+    // insert row in the future, these shouldn't be visible
+    let row = test_row!(ent_path @ [build_log_time(now_plus_one), build_frame_nr(54.into())] => 3; [positions.clone(), colors.clone()]);
+    store.insert_row(&row).unwrap();
+
+    assert!(store
+        .entity_min_time(&timeline_wrong_name, &ent_path)
+        .is_none());
+    assert!(store
+        .entity_min_time(&timeline_wrong_kind, &ent_path)
+        .is_none());
+    assert_eq!(
+        store.entity_min_time(&timeline_frame_nr, &ent_path),
+        Some(TimeInt::from(42))
+    );
+    assert_eq!(
+        store.entity_min_time(&timeline_log_time, &ent_path),
+        Some(TimeInt::from(now))
+    );
+    assert!(store
+        .entity_min_time(&timeline_frame_nr, &wrong_ent_path)
+        .is_none());
+
+    // insert row in the past, these should be visible
+    let row = test_row!(ent_path @ [build_log_time(now_minus_one), build_frame_nr(32.into())] => 3; [positions.clone(), colors.clone()]);
+    store.insert_row(&row).unwrap();
+
+    assert!(store
+        .entity_min_time(&timeline_wrong_name, &ent_path)
+        .is_none());
+    assert!(store
+        .entity_min_time(&timeline_wrong_kind, &ent_path)
+        .is_none());
+    assert_eq!(
+        store.entity_min_time(&timeline_frame_nr, &ent_path),
+        Some(TimeInt::from(32))
+    );
+    assert_eq!(
+        store.entity_min_time(&timeline_log_time, &ent_path),
+        Some(TimeInt::from(now_minus_one))
+    );
+    assert!(store
+        .entity_min_time(&timeline_frame_nr, &wrong_ent_path)
+        .is_none());
+}

crates/re_data_store/src/entity_properties.rs

@@ -237,6 +237,11 @@ impl VisibleHistory {
         to: VisibleHistoryBoundary::AT_CURSOR,
     };
 
+    pub const ALL: Self = Self {
+        from: VisibleHistoryBoundary::Infinite,
+        to: VisibleHistoryBoundary::Infinite,
+    };
+
     pub fn from(&self, cursor: TimeInt) -> TimeInt {
         match self.from {
             VisibleHistoryBoundary::Absolute(value) => TimeInt::from(value),

crates/re_space_view/src/space_view_contents.rs

@@ -253,7 +253,7 @@ impl SpaceViewContents {
     /// Should be called on frame start.
     ///
     /// Propagates any data blueprint changes along the tree.
-    pub fn propagate_individual_to_tree(&mut self) {
+    pub fn propagate_individual_to_tree(&mut self, root_entity_property: &EntityProperties) {
         re_tracing::profile_function!();
 
         // NOTE: We could do this projection only when the entity properties changes
@@ -287,7 +287,7 @@ impl SpaceViewContents {
             }
         }
 
-        project_tree(self, &EntityProperties::default(), self.root_group_handle);
+        project_tree(self, root_entity_property, self.root_group_handle);
     }
 
     /// Adds a list of entity paths to the tree, using grouping as dictated by their entity path hierarchy.

crates/re_space_view_time_series/Cargo.toml

@@ -17,15 +17,15 @@ all-features = true
 
 [dependencies]
 re_arrow_store.workspace = true
+re_format.workspace = true
 re_log_types.workspace = true
+re_query.workspace = true
 re_renderer.workspace = true
 re_space_view.workspace = true
 re_tracing.workspace = true
 re_types.workspace = true
 re_ui.workspace = true
 re_viewer_context.workspace = true
-re_query.workspace = true
-re_format.workspace = true
 
 egui_plot.workspace = true
 egui.workspace = true

crates/re_space_view_time_series/src/space_view_class.rs

@@ -5,21 +5,41 @@ use re_format::next_grid_tick_magnitude_ns;
 use re_log_types::{EntityPath, TimeZone};
 use re_space_view::controls;
 use re_viewer_context::{
-    SpaceViewClass, SpaceViewClassName, SpaceViewClassRegistryError, SpaceViewId,
+    SpaceViewClass, SpaceViewClassName, SpaceViewClassRegistryError, SpaceViewId, SpaceViewState,
     SpaceViewSystemExecutionError, ViewContextCollection, ViewPartCollection, ViewQuery,
     ViewerContext,
 };
 
 use crate::view_part_system::{PlotSeriesKind, TimeSeriesSystem};
 
+#[derive(Clone, Default)]
+pub struct TimerSeriesSpaceViewState {
+    /// track across frames when the user moves the time cursor
+    is_dragging_time_cursor: bool,
+}
+
+impl SpaceViewState for TimerSeriesSpaceViewState {
+    fn as_any(&self) -> &dyn std::any::Any {
+        self
+    }
+
+    fn as_any_mut(&mut self) -> &mut dyn std::any::Any {
+        self
+    }
+}
+
 #[derive(Default)]
 pub struct TimeSeriesSpaceView;
 
+impl TimeSeriesSpaceView {
+    pub const NAME: &'static str = "Time Series";
+}
+
 impl SpaceViewClass for TimeSeriesSpaceView {
-    type State = ();
+    type State = TimerSeriesSpaceViewState;
 
     fn name(&self) -> SpaceViewClassName {
-        "Time Series".into()
+        Self::NAME.into()
     }
 
     fn icon(&self) -> &'static re_ui::Icon {
@@ -84,7 +104,7 @@ impl SpaceViewClass for TimeSeriesSpaceView {
         &self,
         ctx: &mut ViewerContext<'_>,
         ui: &mut egui::Ui,
-        _state: &mut Self::State,
+        state: &mut Self::State,
         _view_ctx: &ViewContextCollection,
         parts: &ViewPartCollection,
         _query: &ViewQuery<'_>,
@@ -101,16 +121,11 @@ impl SpaceViewClass for TimeSeriesSpaceView {
 
         let time_series = parts.get::<TimeSeriesSystem>()?;
 
-        // Compute the minimum time/X value for the entire plot…
-        let min_time = time_series
-            .lines
-            .iter()
-            .flat_map(|line| line.points.iter().map(|p| p.0))
-            .min()
-            .unwrap_or(0);
+        // Get the minimum time/X value for the entire plot…
+        let min_time = time_series.min_time.unwrap_or(0);
 
         // …then use that as an offset to avoid nasty precision issues with
-        // large times (nanos since epoch does not fit into an f64).
+        // large times (nanos since epoch does not fit into a f64).
         let time_offset = if timeline.typ() == TimeType::Time {
             // In order to make the tick-marks on the time axis fall on whole days, hours, minutes etc,
             // we need to round to a whole day:
@@ -200,10 +215,20 @@ impl SpaceViewClass for TimeSeriesSpaceView {
                 }
             }
 
-            current_time.map(|current_time| {
-                let time_x = (current_time - time_offset) as f64;
-                plot_ui.screen_from_plot([time_x, 0.0].into()).x
-            })
+            if state.is_dragging_time_cursor {
+                // Freeze any change to the plot boundaries to avoid weird interaction with the time
+                // cursor.
+                plot_ui.set_plot_bounds(plot_ui.plot_bounds());
+            }
+
+            // decide if the time cursor should be displayed, and if where
+            current_time
+                .map(|current_time| (current_time - time_offset) as f64)
+                .filter(|&x| {
+                    // only display the time cursor when it's actually above the plot area
+                    plot_ui.plot_bounds().min()[0] <= x && x <= plot_ui.plot_bounds().max()[0]
+                })
+                .map(|x| plot_ui.screen_from_plot([x, 0.0].into()).x)
         });
 
         if let Some(time_x) = time_x {
@@ -216,6 +241,7 @@ impl SpaceViewClass for TimeSeriesSpaceView {
                 .interact(line_rect, time_drag_id, egui::Sense::drag())
                 .on_hover_and_drag_cursor(egui::CursorIcon::ResizeHorizontal);
 
+            state.is_dragging_time_cursor = false;
             if response.dragged() {
                 if let Some(pointer_pos) = ui.input(|i| i.pointer.hover_pos()) {
                     let time =
@@ -224,6 +250,8 @@ impl SpaceViewClass for TimeSeriesSpaceView {
                     let time_ctrl = &mut ctx.rec_cfg.time_ctrl;
                     time_ctrl.set_time(time);
                     time_ctrl.pause();
+
+                    state.is_dragging_time_cursor = true;
                 }
             }
 

crates/re_space_view_time_series/src/view_part_system.rs

@@ -62,6 +62,9 @@ pub struct PlotSeries {
 pub struct TimeSeriesSystem {
     pub annotation_map: AnnotationMap,
     pub lines: Vec<PlotSeries>,
+
+    /// Earliest time an entity was recorded at on the current timeline.
+    pub min_time: Option<i64>,
 }
 
 impl NamedViewSystem for TimeSeriesSystem {
@@ -117,18 +120,29 @@ impl TimeSeriesSystem {
 
         let store = ctx.store_db.store();
 
-        for (ent_path, _ent_props) in query.iter_entities_for_system(Self::name()) {
+        for (ent_path, ent_props) in query.iter_entities_for_system(Self::name()) {
             let mut points = Vec::new();
             let annotations = self.annotation_map.find(ent_path);
             let annotation_info = annotations
                 .resolved_class_description(None)
                 .annotation_info();
             let default_color = DefaultColor::EntityPath(ent_path);
 
-            let query = re_arrow_store::RangeQuery::new(
-                query.timeline,
-                TimeRange::new(i64::MIN.into(), i64::MAX.into()),
-            );
+            let visible_history = match query.timeline.typ() {
+                re_log_types::TimeType::Time => ent_props.visible_history.nanos,
+                re_log_types::TimeType::Sequence => ent_props.visible_history.sequences,
+            };
+
+            let (from, to) = if ent_props.visible_history.enabled {
+                (
+                    visible_history.from(query.latest_at),
+                    visible_history.to(query.latest_at),
+                )
+            } else {
+                (i64::MIN.into(), i64::MAX.into())
+            };
+
+            let query = re_arrow_store::RangeQuery::new(query.timeline, TimeRange::new(from, to));
 
             let arch_views = range_archetype::<
                 TimeSeriesScalar,
@@ -171,6 +185,12 @@ impl TimeSeriesSystem {
                 continue;
             }
 
+            let min_time = store
+                .entity_min_time(&query.timeline, ent_path)
+                .map_or(points[0].time, |time| time.as_i64()); // point is not empty
+
+            self.min_time = Some(self.min_time.map_or(min_time, |time| time.min(min_time)));
+
             // If all points within a line share the label (and it isn't `None`), then we use it
             // as the whole line label for the plot legend.
             // Otherwise, we just use the entity path as-is.