crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-29 06:35:48 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

finished statepoint silomesh parser

Browse source
This commit is contained in:
nhorelik 2013-04-08 00:23:53 -04:00
parent 70f7f4c3fa
commit 6b2393975f
1 changed files with 121 additions and 51 deletions
Download patch file Download diff file Expand all files Collapse all files

172
src/utils/statepoint_silomesh.py
View file

@ -1,36 +1,37 @@
#!/usr/bin/env python
import itertools
import re
import warnings
from statepoint import StatePoint
import silomesh # https://github.com/nhorelik/silomesh/
from statepoint import StatePoint
################################################################################
def parse_options():
"""Process command line arguments"""
def tallies_callback(option, opt, value, parser):
"""Option parser function for list of tallies"""
try:
setattr(parser.values, option.dest, [int(v) for v in value.split(',')])
except:
p.print_help()
def tallies_callback(option, opt, value, parser):
"""Option parser function for list of tallies"""
try:
setattr(parser.values, option.dest, [int(v) for v in value.split(',')])
except: p.print_help()
def scores_callback(option, opt, value, parser):
"""Option parser function for list of scores"""
try:
scores = {}
entries = value.split(',')
for e in entries:
tally,score = [int(i) for i in e.split('.')]
if not tally in scores: scores[tally] = []
scores[tally].append(score)
setattr(parser.values, option.dest, scores)
except:
p.print_help()
from optparse import OptionParser
usage = r"""%prog [options] <statepoint_file>
def scores_callback(option, opt, value, parser):
"""Option parser function for list of scores"""
try:
scores = {}
entries = value.split(',')
for e in entries:
tally,score = [int(i) for i in e.split('.')]
if not tally in scores: scores[tally] = []
scores[tally].append(score)
setattr(parser.values, option.dest, scores)
except: p.print_help()
from optparse import OptionParser
usage = r"""%prog [options] <statepoint_file>
The default is to process all tallies and all scores into one silo file. Subsets
can be chosen using the options. For example, to only process tallies 2 and 4
@ -41,55 +42,123 @@ with all scores on tally 2 and only scores 1 and 3 on tally 4:
You can list the available tallies and scores with the -l option:
%prog -l <statepoint_file>"""
p = OptionParser(usage=usage)
p.add_option('-t', '--tallies', dest='tallies', type='string', default=None,
action='callback', callback=tallies_callback,
help='List of tally indices to process, separated by commas.' \
' Default is to process all tallies.')
p.add_option('-s', '--scores', dest='scores', type='string', default=None,
action='callback', callback=scores_callback,
help='List of score indices to process, separated by commas, ' \
'specified as {tallyid}.{scoreid}.' \
' Default is to process all scores in each tally.')
p.add_option('-l', '--list', dest='list', action='store_true',
help='List the tally and score indices available in the file.')
parsed = p.parse_args()
if not parsed[1]: p.print_help()
p = OptionParser(usage=usage)
p.add_option('-t', '--tallies', dest='tallies', type='string', default=None,
action='callback', callback=tallies_callback,
help='List of tally indices to process, separated by commas.' \
' Default is to process all tallies.')
p.add_option('-s', '--scores', dest='scores', type='string', default=None,
action='callback', callback=scores_callback,
help='List of score indices to process, separated by commas, ' \
'specified as {tallyid}.{scoreid}.' \
' Default is to process all scores in each tally.')
p.add_option('-l', '--list', dest='list', action='store_true',
help='List the tally and score indices available in the file.')
p.add_option('-o', '--output', action='store', dest='output',
default='tally.silo', help='path to output SILO file.')
p.add_option('-e', '--error', dest='valerr', default=False,
action='store_true',
help='Flag to extract errors instead of values.')
parsed = p.parse_args()
if not parsed[1]:
p.print_help()
return
return parsed
if parsed[0].valerr:
parsed[0].valerr = 1
else:
parsed[0].valerr = 0
return parsed
################################################################################
def main(file_,o):
def main(file_, o):
"""Main program"""
sp = StatePoint(file_)
sp.read_results()
validate_options(sp,o)
validate_options(sp, o)
if o.list:
print_available(sp)
return
pattern = re.compile(r"[\W_]+")
silomesh.init_silo(o.output)
for tally in sp.tallies:
if o.tallies and not tally.id in o.tallies: continue
if not 'mesh' in tally.filters: continue
nx,ny,nz = get_mesh_dims(sp, tally)
print o.tallies
print o.scores
print "Processing Tally {}...".format(tally.id)
filtercombos = get_filter_combos(tally)
meshparms = get_mesh_parms(sp, tally)
nx,ny,nz = meshparms[0],meshparms[1],meshparms[2]
silomesh.init_mesh('Tally_{}'.format(tally.id),*meshparms)
for sid,score in enumerate(tally.scores):
if o.scores and tally.id in o.scores and not sid in o.scores[tally.id]:
continue
for filterspec in filtercombos:
comboname = "_"+" ".join(["{}_{}".format(filter_,bin)
for filter_,bin in filterspec[1:]])
if len(filterspec[1:]) == 0: comboname = ''
varname = 'Tally_{}_{}{}'.format(tally.id,score,comboname)
varname = pattern.sub('_',varname)
print "\t Score {}.{} {}:\t\t{}".format(tally.id,sid,score,varname)
silomesh.init_var(varname)
for x in range(1,nx+1):
for y in range(1,ny+1):
for z in range(1,nz+1):
filterspec[0][1] = (x,y,z)
val = sp.get_value(tally.id-1,filterspec,sid)[o.valerr]
silomesh.set_value(float(val),x,y,z)
silomesh.finalize_var()
silomesh.finalize_mesh()
silomesh.finalize_silo()
################################################################################
def get_filter_combos(tally):
"""Returns a list of all filter spec combinations, excluding meshes
def get_mesh_dims(sp, tally):
Each combo has the mesh spec as the first element, to be set later.
These filter specs correspond with the second argument to StatePoint.get_value
"""
specs = []
if len(tally.filters) == 1:
return [[['mesh',[1,1,1]]]]
filters = tally.filters.keys()
filters.pop(filters.index('mesh'))
nbins = [tally.filters[f].length for f in filters]
combos = [ [b] for b in range(nbins[0])]
for i,b in enumerate(nbins[1:]):
prod = list(itertools.product(combos,range(b)))
if i == 0:
combos = prod
else:
combos = [[v for v in p[0]]+[p[1]] for p in prod]
for c in combos:
spec = [['mesh',[1,1,1]]]
for i,bin in enumerate(c):
spec.append((filters[i],bin))
specs.append(spec)
return specs
################################################################################
def get_mesh_parms(sp, tally):
meshid = tally.filters['mesh'].bins[0]
for i,m in enumerate(sp.meshes):
if m.id == meshid:
mesh = m
return mesh.dimension
return mesh.dimension + mesh.lower_left + mesh.upper_right
################################################################################
def print_available(sp):
"""Prints available tallies/scores in a statepoint"""
@ -103,6 +172,7 @@ def print_available(sp):
for score in scores:
print "\t\tScore {}".format(score)
################################################################################
def validate_options(sp,o):
"""Validates specified tally/score options for the current statepoint"""
@ -127,15 +197,15 @@ def validate_options(sp,o):
if not otally in available_tallies:
warnings.warn('Tally {} not in statepoint file'.format(otally))
continue
################################################################################
# monkeypatch to suppress the source echo produced by warnings
def formatwarning(message, category, filename, lineno, line):
return "%s:%s: %s: %s\n" % (filename, lineno, category.__name__, message)
warnings.formatwarning = formatwarning
################################################################################
if __name__ == '__main__':
(options, args) = parse_options()
if args:
main(args[0],options)