luminarycloud 0.19.0__py3-none-any.whl → 0.22.0__py3-none-any.whl

luminarycloud/geometry_version.py

@@ -3,6 +3,7 @@ from datetime import datetime
 
 from ._client import get_default_client
 from ._helpers._timestamp_to_datetime import timestamp_to_datetime
+from ._helpers.pagination import PaginationIterator
 from ._proto.api.v0.luminarycloud.geometry import geometry_pb2 as geometrypb
 from ._proto.geometry import geometry_pb2 as gpb
 from ._wrapper import ProtoWrapper, ProtoWrapperBase
@@ -87,7 +88,7 @@ class GeometryVersion(ProtoWrapperBase):
         ]
         return surfaces, volumes
 
-    def copy_to_new_geometry(self, name: str = "") -> Geometry:
+    def copy_to_new_geometry(self, name: str = "", request_id: str = "") -> Geometry:
         """
         Copy this GeometryVersion and create a new Geometry containing only that newly copied version.
 
@@ -95,6 +96,11 @@ class GeometryVersion(ProtoWrapperBase):
         ----------
         name : str, optional
             The name of the new Geometry. If not provided, a default name will be used.
+        request_id : str, optional
+            A deduplication key, useful for idempotency. The first time copy_to_new_geometry() is
+            called with a given request_id, a new geometry will be created. Subsequent calls with
+            the same request_id will return the same geometry. If not provided, no deduplication is
+            done. Max length: 256 characters.
 
         Returns
         -------
@@ -104,6 +110,7 @@ class GeometryVersion(ProtoWrapperBase):
         req = geometrypb.CopyGeometryFromVersionRequest(
             geometry_version_id=self.id,
             name=name,
+            request_id=request_id,
         )
         res: geometrypb.CopyGeometryFromVersionResponse = (
             get_default_client().CopyGeometryFromVersion(req)
@@ -173,14 +180,39 @@ class GeometryVersion(ProtoWrapperBase):
         return res.sdk_code
 
 
+class GeometryVersionIterator(PaginationIterator[GeometryVersion]):
+    """Iterator class for geometry versions that provides length hint."""
+
+    def __init__(self, geometry_id: str, unfiltered: bool, page_size: int):
+        super().__init__(page_size)
+        self._geometry_id: str = geometry_id
+        self._unfiltered: bool = unfiltered
+
+    def _fetch_page(
+        self, page_size: int, page_token: str
+    ) -> tuple[list[GeometryVersion], str, int]:
+        req = geometrypb.ListGeometryVersionsRequest(
+            page_size=page_size,
+            page_token=page_token,
+            geometry_id=self._geometry_id,
+            unfiltered=self._unfiltered,
+        )
+        res = self._client.ListGeometryVersions(req)
+        return (
+            [GeometryVersion(gv) for gv in res.geometry_versions],
+            res.next_page_token,
+            res.total_count,
+        )
+
+
 def get_geometry_version(id: str) -> GeometryVersion:
     """
-    Get a specific geometry version with the given ID.
+    Get a geometry version by its ID.
 
     Parameters
     ----------
     id : str
-        Geometry version ID.
+        ID of the geometry version to get.
 
     Returns
     -------
@@ -191,3 +223,25 @@ def get_geometry_version(id: str) -> GeometryVersion:
     req = geometrypb.GetGeometryVersionRequest(geometry_version_id=id)
     res: geometrypb.GetGeometryVersionResponse = get_default_client().GetGeometryVersion(req)
     return GeometryVersion(res.geometry_version)
+
+
+def update_geometry_version(id: str, name: str) -> GeometryVersion:
+    """
+    Update a geometry version.
+
+    Parameters
+    ----------
+    id : str
+        ID of the geometry version to update.
+    name : str
+        The new name for the geometry version. Pass an empty string to clear the name.
+
+    Returns
+    -------
+    GeometryVersion
+        The updated GeometryVersion.
+    """
+
+    req = geometrypb.UpdateGeometryVersionRequest(geometry_version_id=id, name=name)
+    res: geometrypb.UpdateGeometryVersionResponse = get_default_client().UpdateGeometryVersion(req)
+    return GeometryVersion(res.geometry_version)

luminarycloud/mesh.py

@@ -106,8 +106,7 @@ class Mesh(ProtoWrapperBase):
             interval_seconds=interval_seconds,
             timeout_seconds=timeout_seconds,
         )
-        self._proto = get_mesh(self.id)._proto
-        return self.status
+        return self.refresh().status
 
     def refresh(self) -> "Mesh":
         """

luminarycloud/meshing/mesh_generation_params.py

@@ -108,15 +108,15 @@ class RefinementRegion(PipelineDictable):
         proto.name = self.name
         proto.h_limit = self.h_limit
         if isinstance(self.shape, Sphere):
-            proto.sphere.center.CopyFrom(self.shape.center._to_base_proto())
-            proto.sphere.radius = self.shape.radius
+            proto.sphere.center_float.CopyFrom(self.shape.center._to_base_proto())
+            proto.sphere.radius_float = self.shape.radius
         elif isinstance(self.shape, SphereShell):
             proto.sphere_shell.center.CopyFrom(self.shape.center._to_base_proto())
             proto.sphere_shell.radius = self.shape.radius
             proto.sphere_shell.radius_inner = self.shape.radius_inner
         elif isinstance(self.shape, Cube):
-            proto.cube.min.CopyFrom(self.shape.min._to_base_proto())
-            proto.cube.max.CopyFrom(self.shape.max._to_base_proto())
+            proto.cube.min_float.CopyFrom(self.shape.min._to_base_proto())
+            proto.cube.max_float.CopyFrom(self.shape.max._to_base_proto())
         elif isinstance(self.shape, OrientedCube):
             proto.oriented_cube.min.CopyFrom(self.shape.min._to_base_proto())
             proto.oriented_cube.max.CopyFrom(self.shape.max._to_base_proto())
@@ -124,9 +124,9 @@ class RefinementRegion(PipelineDictable):
             proto.oriented_cube.x_axis.CopyFrom(self.shape.x_axis._to_base_proto())
             proto.oriented_cube.y_axis.CopyFrom(self.shape.y_axis._to_base_proto())
         elif isinstance(self.shape, Cylinder):
-            proto.cylinder.start.CopyFrom(self.shape.start._to_base_proto())
-            proto.cylinder.end.CopyFrom(self.shape.end._to_base_proto())
-            proto.cylinder.radius = self.shape.radius
+            proto.cylinder.start_float.CopyFrom(self.shape.start._to_base_proto())
+            proto.cylinder.end_float.CopyFrom(self.shape.end._to_base_proto())
+            proto.cylinder.radius_float = self.shape.radius
         elif isinstance(self.shape, AnnularCylinder):
             proto.annular_cylinder.start.CopyFrom(self.shape.start._to_base_proto())
             proto.annular_cylinder.end.CopyFrom(self.shape.end._to_base_proto())
@@ -143,7 +143,7 @@ class MeshGenerationParams(PipelineDictable):
 
     geometry_id: str
     "The ID of the geometry to generate a mesh for"
-    sizing_strategy: SizingStrategy = field(default_factory=MaxCount)
+    sizing_strategy: SizingStrategy = field(default_factory=Minimal)
     "The sizing strategy to use"
 
     # Defaults copied from ts/frontend/src/lib/paramDefaults/meshGenerationState.ts