P283370
/
NuMRI
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
Browse Source

mod

master
jeremias 2 years ago
parent
0dd4e01188
commit
2d3873605d
2 changed files with 112 additions and 124 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 171
      codes/Graphics.py
  2. 65
      codes/PostCheck.py

171
codes/Graphics.py
View File

@ -132,7 +132,6 @@ def Plot_Histogram(meshnames,paths,colors,outpath):
ind = 0
for nm in range(len(meshnames)):
if 'box' in meshnames[nm]:
# VOXEL SEQUENCE SIZE
Lx = 64
@ -167,13 +166,11 @@ def Plot_Histogram(meshnames,paths,colors,outpath):
heights_method = False
for k in range(mesh.num_cells()):
# Coordinates of each cell (A,B,C,D)
A = Points[Cells[k]][0]
B = Points[Cells[k]][1]
C = Points[Cells[k]][2]
D = Points[Cells[k]][3]
if heights_method:
hs = np.zeros([4])
for l in range(4):
@ -1412,127 +1409,79 @@ def ROUTINE(options):
if 'Error-curves' in options:
if options['Error-curves']['apply']:
print('--- Error curves ---')
meas = {}
corrs = {}
times = {}
norm = options['Error-curves']['norm']
colorset = options['Error-curves']['colors']
for resol in options['Error-curves']['resol']:
meas[resol] = {}
corrs[resol] = {}
times[resol] = {}
for dts in options['Error-curves']['times']:
path = options['Error-curves']['folder'] + \
resol + '/dt' + dts
try:
corrs[resol][dts] = np.loadtxt(
path + '/w_' + str(norm) + 'norm.txt')
except IOError:
print('no curve for ' + resol + '/' + dts)
try:
meas[resol][dts] = np.loadtxt(
path + '/u_' + str(norm) + 'norm.txt')
times[resol][dts] = np.loadtxt(
path + '/times_' + str(norm) + 'norm.txt')
except IOError:
print('no meas for ' + resol + '/' + dts)
if 'meas' in options['Error-curves']['mode']:
for resol in options['Error-curves']['resol']:
for dts in meas[resol].keys():
if resol == 'H1':
nc = 0
elif resol == 'H2':
nc = 1
elif resol == 'H3':
nc = 2
plt.plot(times[resol][dts], meas[resol]
[dts], color=colorset[nc], label='$' + resol + '/' + dts + '$')
#plt.ylim([0, 1.1])
plt.ylabel('$||u||_2$', fontsize=18)
plt.xlabel('$time \ \ (s)$', fontsize=18)
plt.legend(fontsize=16)
plt.savefig(options['Error-curves']['outpath'] + 'meas_' +dts + '.png',
dpi=500, bbox_inches='tight')
plt.show()
if 'corrs' in options['Error-curves']['mode']:
for resol in options['Error-curves']['resol']:
for dts in corrs[resol].keys():
if resol == 'H1':
nc = 0
elif resol == 'H2':
nc = 1
elif resol == 'H3':
nc = 2
plt.plot(times[resol][dts], corrs[resol]
[dts], color = colorset[nc],label='$' + resol + '/' + dts + '$')
plt.ylim([0, 16.2])
plt.ylabel('$||w||_2$', fontsize=18)
plt.xlabel('$time \ \ (s)$', fontsize=18)
plt.legend(fontsize=16)
plt.savefig(options['Error-curves']['outpath'] + 'corrs_' +dts + '.png',
dpi=500, bbox_inches='tight')
plt.show()
if 'fracs' in options['Error-curves']['mode']:
for resol in options['Error-curves']['resol']:
for dts in corrs[resol].keys():
if resol == 'H1':
nc = 0
elif resol == 'H2':
nc = 1
elif resol == 'H3':
nc = 2
plt.plot(times[resol][dts], corrs[resol][dts]/meas[resol][dts],
color=colorset[nc], label='$' + resol + '/' + dts + '$')
plt.ylim([0, 1])
plt.ylabel('$||w||_2/||u||_2$', fontsize=18)
plt.xlabel('$time \ \ (s)$', fontsize=18)
plt.legend(fontsize=16)
plt.savefig(options['Error-curves']['outpath'] + 'fracs_' +dts + '.png',
dpi=500, bbox_inches='tight')
plt.show()
if 'Components' in options:
if options['Components']['apply']:
print('--- Components analysis ---')
for types in options['Components']['type']:
print('--- Error-curves analysis ---')
ratio = False
for types in options['Error-curves']['type']:
if types=='mean_ratio':
types = 'mean'
ratio = True
if types=='max_ratio':
types = 'max'
ratio = True
nc = 0
for subf in options['Components']['subfolders']:
if len(options['Error-curves']['subfolders'])==0:
ucomp = []
wcomp = []
colorset = options['Components']['colors']
path = options['Components']['folder'] + subf + '/'
colorset = options['Error-curves']['colors']
path = options['Error-curves']['folder']
try:
ucomp = np.loadtxt(path + '/u'+types+'.txt')
wcomp = np.loadtxt(path + '/w'+types+'.txt')
times = np.loadtxt(path + '/times.txt')
except IOError:
print('no components for ' + subf)
plt.plot(
times, ucomp, color=colorset[nc], linestyle='-', label= '$u'+ subf +'$' )
plt.plot(
times, wcomp, color=colorset[nc], linestyle='--', label='$w'+subf+'$')
print('no Error-curves for ' + subf)
if not ratio:
plt.plot(
times, ucomp, color=colorset[nc], linestyle='-', label= '$u$' )
plt.plot(
times, wcomp, color=colorset[nc], linestyle='--', label='$w$')
else:
plt.plot(
times, ucomp, color=colorset[nc], linestyle='-', label= '$u$' )
plt.plot(
times, wcomp, color=colorset[nc], linestyle='--', label='$w$')
nc +=1
else:
for subf in options['Error-curves']['subfolders']:
ucomp = []
wcomp = []
colorset = options['Error-curves']['colors']
labelset = options['Error-curves']['labels']
path = options['Error-curves']['folder'] + subf + '/'
try:
ucomp = np.loadtxt(path + '/u'+types+'.txt')
wcomp = np.loadtxt(path + '/w'+types+'.txt')
times = np.loadtxt(path + '/times.txt')
except IOError:
print('no cError-curves for ' + subf)
if not ratio:
plt.plot(
times, ucomp, color=colorset[nc], linestyle='-', label= '$u'+ subf +'$' )
plt.plot(
times, wcomp, color=colorset[nc], linestyle='--', label='$w'+subf+'$')
else:
wu = wcomp/ucomp
plt.plot(
times, wu, color=colorset[nc], linestyle='-', label= '$'+ labelset[nc] +'$' )
nc +=1
#plt.ylim([0, 170])
plt.ylabel('$velocity \ \ (cm/s)$', fontsize=18)
plt.xlabel('$time \ \ (s)$', fontsize=18)
plt.legend(fontsize=16)
plt.savefig(options['Components']['outpath'] + types + '.png', dpi=500, bbox_inches='tight')
if options['Error-curves']['title']:
plt.title(options['Error-curves']['title'], fontsize=18)
if not ratio:
plt.ylabel('$velocity \ \ (cm/s)$', fontsize=18)
plt.savefig(options['Error-curves']['outpath'] + types + '.png', dpi=500, bbox_inches='tight')
else:
plt.ylabel('$w/u$', fontsize=18)
if 'max' in types:
plt.ylim([0, 1.8])
plt.savefig(options['Error-curves']['outpath'] + types + '_ratio.png', dpi=500, bbox_inches='tight')
plt.show()

65
codes/PostCheck.py
View File

@ -175,6 +175,37 @@ def READcheckpoint(MESH, mode, output_path, checkpoint_path, filename, outname,
for bb in bnds:
QQ[bb] = []
if mode == 'Histograms':
u = Function(W)
if 'w_COR' in filename:
h5name = 'w/vector_0'
else:
h5name = 'u/vector_0'
ValuesTime = np.zeros([len(indexes),len(u.vector().get_local())])
ValuesPeak = np.zeros([len(u.vector().get_local())])
ix = int(0)
for k in indexes:
path = checkpoint_path + str(k) + '/'+filename+'.h5'
hdf = HDF5File(MESH['mesh'].mpi_comm(), path, 'r')
print('Reading checkpoint ',indexes[k])
hdf.read(u, h5name)
time = hdf.attributes(h5name).to_dict()['timestamp']
hdf.close()
uvec = u.vector().get_local()
ValuesTime[ix,:] = np.abs(uvec)
ix +=1
CoordPeak = np.where(ValuesTime==np.max(ValuesTime))
ValuesPeak = np.abs(ValuesTime[CoordPeak[0],:])
freq,edges = np.histogram(ValuesPeak, bins=80, density=True)
#Saving the histogram
print('Saving at ' + output_path)
np.savetxt(output_path + 'hist_frew.txt', freq)
np.savetxt(output_path + 'hist_edges.txt', edges)
if mode == 'perturbation':
u = Function(W)
unew = Function(W)
@ -216,10 +247,10 @@ def READcheckpoint(MESH, mode, output_path, checkpoint_path, filename, outname,
ufactor = options['Perturbation']['type']['phase_contrast']/100
VENC = np.max(np.abs(uvec))*ufactor
gamma = 267.513e6 # rad/Tesla/sec Gyromagnetic ratio for H nuclei
#B0 = 1.5 # Tesla Magnetic Field Strenght
B0 = 1.5 # Tesla Magnetic Field Strenght
TE = 5e-3 # Echo-time
Phi1 = gamma*10*TE + 0*uvec
Phi2 = gamma*10*TE + np.pi*uvec/VENC
Phi1 = 1*B0*TE + 0*uvec
Phi2 = 1*B0*TE + np.pi*uvec/VENC
M1 = np.cos(Phi1) + 1j*np.sin(Phi1)
M2 = np.cos(Phi2) + 1j*np.sin(Phi2)
@ -476,7 +507,10 @@ def ERRORmap(MESH, mode, outpath, reference_path, checkpoint_path, refname,comna
u.rename('meas', 'meas')
w.rename('w', 'w')
hdf_w = HDF5File(MESH['mesh'].mpi_comm(),path_w,'r')
hdf_w.read(w, 'w/vector_0')
if 'w' in comname:
hdf_w.read(w, 'w/vector_0')
else:
hdf_w.read(w, 'u/vector_0')
hdf_u = HDF5File(MESH['mesh'].mpi_comm(), path_u, 'r')
hdf_u.read(u, 'u/vector_0')
hdf_u.close()
@ -917,7 +951,6 @@ def OUTLETwind(MESH, output_path, checkpoint_path, filename, bnds):
unsort_indexes = os.listdir(checkpoint_path)
indexes = [int(x) for x in unsort_indexes]
indexes.sort()
QQ = {}
PP = {}
ds = Measure('ds', domain=MESH['mesh'], subdomain_data=MESH['boundaries'])
@ -930,11 +963,6 @@ def OUTLETwind(MESH, output_path, checkpoint_path, filename, bnds):
for k in range(0, len(indexes)):
path_u = checkpoint_path + str(indexes[k]) + '/'+filename[0]+'.h5'
path_p = checkpoint_path + str(indexes[k]) + '/'+filename[1]+'.h5'
if indexes[k] < 10:
path_u = checkpoint_path + '0' + \
str(indexes[k]) + '/'+filename[0]+'.h5'
path_p = checkpoint_path + '0' + \
str(indexes[k]) + '/'+filename[1]+'.h5'
u = Function(V)
p = Function(W)
@ -946,7 +974,7 @@ def OUTLETwind(MESH, output_path, checkpoint_path, filename, bnds):
hdf_p.read(p, 'p/vector_0')
hdf_p.close()
print('Computing flows and pressures at timestep number ' + str(k))
print('Computing flows and pressures at timestep number ' + str(indexes[k]))
for bb in bnds:
if bb == 2:
QQ[bb].append(-assemble(dot(u, n)*ds(bb)))
@ -1134,10 +1162,21 @@ def ROUTINE(options):
MESH = LOADmesh(options['Perturbation']['meshpath'])
checkpath = options['Perturbation']['checkpath'] + 'checkpoint/'
out_check = options['Perturbation']['checkpath'] + 'Perturbation/'
name_folder = 'SNR' +options['Perturbation']['type']['SNR'] + 'V' + str(options['Perturbation']['type']['phase_contrast'])
out_check = options['Perturbation']['checkpath'] + name_folder + '/'
READcheckpoint(MESH,'perturbation', out_check,checkpath,'u','u',options)
if 'Histograms' in options:
if options['Histograms']['apply']:
if rank==0:
print('--- Histograms in measurements ---')
MESH = LOADmesh(options['Histograms']['meshpath'])
checkpath = options['Histograms']['checkpath'] + 'checkpoint/'
outpath = options['Histograms']['checkpath']
field_name = options['Histograms']['field_name']
READcheckpoint(MESH,'Histograms', outpath, checkpath,field_name,field_name,options)
if __name__ == '__main__':

Write Preview
Loading…
Cancel
Save