attach a widget (not working well yet)

src/Vtk/HEP/Detectors/vtkDetectorChamber.cxx

@@ -23,18 +23,45 @@
 
 //////////////////////////////////////////////////////////////////////////////*/
 
+#include <vtkAxes.h>
+#include <vtkBoxWidget.h>
+#include <vtkCubeSource.h>
+#include <vtkLineSource.h>
+#include <vtkMatrix4x4.h>
+#include <vtkPolyDataMapper.h>
+#include <vtkPropPicker.h>
+#include <vtkProperty.h>
+#include <vtkRenderWindow.h>
+#include <vtkRenderWindowInteractor.h>
+#include <vtkRenderer.h>
+#include <vtkRendererCollection.h>
+#include <vtkSmartPointer.h>
+#include <vtkTransform.h>
+
 #include "Vtk/HEP/Detectors/vtkDetectorChamber.h"
-#include "Vtk/vtkContainerBox.h"
 
 namespace uLib {
 namespace Vtk {
 
 vtkDetectorChamber::vtkDetectorChamber(DetectorChamber *content)
     : vtkContainerBox(content) {
-  InstallPipe();
+  m_Widget = vtkBoxWidget::New();
+  m_Callback = vtkWidgetCallback::New();
+  m_PickerCallback = vtkSelectionCallback::New();
+
+  m_Callback->SetChamber(this);
+  m_PickerCallback->SetChamber(this);
+  m_Widget->AddObserver(vtkCommand::InteractionEvent, m_Callback);
+  m_Widget->SetRotationEnabled(1);
+  m_Widget->SetTranslationEnabled(1);
+  m_Widget->SetScalingEnabled(0);
 }
 
-vtkDetectorChamber::~vtkDetectorChamber() {}
+vtkDetectorChamber::~vtkDetectorChamber() {
+  m_Widget->Delete();
+  m_Callback->Delete();
+  m_PickerCallback->Delete();
+}
 
 DetectorChamber *vtkDetectorChamber::GetContent() {
   return static_cast<DetectorChamber *>(m_Content);
@@ -44,7 +71,130 @@ void vtkDetectorChamber::PrintSelf(std::ostream &o) const {
   vtkContainerBox::PrintSelf(o);
 }
 
-void vtkDetectorChamber::InstallPipe() { vtkContainerBox::InstallPipe(); }
+void vtkDetectorChamber::ConnectInteractor(
+    vtkRenderWindowInteractor *interactor) {
+  if (!interactor)
+    return;
+  m_Widget->SetInteractor(interactor);
+  m_Widget->SetProp3D(m_Cube);
+  m_Widget->PlaceWidget();
+  interactor->AddObserver(vtkCommand::LeftButtonPressEvent, m_PickerCallback);
+}
+
+void vtkDetectorChamber::Update() {
+  if (!m_Content)
+    return;
+
+  Matrix4f mat = static_cast<Content *>(m_Content)->GetWorldMatrix();
+  vtkSmartPointer<vtkMatrix4x4> vmat = vtkSmartPointer<vtkMatrix4x4>::New();
+  for (int i = 0; i < 4; ++i)
+    for (int j = 0; j < 4; ++j)
+      vmat->SetElement(i, j, mat(i, j));
+
+  m_Cube->SetUserMatrix(vmat);
+  m_Axes->SetUserMatrix(vmat);
+  m_Pivot->SetUserMatrix(vmat);
+}
+
+void vtkDetectorChamber::InstallPipe() {
+  if (!m_Content)
+    return;
+
+  // Cube: unit box (0,0,0) -> (1,1,1)
+  vtkSmartPointer<vtkCubeSource> cube = vtkSmartPointer<vtkCubeSource>::New();
+  cube->SetBounds(0, 1, 0, 1, 0, 1);
+  vtkSmartPointer<vtkPolyDataMapper> mapper =
+      vtkSmartPointer<vtkPolyDataMapper>::New();
+  mapper->SetInputConnection(cube->GetOutputPort());
+  m_Cube->SetMapper(mapper);
+  m_Cube->GetProperty()->SetRepresentationToWireframe();
+  m_Cube->GetProperty()->SetAmbient(0.7);
+
+  // Axes: corner origin
+  vtkSmartPointer<vtkAxes> axes = vtkSmartPointer<vtkAxes>::New();
+  axes->SetOrigin(0, 0, 0);
+  axes->SetScaleFactor(0.2);
+  mapper = vtkSmartPointer<vtkPolyDataMapper>::New();
+  mapper->SetInputConnection(axes->GetOutputPort());
+  m_Axes->SetMapper(mapper);
+  m_Axes->GetProperty()->SetLineWidth(3);
+  m_Axes->GetProperty()->SetAmbient(0.4);
+
+  // Pivot: center of unit box
+  vtkSmartPointer<vtkAxes> pivot = vtkSmartPointer<vtkAxes>::New();
+  pivot->SetOrigin(0.5, 0.5, 0.5);
+  pivot->SetScaleFactor(0.1);
+  mapper = vtkSmartPointer<vtkPolyDataMapper>::New();
+  mapper->SetInputConnection(pivot->GetOutputPort());
+  m_Pivot->SetMapper(mapper);
+  m_Pivot->GetProperty()->SetLineWidth(3);
+  m_Pivot->GetProperty()->SetAmbient(0.4);
+
+  this->SetProp(m_Cube);
+  this->SetProp(m_Axes);
+  this->SetProp(m_Pivot);
+
+  this->Update();
+}
+
+void vtkDetectorChamber::vtkWidgetCallback::Execute(vtkObject *caller,
+                                                    unsigned long, void *) {
+  vtkBoxWidget *widget = reinterpret_cast<vtkBoxWidget *>(caller);
+  vtkSmartPointer<vtkTransform> t = vtkSmartPointer<vtkTransform>::New();
+  widget->GetTransform(t);
+
+  vtkMatrix4x4 *vmat = t->GetMatrix();
+  Matrix4f mat;
+  for (int i = 0; i < 4; ++i)
+    for (int j = 0; j < 4; ++j)
+      mat(i, j) = vmat->GetElement(i, j);
+
+  chamber->GetContent()->SetMatrix(mat);
+
+  // Reset internal matrix to identity to avoid double-transformation
+  vtkProp3D *prop = widget->GetProp3D();
+  if (prop) {
+    prop->SetPosition(0, 0, 0);
+    prop->SetOrientation(0, 0, 0);
+    prop->SetScale(1, 1, 1);
+  }
+
+  chamber->Update();
+}
+
+void vtkDetectorChamber::vtkSelectionCallback::Execute(vtkObject *caller,
+                                                       unsigned long, void *) {
+  vtkRenderWindowInteractor *interactor =
+      reinterpret_cast<vtkRenderWindowInteractor *>(caller);
+  vtkSmartPointer<vtkPropPicker> picker = vtkSmartPointer<vtkPropPicker>::New();
+  int *pos = interactor->GetEventPosition();
+  picker->Pick(
+      pos[0], pos[1], 0,
+      interactor->GetRenderWindow()->GetRenderers()->GetFirstRenderer());
+
+  vtkProp *picked = picker->GetViewProp();
+  if (picked == chamber->m_Cube || picked == chamber->m_Axes ||
+      picked == chamber->m_Pivot) {
+    if (!chamber->m_Widget->GetEnabled()) {
+      // Register current position/rotation/scale from content //
+      Matrix4f mat = chamber->GetContent()->GetWorldMatrix();
+      vtkSmartPointer<vtkMatrix4x4> vmat = vtkSmartPointer<vtkMatrix4x4>::New();
+      for (int i = 0; i < 4; ++i)
+        for (int j = 0; j < 4; ++j)
+          vmat->SetElement(i, j, mat(i, j));
+
+      vtkSmartPointer<vtkTransform> t = vtkSmartPointer<vtkTransform>::New();
+      t->SetMatrix(vmat);
+      chamber->m_Widget->SetTransform(t);
+      chamber->m_Widget->PlaceWidget();
+      chamber->m_Widget->On();
+    }
+  } else {
+    if (chamber->m_Widget->GetEnabled()) {
+      chamber->m_Widget->Off();
+    }
+  }
+}
 
 } // namespace Vtk
 } // namespace uLib