marcel/pylot
3
0
Fork 0
Code Issues 7 Pull Requests 1 Releases Wiki Activity

changed vtk file structure for vgrids, making it way faster and smaller and enabling paraview to display surfaces instead of points

Browse Source
This commit is contained in:
Marcel Paffrath 2015-10-01 17:30:57 +02:00
parent 3d3cac7085
commit 7120a20499
1 changed files with 7 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
pylot/core/active/fmtomo2vtk.py
View File

@ -74,33 +74,20 @@ def vgrids2VTK(inputfile = 'vgrids.in', outputfile = 'vgrids.vtk'):
dX = getDistance(np.rad2deg(dPhi))
dY = getDistance(np.rad2deg(dTheta))
xGrid = np.linspace(sX, sX + (dX * nX), nX)
yGrid = np.linspace(sZ, sZ + (nY * dY), nY)
zGrid = np.linspace(sZ, sZ + (nR * dR), nR)
nPoints = len(xGrid) * len(yGrid) * len(zGrid)
nPoints = nX * nY * nZ
dZ = dR
# write header
print("Writing header for VTK file...")
outfile.writelines('# vtk DataFile Version 3.1\n')
outfile.writelines('Velocity on FMTOMO vgrids.in points\n')
outfile.writelines('ASCII\n')
outfile.writelines('DATASET POLYDATA\n')
outfile.writelines('POINTS %15d float\n' %(nPoints))
outfile.writelines('DATASET STRUCTURED_POINTS\n')
# write coordinates
print("Writing coordinates to VTK file...")
for z in zGrid:
for y in yGrid:
for x in xGrid:
outfile.writelines('%10f %10f %10f \n' %(x, y, z))
outfile.writelines('VERTICES %15d %15d\n' %(nPoints, 2 * nPoints))
# write indices
print("Writing indices to VTK file...")
for index in range(nPoints):
outfile.writelines('%10d %10d\n' %(1, index))
outfile.writelines('DIMENSIONS %d %d %d\n' %(nX, nY, nZ))
outfile.writelines('ORIGIN %f %f %f\n' %(sX, sY, sZ))
outfile.writelines('SPACING %f %f %f\n' %(dX, dY, dZ))
outfile.writelines('POINT_DATA %15d\n' %(nPoints))
outfile.writelines('SCALARS velocity float %d\n' %(1))