Adjusting source write for unstructured mesh

pshriwise · Apr 8, 2024 · 84d774c · 84d774c
1 parent e3c3e5a
commit 84d774c
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/openmc/source.py b/openmc/source.py
@@ -500,9 +500,13 @@ def populate_xml_element(self, elem: ET.Element):
         elem.set("mesh", str(self.mesh.id))
 
         # write in the order of mesh indices
-        for idx in self.mesh.indices:
-            idx = tuple(i - 1 for i in idx)
-            elem.append(self.sources[idx].to_xml_element())
+        if isinstance(self.mesh, openmc.UnstructuredMesh):
+            for s in self.sources:
+                elem.append(s.to_xml_element())
+        else:
+            for idx in self.mesh.indices:
+                idx = tuple(i - 1 for i in idx)
+                elem.append(self.sources[idx].to_xml_element())
 
     @classmethod
     def from_xml_element(cls, elem: ET.Element, meshes) -> openmc.MeshSource: