crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-27 21:55:41 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

Added optimization option for OpenCG compatibility module

Browse source
This commit is contained in:
Will Boyd 2016-08-18 10:51:22 -04:00
parent b34ba8c647
commit 4acfa57062
1 changed files with 11 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

20
openmc/opencg_compatible.py
View file

@ -997,13 +997,17 @@ def get_opencg_geometry(openmc_geometry):
return opencg_geometry
def get_openmc_geometry(opencg_geometry):
def get_openmc_geometry(opencg_geometry, compatible=True):
"""Return an OpenMC geometry corresponding to an OpenCG geometry.
Parameters
----------
opencg_geometry : opencg.Geometry
OpenCG geometry
compatible : bool
Whether the OpenCG geometry is compatible with OpenMOC's geometric
primitives. This should be set to False if the OpenCG geometry
uses SquarePrism surfaces. True by default as an optimization.
Returns
-------
@ -1019,9 +1023,6 @@ def get_openmc_geometry(opencg_geometry):
opencg_geometry.assign_auto_ids()
opencg_geometry = copy.deepcopy(opencg_geometry)
# Update Cell bounding boxes in Geometry
opencg_geometry.update_bounding_boxes()
# Clear dictionaries and auto-generated ID
OPENMC_SURFACES.clear()
OPENCG_SURFACES.clear()
@ -1033,12 +1034,13 @@ def get_openmc_geometry(opencg_geometry):
OPENCG_LATTICES.clear()
# Make the entire geometry "compatible" before assigning auto IDs
universes = opencg_geometry.get_all_universes()
for universe in universes.values():
if not isinstance(universe, opencg.Lattice):
make_opencg_cells_compatible(universe)
if not compatible:
universes = opencg_geometry.get_all_universes()
for universe in universes.values():
if not isinstance(universe, opencg.Lattice):
make_opencg_cells_compatible(universe)
opencg_geometry.assign_auto_ids()
opencg_geometry.assign_auto_ids()
opencg_root_universe = opencg_geometry.root_universe
openmc_root_universe = get_openmc_universe(opencg_root_universe)