Set visual 2D bounds from code

crates/re_entity_db/src/entity_db.rs

@@ -232,6 +232,14 @@ impl EntityDb {
         }
     }
 
+    /// Get the latest index and value for a given dense [`re_types_core::Component`].
+    ///
+    /// This assumes that the row we get from the store contains at most one instance for this
+    /// component; it will log a warning otherwise.
+    ///
+    /// This should only be used for "mono-components" such as `Transform` and `Tensor`.
+    ///
+    /// This is a best-effort helper, it will merely log errors on failure.
     #[inline]
     pub fn latest_at_component<C: re_types_core::Component>(
         &self,
@@ -246,6 +254,14 @@ impl EntityDb {
         )
     }
 
+    /// Get the latest index and value for a given dense [`re_types_core::Component`].
+    ///
+    /// This assumes that the row we get from the store contains at most one instance for this
+    /// component; it will log a warning otherwise.
+    ///
+    /// This should only be used for "mono-components" such as `Transform` and `Tensor`.
+    ///
+    /// This is a best-effort helper, and will quietly swallow any errors.
     #[inline]
     pub fn latest_at_component_quiet<C: re_types_core::Component>(
         &self,

crates/re_query/src/latest_at/helpers.rs

@@ -62,7 +62,7 @@ impl LatestAtComponentResults {
 
     /// Returns the component data of the single instance.
     ///
-    /// This assumes that the row we get from the store only contains a single instance for this
+    /// This assumes that the row we get from the store contains at most one instance for this
     /// component; it will log a warning otherwise.
     ///
     /// This should only be used for "mono-components" such as `Transform` and `Tensor`.
@@ -77,14 +77,19 @@ impl LatestAtComponentResults {
                 re_log::debug_once!("Couldn't deserialize {component_name}: promise still pending",);
                 None
             }
-            PromiseResult::Ready(data) if data.len() == 1 => Some(data[0].clone()),
             PromiseResult::Ready(data) => {
-                re_log::log_once!(
-                    level,
-                    "Couldn't deserialize {component_name}: not a mono-batch (length: {})",
-                    data.len(),
-                );
-                None
+                match data.len() {
+                    0 => {
+                        None // Empty list = no data.
+                    }
+                    1 => Some(data[0].clone()),
+                    len => {
+                        re_log::log_once!(level,
+                            "Couldn't deserialize {component_name}: not a mono-batch (length: {len})"
+                        );
+                        None
+                    }
+                }
             }
             PromiseResult::Error(err) => {
                 re_log::log_once!(
@@ -99,7 +104,7 @@ impl LatestAtComponentResults {
 
     /// Returns the component data of the single instance as an arrow array.
     ///
-    /// This assumes that the row we get from the store only contains a single instance for this
+    /// This assumes that the row we get from the store contains at most one instance for this
     /// component; it will log a warning otherwise.
     ///
     /// This should only be used for "mono-components" such as `Transform` and `Tensor`.
@@ -118,16 +123,20 @@ impl LatestAtComponentResults {
                 re_log::debug_once!("Couldn't get {component_name}: promise still pending");
                 None
             }
-            PromiseResult::Ready(cell) if cell.as_arrow_ref().len() == 1 => {
-                Some(cell.as_arrow_ref().sliced(0, 1))
-            }
             PromiseResult::Ready(cell) => {
-                re_log::log_once!(
-                    level,
-                    "Couldn't get {component_name}: not a mono-batch (length: {})",
-                    cell.as_arrow_ref().len(),
-                );
-                None
+                match cell.as_arrow_ref().len() {
+                    0 => {
+                        None // Empty list = no data.
+                    }
+                    1 => Some(cell.as_arrow_ref().sliced(0, 1)),
+                    len => {
+                        re_log::log_once!(
+                            level,
+                            "Couldn't get {component_name}: not a mono-batch (length: {len})",
+                        );
+                        None
+                    }
+                }
             }
             PromiseResult::Error(err) => {
                 re_log::log_once!(
@@ -272,7 +281,7 @@ impl Caches {
     ///
     /// Returns `None` if the data is a promise that has yet to be resolved.
     ///
-    /// This assumes that the row we get from the store only contains a single instance for this
+    /// This assumes that the row we get from the store contains at most one instance for this
     /// component; it will generate a log message of `level` otherwise.
     ///
     /// This should only be used for "mono-components" such as `Transform` and `Tensor`.
@@ -303,18 +312,24 @@ impl Caches {
                 );
                 None
             }
-            PromiseResult::Ready(data) if data.len() == 1 => Some(LatestAtMonoResult {
-                index,
-                value: data[0].clone(),
-            }),
             PromiseResult::Ready(data) => {
-                re_log::log_once!(
-                    level,
-                    "Couldn't deserialize {entity_path}:{} @ {data_time:?}#{row_id}: not a mono-batch (length: {})",
-                    C::name(),
-                    data.len(),
-                );
-                None
+                match data.len() {
+                    0 => {
+                        None // Empty list = no data.
+                    }
+                    1 => Some(LatestAtMonoResult {
+                        index,
+                        value: data[0].clone(),
+                    }),
+                    len => {
+                        re_log::log_once!(
+                            level,
+                            "Couldn't deserialize {entity_path}:{} @ {data_time:?}#{row_id}: not a mono-batch (length: {len})",
+                            C::name(),
+                        );
+                        None
+                    }
+                }
             }
             PromiseResult::Error(err) => {
                 re_log::log_once!(
@@ -330,7 +345,7 @@ impl Caches {
 
     /// Get the latest index and value for a given dense [`re_types_core::Component`].
     ///
-    /// This assumes that the row we get from the store only contains a single instance for this
+    /// This assumes that the row we get from the store contains at most one instance for this
     /// component; it will log a warning otherwise.
     ///
     /// This should only be used for "mono-components" such as `Transform` and `Tensor`.
@@ -355,7 +370,7 @@ impl Caches {
 
     /// Get the latest index and value for a given dense [`re_types_core::Component`].
     ///
-    /// This assumes that the row we get from the store only contains a single instance for this
+    /// This assumes that the row we get from the store contains at most one instance for this
     /// component; it will return None and log a debug message otherwise.
     ///
     /// This should only be used for "mono-components" such as `Transform` and `Tensor`.

crates/re_space_view/src/lib.rs

@@ -18,8 +18,9 @@ pub use screenshot::ScreenshotMode;
 pub use space_view::SpaceViewBlueprint;
 pub use space_view_contents::SpaceViewContents;
 pub use sub_archetypes::{
-    entity_path_for_space_view_sub_archetype, query_space_view_sub_archetype,
-    query_space_view_sub_archetype_or_default, space_view_sub_archetype,
+    edit_blueprint_component, entity_path_for_space_view_sub_archetype, get_blueprint_component,
+    query_space_view_sub_archetype, query_space_view_sub_archetype_or_default,
+    space_view_sub_archetype,
 };
 pub use visual_time_range::{
     query_visual_history, time_range_boundary_to_visible_history_boundary,

crates/re_space_view/src/space_view.rs

@@ -553,6 +553,7 @@ mod tests {
             let ctx = StoreContext {
                 app_id: re_log_types::ApplicationId::unknown(),
                 blueprint: &blueprint,
+                default_blueprint: None,
                 recording: &recording,
                 bundle: &Default::default(),
                 hub: &re_viewer_context::StoreHub::test_hub(),
@@ -597,6 +598,7 @@ mod tests {
             let ctx = StoreContext {
                 app_id: re_log_types::ApplicationId::unknown(),
                 blueprint: &blueprint,
+                default_blueprint: None,
                 recording: &recording,
                 bundle: &Default::default(),
                 hub: &re_viewer_context::StoreHub::test_hub(),
@@ -647,6 +649,7 @@ mod tests {
             let ctx = StoreContext {
                 app_id: re_log_types::ApplicationId::unknown(),
                 blueprint: &blueprint,
+                default_blueprint: None,
                 recording: &recording,
                 bundle: &Default::default(),
                 hub: &re_viewer_context::StoreHub::test_hub(),
@@ -920,6 +923,7 @@ mod tests {
             let ctx = StoreContext {
                 app_id: re_log_types::ApplicationId::unknown(),
                 blueprint: &blueprint,
+                default_blueprint: None,
                 recording: &recording,
                 bundle: &Default::default(),
                 hub: &re_viewer_context::StoreHub::test_hub(),

crates/re_space_view/src/space_view_contents.rs

@@ -695,6 +695,7 @@ mod tests {
         let ctx = StoreContext {
             app_id: re_log_types::ApplicationId::unknown(),
             blueprint: &blueprint,
+            default_blueprint: None,
             recording: &recording,
             bundle: &Default::default(),
             hub: &StoreHub::test_hub(),

crates/re_space_view/src/sub_archetypes.rs

@@ -5,7 +5,7 @@ use re_entity_db::{
 };
 use re_log_types::EntityPath;
 use re_types::Archetype;
-use re_viewer_context::{external::re_entity_db::EntityTree, SpaceViewId};
+use re_viewer_context::{external::re_entity_db::EntityTree, SpaceViewId, ViewerContext};
 
 pub fn entity_path_for_space_view_sub_archetype<T: Archetype>(
     space_view_id: SpaceViewId,
@@ -69,3 +69,59 @@ where
     let (arch, path) = query_space_view_sub_archetype(space_view_id, blueprint_db, query);
     (arch.ok().flatten().unwrap_or_default(), path)
 }
+
+/// Read a single component of a blueprint archetype in a space view.
+pub fn get_blueprint_component<A: re_types::Archetype, C: re_types::Component>(
+    ctx: &ViewerContext<'_>,
+    space_view_id: SpaceViewId,
+) -> Option<C> {
+    let blueprint_db = ctx.store_context.blueprint;
+    let query = ctx.blueprint_query;
+    let path = entity_path_for_space_view_sub_archetype::<A>(space_view_id, blueprint_db.tree());
+    blueprint_db
+        .latest_at_component::<C>(&path, query)
+        .map(|x| x.value)
+}
+
+/// Edit a single component of a blueprint archetype in a space view.
+///
+/// Set to `None` to reset the value to the value in the default blueprint, if any,
+/// else will just store `None` (an empty component list) in the store.
+pub fn edit_blueprint_component<A: re_types::Archetype, C: re_types::Component + PartialEq, R>(
+    ctx: &ViewerContext<'_>,
+    space_view_id: SpaceViewId,
+    edit_component: impl FnOnce(&mut Option<C>) -> R,
+) -> R {
+    let active_blueprint = ctx.store_context.blueprint;
+    let active_path =
+        entity_path_for_space_view_sub_archetype::<A>(space_view_id, active_blueprint.tree());
+    let original_value: Option<C> = active_blueprint
+        .latest_at_component::<C>(&active_path, ctx.blueprint_query)
+        .map(|x| x.value);
+
+    let mut edited_value = original_value.clone();
+    let ret = edit_component(&mut edited_value);
+
+    if edited_value != original_value {
+        if let Some(edited) = edited_value {
+            ctx.save_blueprint_component(&active_path, &edited);
+        } else {
+            // Reset to the value in the default blueprint, if any.
+            let default_value = ctx
+                .store_context
+                .default_blueprint
+                .and_then(|default_blueprint| {
+                    let default_path = entity_path_for_space_view_sub_archetype::<A>(
+                        space_view_id,
+                        default_blueprint.tree(),
+                    );
+                    default_blueprint
+                        .latest_at_component::<C>(&default_path, ctx.blueprint_query)
+                        .map(|x| x.value)
+                });
+            ctx.save_blueprint_component(&active_path, &default_value);
+        }
+    }
+
+    ret
+}