145 lines
4.8 KiB
C++
145 lines
4.8 KiB
C++
/*//////////////////////////////////////////////////////////////////////////////
|
|
// 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.
|
|
|
|
//////////////////////////////////////////////////////////////////////////////*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include "config.h"
|
|
#endif
|
|
|
|
#include "Math/Dense.h"
|
|
#include "Vtk/HEP/Detectors/vtkMuonScatter.h"
|
|
|
|
namespace uLib {
|
|
namespace Vtk {
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
///// VTK MUON Scatter ///////////////////////////////////////////////////////
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
vtkMuonScatter::vtkMuonScatter(MuonScatter &content)
|
|
: m_Content(&content), m_LineIn(vtkLineSource::New()),
|
|
m_LineOut(vtkLineSource::New()), m_PolyData(vtkPolyData::New()),
|
|
m_SpherePoca(NULL) {
|
|
InstallPipe();
|
|
}
|
|
|
|
vtkMuonScatter::vtkMuonScatter(const MuonScatter &content)
|
|
: m_Content(const_cast<MuonScatter *>(&content)),
|
|
m_LineIn(vtkLineSource::New()), m_LineOut(vtkLineSource::New()),
|
|
m_PolyData(vtkPolyData::New()), m_SpherePoca(NULL) {
|
|
InstallPipe();
|
|
}
|
|
|
|
vtkMuonScatter::~vtkMuonScatter() {
|
|
m_LineIn->Delete();
|
|
m_LineOut->Delete();
|
|
if (m_SpherePoca)
|
|
m_SpherePoca->Delete();
|
|
}
|
|
|
|
vtkMuonScatter::Content &vtkMuonScatter::GetContent() { return *m_Content; }
|
|
|
|
void vtkMuonScatter::PrintSelf(std::ostream &o) const {}
|
|
|
|
void vtkMuonScatter::InstallPipe() {
|
|
if (m_Content) {
|
|
vtkLineSource *line_in = m_LineIn;
|
|
vtkLineSource *line_out = m_LineOut;
|
|
|
|
float distance =
|
|
(m_Content->LineIn().origin - m_Content->LineOut().origin).norm() / 10;
|
|
|
|
HPoint3f pt;
|
|
pt = m_Content->LineIn().origin;
|
|
line_in->SetPoint1(pt(0), pt(1), pt(2));
|
|
pt = m_Content->LineIn().origin + m_Content->LineIn().direction * distance;
|
|
line_in->SetPoint2(pt(0), pt(1), pt(2));
|
|
pt = m_Content->LineOut().origin;
|
|
line_out->SetPoint1(pt(0), pt(1), pt(2));
|
|
pt =
|
|
m_Content->LineOut().origin + m_Content->LineOut().direction * distance;
|
|
line_out->SetPoint2(pt(0), pt(1), pt(2));
|
|
}
|
|
|
|
vtkSmartPointer<vtkPolyDataMapper> mapper =
|
|
vtkSmartPointer<vtkPolyDataMapper>::New();
|
|
mapper->SetInputConnection(m_LineIn->GetOutputPort());
|
|
vtkSmartPointer<vtkActor> actor = vtkSmartPointer<vtkActor>::New();
|
|
actor->SetMapper(mapper);
|
|
this->SetProp(actor);
|
|
|
|
mapper = vtkSmartPointer<vtkPolyDataMapper>::New();
|
|
mapper->SetInputConnection(m_LineOut->GetOutputPort());
|
|
actor = vtkSmartPointer<vtkActor>::New();
|
|
actor->SetMapper(mapper);
|
|
this->SetProp(actor);
|
|
}
|
|
|
|
vtkPolyData *vtkMuonScatter::GetPolyData() const {
|
|
vtkSmartPointer<vtkAppendPolyData> append =
|
|
vtkSmartPointer<vtkAppendPolyData>::New();
|
|
append->AddInputConnection(m_LineIn->GetOutputPort());
|
|
append->AddInputConnection(m_LineOut->GetOutputPort());
|
|
if (m_SpherePoca)
|
|
append->AddInputConnection(m_SpherePoca->GetOutputPort());
|
|
append->Update();
|
|
m_PolyData->DeepCopy(append->GetOutput());
|
|
return m_PolyData;
|
|
}
|
|
|
|
void vtkMuonScatter::AddPocaPoint(HPoint3f poca) {
|
|
vtkSphereSource *sphere = vtkSphereSource::New();
|
|
float size =
|
|
(m_Content->LineIn().origin - m_Content->LineOut().origin).head(3).norm();
|
|
size /= 100;
|
|
sphere->SetRadius(size);
|
|
sphere->SetCenter(poca(0), poca(1), poca(2));
|
|
sphere->Update();
|
|
m_SpherePoca = sphere;
|
|
|
|
vtkSmartPointer<vtkPolyDataMapper> mapper =
|
|
vtkSmartPointer<vtkPolyDataMapper>::New();
|
|
mapper->SetInputConnection(m_SpherePoca->GetOutputPort());
|
|
vtkSmartPointer<vtkActor> actor = vtkSmartPointer<vtkActor>::New();
|
|
actor->SetMapper(mapper);
|
|
this->SetProp(actor);
|
|
}
|
|
|
|
HPoint3f vtkMuonScatter::GetPocaPoint() {
|
|
double center[3];
|
|
if (m_SpherePoca) {
|
|
m_SpherePoca->GetCenter(center);
|
|
return HPoint3f(center[0], center[1], center[2]);
|
|
} else {
|
|
return HPoint3f(0, 0, 0);
|
|
}
|
|
}
|
|
|
|
void vtkMuonScatter::ConnectInteractor(vtkRenderWindowInteractor *interactor) {
|
|
// TODO
|
|
}
|
|
|
|
} // namespace Vtk
|
|
} // namespace uLib
|