wassname/simpeg
1
0
Fork 0
mirror of https://github.com/wassname/simpeg.git synced 2026-08-11 11:26:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixed vtkTools and vtkView to work for all cell,faces, edges.

Browse Source 
Changed index for cells, faces and edges to be C,F,E respectively.
Made viewprop, limits, extent as @properties, with error checks in the set methods. Need to finish the limits error checks.
This commit is contained in:
Gudni Karl Rosenkjaer
2013-11-26 21:07:30 -08:00
parent 4b292b574d
commit 1ec3c5cd09
2 changed files with 134 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files
SimPEG/visualize/vtk/vtkTools.py
+4 -2
View File
@@ -61,6 +61,7 @@ class vtkTools(object):
vtkObj.GetCellData().AddArray(vtkDoubleArr)
vtkObj.GetCellData().SetActiveScalars(model.keys()[0])
vtkObj.Update()
return vtkObj
@staticmethod
@@ -104,7 +105,7 @@ class vtkTools(object):
# Cells -cell array
FCellArr = vtk.vtkCellArray()
FCellArr.SetNumberOfCells(mesh.nF)
FCellArr.SetCells(mesh.nF*5,npsup.numpy_to_vtkIdTypeArray(np.vstack([FxCellBlock,FyCellBlock,FzCellBlock]),deep=1))
FCellArr.SetCells(mesh.nF,npsup.numpy_to_vtkIdTypeArray(np.vstack([FxCellBlock,FyCellBlock,FzCellBlock]),deep=1))
# Cell type
FCellType = npsup.numpy_to_vtk(vtk.VTK_QUAD*np.ones(mesh.nF,dtype='uint8'),deep=1)
# Cell location
@@ -166,7 +167,7 @@ class vtkTools(object):
# Cells -cell array
ECellArr = vtk.vtkCellArray()
ECellArr.SetNumberOfCells(mesh.nE)
ECellArr.SetCells(mesh.nE*3,npsup.numpy_to_vtkIdTypeArray(np.vstack([ExCellBlock,EyCellBlock,EzCellBlock]),deep=1))
ECellArr.SetCells(mesh.nE,npsup.numpy_to_vtkIdTypeArray(np.vstack([ExCellBlock,EyCellBlock,EzCellBlock]),deep=1))
# Cell type
ECellType = npsup.numpy_to_vtk(vtk.VTK_LINE*np.ones(mesh.nE,dtype='uint8'),deep=1)
# Cell location
@@ -186,6 +187,7 @@ class vtkTools(object):
vtkObj.GetCellData().AddArray(vtkDoubleArr)
vtkObj.GetCellData().SetActiveScalars(model.keys()[0])
vtkObj.Update()
return vtkObj
@staticmethod