84 lines
2.7 KiB
Python
84 lines
2.7 KiB
Python
import os
|
|
import numpy as np
|
|
import vtk
|
|
import vtkmodules
|
|
#from vtk.util import numpy_support
|
|
|
|
class VTK: # capitals needed/preferred?
|
|
"""
|
|
Manage vtk files.
|
|
|
|
tbd
|
|
"""
|
|
|
|
def __init__(self,geom):
|
|
"""tbd."""
|
|
self.geom = geom
|
|
|
|
@staticmethod
|
|
def from_rectilinearGrid(grid,size,origin=np.zeros(3)):
|
|
"""Check https://blog.kitware.com/ghost-and-blanking-visibility-changes/ for missing data."""
|
|
coordArray = [vtk.vtkDoubleArray(),vtk.vtkDoubleArray(),vtk.vtkDoubleArray()]
|
|
for dim in [0,1,2]:
|
|
for c in np.linspace(0,size[dim],1+grid[dim]):
|
|
coordArray[dim].InsertNextValue(c)
|
|
|
|
geom = vtk.vtkRectilinearGrid()
|
|
geom.SetDimensions(*(grid+1))
|
|
geom.SetXCoordinates(coordArray[0])
|
|
geom.SetYCoordinates(coordArray[1])
|
|
geom.SetZCoordinates(coordArray[2])
|
|
|
|
return VTK(geom)
|
|
|
|
|
|
@staticmethod
|
|
def from_unstructuredGrid(nodes,connectivity,elem):
|
|
geom = vtk.vtkUnstructuredGrid()
|
|
geom.SetPoints(nodes)
|
|
geom.Allocate(connectivity.shape[0])
|
|
|
|
if elem == 'TRIANGLE':
|
|
vtk_type = vtk.VTK_TRIANGLE
|
|
n_nodes = 3
|
|
elif elem == 'QUAD':
|
|
vtk_type = vtk.VTK_QUAD
|
|
n_nodes = 4
|
|
elif elem == 'TETRA':
|
|
vtk_type = vtk.VTK_TETRA
|
|
n_nodes = 4
|
|
elif elem == 'HEXAHEDRON':
|
|
vtk_type = vtk.VTK_HEXAHEDRON
|
|
n_nodes = 8
|
|
|
|
for i in connectivity:
|
|
geom.InsertNextCell(vtk_type,n_nodes,i-1)
|
|
|
|
return VTK(geom)
|
|
|
|
|
|
def write(self,fname): #ToDo: Discuss how to handle consistently filename extensions
|
|
if (isinstance(self.geom,vtkmodules.vtkCommonDataModel.vtkRectilinearGrid)):
|
|
writer = vtk.vtkXMLRectilinearGridWriter()
|
|
elif(isinstance(self.geom,vtkmodules.vtkCommonDataModel.vtkUnstructuredGrid)):
|
|
writer = vtk.vtkUnstructuredGrid()
|
|
elif(isinstance(self.geom,vtkmodules.vtkCommonDataModel.vtkPolyData)):
|
|
writer = vtk.vtkXMLPolyDataWriter()
|
|
|
|
writer.SetFileName('{}.{}'.format(os.path.splitext(fname)[0],
|
|
writer.GetDefaultFileExtension()))
|
|
writer.SetCompressorTypeToZLib()
|
|
writer.SetDataModeToBinary()
|
|
writer.SetInputData(self.geom)
|
|
|
|
writer.Write()
|
|
|
|
def __repr__(self):
|
|
"""ASCII representation of the VTK data."""
|
|
writer = vtk.vtkDataSetWriter()
|
|
#writer.SetHeader('damask.Geom '+version)
|
|
writer.WriteToOutputStringOn()
|
|
writer.SetInputData(self.geom)
|
|
writer.Write()
|
|
return writer.GetOutputString()
|