Homec4science
Diffusion futils (master) Browse

rFUTILS/examples/prof2.py
4ec579bf8683master

Last Change
Raw File

prof2.py
View Options

#
# @file prof2.py
#
# @brief
#
# @copyright
# Copyright (©) 2021 EPFL (Ecole Polytechnique Fédérale de Lausanne)
# SPC (Swiss Plasma Center)
#
# futils is free software: you can redistribute it and/or modify it under
# the terms of the GNU Lesser General Public License as published by the Free
# Software Foundation, either version 3 of the License, or (at your option)
# any later version.
#
# futils is distributed in the hope that it will be useful, but WITHOUT ANY
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
# FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
#
# @authors
# (in alphabetical order)
# @author Trach-Minh Tran <trach-minh.tran@epfl.ch>
#
#!/usr/bin/env python
from pylab import *
import h5py
import time
ion()
f=h5py.File('pprof.h5', 'r')
timeid=f['/profile_2d/time']
xgid=f['/profile_2d/xg']
ygid=f['/profile_2d/yg']
phiid=f['/profile_2d/phi']
title_ann = f['/'].attrs['title']
time_ann = timeid.attrs['title']
xg_ann = xgid.attrs['title']
yg_ann = ygid.attrs['title']
phi_ann = phiid.attrs['title']
nsteps=len(timeid)
extent=(xgid[0],xgid[-1],ygid[0],ygid[-1])
im=imshow(phiid[0,:,:], extent=extent,
interpolation='nearest')
# interpolation='nearest', cmap=cm.hot)
xlabel(xg_ann)
ylabel(yg_ann)
colorbar(orientation='horizontal')
tstart = time.time() # for profiling
msteps=min(nsteps,200)
for i in range(msteps):
label="t=%.1f" % (timeid[i],)
title(label)
im.set_data(phiid[i,:,:])
draw()
print 'FPS:' , nsteps/(time.time()-tstart)
c4science · Help