diff --git a/openmc/geometry.py b/openmc/geometry.py index 260bb9e83c..61032d9a80 100644 --- a/openmc/geometry.py +++ b/openmc/geometry.py @@ -10,6 +10,7 @@ import openmc import openmc._xml as xml from openmc.checkvalue import check_type +from .geomtrack import ElementTracker class Geometry(object): """Geometry representing a collection of surfaces, cells, and universes. @@ -87,6 +88,10 @@ class Geometry(object): """ # Create XML representation + + et = ElementTracker() + et.reset() + root_element = ET.Element("geometry") self.root_universe.create_xml_subelement(root_element) diff --git a/openmc/geomtrack.py b/openmc/geomtrack.py index bf3fcac121..f2230ce7d3 100644 --- a/openmc/geomtrack.py +++ b/openmc/geomtrack.py @@ -29,6 +29,12 @@ class ElementTracker: def add_universe(self, universe_id): self.update("universe", universe_id) + def reset(self): + self.cells = set() + self.surfaces = set() + self.lattices = set() + self.universes = set() + instance = None def __init__(self):