mirror of
https://github.com/openmc-dev/openmc.git
synced 2026-07-22 06:55:35 -04:00
155 lines
4.7 KiB
Python
155 lines
4.7 KiB
Python
import os
|
|
import xml.etree.ElementTree as ET
|
|
|
|
import h5py
|
|
|
|
from openmc.mixin import EqualityMixin
|
|
from openmc.clean_xml import clean_xml_indentation
|
|
from openmc.checkvalue import check_type
|
|
|
|
|
|
class DataLibrary(EqualityMixin):
|
|
"""Collection of cross section data libraries.
|
|
|
|
Attributes
|
|
----------
|
|
libraries : list of dict
|
|
List in which each item is a dictionary summarizing cross section data
|
|
from a single file. The dictionary has keys 'path', 'type', and
|
|
'materials'.
|
|
|
|
"""
|
|
|
|
def __init__(self):
|
|
self.libraries = []
|
|
|
|
def get_by_material(self, name):
|
|
"""Return the library dictionary containing a given material.
|
|
|
|
Parameters
|
|
----------
|
|
name : str
|
|
Name of material, e.g. 'Am241'
|
|
|
|
Returns
|
|
-------
|
|
library : dict or None
|
|
Dictionary summarizing cross section data from a single file;
|
|
the dictionary has keys 'path', 'type', and 'materials'.
|
|
|
|
"""
|
|
for library in self.libraries:
|
|
if name in library['materials']:
|
|
return library
|
|
return None
|
|
|
|
def register_file(self, filename):
|
|
"""Register a file with the data library.
|
|
|
|
Parameters
|
|
----------
|
|
filename : str
|
|
Path to the file to be registered.
|
|
|
|
"""
|
|
with h5py.File(filename, 'r') as h5file:
|
|
|
|
materials = []
|
|
if 'filetype' in h5file.attrs:
|
|
filetype = h5file.attrs['filetype'].decode().lstrip('data_')
|
|
else:
|
|
filetype = 'neutron'
|
|
for name in h5file:
|
|
if name.startswith('c_'):
|
|
filetype = 'thermal'
|
|
materials.append(name)
|
|
|
|
library = {'path': filename, 'type': filetype, 'materials': materials}
|
|
self.libraries.append(library)
|
|
|
|
def export_to_xml(self, path='cross_sections.xml'):
|
|
"""Export cross section data library to an XML file.
|
|
|
|
Parameters
|
|
----------
|
|
path : str
|
|
Path to file to write. Defaults to 'cross_sections.xml'.
|
|
append : bool
|
|
Whether to append to an existing file, if it exists.
|
|
Defaults to False.
|
|
|
|
"""
|
|
root = ET.Element('cross_sections')
|
|
|
|
# Determine common directory for library paths
|
|
common_dir = os.path.dirname(os.path.commonprefix(
|
|
[lib['path'] for lib in self.libraries]))
|
|
if common_dir == '':
|
|
common_dir = '.'
|
|
|
|
directory = os.path.relpath(common_dir, os.path.dirname(path))
|
|
if directory != '.':
|
|
dir_element = ET.SubElement(root, "directory")
|
|
dir_element.text = directory
|
|
|
|
for library in self.libraries:
|
|
lib_element = ET.SubElement(root, "library")
|
|
lib_element.set('materials', ' '.join(library['materials']))
|
|
lib_element.set('path', os.path.relpath(library['path'], common_dir))
|
|
lib_element.set('type', library['type'])
|
|
|
|
# Clean the indentation to be user-readable
|
|
clean_xml_indentation(root)
|
|
|
|
# Write XML file
|
|
tree = ET.ElementTree(root)
|
|
tree.write(path, xml_declaration=True, encoding='utf-8',
|
|
method='xml')
|
|
|
|
@classmethod
|
|
def from_xml(cls, path=None):
|
|
"""Read cross section data library from an XML file.
|
|
|
|
Parameters
|
|
----------
|
|
path : str, optional
|
|
Path to XML file to read. If not provided, the
|
|
`OPENMC_CROSS_SECTIONS` environment variable will be used.
|
|
|
|
Returns
|
|
-------
|
|
data : openmc.data.DataLibrary
|
|
Data library object initialized from the provided XML
|
|
|
|
"""
|
|
|
|
data = cls()
|
|
|
|
# If path is None, get the cross sections from the
|
|
# OPENMC_CROSS_SECTIONS environment variable
|
|
if path is None:
|
|
path = os.environ.get('OPENMC_CROSS_SECTIONS')
|
|
|
|
# Check to make sure there was an environmental variable.
|
|
if path is None:
|
|
raise ValueError("Either path or OPENMC_CROSS_SECTIONS "
|
|
"environmental variable must be set")
|
|
|
|
check_type('path', path, str)
|
|
|
|
tree = ET.parse(path)
|
|
root = tree.getroot()
|
|
if root.find('directory') is not None:
|
|
directory = root.find('directory').text
|
|
else:
|
|
directory = os.path.dirname(path)
|
|
|
|
for lib_element in root.findall('library'):
|
|
filename = os.path.join(directory, lib_element.attrib['path'])
|
|
filetype = lib_element.attrib['type']
|
|
materials = lib_element.attrib['materials'].split()
|
|
library = {'path': filename, 'type': filetype,
|
|
'materials': materials}
|
|
data.libraries.append(library)
|
|
|
|
return data
|