DISPATCH
demo.py
1 # -*- coding: utf-8 -*-
2 """
3 Created on Sun Jul 29 21:01:44 2018
4 
5 @author: Aake
6 """
7 
8 import matplotlib.pyplot as pl
9 
10 import dispatch
11 import dispatch.select
12 
13 from dispatch.graphics._graphics import *
14 
15 def demo(iout=1,run='.',data='../data',dir=0,iv=1):
16  snapfile=os.path.join(data,run,'snapshots.dat')
17  assert os.path.isfile(snapfile), 'the file '+snapfile+' must exist'
18  #
19  pl.figure(1)
20  pl.clf()
21  s=dispatch.snapshot(iout,run,data)
22  imshow(dispatch.select.unigrid_plane(s,dir=dir,iv=iv))
23  pl.title('dispatch.graphics.imshow')
24  #
25  pl.figure(2)
26  pl.clf()
27  pt=[0.5,0.5,0.5]
28  pp=dispatch.snapshot(iout,run,data).patches
29  plot_values_along(pt,pp,dir=dir,iv=iv)
30  pl.title('dispatch.graphics.plot_values_along')
31  #
32  pl.figure(3)
33  pl.clf()
34  pt=[0.5,0.5,0.5]
35  plot_patch_values_along(pt,pp,dir=dir,iv=iv,all=True,marker='o',verbose=2)
36  pl.title('dispatch.graphics.plot_patch_values_along')
37  #
38  pl.figure(4)
39  pl.clf()
40  pt=[0.5,0.5,0.5]
41  dir=1; v='d'
42  plot_patch_values_along(pt,pp,dir=dir,var=v,all=1,marker='o')
43  pl.title('dispatch.graphics.plot_patch_values_along')
44  sdir=['x','y','z']
45  pl.xlabel(sdir[dir])
46  pl.ylabel(v)
47  pl.tight_layout()
48 
49 if __name__ == '__main__':
50  demo()