plot.py
a2c08147a0efrstcheck

plot.py
View Options


			import sys, os, re
			import h5py
			import numpy as np
			import matplotlib.pyplot as plt
			import goosempl as gplt
			import matplotlib as mpl

			plt.style.use(['goose','goose-latex'])

			# --------------------------------------------------------------------------------------------------

			name = 'example.hdf5'
			f = h5py.File(name,'r')

			fig, axes = plt.subplots(figsize=(20,10), ncols=2)

			# --------------------------------------------------------------------------------------------------

			ax = axes[0]

			t = f['/global/t' ][:]
			E = f['/global/Ekin'][:] + f['/global/Epot'][:]; ax.plot(t,E,color='k',label=r'$E_\mathrm{pot} + E_\mathrm{kin}$')
			E = f['/global/Epot'][:] ; ax.plot(t,E,color='r',label=r'$E_\mathrm{pot}$')
			E = f['/global/Ekin'][:] ; ax.plot(t,E,color='b',label=r'$E_\mathrm{kin}$')

			ax.legend(loc='lower right')

			ax.set_xlabel(r'$t$')
			ax.set_ylabel(r'$E$')

			# --------------------------------------------------------------------------------------------------

			ax = axes[1]

			coor = f['/mesh/coor'][:]
			conn = f['/mesh/conn'][:]
			u = f['/mesh/disp'][:]


			im = gplt.patch(coor=coor+u,conn=conn)

			Lx = np.max(coor[:,0])-np.min(coor[:,0])
			Ly = np.max(coor[:,1])-np.min(coor[:,1])

			ax.set_xlim([-0.1Lx,1.3Lx])
			ax.set_ylim([-0.1Ly,1.3Ly])

			ax.set_aspect('equal')

			plt.savefig(re.sub(r'(.*)(\.hdf5)',r'\1.svg',name))
			plt.show()