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 <Math/Vector3D.h>
#include <TObject.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_positionX(0.), m_positionY(0.), m_positionZ(0.), m_momentumX(0.),
      m_momentumY(0.), m_momentumZ(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,
                const ROOT::Math::XYZVector& position,
                const ROOT::Math::XYZVector& 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_t(t), m_E_start(E_start), m_E_end(E_end), m_energyDeposit(eDep),
      m_trackLength(trackLength), m_neutronWeight(nWeight)
    {
      setPosition(position);
      setMomentum(momentum);
    }
 
    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;}
 
    ROOT::Math::XYZVector getPosition() const
    {
      return ROOT::Math::XYZVector(m_positionX, m_positionY, m_positionZ);
    }
 
    ROOT::Math::XYZVector getMomentum() const
    {
      return ROOT::Math::XYZVector(m_momentumX, m_momentumY, m_momentumZ);
    }
 
    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(ROOT::Math::XYZVector position)
    {
      m_positionX = position.X();
      m_positionY = position.Y();
      m_positionZ = position.Z();
    }
 
    void setMomentum(ROOT::Math::XYZVector momentum)
    {
      m_momentumX = momentum.X();
      m_momentumY = momentum.Y();
      m_momentumZ = momentum.Z();
    }
 
    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;
 
    float m_positionX;
 
    float m_positionY;
 
    float m_positionZ;
 
    float m_momentumX;
 
    float m_momentumY;
 
    float m_momentumZ;
 
    float m_t;
 
    float m_E_start;
 
    float m_E_end;
 
    float m_energyDeposit;
 
    float m_trackLength;
 
    float m_neutronWeight;
 
    ClassDef(BeamBackHit, 2);
 
  };
 
} // end namespace Belle2