Python Components now implement the ComponentBatchLike interface

crates/re_types_builder/src/codegen/python/mod.rs

@@ -390,7 +390,8 @@ impl PythonCodeGenerator {
                 Archetype,
                 BaseExtensionType,
                 BaseBatch,
-                ComponentBatchMixin
+                ComponentBatchMixin,
+                ComponentMixin,
             )
             from {rerun_path}_converters import (
                 int_or_none,
@@ -612,6 +613,10 @@ fn code_for_struct(
         ));
     }
 
+    if *kind == ObjectKind::Component {
+        superclasses.push("ComponentMixin".to_owned());
+    }
+
     if let Some(deprecation_notice) = obj.deprecation_notice() {
         code.push_unindented(format!(r#"@deprecated("""{deprecation_notice}""")"#), 1);
     }
@@ -634,6 +639,10 @@ fn code_for_struct(
 
     code.push_indented(1, quote_obj_docs(obj), 0);
 
+    if *kind == ObjectKind::Component {
+        code.push_indented(1, "_BATCH_TYPE = None", 1);
+    }
+
     if ext_class.has_init {
         code.push_indented(
             1,
@@ -775,7 +784,23 @@ fn code_for_struct(
 
     match kind {
         ObjectKind::Archetype => (),
-        ObjectKind::Datatype | ObjectKind::Component => {
+        ObjectKind::Component => {
+            code.push_indented(
+                0,
+                quote_arrow_support_from_obj(reporter, arrow_registry, ext_class, objects, obj),
+                1,
+            );
+
+            code.push_indented(
+                0,
+                format!(
+                    "# This is patched in late to avoid circular dependencies.
+{name}._BATCH_TYPE = {name}Batch  # type: ignore[assignment]"
+                ),
+                1,
+            );
+        }
+        ObjectKind::Datatype => {
             code.push_indented(
                 0,
                 quote_arrow_support_from_obj(reporter, arrow_registry, ext_class, objects, obj),

rerun_py/rerun_sdk/rerun/_baseclasses.py

@@ -275,6 +275,32 @@ def component_name(self) -> str:
         return self._ARROW_TYPE._TYPE_NAME  # type: ignore[attr-defined, no-any-return]
 
 
+class ComponentMixin(ComponentBatchLike):
+    """
+    Makes components adhere to the ComponentBatchLike interface.
+
+    A single component will always map to a batch of size 1.
+
+    The class using the mixin must define the `_BATCH_TYPE` field, which should be a subclass of `BaseBatch`.
+    """
+
+    def component_name(self) -> str:
+        """
+        The name of the component.
+
+        Part of the `ComponentBatchLike` logging interface.
+        """
+        return self._BATCH_TYPE._ARROW_TYPE._TYPE_NAME  # type: ignore[attr-defined, no-any-return]
+
+    def as_arrow_array(self) -> pa.Array:
+        """
+        The component as an arrow batch.
+
+        Part of the `ComponentBatchLike` logging interface.
+        """
+        return self._BATCH_TYPE([self]).as_arrow_array()  # type: ignore[attr-defined, no-any-return]
+
+
 @catch_and_log_exceptions(context="creating empty array")
 def _empty_pa_array(type: pa.DataType) -> pa.Array:
     if type == pa.null():