Merge pull request #482 from danlipsa/add-vertex-cells

Add vertex cells.
Kitware · May 19, 2021 · 36b6a09 · 36b6a09
2 parents b800f07 + d7a76cd
commit 36b6a09
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/gui/WorldView.cxx b/gui/WorldView.cxx
@@ -90,6 +90,7 @@
 #include <vtkTimeStamp.h>
 #include <vtkUnsignedCharArray.h>
 #include <vtkUnsignedIntArray.h>
+#include <vtkVertexGlyphFilter.h>
 #include <vtkXMLImageDataReader.h>
 #include <vtkXMLPolyDataWriter.h>
 
@@ -1034,6 +1035,14 @@ void WorldView::setMesh(vtkSmartPointer<vtkPolyData> mesh)
 {
   QTE_D();
 
+  if (mesh->GetNumberOfCells() == 0)
+  {
+    vtkNew<vtkVertexGlyphFilter> vertexFilter;
+    vertexFilter->SetInputData(mesh);
+    vertexFilter->Update();
+    mesh = vertexFilter->GetOutput();
+  }
+
   d->mesh = mesh;
 
   // Create mapper