widget in viewport

src/Vtk/uLibVtkInterface.cxx

@@ -53,9 +53,12 @@
 #include <vtkRenderer.h>
 #include <vtkProperty.h>
 #include <vtkCamera.h>
-
+#include <vtkPolyData.h>
+#include <vtkFeatureEdges.h>
+#include <vtkTransform.h>
 
 #include "uLibVtkInterface.h"
+#include "vtkHandlerWidget.h"
 
 
 
@@ -78,36 +81,30 @@ namespace Vtk {
 class PuppetData {
 public:
     PuppetData() :
-        m_Renderers(vtkRendererCollection::New()),
-        m_Assembly(vtkPropAssembly::New()),
-        m_OutlineSource(NULL),
-        m_OutlineActor(NULL),
-        m_CubeAxesActor(NULL),
+        m_Renderers(vtkSmartPointer<vtkRendererCollection>::New()),
+        m_Assembly(vtkSmartPointer<vtkPropAssembly>::New()),
         m_ShowBoundingBox(false),
         m_ShowScaleMeasures(false),
         m_Representation(-1),
-        m_Opacity(-1.0)
+        m_Opacity(-1.0),
+        m_Selectable(true),
+        m_Selected(false)
     {
         m_Color[0] = m_Color[1] = m_Color[2] = -1.0;
     }
 
     ~PuppetData() {
-        m_Renderers->RemoveAllItems();
-        m_Assembly->GetParts()->RemoveAllItems();
-        m_Renderers->Delete();
-        m_Assembly->Delete();
-        if (m_OutlineSource) m_OutlineSource->Delete();
-        if (m_OutlineActor) m_OutlineActor->Delete();
-        if (m_CubeAxesActor) m_CubeAxesActor->Delete();
+        // No manual Delete needed for smart pointers
     }
 
     // members //
-    vtkRendererCollection *m_Renderers;
-    vtkPropAssembly       *m_Assembly;
+    vtkSmartPointer<vtkRendererCollection> m_Renderers;
+    vtkSmartPointer<vtkPropAssembly>       m_Assembly;
 
-    vtkOutlineSource      *m_OutlineSource;
-    vtkActor              *m_OutlineActor;
-    vtkCubeAxesActor      *m_CubeAxesActor;
+    vtkSmartPointer<vtkOutlineSource>      m_OutlineSource;
+    vtkSmartPointer<vtkActor>              m_OutlineActor;
+    vtkSmartPointer<vtkCubeAxesActor>      m_CubeAxesActor;
+    vtkSmartPointer<vtkActor>              m_HighlightActor;
 
     bool m_ShowBoundingBox;
     bool m_ShowScaleMeasures;
@@ -132,6 +129,69 @@ public:
             actor->GetProperty()->SetOpacity(m_Opacity);
         }
     }
+
+    void UpdateHighlight() {
+        if (m_Selected) {
+            if (!m_HighlightActor) {
+                vtkSmartPointer<vtkFeatureEdges> edges = vtkSmartPointer<vtkFeatureEdges>::New();
+                edges->BoundaryEdgesOn();
+                edges->FeatureEdgesOn();
+                edges->SetFeatureAngle(30);
+                edges->NonManifoldEdgesOn();
+                edges->ManifoldEdgesOff();
+
+                // Find first polydata in assembly to highlight
+                vtkPropCollection *parts = m_Assembly->GetParts();
+                parts->InitTraversal();
+                for (int i = 0; i < parts->GetNumberOfItems(); ++i) {
+                    vtkActor *actor = vtkActor::SafeDownCast(parts->GetNextProp());
+                    if (actor && actor->GetMapper() && actor->GetMapper()->GetDataSetInput()) {
+                        edges->SetInputData(vtkPolyData::SafeDownCast(actor->GetMapper()->GetDataSetInput()));
+                        break;
+                    }
+                }
+
+                m_HighlightActor = vtkSmartPointer<vtkActor>::New();
+                vtkSmartPointer<vtkPolyDataMapper> mapper = vtkSmartPointer<vtkPolyDataMapper>::New();
+                mapper->SetInputConnection(edges->GetOutputPort());
+                m_HighlightActor->SetMapper(mapper);
+                m_HighlightActor->GetProperty()->SetColor(1.0, 0.5, 0.0); // Orange
+                m_HighlightActor->GetProperty()->SetLineWidth(2.0);
+                m_HighlightActor->GetProperty()->SetLighting(0);
+            }
+
+            // Update highlight data and transform from first actor
+            vtkPropCollection *parts = m_Assembly->GetParts();
+            parts->InitTraversal();
+            for (int i = 0; i < parts->GetNumberOfItems(); ++i) {
+                vtkActor *actor = vtkActor::SafeDownCast(parts->GetNextProp());
+                if (actor) {
+                    // Sync transform
+                    m_HighlightActor->SetUserTransform(actor->GetUserTransform());
+                    m_HighlightActor->SetPosition(actor->GetPosition());
+                    m_HighlightActor->SetOrientation(actor->GetOrientation());
+                    m_HighlightActor->SetScale(actor->GetScale());
+                    break;
+                }
+            }
+
+            m_Renderers->InitTraversal();
+            for (int i = 0; i < m_Renderers->GetNumberOfItems(); ++i) {
+                vtkRenderer *ren = m_Renderers->GetNextItem();
+                ren->AddActor(m_HighlightActor);
+            }
+        } else {
+            if (m_HighlightActor) {
+                m_Renderers->InitTraversal();
+                for (int i = 0; i < m_Renderers->GetNumberOfItems(); ++i) {
+                    m_Renderers->GetNextItem()->RemoveActor(m_HighlightActor);
+                }
+            }
+        }
+    }
+
+    bool m_Selectable;
+    bool m_Selected;
 };
 
 // -------------------------------------------------------------------------- //
@@ -195,6 +255,10 @@ void Puppet::ConnectRenderer(vtkRenderer *renderer)
             d->m_CubeAxesActor->SetCamera(renderer->GetActiveCamera());
             renderer->AddActor(d->m_CubeAxesActor);
         }
+
+        if (d->m_Selected && d->m_HighlightActor) {
+            renderer->AddActor(d->m_HighlightActor);
+        }
     }
 }
 
@@ -248,8 +312,8 @@ void Puppet::ShowBoundingBox(bool show)
     d->m_ShowBoundingBox = show;
     if (show) {
         if (!d->m_OutlineActor) {
-            d->m_OutlineSource = vtkOutlineSource::New();
-            d->m_OutlineActor = vtkActor::New();
+            d->m_OutlineSource = vtkSmartPointer<vtkOutlineSource>::New();
+            d->m_OutlineActor = vtkSmartPointer<vtkActor>::New();
             vtkSmartPointer<vtkPolyDataMapper> mapper = vtkSmartPointer<vtkPolyDataMapper>::New();
             mapper->SetInputConnection(d->m_OutlineSource->GetOutputPort());
             d->m_OutlineActor->SetMapper(mapper);
@@ -282,7 +346,7 @@ void Puppet::ShowScaleMeasures(bool show)
     d->m_ShowScaleMeasures = show;
     if (show) {
         if (!d->m_CubeAxesActor) {
-            d->m_CubeAxesActor = vtkCubeAxesActor::New();
+            d->m_CubeAxesActor = vtkSmartPointer<vtkCubeAxesActor>::New();
             d->m_CubeAxesActor->SetFlyModeToOuterEdges();
             d->m_CubeAxesActor->SetUseTextActor3D(1);
             d->m_CubeAxesActor->GetProperty()->SetColor(1.0, 1.0, 1.0);
@@ -363,8 +427,32 @@ void Puppet::SetOpacity(double alpha)
 
 
 
+void Puppet::SetSelectable(bool selectable)
+{
+    d->m_Selectable = selectable;
+}
 
+bool Puppet::IsSelectable() const
+{
+    return d->m_Selectable;
+}
 
+void Puppet::SetSelected(bool selected)
+{
+    if (!d->m_Selectable) return;
+    if (d->m_Selected == selected) return;
+    d->m_Selected = selected;
+    d->UpdateHighlight();
+}
 
-} // Vtk
-} // uLib
+bool Puppet::IsSelected() const
+{
+    return d->m_Selected;
+}
+
+void Puppet::ConnectInteractor(vtkRenderWindowInteractor *interactor)
+{
+}
+
+} // namespace Vtk
+} // namespace uLib