reorganization of sources, moving cmt pertaining structures into HEP folder

2026-03-07 08:58:31 +00:00
parent 49cf0aeedd
commit e8f8e96521
70 changed files with 1468 additions and 1627 deletions
--- a/src/Vtk/HEP/MuonTomography/testing/vtkVoxRaytracerTest.cpp
+++ b/src/Vtk/HEP/MuonTomography/testing/vtkVoxRaytracerTest.cpp
@@ -0,0 +1,132 @@
+/*//////////////////////////////////////////////////////////////////////////////
+// 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 <vtkBoxWidget.h>
+#include <vtkCommand.h>
+#include <vtkSmartPointer.h>
+
+#include <vtkCornerAnnotation.h>
+
+#include <HEP/Detectors/MuonScatter.h>
+#include <Math/VoxRaytracer.h>
+
+#include "Vtk/HEP/Detectors/vtkMuonScatter.h"
+#include "Vtk/HEP/MuonTomography/vtkVoxRaytracerRepresentation.h"
+#include "Vtk/Math/vtkStructuredGrid.h"
+#include "Vtk/uLibVtkViewer.h"
+
+#define BOOST_TEST_MODULE VtkVoxRaytracerTest
+#include <boost/test/unit_test.hpp>
+
+using namespace uLib;
+
+class vtkWidgetCallback : public vtkCommand {
+public:
+  static vtkWidgetCallback *New() { return new vtkWidgetCallback; }
+
+  void SetTracer(Vtk::vtkVoxRaytracerRepresentation *parent) {
+    this->vtk_raytr = parent;
+  }
+
+  void SetMuon(uLib::MuonScatter *muon) { this->muon = muon; }
+
+  void SetAnnotation(vtkCornerAnnotation *annotation) {
+    this->annotation = annotation;
+  }
+
+  virtual void Execute(vtkObject *caller, unsigned long, void *) {
+    char str[40];
+    vtk_raytr->SetMuon(*muon);
+    if (annotation) {
+      sprintf(str, "total length = %f", vtk_raytr->GetRay().TotalLength());
+      annotation->SetText(1, str);
+      for (int i = 0; i < vtk_raytr->GetRay().Data().size(); ++i) {
+        std::cout << "L[" << i << "] = " << vtk_raytr->GetRay().Data().at(i).L
+                  << "\n";
+      }
+      std::cout << "\n";
+    }
+  }
+
+private:
+  vtkWidgetCallback() : vtk_raytr(NULL), muon(NULL), annotation(NULL) {}
+
+  uLib::VoxRaytracer *raytracer;
+  Vtk::vtkVoxRaytracerRepresentation *vtk_raytr;
+  uLib::MuonScatter *muon;
+  vtkCornerAnnotation *annotation;
+};
+
+BOOST_AUTO_TEST_CASE(vtkVoxRaytracerRepresentationTest) {
+  // muon scatter //
+  MuonScatter muon;
+  muon.LineIn().origin << -6, 12, -6, 1;
+  muon.LineIn().direction << 1, -1, 1, 0;
+  muon.LineOut().origin << 6, -4, 6, 1;
+  muon.LineOut().direction << 1, -1, 1, 0;
+
+  Vtk::vtkMuonScatter v_muon(muon);
+
+  // structured grid //
+  StructuredGrid grid(Vector3i(12, 10, 12));
+  grid.SetSpacing(Vector3f(1, 1, 1));
+  grid.SetPosition(Vector3f(0, 0, 0));
+
+  Vtk::vtkStructuredGrid v_grid(grid);
+
+  // voxraytracer //
+  VoxRaytracer rt(grid);
+  HPoint3f pt;
+  rt.GetEntryPoint(muon.LineIn(), pt);
+  std::cout << pt.transpose() << "\n";
+
+  Vtk::vtkVoxRaytracerRepresentation v_rt(rt);
+  v_rt.SetMuon(muon);
+  v_rt.SetRayColor(Vector4f(1, 0, 0, 1));
+
+  if (std::getenv("CTEST_PROJECT_NAME") == nullptr) {
+    // renderer //
+    Vtk::Viewer viewer;
+
+    // widget //
+    vtkBoxWidget *widget = v_grid.GetWidget();
+
+    vtkWidgetCallback *cbk = vtkWidgetCallback::New();
+    cbk->SetTracer(&v_rt);
+    cbk->SetMuon(&muon);
+    cbk->SetAnnotation(viewer.GetAnnotation());
+    widget->AddObserver(vtkCommand::InteractionEvent, cbk);
+    widget->SetInteractor(viewer.GetInteractor());
+    widget->PlaceWidget();
+    widget->On();
+
+    viewer.AddPuppet(v_grid);
+    viewer.AddPuppet(v_rt);
+    viewer.AddPuppet(v_muon);
+    viewer.Start();
+  }
+
+  BOOST_CHECK(v_rt.GetRay().Count() > 0);
+}