use ingredient stored for compartment

mesoscope · Oct 11, 2023 · 25d9839 · 25d9839
1 parent 4a887de
commit 25d9839
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/cellpack/autopack/Environment.py b/cellpack/autopack/Environment.py
@@ -3151,6 +3151,7 @@ def create_voxelization(self, image_data, image_size, voxel_size, hollow=False):
         channel_colors = []
 
         for obj in self.packed_objects.get_all():
+            mesh_store = None
             if obj.name not in image_data:
                 image_data[obj.name] = numpy.zeros(image_size, dtype=numpy.uint8)
                 if obj.color is not None:
@@ -3159,12 +3160,9 @@ def create_voxelization(self, image_data, image_size, voxel_size, hollow=False):
                         color = [int(col * 255) for col in obj.color]
                     channel_colors.append(color)
             if obj.is_compartment:
-                obj_instance = self.get_compartment_object_by_name(obj.name)
                 mesh_store = self.mesh_store
-            else: 
-                obj_instance = obj.ingredient
-                mesh_store = None
-            image_data[obj.name] = obj_instance.create_voxelization(
+
+            image_data[obj.name] = obj.ingredient.create_voxelization(
                 image_data=image_data[obj.name],
                 bounding_box=self.boundingBox,
                 voxel_size=voxel_size,