add QCanvas Root and viewport pane in gcompose

src/Vtk/HEP/Detectors/testing/CMakeLists.txt

@@ -2,6 +2,7 @@
 set(TESTS
     vtkMuonScatterTest
     vtkDetectorChamberTest
+    vtkDetectorMuonProjectionTest
 )
 
 set(LIBRARIES

src/Vtk/HEP/Detectors/testing/vtkDetectorMuonProjectionTest.cpp

@@ -0,0 +1,156 @@
+/*//////////////////////////////////////////////////////////////////////////////
+// CMT Cosmic Muon Tomography project //////////////////////////////////////////
+////////////////////////////////////////////////////////////////////////////////
+
+  Copyright (c) 2014, Universita' degli Studi di Padova, INFN sez. di Padova
+  All rights reserved
+
+  Authors: Andrea Rigoni Garola < andrea.rigoni@pd.infn.it >
+
+  ------------------------------------------------------------------
+  This library is free software;  you  can  redistribute  it  and/or
+  modify it  under the  terms  of  the  GNU  Lesser  General  Public
+  License as published  by  the  Free  Software  Foundation;  either
+  version 3.0 of the License, or (at your option) any later version.
+
+  This library is  distributed in  the hope that it will  be useful,
+  but  WITHOUT ANY WARRANTY;  without  even  the implied warranty of
+  MERCHANTABILITY  or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+  Lesser General Public License for more details.
+
+  You should have received a copy of  the GNU Lesser General  Public
+  License along with this library.
+
+//////////////////////////////////////////////////////////////////////////////*/
+
+#include "HEP/Detectors/DetectorChamber.h"
+#include "HEP/Detectors/MuonScatter.h"
+#include "Math/Units.h"
+#include "Vtk/HEP/Detectors/vtkDetectorChamber.h"
+#include "Vtk/HEP/Detectors/vtkMuonScatter.h"
+#include "Vtk/uLibVtkViewer.h"
+
+#include <vtkActor.h>
+#include <vtkArrowSource.h>
+#include <vtkMath.h>
+#include <vtkPolyDataMapper.h>
+#include <vtkProperty.h>
+#include <vtkTransform.h>
+
+#define BOOST_TEST_MODULE vtkDetectorMuonProjectionTest
+#include <boost/test/unit_test.hpp>
+
+using namespace uLib;
+
+// A simple puppet class to represent an arrow indicative of a projected muon hit
+class vtkArrowPuppet : public Vtk::Puppet {
+public:
+  vtkArrowPuppet() : m_Actor(vtkActor::New()) {
+    vtkNew<vtkArrowSource> arrow;
+    vtkNew<vtkPolyDataMapper> mapper;
+    mapper->SetInputConnection(arrow->GetOutputPort());
+    m_Actor->SetMapper(mapper);
+    this->SetProp(m_Actor);
+    m_Actor->GetProperty()->SetColor(1, 1, 0); // Yellow color for visibility
+  }
+
+  virtual ~vtkArrowPuppet() { m_Actor->Delete(); }
+
+  void SetProjection(const HLine3f &line) {
+    HPoint3f p = line.origin;
+    HVector3f d = line.direction;
+    d.normalize();
+
+    vtkNew<vtkTransform> t;
+    t->Translate(p.x(), p.y(), p.z());
+
+    // vtkArrowSource initially points along X+ (1, 0, 0)
+    double vtk_x[3] = {1, 0, 0};
+    double vtk_d[3] = {d.x(), d.y(), d.z()};
+
+    double axis[3];
+    vtkMath::Cross(vtk_x, vtk_d, axis);
+    double dot = vtkMath::Dot(vtk_x, vtk_d);
+    double angle = vtkMath::DegreesFromRadians(acos(dot));
+
+    // Handle parallel or anti-parallel cases
+    if (vtkMath::Norm(axis) < 1e-6) {
+      if (dot < 0) {
+        t->RotateY(180); // Reverse direction if anti-parallel
+      }
+    } else {
+      t->RotateWXYZ(angle, axis);
+    }
+
+    t->Scale(100, 100, 100); // Scale the arrow for visual prominence
+    m_Actor->SetUserTransform(t);
+  }
+
+private:
+  vtkActor *m_Actor;
+};
+
+BOOST_AUTO_TEST_CASE(vtkDetectorMuonProjectionTest) {
+  using namespace uLib::literals;
+
+  // 1. Prepare a scene with two detector chambers
+  DetectorChamber d1, d2;
+  d1.Scale(Vector3f(1_m, 1_m, 10_cm));
+  d1.Translate(Vector3f(0, 0, -0.5_m));
+
+  
+  d2.Scale(Vector3f(1_m, 1_m, 10_cm));
+  d2.Translate(Vector3f(0, 0, 0.5_m));
+
+  // 2. Setup a muon event that crosses the detectors
+  MuonScatter event;
+  // Set an origin slightly offset and point it toward (0,0,0)
+  HPoint3f mu_origin(20_cm, 30_cm, -1_m);
+  event.LineIn().origin = mu_origin;
+  
+  // Direction is toward (0,0,0)
+  HVector3f mu_dir = (HPoint3f(0,0,0) - mu_origin);
+  mu_dir.normalize();
+  event.LineIn().direction = mu_dir;
+  
+  // Let's make the track continue straight for a simple projection test
+  event.LineOut().origin = mu_origin - mu_dir * 3_m; 
+  event.LineOut().direction = -mu_dir;
+
+  MuonEvent mu_event;
+  mu_event.LineIn() = event.LineIn();
+  mu_event.LineOut() = event.LineOut();
+
+  // 3. Project the muon on each detector to find the projection points
+  MuonEvent mu_proj1 = d1.ProjectMuonEvent(mu_event);
+  MuonEvent mu_proj2 = d2.ProjectMuonEvent(mu_event);
+
+  Vtk::vtkDetectorChamber v_d1(&d1);
+  Vtk::vtkDetectorChamber v_d2(&d2);
+  Vtk::vtkMuonScatter v_event(event);
+  v_event.AddPocaPoint(HPoint3f(0, 0, 0));
+  v_event.SetColor(1, 0, 0); // Red muon event
+
+  v_d1.SetRepresentation(Vtk::Puppet::Surface);
+  v_d1.SetOpacity(0.3);
+  v_d2.SetRepresentation(Vtk::Puppet::Surface);
+  v_d2.SetOpacity(0.3);
+
+  // 5. Add two arrows to mark where the projection is located on the chambers
+  vtkArrowPuppet v_p1, v_p2;
+  v_p1.SetProjection(mu_proj1.LineIn());
+  v_p2.SetProjection(mu_proj2.LineIn());
+
+  if (std::getenv("CTEST_PROJECT_NAME") == nullptr) {
+    Vtk::Viewer viewer;
+    viewer.SetGridAxis(Vtk::Viewport::Z);
+    viewer.AddPuppet(v_d1);
+    viewer.AddPuppet(v_d2);
+    viewer.AddPuppet(v_event);
+    viewer.AddPuppet(v_p1);
+    viewer.AddPuppet(v_p2);
+    viewer.Start();
+  }
+
+  BOOST_CHECK(true);
+}

src/Vtk/vtkViewport.h

@@ -61,6 +61,7 @@ public:
     virtual vtkRenderWindowInteractor* GetInteractor() = 0;
     vtkCornerAnnotation*        GetAnnotation()  { return m_Annotation; }
     vtkCameraOrientationWidget* GetCameraWidget(){ return m_CameraWidget; }
+    const std::vector<Puppet*>& getPuppets() const { return m_Puppets; }
 
     // Grid control
     void SetGridVisible(bool visible);