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

Merge branch 'mgxs-longer-names' into mgxs

Browse source
This commit is contained in:
Will Boyd 2015-10-13 13:26:03 -04:00
commit 23e26ea4e9
1 changed files with 59 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

59
openmc/universe.py
View file

@ -300,6 +300,27 @@ class Cell(object):
return cells
def get_all_materials(self):
"""Return all materials that are contained within the cell
Returns
-------
materials : dict
Dictionary whose keys are material IDs and values are Material instances
"""
materials = OrderedDict()
if self.fill_type == 'material':
materials[self.fill.id] = self.fill
# Append all Cells in each Cell in the Universe to the dictionary
cells = self.get_all_cells()
for cell_id, cell in cells.items():
materials.update(cell.get_all_materials())
return materials
def get_all_universes(self):
"""Return all universes that are contained within this one if any of
its cells are filled with a universe or lattice.
@ -570,6 +591,25 @@ class Universe(object):
return cells
def get_all_materials(self):
"""Return all materials that are contained within the universe
Returns
-------
materials : dict
Dictionary whose keys are material IDs and values are Material instances
"""
materials = OrderedDict()
# Append all Cells in each Cell in the Universe to the dictionary
cells = self.get_all_cells()
for cell_id, cell in cells.items():
materials.update(cell.get_all_materials())
return materials
def get_all_universes(self):
"""Return all universes that are contained within this one.
@ -775,6 +815,25 @@ class Lattice(object):
return cells
def get_all_materials(self):
"""Return all materials that are contained within the lattice
Returns
-------
materials : dict
Dictionary whose keys are material IDs and values are Material instances
"""
materials = OrderedDict()
# Append all Cells in each Cell in the Universe to the dictionary
cells = self.get_all_cells()
for cell_id, cell in cells.items():
materials.update(cell.get_all_materials())
return materials
def get_all_universes(self):
"""Return all universes that are contained within the lattice