crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-28 14:15:42 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

Merge pull request #707 from wbinventor/speedup-opencg

Browse source 
Optimization to Speedup OpenCG Conversion
This commit is contained in:
Paul Romano 2016-08-18 17:27:16 -05:00 • committed by GitHub
commit ecf6340825
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=False):
"""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 OpenMC's geometric
primitives. This should be set to False if the OpenCG geometry
uses SquarePrism surfaces. False by default.
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)