Improve transform performance (by caching affine transforms resulting from transform components)

crates/store/re_chunk/src/helpers.rs

@@ -349,7 +349,7 @@ impl UnitChunkShared {
 
     /// Returns the raw data for the specified component, assuming a mono-batch.
     ///
-    /// Returns an error if the underlying batch is not of unit length.
+    /// Returns none if the underlying batch is not of unit length.
     #[inline]
     pub fn component_mono_raw(
         &self,
@@ -369,7 +369,7 @@ impl UnitChunkShared {
 
     /// Returns the deserialized data for the specified component, assuming a mono-batch.
     ///
-    /// Returns an error if the data cannot be deserialized, or if the underlying batch is not of unit length.
+    /// Returns none if the data cannot be deserialized, or if the underlying batch is not of unit length.
     #[inline]
     pub fn component_mono<C: Component>(&self) -> Option<ChunkResult<C>> {
         let res = self.component_mono_raw(&C::name())?;

crates/store/re_query/src/latest_at.rs

@@ -508,15 +508,15 @@ impl LatestAtResults {
 
     /// Returns the deserialized data for the specified component, assuming a mono-batch.
     ///
-    /// Returns an error if the data cannot be deserialized, or if the underlying batch is not of unit length.
+    /// Logs an error if the data cannot be deserialized, or if the underlying batch is not of unit length.
     #[inline]
     pub fn component_mono<C: Component>(&self) -> Option<C> {
         self.component_mono_with_log_level(re_log::Level::Error)
     }
 
     /// Returns the deserialized data for the specified component, assuming a mono-batch.
     ///
-    /// Returns an error if the data cannot be deserialized, or if the underlying batch is not of unit length.
+    /// Returns none if the data cannot be deserialized, or if the underlying batch is not of unit length.
     #[inline]
     pub fn component_mono_quiet<C: Component>(&self) -> Option<C> {
         self.components

crates/store/re_types/src/archetypes/pinhole_ext.rs

@@ -1,8 +1,16 @@
-use crate::datatypes::Vec2D;
+use crate::{components::ViewCoordinates, datatypes::Vec2D};
 
 use super::Pinhole;
 
 impl Pinhole {
+    /// Camera orientation used when there's no camera orientation explicitly logged.
+    ///
+    /// - x pointing right
+    /// - y pointing down
+    /// - z pointing into the image plane
+    ///   (this is convenient for reading out a depth image which has typically positive z values)
+    pub const DEFAULT_CAMERA_XYZ: ViewCoordinates = ViewCoordinates::RDF;
+
     /// Creates a pinhole from the camera focal length and resolution, both specified in pixels.
     ///
     /// The focal length is the diagonal of the projection matrix.

crates/viewer/re_view/src/query.rs

@@ -215,6 +215,12 @@ pub trait DataResultQuery {
         latest_at_query: &'a LatestAtQuery,
     ) -> HybridLatestAtResults<'a>;
 
+    fn latest_at_with_blueprint_resolved_data_for_component<'a, C: re_types_core::Component>(
+        &'a self,
+        ctx: &'a ViewContext<'a>,
+        latest_at_query: &'a LatestAtQuery,
+    ) -> HybridLatestAtResults<'a>;
+
     fn query_archetype_with_history<'a, A: re_types_core::Archetype>(
         &'a self,
         ctx: &'a ViewContext<'a>,
@@ -235,14 +241,30 @@ impl DataResultQuery for DataResult {
         ctx: &'a ViewContext<'a>,
         latest_at_query: &'a LatestAtQuery,
     ) -> HybridLatestAtResults<'a> {
-        let query_shadowed_defaults = false;
+        let query_shadowed_components = false;
         latest_at_with_blueprint_resolved_data(
             ctx,
             None,
             latest_at_query,
             self,
             A::all_components().iter().map(|descr| descr.component_name),
-            query_shadowed_defaults,
+            query_shadowed_components,
+        )
+    }
+
+    fn latest_at_with_blueprint_resolved_data_for_component<'a, C: re_types_core::Component>(
+        &'a self,
+        ctx: &'a ViewContext<'a>,
+        latest_at_query: &'a LatestAtQuery,
+    ) -> HybridLatestAtResults<'a> {
+        let query_shadowed_components = false;
+        latest_at_with_blueprint_resolved_data(
+            ctx,
+            None,
+            latest_at_query,
+            self,
+            std::iter::once(C::name()),
+            query_shadowed_components,
         )
     }
 

crates/viewer/re_view_spatial/src/contexts/mod.rs

@@ -1,10 +1,10 @@
 mod depth_offsets;
-mod transform_context;
+mod transform_tree_context;
 
 pub use depth_offsets::EntityDepthOffsets;
 use re_types::ViewClassIdentifier;
 use re_view::AnnotationSceneContext;
-pub use transform_context::{TransformContext, TransformInfo, TwoDInThreeDTransformInfo};
+pub use transform_tree_context::{TransformInfo, TransformTreeContext, TwoDInThreeDTransformInfo};
 
 // -----------------------------------------------------------------------------
 
@@ -24,7 +24,7 @@ pub struct SpatialSceneEntityContext<'a> {
 pub fn register_spatial_contexts(
     system_registry: &mut re_viewer_context::ViewSystemRegistrator<'_>,
 ) -> Result<(), ViewClassRegistryError> {
-    system_registry.register_context_system::<TransformContext>()?;
+    system_registry.register_context_system::<TransformTreeContext>()?;
     system_registry.register_context_system::<EntityDepthOffsets>()?;
     system_registry.register_context_system::<AnnotationSceneContext>()?;
     Ok(())