Version 1.12.1

NVlabs · Jan 10, 2025 · 829e30d · 829e30d
1 parent dfd1d14
commit 829e30d
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/src/scene_synthesizer/scene.py b/src/scene_synthesizer/scene.py
@@ -1008,6 +1008,7 @@ def add_walls(
         hole_extents=(0, 0),
         hole_offset=(0, 0),
         use_primitives=True,
+        object_ids=None,
     ):
         """Add walls/box assets to any combination of the six faces of the scene volume.
 
@@ -1018,9 +1019,11 @@ def add_walls(
             hole_extents (list[float], optional): 2-dim extents of the hole in the wall. Defaults to (0, 0).
             hole_offset (tuple[float], optional): 2-dim position offset of the hole in the wall. Defaults to (0, 0).
             use_primitives (bool, optional): Whether to create a mesh or use a primitive. Defaults to True.
+            object_ids (list[str], optional): A list of object names used for adding walls, needs to have same length as dimensions. If None, names will be "wall_{dim}". Defaults to None
 
         Raises:
             ValueError: If overhang is not a float, or a list of length 3 or 6.
+            ValueError: If object_ids is not None and len(object_ids) != len(dimensions).
         """
         overhangs = None
         if isinstance(overhang, float):
@@ -1040,6 +1043,9 @@ def add_walls(
         if overhangs is None:
             raise ValueError("Overhang needs to be a float or a list with 3 or 6 floats.")
 
+        if object_ids is not None and len(object_ids) != len(dimensions):
+            raise ValueError("If object_ids is set it needs to have as many elements as dimensions.")
+
         extents = {
             "x": [
                 thickness,
@@ -1085,7 +1091,11 @@ def add_walls(
             "z": ("center", "center", "bottom"),
             "-z": ("center", "center", "top"),
         }
-        for dim in dimensions:
+
+        if object_ids is None:
+            object_ids = [f"wall_{dim}" for dim in dimensions]
+
+        for dim, obj_id in zip(dimensions, object_ids):
             if all(hole_extents):
                 wall = BoxWithHoleAsset(
                     *extents[dim],
@@ -1098,7 +1108,7 @@ def add_walls(
 
             self.add_object(
                 wall,
-                f"wall_{dim}",
+                obj_id,
                 connect_parent_id=self._scene.graph.base_frame,
                 connect_parent_anchor=parent_achors[dim],
                 connect_obj_anchor=obj_anchors[dim],