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