DetectorChamber vtk handler

This commit is contained in:
AndreaRigoni
2026-03-10 08:18:17 +00:00
parent 00275ac56d
commit f2133c31d5
14 changed files with 358 additions and 272 deletions

View File

@@ -13,3 +13,4 @@ set(LIBRARIES
)
uLib_add_tests(VtkDetectors)

View File

@@ -33,11 +33,16 @@
BOOST_AUTO_TEST_CASE(vtkDetectorChamberTest) {
uLib::DetectorChamber d1, d2;
d1.SetSize(uLib::Vector3f(10, 20, 2));
d1.SetSize(uLib::Vector3f(1, 1, 1));
d1.SetPosition(uLib::Vector3f(0, 0, 0));
d1.Scale(uLib::Vector3f(5, 10, 2));
d1.Translate(uLib::Vector3f(0, 0, 0));
d2.SetSize(uLib::Vector3f(1, 1, 1));
d2.SetPosition(uLib::Vector3f(0, 0, 0));
d2.Scale(uLib::Vector3f(5, 10, 2));
d2.Translate(uLib::Vector3f(0, 0, 10));
d2.SetSize(uLib::Vector3f(10, 20, 2));
d2.SetPosition(uLib::Vector3f(0, 0, 20));
uLib::Vtk::vtkDetectorChamber vtkDetectorChamber(&d1);
uLib::Vtk::vtkDetectorChamber vtkDetectorChamber2(&d2);

View File

@@ -23,6 +23,7 @@
//////////////////////////////////////////////////////////////////////////////*/
#include <vtkAbstractTransform.h>
#include <vtkAxes.h>
#include <vtkCubeSource.h>
#include <vtkLineSource.h>
@@ -38,23 +39,31 @@
#include <vtkTransform.h>
#include "Vtk/HEP/Detectors/vtkDetectorChamber.h"
#include "Vtk/vtkHandlerWidget.h"
#include <vtkBoxWidget.h>
#include <vtkTransformPolyDataFilter.h>
namespace uLib {
namespace Vtk {
vtkDetectorChamber::vtkDetectorChamber(DetectorChamber *content)
: vtkContainerBox(content) {
m_Widget = vtkHandlerWidget::New();
: vtkContainerBox(content), m_Actor(vtkActor::New()),
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_InitialTransform = vtkSmartPointer<vtkTransform>::New();
m_RelativeTransform = vtkSmartPointer<vtkTransform>::New();
m_TotalTransform = vtkSmartPointer<vtkTransform>::New();
this->InstallPipe();
}
vtkDetectorChamber::~vtkDetectorChamber() {
m_Actor->Delete();
m_Widget->Delete();
m_Callback->Delete();
m_PickerCallback->Delete();
@@ -68,94 +77,97 @@ void vtkDetectorChamber::PrintSelf(std::ostream &o) const {
vtkContainerBox::PrintSelf(o);
}
/**
* Connect the interactor to the widget
*/
void vtkDetectorChamber::ConnectInteractor(
vtkRenderWindowInteractor *interactor) {
if (!interactor)
return;
m_Widget->SetInteractor(interactor);
m_Widget->SetProp3D(m_Cube);
m_Widget->PlaceWidget();
m_Widget->SetProp3D(m_Actor);
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();
void vtkDetectorChamber::SetTransform(vtkTransform *t) {
m_RelativeTransform->SetMatrix(t->GetMatrix());
m_RelativeTransform->Update();
// Set content global transform (BaseClass of ContainerBox) //
vtkMatrix4x4 *vmat = m_TotalTransform->GetMatrix();
Matrix4f transform;
for (int i = 0; i < 4; ++i)
for (int j = 0; j < 4; ++j)
vmat->SetElement(i, j, mat(i, j));
transform(i, j) = vmat->GetElement(i, j);
this->GetContent()->SetMatrix(transform);
m_Cube->SetUserMatrix(vmat);
m_Axes->SetUserMatrix(vmat);
m_Pivot->SetUserMatrix(vmat);
this->Update();
}
vtkBoxWidget *vtkDetectorChamber::GetWidget() { return m_Widget; }
void vtkDetectorChamber::Update() {
if (m_Actor->GetMapper())
m_Actor->GetMapper()->Update();
BaseClass::Update();
}
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);
// 1. Initialize Global Transform (m_Transform) from Content's matrix (Base
// class AffineTransform)
vtkSmartPointer<vtkMatrix4x4> vmatGlobal =
vtkSmartPointer<vtkMatrix4x4>::New();
Matrix4f matGlobal = this->GetContent()->GetMatrix();
for (int i = 0; i < 4; ++i)
for (int j = 0; j < 4; ++j)
vmatGlobal->SetElement(i, j, matGlobal(i, j));
m_InitialTransform->SetMatrix(vmatGlobal);
m_InitialTransform->Update();
vtkSmartPointer<vtkPolyDataMapper> mapper =
vtkSmartPointer<vtkPolyDataMapper>::New();
mapper->SetInputConnection(cube->GetOutputPort());
m_Cube->SetMapper(mapper);
m_Cube->GetProperty()->SetRepresentationToWireframe();
m_Cube->GetProperty()->SetAmbient(0.7);
m_Actor->SetMapper(mapper);
// 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);
m_Actor->GetProperty()->SetRepresentationToSurface();
m_Actor->GetProperty()->SetEdgeVisibility(true);
m_Actor->GetProperty()->SetOpacity(0.4);
m_Actor->GetProperty()->SetAmbient(0.7);
// 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);
// Temporarily disable UserTransform to place widget on local base
m_Widget->SetProp3D(m_Actor);
this->SetProp(m_Cube);
this->SetProp(m_Axes);
this->SetProp(m_Pivot);
m_TotalTransform->SetInput(m_RelativeTransform);
m_TotalTransform->Concatenate(m_InitialTransform);
m_Actor->SetUserTransform(m_TotalTransform);
m_TotalTransform->Update();
m_Widget->PlaceWidget();
m_Widget->SetPlaceFactor(2);
this->SetProp(m_Actor);
this->Update();
}
void vtkDetectorChamber::vtkWidgetCallback::Execute(vtkObject *caller,
unsigned long, void *) {
vtkHandlerWidget *widget = reinterpret_cast<vtkHandlerWidget *>(caller);
vtkBoxWidget *widget = reinterpret_cast<vtkBoxWidget *>(caller);
// Get the Relative transform from the widget //
vtkSmartPointer<vtkTransform> t = vtkSmartPointer<vtkTransform>::New();
widget->GetTransform(t);
chamber->SetTransform(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);
}
// Apply to both the content and the actor state //
chamber->Update();
}
@@ -170,22 +182,9 @@ void vtkDetectorChamber::vtkSelectionCallback::Execute(vtkObject *caller,
interactor->GetRenderWindow()->GetRenderers()->GetFirstRenderer());
vtkProp *picked = picker->GetViewProp();
if (picked == chamber->m_Cube || picked == chamber->m_Axes ||
picked == chamber->m_Pivot) {
if (picked == chamber->m_Actor) {
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->SetCurrentRenderer(
interactor->GetRenderWindow()->GetRenderers()->GetFirstRenderer());
chamber->m_Widget->SetTransform(t);
chamber->m_Widget->PlaceWidget();
chamber->m_Widget->SetInteractor(interactor);
chamber->m_Widget->On();
}
} else {

View File

@@ -35,14 +35,18 @@
#include "Math/Dense.h"
#include "Vtk/uLibVtkInterface.h"
#include "Vtk/vtkContainerBox.h"
#include <vtkActor.h>
#include <vtkBoxWidget.h>
#include <vtkTransformPolyDataFilter.h>
namespace uLib {
namespace Vtk {
class vtkHandlerWidget;
// class vtkHandlerWidget; // Removed as we use vtkBoxWidget now
class vtkDetectorChamber : public vtkContainerBox {
typedef DetectorChamber Content;
typedef vtkContainerBox BaseClass;
public:
vtkDetectorChamber(DetectorChamber *content);
@@ -50,6 +54,10 @@ public:
Content *GetContent();
void SetTransform(class vtkTransform *t);
class vtkBoxWidget *GetWidget();
void Update();
void ConnectInteractor(vtkRenderWindowInteractor *interactor) override;
@@ -80,9 +88,14 @@ private:
uLib::Vtk::vtkDetectorChamber *chamber;
};
vtkHandlerWidget *m_Widget;
vtkActor *m_Actor;
vtkBoxWidget *m_Widget;
vtkWidgetCallback *m_Callback;
vtkSelectionCallback *m_PickerCallback;
vtkSmartPointer<vtkTransform> m_InitialTransform;
vtkSmartPointer<vtkTransform> m_RelativeTransform;
vtkSmartPointer<vtkTransform> m_TotalTransform;
};
} // namespace Vtk

View File

@@ -1,93 +0,0 @@
#if VTK_5_x
_VTK_LIBS = -lQVTK \
-lLSDyna \
-lMapReduceMPI \
-lmpistubs \
-lvtkalglib \
-lvtkCharts \
-lvtkCommon \
-lvtkDICOMParser \
-lvtkexoIIc \
-lvtkFiltering \
-lvtkftgl \
-lvtkGenericFiltering \
-lvtkGraphics \
-lvtkHybrid \
-lvtkImaging \
-lvtkInfovis \
-lvtkmetaio \
-lvtkNetCDF_cxx \
-lvtkNetCDF \
-lvtkproj4 \
-lvtkRendering \
-lvtksqlite \
-lvtksys \
-lvtkverdict \
-lvtkViews \
-lvtkVolumeRendering \
-lvtkWidgets
AM_CFLAGS += $(VTK_CFLAGS)
AM_CXXFLAGS += $(VTK_CXXFLAGS)
AM_LDFLAGS += $(VTK_LDFLAGS) $(_VTK_LIBS)
#else
#_VTK_LIBS = \
# -lvtkRenderingOpenGL-6.0 \
# -lvtkImagingHybrid-6.0 \
# -lvtkIOImage-6.0 \
# -lvtkCommonDataModel-6.0 \
# -lvtkCommonMath-6.0 \
# -lvtkCommonCore-6.0 \
# -lvtksys-6.0 \
# -lvtkCommonMisc-6.0 \
# -lvtkCommonSystem-6.0 \
# -lvtkCommonTransforms-6.0 \
# -lvtkCommonExecutionModel-6.0 \
# -lvtkDICOMParser-6.0 \
# -lvtkIOCore-6.0 \
# -lvtkzlib-6.0 \
# -lvtkmetaio-6.0 \
# -lvtkjpeg-6.0 \
# -lvtkpng-6.0 \
# -lvtktiff-6.0 \
# -lvtkImagingCore-6.0 \
# -lvtkRenderingCore-6.0 \
# -lvtkFiltersExtraction-6.0 \
# -lvtkFiltersCore-6.0 \
# -lvtkFiltersGeneral-6.0 \
# -lvtkCommonComputationalGeometry-6.0 \
# -lvtkFiltersStatistics-6.0 \
# -lvtkImagingFourier-6.0 \
# -lvtkalglib-6.0 \
# -lvtkFiltersGeometry-6.0 \
# -lvtkFiltersSources-6.0 \
# -lvtkIOXMLParser-6.0 \
# -lvtkexpat-6.0 \
# -lvtkRenderingAnnotation-6.0 \
# -lvtkImagingColor-6.0 \
# -lvtkRenderingFreeType-6.0 \
# -lvtkfreetype-6.0 \
# -lvtkftgl-6.0 \
# -lvtkInteractionStyle-6.0 \
# -lvtkRenderingVolumeOpenGL-6.0 \
# -lvtkRenderingVolume-6.0 \
# -lvtkIOXML-6.0 \
# -lvtkIOGeometry-6.0 \
# -lvtkjsoncpp-6.0 \
# -lvtkInteractionWidgets-6.0 \
# -lvtkFiltersHybrid-6.0 \
# -lvtkImagingSources-6.0 \
# -lvtkFiltersModeling-6.0 \
# -lvtkImagingGeneral-6.0 \
# -lvtkIOLegacy-6.0 \
# -lvtkRenderingFreeTypeOpenGL-6.0
#AM_CFLAGS += $(VTK_CFLAGS)
#AM_CXXFLAGS += $(VTK_CXXFLAGS) -rdynamic
#AM_LDFLAGS += $(VTK_LDFLAGS) $(_VTK_LIBS) -lGLU -lSM -lICE -lX11 -lXext -lSM -lICE -lX11 -lXext -lXt -lm -ldl -lGL
#endif

View File

@@ -23,17 +23,11 @@
//////////////////////////////////////////////////////////////////////////////*/
#ifndef ULIBVTKINTERFACE_H
#define ULIBVTKINTERFACE_H
#include <ostream>
#include <iomanip>
#include <ostream>
// vtk classes forward declaration //
class vtkProp;
@@ -42,66 +36,57 @@ class vtkPropCollection;
class vtkRenderer;
class vtkRendererCollection;
namespace uLib {
namespace Vtk {
class Puppet
{
class Puppet {
public:
Puppet();
~Puppet();
Puppet();
~Puppet();
virtual vtkProp * GetProp();
virtual vtkProp *GetProp();
virtual vtkPropCollection * GetProps();
virtual vtkPropCollection *GetProps();
void ConnectRenderer(vtkRenderer *renderer);
void ConnectRenderer(vtkRenderer *renderer);
void DisconnectRenderer(vtkRenderer *renderer);
void DisconnectRenderer(vtkRenderer *renderer);
void ConnectViewer(class Viewer *viewer);
void ConnectViewer(class Viewer *viewer);
void DisonnectViewer(class Viewer *viewer);
void DisonnectViewer(class Viewer *viewer);
vtkRendererCollection * GetRenderers() const;
vtkRendererCollection *GetRenderers() const;
virtual void PrintSelf(std::ostream &o) const;
virtual void PrintSelf(std::ostream &o) const;
virtual void ConnectInteractor(class vtkRenderWindowInteractor *interactor) { (void)interactor; }
virtual void ConnectInteractor(class vtkRenderWindowInteractor *interactor) {
(void)interactor;
}
protected:
void SetProp(vtkProp *prop);
void SetProp(vtkProp *prop);
void RemoveProp(vtkProp *prop);
void RemoveProp(vtkProp *prop);
private:
friend class PuppetData;
class PuppetData *d;
friend class PuppetData;
class PuppetData *d;
};
class Polydata {
public:
virtual vtkPolyData *GetPolyData() const { return NULL; }
virtual vtkPolyData * GetPolyData() const { return NULL; }
virtual void SaveToFile(const char *vtk_file);
virtual void SaveToFile(const char *vtk_file);
virtual void SaveToXMLFile(const char *vtp_file);
virtual void SaveToXMLFile(const char *vtp_file);
protected:
virtual ~Polydata() {}
virtual ~Polydata() {}
};
} // vtk
} // uLib
} // namespace Vtk
} // namespace uLib
#endif // ULIBVTKINTERFACE_H

View File

@@ -86,6 +86,7 @@ Viewer::~Viewer() {
void Viewer::InstallPipe() {
m_RenderWindow->AddRenderer(m_Renderer);
m_RenderWindow->SetSize(600,600);
vtkSmartPointer<vtkRenderWindowInteractor> renderWindowInteractor =
vtkSmartPointer<vtkRenderWindowInteractor>::New();
renderWindowInteractor->SetRenderWindow(m_RenderWindow);

View File

@@ -27,23 +27,28 @@
#include "config.h"
#endif
#include <iostream>
#include "vtkContainerBox.h"
#include <vtkActor.h>
#include <vtkAssembly.h>
#include <vtkAxes.h>
#include <vtkCubeSource.h>
#include <vtkLineSource.h>
#include <vtkMatrix4x4.h>
#include <vtkPolyDataMapper.h>
#include <vtkProperty.h>
#include <vtkSmartPointer.h>
#include <vtkTransform.h>
#include "vtkContainerBox.h"
namespace uLib {
namespace Vtk {
vtkContainerBox::vtkContainerBox(vtkContainerBox::Content *content)
: m_Cube(vtkActor::New()), m_Axes(vtkActor::New()),
m_Pivot(vtkActor::New()), m_Content(content) {
m_Pivot(vtkActor::New()),
m_Content(content) {
this->InstallPipe();
}
@@ -58,6 +63,28 @@ vtkPolyData *vtkContainerBox::GetPolyData() const {
return NULL;
}
void vtkContainerBox::Update() {
if (!m_Content)
return;
vtkSmartPointer<vtkMatrix4x4> vmat = vtkSmartPointer<vtkMatrix4x4>::New();
Matrix4f transform = m_Content->GetMatrix();
for (int i = 0; i < 4; ++i)
for (int j = 0; j < 4; ++j)
vmat->SetElement(i, j, transform(i, j));
std::cout << "transform: " << transform << std::endl;
// m_RelativeTransform->SetMatrix(vmat);
// m_RelativeTransform->Update();
m_Cube->SetUserMatrix(vmat);
m_Axes->SetUserMatrix(vmat);
}
void vtkContainerBox::InstallPipe() {
if (!m_Content)
return;
@@ -65,17 +92,27 @@ void vtkContainerBox::InstallPipe() {
// CUBE
vtkSmartPointer<vtkCubeSource> cube = vtkSmartPointer<vtkCubeSource>::New();
Vector3f p = c->GetPosition();
cube->SetCenter(p(0), p(1), p(2));
Vector4f p1 = c->GetWorldPoint(HPoint3f(0, 0, 0));
Vector4f p2 = c->GetWorldPoint(HPoint3f(1, 1, 1));
vtkSmartPointer<vtkLineSource> line = vtkSmartPointer<vtkLineSource>::New();
line->SetPoint1(p1(0), p1(1), p1(2));
line->SetPoint2(p2(0), p2(1), p2(2));
line->Update();
cube->SetBounds(line->GetOutput()->GetBounds());
// cube->SetCenter(p(0), p(1), p(2));
// Vector4f p1 = c->GetWorldPoint(HPoint3f(0, 0, 0));
// Vector4f p2 = c->GetWorldPoint(HPoint3f(1, 1, 1));
// vtkSmartPointer<vtkLineSource> line =
// vtkSmartPointer<vtkLineSource>::New(); line->SetPoint1(p1(0), p1(1),
// p1(2)); line->SetPoint2(p2(0), p2(1), p2(2)); line->Update();
// cube->SetBounds(line->GetOutput()->GetBounds());
vtkSmartPointer<vtkMatrix4x4> vmat = vtkSmartPointer<vtkMatrix4x4>::New();
Matrix4f transform = c->GetMatrix();
for (int i = 0; i < 4; ++i)
for (int j = 0; j < 4; ++j)
vmat->SetElement(i, j, transform(i, j));
m_Cube->SetUserMatrix(vmat);
vtkSmartPointer<vtkPolyDataMapper> mapper =
vtkSmartPointer<vtkPolyDataMapper>::New();
cube->SetBounds(0, 1, 0, 1, 0, 1);
mapper->SetInputConnection(cube->GetOutputPort());
mapper->Update();
m_Cube->SetMapper(mapper);
@@ -84,11 +121,12 @@ void vtkContainerBox::InstallPipe() {
// AXES //
vtkSmartPointer<vtkAxes> axes = vtkSmartPointer<vtkAxes>::New();
axes->SetOrigin(p1(0), p1(1), p1(2));
axes->SetOrigin(0, 0, 0);
mapper = vtkSmartPointer<vtkPolyDataMapper>::New();
mapper->SetInputConnection(axes->GetOutputPort());
mapper->Update();
m_Axes->SetMapper(mapper);
m_Axes->SetUserMatrix(vmat);
Vector3f s = c->GetSize();
// m_Axes->SetScale(s(0),s(1),s(2));
m_Axes->GetProperty()->SetLineWidth(3);
@@ -97,11 +135,19 @@ void vtkContainerBox::InstallPipe() {
// PIVOT //
axes = vtkSmartPointer<vtkAxes>::New();
axes->SetOrigin(p(0), p(1), p(2));
axes->SetOrigin(0, 0, 0);
mapper = vtkSmartPointer<vtkPolyDataMapper>::New();
mapper->SetInputConnection(axes->GetOutputPort());
mapper->Update();
m_Pivot->SetUserMatrix(vmat);
m_Pivot->SetMapper(mapper);
Matrix4f pivotTransform = c->AffineTransform::GetWorldMatrix();
vtkSmartPointer<vtkMatrix4x4> pmat = vtkSmartPointer<vtkMatrix4x4>::New();
for (int i = 0; i < 4; ++i)
for (int j = 0; j < 4; ++j)
pmat->SetElement(i, j, pivotTransform(i, j));
m_Pivot->SetUserMatrix(pmat);
s = c->GetScale();
// m_Pivot->SetScale(s(0),s(1),s(2));
m_Pivot->GetProperty()->SetLineWidth(3);

View File

@@ -28,6 +28,7 @@
#include "Math/ContainerBox.h"
#include "uLibVtkInterface.h"
#include <vtkActor.h>
namespace uLib {
namespace Vtk {
@@ -41,6 +42,9 @@ public:
virtual class vtkPolyData *GetPolyData() const;
virtual void Update();
protected:
virtual void InstallPipe();