TOPGeoModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2016 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Marko Staric                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <top/dbobjects/TOPGeoBase.h>
#include <top/dbobjects/TOPGeoBarSegment.h>
#include <top/dbobjects/TOPGeoMirrorSegment.h>
#include <top/dbobjects/TOPGeoPrism.h>
#include <top/dbobjects/TOPGeoPMTArray.h>
#include <top/dbobjects/TOPGeoPMTArrayDisplacement.h>
#include <top/dbobjects/TOPGeoModuleDisplacement.h>
#include <TVector3.h>
#include <TRotation.h>
 
namespace Belle2 {
  class TOPGeoModule: public TOPGeoBase {
  public:
 
    TOPGeoModule()
    {}
 
    TOPGeoModule(int moduleID, double radius, double phi, double backwardZ,
                 const std::string& name = "TOPModule"): TOPGeoBase(name),
      m_moduleID(moduleID), m_radius(radius), m_phi(phi), m_backwardZ(backwardZ)
    {}
 
    TOPGeoModule(const TOPGeoModule& module): TOPGeoBase(module.getName())
    {
      *this = module;
      m_rotation = 0;
      m_rotationInverse = 0;
      m_translation = 0;
    }
 
    TOPGeoModule& operator=(const TOPGeoModule& module)
    {
      if (this != &module) {
        TOPGeoBase::operator=(module);
        m_moduleID = module.getModuleID();
        m_radius = module.getRadius();
        m_phi = module.getPhi();
        m_backwardZ = module.getBackwardZ();
        m_moduleCNumber = module.getModuleCNumber();
        m_bar1 = module.getBarSegment1();
        m_bar2 = module.getBarSegment2();
        m_mirror = module.getMirrorSegment();
        m_prism = module.getPrism();
        m_pmtArray = module.getPMTArray();
        m_arrayDisplacement = module.getPMTArrayDisplacement();
        m_moduleDisplacement = module.getModuleDisplacement();
        if (m_rotation) delete m_rotation;
        if (m_rotationInverse) delete m_rotationInverse;
        if (m_translation) delete m_translation;
        m_rotation = 0;
        m_rotationInverse = 0;
        m_translation = 0;
      }
      return *this;
    }
 
    ~TOPGeoModule()
    {
      if (m_rotation) delete m_rotation;
      if (m_rotationInverse) delete m_rotationInverse;
      if (m_translation) delete m_translation;
    }
 
    void setModuleCNumber(unsigned moduleCNumber) {m_moduleCNumber = moduleCNumber;}
 
    void setBarSegment1(const TOPGeoBarSegment& bar) {m_bar1 = bar;}
 
    void setBarSegment2(const TOPGeoBarSegment& bar) {m_bar2 = bar;}
 
    void setMirrorSegment(const TOPGeoMirrorSegment& mirror) {m_mirror = mirror;}
 
    void setPrism(const TOPGeoPrism& prism) {m_prism = prism;}
 
    void setPMTArray(const TOPGeoPMTArray& array) {m_pmtArray = array;}
 
    void setDecoupledPMT(unsigned pmtID) {m_pmtArray.setDecoupledPMT(pmtID);}
 
    void generateDecoupledPMTs(double fraction)
    {
      m_pmtArray.generateDecoupledPMTs(fraction);
    }
 
    void setBrokenGlue(int glueID, double fraction, double angle,
                       const std::string& material);
 
    void setPeelOffRegions(double thickness, const std::string& material);
 
    void appendPeelOffRegion(unsigned ID, double fraction, double angle)
    {
      m_prism.appendPeelOffRegion(ID, fraction, angle);
    }
 
 
    void setPMTArrayDisplacement(const TOPGeoPMTArrayDisplacement& displ)
    {
      m_arrayDisplacement = displ;
    }
 
    void setModuleDisplacement(const TOPGeoModuleDisplacement& displ)
    {
      m_moduleDisplacement = displ;
    }
 
    int getModuleID() const {return m_moduleID;}
 
    double getRadius() const {return m_radius / s_unit;}
 
    double getPhi() const {return m_phi;}
 
    double getBackwardZ() const {return m_backwardZ / s_unit;}
 
    unsigned getModuleCNumber() const {return m_moduleCNumber;}
 
    const TOPGeoBarSegment& getBarSegment1() const {return m_bar1;}
 
    const TOPGeoBarSegment& getBarSegment2() const {return m_bar2;}
 
    const TOPGeoMirrorSegment& getMirrorSegment() const {return m_mirror;}
 
    const TOPGeoPrism& getPrism() const {return m_prism;}
 
    const TOPGeoPMTArray& getPMTArray() const {return m_pmtArray;}
 
    const TOPGeoPMTArrayDisplacement& getPMTArrayDisplacement() const
    {
      return m_arrayDisplacement;
    }
 
    const TOPGeoModuleDisplacement& getModuleDisplacement() const
    {
      return m_moduleDisplacement;
    }
 
    double getBarLength() const
    {
      return m_bar1.getFullLength() + m_bar2.getFullLength() + m_mirror.getFullLength();
    }
 
    double getFullLength() const {return getBarLength() + m_prism.getFullLength();}
 
    double getForwardZ() const {return getBackwardZ() + getBarLength();}
 
    double getBarThickness() const
    {
      return (m_bar1.getThickness() * m_bar1.getFullLength() +
              m_bar2.getThickness() * m_bar2.getFullLength() +
              m_mirror.getThickness() * m_mirror.getFullLength()) / getBarLength();
    }
 
    double getBarWidth() const
    {
      return (m_bar1.getWidth() * m_bar1.getFullLength() +
              m_bar2.getWidth() * m_bar2.getFullLength() +
              m_mirror.getWidth() * m_mirror.getFullLength()) / getBarLength();
    }
 
    double getSurfaceReflectivity(double energy) const
    {
      return (m_bar1.getSurfaceReflectivity(energy) * m_bar1.getFullLength() +
              m_bar2.getSurfaceReflectivity(energy) * m_bar2.getFullLength() +
              m_mirror.getSurfaceReflectivity(energy) * m_mirror.getFullLength()) / getBarLength();
    }
 
    double getZc() const {return (getBackwardZ() + getForwardZ()) / 2;}
 
    double getInnerRadius() const {return getRadius() - getBarThickness() / 2;}
 
    TVector3 pointToLocal(const TVector3& point) const;
 
    TVector3 momentumToLocal(const TVector3& momentum) const;
 
    TVector3 pointToGlobal(const TVector3& point) const;
 
    TVector3 momentumToGlobal(const TVector3& momentum) const;
 
    bool isConsistent() const override;
 
    void print(const std::string& title = "Module geometry parameters") const override;
 
 
  private:
 
    void setTransformation() const;
 
    int m_moduleID = 0; 
    float m_radius = 0; 
    float m_phi = 0; 
    float m_backwardZ = 0; 
    unsigned m_moduleCNumber = 0; 
    TOPGeoBarSegment m_bar1; 
    TOPGeoBarSegment m_bar2; 
    TOPGeoMirrorSegment m_mirror; 
    TOPGeoPrism m_prism; 
    TOPGeoPMTArray m_pmtArray;  
    TOPGeoPMTArrayDisplacement m_arrayDisplacement;  
    TOPGeoModuleDisplacement m_moduleDisplacement;   
    mutable TRotation* m_rotation = 0;    
 
    mutable TRotation* m_rotationInverse = 0;    
 
    mutable TVector3* m_translation = 0;  
 
    ClassDefOverride(TOPGeoModule, 3); 
  };
 
} // end namespace Belle2