223031012
/
DAMASK_EICMD
mirror of https://github.com/achalhp/DAMASK_EICMD.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix: nodes need to be converted to vtk type 

some stub definitions
This commit is contained in:
Martin Diehl 2020-03-11 18:07:31 +01:00
parent c92a6ad459
commit f324e67f7b
1 changed files with 18 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
python/damask/ktv.py
View File

@ -1,7 +1,12 @@
import os import os
import pandas as pd
import numpy as np import numpy as np
import vtk import vtk
#from vtk.util import numpy_support from vtk.util import numpy_support
from . import table
from . import version
class VTK: # capitals needed/preferred? class VTK: # capitals needed/preferred?
""" """
@ -34,7 +39,7 @@ class VTK: # capitals needed/preferred?
@staticmethod @staticmethod
def from_unstructuredGrid(nodes,connectivity,elem): def from_unstructuredGrid(nodes,connectivity,elem):
geom = vtk.vtkUnstructuredGrid() geom = vtk.vtkUnstructuredGrid()
geom.SetPoints(nodes) geom.SetPoints(numpy_support.numpy_to_vtk(nodes)) #,deep=True)
geom.Allocate(connectivity.shape[0]) geom.Allocate(connectivity.shape[0])
if elem == 'TRIANGLE': if elem == 'TRIANGLE':
@ -72,10 +77,20 @@ class VTK: # capitals needed/preferred?
writer.Write() writer.Write()
def add(data,label=None):
if isinstance(data,np.ndarray):
pass
elif isinstance(data,pd.DataFrame):
pass
elif isinstance(data,table):
pass
def __repr__(self): def __repr__(self):
"""ASCII representation of the VTK data.""" """ASCII representation of the VTK data."""
writer = vtk.vtkDataSetWriter() writer = vtk.vtkDataSetWriter()
#writer.SetHeader('damask.Geom '+version) writer.SetHeader('DAMASK.VTK v{}'.format(version))
writer.WriteToOutputStringOn() writer.WriteToOutputStringOn()
writer.SetInputData(self.geom) writer.SetInputData(self.geom)
writer.Write() writer.Write()