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 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71
|
#!/usr/bin/env python
## Program: VMTK
## Module: $RCSfile: vmtkmeshtetrahedralize2.py,v $
## Language: Python
## Date: $Date: 2006/07/17 09:53:14 $
## Version: $Revision: 1.1 $
## Copyright (c) Luca Antiga, David Steinman. All rights reserved.
## See LICENCE file for details.
## This software is distributed WITHOUT ANY WARRANTY; without even
## the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
## PURPOSE. See the above copyright notices for more information.
## Note: this class was contributed by
## Tangui Morvan
## Kalkulo AS
## Simula Research Laboratory
## Bug fix for VTK 5.0
import vtk
import sys
import pypes
from vmtk import vtkvmtk
vmtkmeshtetrahedralize2 = 'vmtkMeshTetrahedralize2'
class vmtkMeshTetrahedralize2(pypes.pypeScript):
def __init__(self):
pypes.pypeScript.__init__(self)
self.Mesh = None
self.TetrahedraOnly = 0
self.SetScriptName('vmtkmeshtetrahedralize')
self.SetScriptDoc('convert the elements of a mesh to linear')
self.SetInputMembers([
['Mesh','i','vtkUnstructuredGrid',1,'','the input mesh','vmtkmeshreader'],
['TetrahedraOnly','tetonly','bool',1,'','toggle suppression of 1D and 2D cells']
])
self.SetOutputMembers([
['Mesh','o','vtkUnstructuredGrid',1,'','the output mesh','vmtkmeshwriter']])
def Execute(self):
if self.Mesh == None:
self.PrintError('Error: No input mesh.')
tetrahedralizeFilter = vtk.vtkDataSetTriangleFilter()
tetrahedralizeFilter.SetInput(self.Mesh)
if ((vtk.vtkVersion.GetVTKMajorVersion()>=5) and (vtk.vtkVersion.GetVTKMinorVersion()>0)):
tetrahedralizeFilter.SetTetrahedraOnly(self.TetrahedraOnly)
tetrahedralizeFilter.Update()
self.Mesh = tetrahedralizeFilter.GetOutput()
if self.Mesh.GetSource():
self.Mesh.GetSource().UnRegisterAllOutputs()
if __name__=='__main__':
main = pypes.pypeMain()
main.Arguments = sys.argv
main.Execute()
|