ESS-Estia/Estia-MCNP
1
0
Fork 0
Code Actions Packages Releases Activity

Update tally plotting to use subfolder and new color range

Browse Source
This commit is contained in:
glavic_a
2019-02-22 09:27:15 +01:00
parent 64dd197753
commit 447876ffd7
1 changed files with 63 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

75
plot_tally.py
View File

@@ -13,7 +13,7 @@ from pylab import *
from matplotlib.colors import LogNorm
COLORMAP='tab20c'
SCALING=1.56e16/8.7374e15
SCALING=1.0
def read_tally(fname, use_tally=None):
if fname.endswith('.gz'):
@@ -68,9 +68,16 @@ def plot_xy(outfile, res, zidx=5, Imin=1e-6, Imax=1e4, cticks=None):
#fig.savefig(outfile)
#pc=p.get_facecolors()
#pc[:,3]=where(I[:,:,zidx].T.flatten()>0, 0.9-0.25*dI[:,:,zidx].T.flatten(), 0.)
ax.add_patch(Circle((0,0), radius=5.50, color='black', fill=False, lw=2))
ax.add_patch(Circle((0,0), radius=11.50, color='black', fill=False, lw=2))
ax.add_patch(Circle((0,0), radius=15.00, color='black', fill=False, lw=2))
ax.add_patch(Circle((-0.151,0.021), radius=5.50, color='black', fill=False, lw=2))
ax.add_patch(Circle((-0.151,0.021), radius=11.50, color='black', fill=False, lw=2))
ax.add_patch(Circle((-0.151,0.021), radius=15.00, color='black', fill=False, lw=2))
# shielding walls outside bunker
ax.add_line(Line2D([21.12-0.151, 21.12-0.151], [-1.5,1.5], color='black', lw=2))
ax.add_line(Line2D([14.845,19.426], [-0.541,-0.713], color='black', lw=2))
ax.add_line(Line2D([19.426,19.405], [-0.713,-1.248], color='black', lw=2))
ax.add_line(Line2D([19.405,21.205], [-1.248,-1.460], color='black', lw=2))
ax.add_line(Line2D([14.723, 22.740], [2.232, 3.084], color='black', lw=2))
ax.add_line(Line2D([14.801, 20.894], [-1.210, -1.724], color='black', lw=2))
if outfile is not None:
fig.savefig(outfile, transparent=False, dpi=300)
@@ -89,9 +96,12 @@ def plot_xz(outfile, res, yidx=5, Imin=1e-6, Imax=1e4, cticks=None):
#fig.savefig(outfile)
#pc=p.get_facecolors()
#pc[:,3]=where(I[:,yidx].T.flatten()>0, 0.9-0.25*dI[:,yidx].T.flatten(), 0.)
ax.plot([sqrt(5.5**2-ypos**2), sqrt(5.5**2-ypos**2)], [z.min()/100., z.max()/100.], color='black', lw=2)
ax.plot([sqrt(11.5**2-ypos**2), sqrt(11.5**2-ypos**2)], [z.min()/100., z.max()/100.], color='black', lw=2)
ax.plot([sqrt(15.0**2-ypos**2), sqrt(15.0**2-ypos**2)], [z.min()/100., z.max()/100.], color='black', lw=2)
ax.plot([sqrt(5.5**2-ypos**2)-0.151, sqrt(5.5**2-ypos**2)-0.151],
[z.min()/100., z.max()/100.], color='black', lw=2)
ax.plot([sqrt(11.5**2-ypos**2)-0.151, sqrt(11.5**2-ypos**2)-0.151],
[z.min()/100., z.max()/100.], color='black', lw=2)
ax.plot([sqrt(15.0**2-ypos**2)-0.151, sqrt(15.0**2-ypos**2)-0.151],
[z.min()/100., z.max()/100.], color='black', lw=2)
if outfile is not None:
fig.savefig(outfile, transparent=False, dpi=300)
@@ -112,26 +122,67 @@ def plot_yz(outfile, res, xidx=5, Imin=1e-6, Imax=1e4, cticks=None):
if outfile is not None:
fig.savefig(outfile, transparent=False, dpi=300)
def get_name(fname):
# extract tag-name and version info from filename
stripped=os.path.basename(fname)[9:].split('.')[0]
tag, version=stripped.split('-', 1)
return tag, version
if __name__=='__main__':
import sys, os
res=read_tally(sys.argv[1], use_tally=1)
if '-o' in sys.argv:
idx=sys.argv.index('-o')
sys.argv.pop(idx)
prefix=sys.argv.pop(idx)
else:
prefix='estia_dosemap'
print sys.argv[1]
if ':' in sys.argv[1]:
fname, tally=sys.argv[1].split(':',2)
tally=int(tally)
else:
fname=sys.argv[1]
tally=None
res=read_tally(fname, use_tally=tally)
x,y,z,I,dI=res
I*=SCALING
fpre=os.path.join('.', 'images', os.path.basename(sys.argv[1])[:-4])
fpath=os.path.join('.', 'images', get_name(fname)[0]+'-'+get_name(fname)[1])
if tally is not None:
fpath+='[%s]'%tally
if len(sys.argv)>2:
for fi in sys.argv[2:]:
print fi
if ':' in fi:
fi, tally=fi.split(':',2)
tally=int(tally)
else:
tally=None
resi=read_tally(fi, use_tally=tally)
xi,yi,zi,Ii,dIi=resi
Ii*=SCALING
I+=Ii
dI=sqrt(dI**2+dIi**2)
res=(x,y,z,I,dI)
fpath+='+'+get_name(fi)[1]
if tally is not None:
fpath+='[%s]'%tally
os.mkdir(fpath)
fpre=os.path.join(fpath, prefix)
xi, yi, zi=res[3].shape
for zidx in range(zi):
print fpre+'_xy_%02i.png'%zidx
figure(figsize=(12,8))
plot_xy(fpre+'_xy_%02i.png'%zidx, res, zidx=zidx,
Imin=1e-8, Imax=1e2, cticks=[1e-8, 1e-6, 1e-4, 1e-2, 1, 100])
Imin=1e-7, Imax=1e-2, cticks=[1e-7, 1e-6, 1e-5, 1e-4, 1e-3, 1e-2])
for yidx in range(yi):
print fpre+'_xz_%02i.png'%yidx
figure(figsize=(12,8))
plot_xz(fpre+'_xz_%02i.png'%yidx, res, yidx=yidx,
Imin=1e-8, Imax=1e2, cticks=[1e-8, 1e-6, 1e-4, 1e-2, 1, 100])
Imin=1e-7, Imax=1e-2, cticks=[1e-7, 1e-6, 1e-5, 1e-4, 1e-3, 1e-2])
for xidx in range(xi):
print fpre+'_yz_%02i.png'%xidx
figure(figsize=(8,10))
plot_yz(fpre+'_yz_%02i.png'%xidx, res, xidx=xidx,
Imin=1e-8, Imax=1e2, cticks=[1e-8, 1e-6, 1e-4, 1e-2, 1, 100])
Imin=1e-7, Imax=1e-2, cticks=[1e-7, 1e-6, 1e-5, 1e-4, 1e-3, 1e-2])