Set lighting propreties when pressing 's' or 'w' in a PCLVisualizer and using setRepresentationTo*ForAllActors()

Author: VictorLamoine

visualization/src/interactor_style.cpp

@@ -831,6 +831,24 @@ pcl::visualization::PCLVisualizerInteractorStyle::OnKeyDown ()
       }
       break;
     }
+
+    // Switch representation to wireframe (override default behavior)
+    case 'w': case 'W':
+    {
+      vtkSmartPointer<vtkActorCollection> ac = CurrentRenderer->GetActors ();
+      vtkCollectionSimpleIterator ait;
+      for (ac->InitTraversal (ait); vtkActor* actor = ac->GetNextActor (ait); )
+      {
+        for (actor->InitPathTraversal (); vtkAssemblyPath* path = actor->GetNextPath (); )
+        {
+          vtkSmartPointer<vtkActor> apart = reinterpret_cast <vtkActor*> (path->GetLastNode ()->GetViewProp ());
+          apart->GetProperty ()->SetRepresentationToWireframe ();
+          apart->GetProperty ()->SetLighting (false);
+        }
+      }
+      break;
+    }
+
     // Save a PNG snapshot with the current screen
     case 'j': case 'J':
     {
@@ -992,7 +1010,20 @@ pcl::visualization::PCLVisualizerInteractorStyle::OnKeyDown ()
         Interactor->Render ();
       }
       else
-        Superclass::OnKeyDown ();
+      {
+        Superclass::OnKeyDown();
+        vtkSmartPointer<vtkActorCollection> ac = CurrentRenderer->GetActors();
+        vtkCollectionSimpleIterator ait;
+        for (ac->InitTraversal(ait); vtkActor* actor = ac->GetNextActor(ait);)
+        {
+          for (actor->InitPathTraversal(); vtkAssemblyPath* path = actor->GetNextPath();)
+          {
+            vtkSmartPointer<vtkActor> apart = reinterpret_cast<vtkActor*>(path->GetLastNode()->GetViewProp());
+            apart->GetProperty()->SetRepresentationToSurface();
+            apart->GetProperty()->SetLighting(true);
+          }
+        }
+      }
       break;
     }
 

visualization/src/pcl_visualizer.cpp

@@ -3238,6 +3238,7 @@ pcl::visualization::PCLVisualizer::setRepresentationToSurfaceForAllActors ()
     while ((actor = actors->GetNextActor ()) != NULL)
     {
       actor->GetProperty ()->SetRepresentationToSurface ();
+      actor->GetProperty ()->SetLighting(true);
     }
   }
 }
@@ -3276,6 +3277,7 @@ pcl::visualization::PCLVisualizer::setRepresentationToWireframeForAllActors ()
     while ((actor = actors->GetNextActor ()) != NULL)
     {
       actor->GetProperty ()->SetRepresentationToWireframe ();
+      actor->GetProperty ()->SetLighting(false);
     }
   }
 }