1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
#!/usr/bin/env python
import vtktools
from numpy import concatenate, arange, newaxis, ones, array
from pylab import *
import glob
from fluidity_tools import stat_parser
colx=ones((21,1))*0.1
colz=zeros((21,1))
coly=arange(-0.5,0.51,0.05)[:,newaxis]
coordinates=concatenate((colx,coly,colz),1)
vtus=glob.glob1('./','*.vtu')
for vtu in vtus:
vtunumber=vtu[19:]
vtunumber=vtunumber[:-4]
vtufile=vtktools.vtu('./'+vtu)
water=vtktools.vtu.ProbeData(vtufile,coordinates,'Water::MaterialVolumeFraction')
droplet=vtktools.vtu.ProbeData(vtufile,coordinates,'Droplet::MaterialVolumeFraction')
air=vtktools.vtu.ProbeData(vtufile,coordinates,'Air::MaterialVolumeFraction')
waterdropletsum=vtktools.vtu.ProbeData(vtufile,coordinates,'Water::SumMaterialVolumeFractions')
plot(coordinates[:,1],water,coordinates[:,1],droplet,coordinates[:,1],air,coordinates[:,1],waterdropletsum)
axis([-0.55, 0.55, -0.5, 1.5])
xlabel('Distance')
ylabel('VolumeFraction')
legend(('Water', 'Droplet', 'Air','Sum'), loc='center left')
if(len(vtunumber)==1):
vtunumber='000'+vtunumber
elif(len(vtunumber)==2):
vtunumber='00'+vtunumber
elif(len(vtunumber)==3):
vtunumber='0'+vtunumber
savefig('./'+vtu[:19]+vtunumber+'.png')
cla()
|