add cylinder

2026-03-24 15:22:50 +00:00
parent b45cde0bad
commit 51e6dbb4f5
11 changed files with 523 additions and 121 deletions
--- a/src/Vtk/Math/vtkCylinder.cpp
+++ b/src/Vtk/Math/vtkCylinder.cpp
@@ -0,0 +1,144 @@
+/*//////////////////////////////////////////////////////////////////////////////
+// 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 "Vtk/Math/vtkCylinder.h"
+#include <vtkActor.h>
+#include <vtkCylinderSource.h>
+#include <vtkMatrix4x4.h>
+#include <vtkPolyDataMapper.h>
+#include <vtkProperty.h>
+#include <vtkSmartPointer.h>
+#include <vtkTransform.h>
+#include "Math/vtkDense.h"
+
+namespace uLib {
+namespace Vtk {
+
+vtkCylinder::vtkCylinder(vtkCylinder::Content *content)
+    : m_Actor(vtkActor::New()), m_Content(content) {
+  this->InstallPipe();
+  Object::connect(m_Content, &Content::Updated, this, &vtkCylinder::contentUpdate);
+}
+
+vtkCylinder::~vtkCylinder() {
+  m_Actor->Delete();
+}
+
+void vtkCylinder::contentUpdate() {
+  if (!m_Content)
+    return;
+
+  vtkProp3D* root = vtkProp3D::SafeDownCast(this->GetProp());
+  if (!root) return;
+
+  vtkMatrix4x4* vmat = root->GetUserMatrix();
+  if (!vmat) {
+    vtkNew<vtkMatrix4x4> mat;
+    root->SetUserMatrix(mat);
+    vmat = mat;
+  }
+
+  // Multiply the placement matrix by the volume scaling (Radius, Radius, Height)
+  Matrix4f transform = m_Content->GetMatrix() * m_Content->GetLocalMatrix();
+  Matrix4fToVtk(transform, vmat);
+
+  // Update internal alignment based on active axis
+  vtkTransform* alignment = vtkTransform::SafeDownCast(m_Actor->GetUserTransform());
+  if (alignment) {
+      alignment->Identity();
+      int axis = m_Content->GetAxis();
+      if (axis == 0) alignment->RotateZ(-90); // Y -> X
+      else if (axis == 1) ;                   // Y -> Y (identity)
+      else if (axis == 2) alignment->RotateX(90);  // Y -> Z
+      
+      // We keep it centered as per latest user preference in Step 677
+      // alignment->Translate(0, 0, 0); // Implicit
+  }
+
+  root->Modified();
+  Puppet::Update();
+}
+
+void vtkCylinder::Update() {
+  if (!m_Content) return;
+
+  vtkProp3D* root = vtkProp3D::SafeDownCast(this->GetProp());
+  if (!root) return;
+
+  vtkMatrix4x4* vmat = root->GetUserMatrix();
+  if (!vmat) return;
+  
+  Matrix4f fullTransform = VtkToMatrix4f(vmat);
+  Matrix4f placementScale = m_Content->GetLocalMatrix().inverse();
+  Matrix4f transform = fullTransform * placementScale;
+
+  if (m_Content->GetParent()) {
+    Matrix4f localT = m_Content->GetParent()->GetWorldMatrix().inverse() * transform;
+    m_Content->SetMatrix(localT);
+  } else {
+    m_Content->SetMatrix(transform);
+  }
+
+  m_Content->Updated();
+}
+
+void vtkCylinder::InstallPipe() {
+  if (!m_Content)
+    return;
+
+  vtkNew<vtkCylinderSource> cylinder;
+  cylinder->SetRadius(1.0);
+  cylinder->SetHeight(1.0);
+  cylinder->SetResolution(32);
+
+  vtkNew<vtkTransform> alignment;
+  alignment->Identity();
+  alignment->Translate(0, 0, -0.5);
+
+  // Default to Y alignment (Identity) as per latest request
+  int axis = m_Content->GetAxis();
+  if (axis == 0) alignment->RotateZ(-90);
+  else if (axis == 2) alignment->RotateX(90);
+
+  vtkNew<vtkPolyDataMapper> mapper;
+  mapper->SetInputConnection(cylinder->GetOutputPort());
+  
+  m_Actor->SetMapper(mapper);
+  m_Actor->SetUserTransform(alignment);
+  m_Actor->GetProperty()->SetRepresentationToWireframe();
+  m_Actor->GetProperty()->SetAmbient(0.6);
+
+  this->SetProp(m_Actor);
+  
+  vtkProp3D* root = vtkProp3D::SafeDownCast(this->GetProp());
+  if (root) {
+      vtkNew<vtkMatrix4x4> vmat;
+      Matrix4fToVtk(m_Content->GetMatrix() * m_Content->GetLocalMatrix(), vmat);
+      root->SetUserMatrix(vmat);
+  }
+}
+
+} // namespace Vtk
+} // namespace uLib