Automatic codegen of AABB2D and RotationAxisAngle

rerun-io · Apr 29, 2024 · a41029a · a41029a
1 parent a444720
commit a41029a
Show file tree

Hide file tree

Showing 4 changed files with 26 additions and 52 deletions.
diff --git a/rerun_py/rerun_sdk/rerun/datatypes/aabb2d.py b/rerun_py/rerun_sdk/rerun/datatypes/aabb2d.py
diff --git a/rerun_py/rerun_sdk/rerun/datatypes/aabb2d_ext.py b/rerun_py/rerun_sdk/rerun/datatypes/aabb2d_ext.py
diff --git a/rerun_py/rerun_sdk/rerun/datatypes/rotation_axis_angle.py b/rerun_py/rerun_sdk/rerun/datatypes/rotation_axis_angle.py
diff --git a/rerun_py/rerun_sdk/rerun/datatypes/rotation_axis_angle_ext.py b/rerun_py/rerun_sdk/rerun/datatypes/rotation_axis_angle_ext.py
@@ -2,10 +2,8 @@
 
 from typing import TYPE_CHECKING, Any
 
-import pyarrow as pa
-
 if TYPE_CHECKING:
- from . import Angle, AngleLike, RotationAxisAngleArrayLike, Vec3DLike
+ from . import Angle, AngleLike, Vec3DLike
 
 
 class RotationAxisAngleExt:
@@ -58,21 +56,3 @@ def angle__field_converter_override(x: AngleLike) -> Angle:
  return x
  else:
  return Angle(rad=x)
-
- @staticmethod
- def native_to_pa_array_override(data: RotationAxisAngleArrayLike, data_type: pa.DataType) -> pa.Array:
- from . import AngleBatch, RotationAxisAngle, Vec3DBatch
-
- if isinstance(data, RotationAxisAngle):
- data = [data]
-
- axis_pa_array = Vec3DBatch._native_to_pa_array([rotation.axis for rotation in data], data_type["axis"].type)
- angle_pa_arr = AngleBatch._native_to_pa_array([rotation.angle for rotation in data], data_type["angle"].type)
-
- return pa.StructArray.from_arrays(
- [
- axis_pa_array,
- angle_pa_arr,
- ],
- fields=list(data_type),
- )