54 lines
1.8 KiB
Python
54 lines
1.8 KiB
Python
|
#!/usr/bin/python
|
||
|
|
|
||
|
|
import pylab
|
||
|
|
import re
|
||
|
|
import pickle
|
||
|
|
|
||
|
|
forceRegex=r"([0-9.Ee\-+]+)\s+\(+([0-9.Ee\-+]+)\s([0-9.Ee\-+]+)\s([0-9.Ee\-+]+)\)\s\(([0-9.Ee\-+]+)\s([0-9.Ee\-+]+)\s([0-9 .Ee\-+]+)\)+\s\(+([0-9.Ee\-+]+)\s([0-9.Ee\-+]+)\s([0-9.Ee\-+]+)\)\s\(([0-9.Ee\-+]+)\s([0-9.Ee\-+]+)\s([0-9.Ee\-+]+)\)+"
|
||
|
|
t = []
|
||
|
|
fpx = []; fpy = []; fpz = []
|
||
|
|
fvx = []; fvy = []; fvz = []
|
||
|
|
mpx = []; mpy = []; mpz = []
|
||
|
|
mvx = []; mvy = []; mvz = []
|
||
|
|
pipefile=open('forces/0/forces.dat','r')
|
||
|
|
lines = pipefile.readlines()
|
||
|
|
for line in lines:
|
||
|
|
match=re.search(forceRegex,line)
|
||
|
|
if match:
|
||
|
|
t.append(float(match.group(1)))
|
||
|
|
fpx.append(float(match.group(2)))
|
||
|
|
fpy.append(float(match.group(3)))
|
||
|
|
fpz.append(float(match.group(4)))
|
||
|
|
fvx.append(float(match.group(5)))
|
||
|
|
fvy.append(float(match.group(6)))
|
||
|
|
fvz.append(float(match.group(7)))
|
||
|
|
mpx.append(float(match.group(8)))
|
||
|
|
mpy.append(float(match.group(9)))
|
||
|
|
mpz.append(float(match.group(10)))
|
||
|
|
mvx.append(float(match.group(11)))
|
||
|
|
mvy.append(float(match.group(12)))
|
||
|
|
mvz.append(float(match.group(13)))
|
||
|
|
|
||
|
|
# Calculate total forces
|
||
|
|
fx = [x + y for x, y in zip(fpx, fvx)]
|
||
|
|
fy = [x + y for x, y in zip(fpy, fvy)]
|
||
|
|
fz = [x + y for x, y in zip(fpz, fvz)]
|
||
|
|
|
||
|
|
for i in range(len(t)):
|
||
|
|
fx[i] = fpx[i] + fvx[i]
|
||
|
|
|
||
|
|
for i in range(len(t)):
|
||
|
|
fx[i] = fpx[i] + fvx[i]
|
||
|
|
|
||
|
|
with open('pressureForces.dat', 'w') as f:
|
||
|
|
for f1, f2, f3 in zip(t, fpx, fpy):
|
||
|
|
print >> f, f1, f2, f3
|
||
|
|
|
||
|
|
with open('viscousForces.dat', 'w') as f:
|
||
|
|
for f1, f2, f3 in zip(t, fvx, fvy):
|
||
|
|
print >> f, f1, f2, f3
|
||
|
|
|
||
|
|
with open('totalForces.dat', 'w') as f:
|
||
|
|
for f1, f2, f3 in zip(t, fx, fy):
|
||
|
|
print >> f, f1, f2, f3
|