Merge pull request #2 from Geode-solutions/fix/pipeline

composite mapper
Geode-solutions · Aug 31, 2023 · 30c3004 · 30c3004
2 parents bf1652a + 7e7d79c
commit 30c3004
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/vtk_protocol.py b/vtk_protocol.py
@@ -36,21 +36,22 @@ def create_object_pipeline(self, params):
             file_name = params["file_name"]
 
             actor = vtk.vtkActor()
-            mapper = vtk.vtkDataSetMapper()
-            actor.SetMapper(mapper)
             if ".vtm" in file_name:
                 reader = vtk.vtkXMLMultiBlockDataReader()
-                filter = vtk.vtkCompositeDataGeometryFilter()
+                filter = vtk.vtkGeometryFilter()
                 filter.SetInputConnection(reader.GetOutputPort())
+                mapper = vtk.vtkCompositePolyDataMapper()
                 mapper.SetInputConnection(filter.GetOutputPort())
                 self.register_object(id, reader, filter, actor, mapper, {})
             else:
                 reader = vtk.vtkXMLGenericDataObjectReader()
+                mapper = vtk.vtkDataSetMapper()
                 mapper.SetInputConnection(reader.GetOutputPort())
                 self.register_object(id, reader, {}, actor, mapper, {})
 
             reader.SetFileName(f"/data/{file_name}")
 
+            actor.SetMapper(mapper)
             mapper.SetColorModeToMapScalars()
             mapper.SetResolveCoincidentTopologyLineOffsetParameters(1, -0.1)
             mapper.SetResolveCoincidentTopologyPolygonOffsetParameters(2, 0)