zhyang-dev/LPP
1
0
Fork 0
mirror of https://github.com/ParticulateFlow/LPP.git synced 2025-12-08 06:37:46 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

extracted filename-generation to be a function

Browse Source 
s.t. it can be called from other modules
This commit is contained in:
Stefan A
2012-04-13 14:58:38 +02:00
parent 94d3208aeb
commit 7b68703319
1 changed files with 27 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
src/vtk.py
View File

@ -151,22 +151,7 @@ class vtk:
surfflag = 1
surface(tris)
if fileNos[n] < 10:
file = root + "000" + str(fileNos[n]) + ".vtk"
file_bb= root + "000" + str(fileNos[n]) + "_boundingBox.vtk"
file_walls= root + "000" + str(fileNos[n]) + "_walls.vtk"
elif fileNos[n] < 100:
file = root + "00" + str(fileNos[n]) + ".vtk"
file_bb= root + "00" + str(fileNos[n]) + "_boundingBox.vtk"
file_walls= root + "00" + str(fileNos[n]) + "_walls.vtk"
elif fileNos[n] < 1000:
file = root + "0" + str(fileNos[n]) + ".vtk"
file_bb= root + "0" + str(fileNos[n]) + "_boundingBox.vtk"
file_walls= root + "0" + str(fileNos[n]) + "_walls.vtk"
else:
file = root + str(fileNos[n]) + ".vtk"
file_bb= root + str(fileNos[n]) + "_boundingBox.vtk"
file_walls= root + str(fileNos[n]) + "_walls.vtk"
file, file_bb, file_walls = generateFilename(root,fileNos,n)
boundingBox(file_bb,xlo,xhi,ylo,yhi,zlo,zhi)
nvalues = 0
@ -191,6 +176,32 @@ class vtk:
if len(tris): surface(tris)
particle(file,atoms)
# ----------------------------------------------------------------------------
# generates the filename of the output-vtk-files from
# - a root string,
# - the a string of numbers (timestamps) and
# - the index, i.e. which of those numbers is going to be used
# ----------------------------------------------------------------------------
def generateFilename(root,fileNos,n):
if fileNos[n] < 10:
file = root + "000" + str(fileNos[n]) + ".vtk"
file_bb= root + "000" + str(fileNos[n]) + "_boundingBox.vtk"
file_walls= root + "000" + str(fileNos[n]) + "_walls.vtk"
elif fileNos[n] < 100:
file = root + "00" + str(fileNos[n]) + ".vtk"
file_bb= root + "00" + str(fileNos[n]) + "_boundingBox.vtk"
file_walls= root + "00" + str(fileNos[n]) + "_walls.vtk"
elif fileNos[n] < 1000:
file = root + "0" + str(fileNos[n]) + ".vtk"
file_bb= root + "0" + str(fileNos[n]) + "_boundingBox.vtk"
file_walls= root + "0" + str(fileNos[n]) + "_walls.vtk"
else:
file = root + str(fileNos[n]) + ".vtk"
file_bb= root + str(fileNos[n]) + "_boundingBox.vtk"
file_walls= root + str(fileNos[n]) + "_walls.vtk"
return (file, file_bb, file_walls)
# --------------------------------------------------------------------
# write list of triangles into VTK surface files: SURF1.vtk, SURF2.vtk, ...
# all triangles of one type constitute 1 surface = 1 file