uLib/src/Vtk/HEP/Detectors/vtkMuonScatter.cxx

/*//////////////////////////////////////////////////////////////////////////////
// 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), m_Asm(vtkAssembly::New()) {
  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), m_Asm(vtkAssembly::New()) {
  InstallPipe();
}

vtkMuonScatter::~vtkMuonScatter() {
  m_LineIn->Delete();
  m_LineOut->Delete();
  m_Asm->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);
  m_Asm->AddPart(actor);

  mapper = vtkSmartPointer<vtkPolyDataMapper>::New();
  mapper->SetInputConnection(m_LineOut->GetOutputPort());
  actor = vtkSmartPointer<vtkActor>::New();
  actor->SetMapper(mapper);
  m_Asm->AddPart(actor);

  this->SetProp(m_Asm);
}

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);
  m_Asm->AddPart(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