BeamBackHit.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <TObject.h>
#include <TVector3.h>
#include <TString.h>
 
namespace Belle2 {
 
  class BeamBackHit : public TObject {
  public:
 
    BeamBackHit(): m_subDet(0), m_identifier(0), m_PDG(0), m_trackID(0),
      m_position(0., 0., 0.), m_momentum(0., 0., 0.), m_t(0.), m_E_start(0.),
      m_E_end(0.), m_energyDeposit(0), m_trackLength(0.), m_neutronWeight(0.)
    {}
 
    BeamBackHit(int subDet, int iden, int PDG, int trackID, TVector3 position,
                TVector3 momentum, double t, double E_start, double E_end,
                double eDep, double trackLength, double nWeight):
      m_subDet(subDet), m_identifier(iden), m_PDG(PDG), m_trackID(trackID),
      m_position(position), m_momentum(momentum), m_t(t), m_E_start(E_start),
      m_E_end(E_end), m_energyDeposit(eDep), m_trackLength(trackLength),
      m_neutronWeight(nWeight)
    {}
 
    TString getSubDetName() const
    {
      switch (m_subDet) {
        case 0: return "IR";
        case 1: return "PDX";
        case 2: return "SVD";
        case 3: return "CDC";
        case 4: return "ARICH";
        case 5: return "TOP";
        case 6: return "ECL";
        case 7: return "EKLM";
        case 8: return "BKLM";
      }
      return "";
    }
 
    int getIdentifier() const { return m_identifier;    }
 
    int getSubDet() const {return m_subDet; }
 
    int getPDG() const { return m_PDG; }
 
    int getTrackID() const { return m_trackID;}
 
    const TVector3& getPosition() const { return m_position; }
 
    const TVector3& getMomentum() const { return m_momentum; }
 
    double getTime() const { return m_t; }
 
    double getEnergy() const { return m_E_start; }
 
    double getEnergyAtExit() const { return m_E_end; }
 
    double getEnergyDeposit() const { return m_energyDeposit; }
 
    double getTrackLength() const { return m_trackLength; }
 
    double getNeutronWeight() const {return m_neutronWeight;}
 
    void setSubDet(int subDet) { m_subDet = subDet; }
 
    void setIdentifier(int iden) { m_identifier = iden; }
 
    void setPDG(int PDG)  { m_PDG = PDG; }
 
    void setTrackID(int trackID) {m_trackID = trackID;}
 
    void setPosition(TVector3 position)  {  m_position = position; }
 
    void setMomentum(TVector3 momentum)  {  m_momentum = momentum; }
 
    void setTime(double t)  {  m_t = t; }
 
    void setEnergy(double E)  {  m_E_start = E; }
 
    void setEnergyAtExit(double E) {m_E_end = E;}
 
    void setEnergyDeposit(double eDep) { m_energyDeposit = eDep; }
 
    void setTrackLength(double l) {m_trackLength = l;}
 
    void setNeutronWeight(double w) {m_neutronWeight = w;}
 
    void shiftInTime(double time) { m_t += time; }
 
  private:
    int m_subDet;               
    int m_identifier;           
    int m_PDG;                  
    int m_trackID;                
    TVector3 m_position;        
    TVector3 m_momentum;        
    double m_t;                 
    double m_E_start;           
    double m_E_end;             
    double m_energyDeposit;     
    double m_trackLength;       
    double m_neutronWeight;     
    ClassDef(BeamBackHit, 1);   
  };
 
} // end namespace Belle2