TPCG4TrackInfo.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.                  *
 **************************************************************************/
 
#ifndef TPCTRACKINFO_H
#define TPCTRACKINFO_H
 
#include <simulation/dataobjects/SimHitBase.h>
 
// ROOT
#include <TVector3.h>
 
namespace Belle2 {
  class TPCG4TrackInfo : public SimHitBase {
  public:
 
    TPCG4TrackInfo(): m_trackID(0), m_PDG(0), m_Mass(0), m_Energy(0), m_vtx(), m_mom() {}
 
 
    TPCG4TrackInfo(int trackID, int PDG, float Mass, float Energy, const float* vtx, const float* mom):
      SimHitBase(), m_trackID(trackID), m_PDG(PDG), m_Mass(Mass), m_Energy(Energy)
    {
      std::copy(vtx, vtx + 3, m_vtx);
      std::copy(mom, mom + 3, m_mom);
    }
 
    int getTrackID() const { return m_trackID; }
    int getPDG()      const { return m_PDG; }
    float getMass()      const { return m_Mass; }
    float getEnergy()      const { return m_Energy; }
    TVector3 getProductionVertex()  const { return TVector3(m_vtx[0], m_vtx[1], m_vtx[2]); }
    TVector3 getMomentum() const { return TVector3(m_mom[0], m_mom[1], m_mom[2]); }
    const float* getProductionVertexArray() const { return m_vtx; }
    const float* getMomentumArray() const { return m_mom; }
 
  private:
    int m_trackID;
    int m_PDG;
    float m_Mass;
    float m_Energy;
    float m_vtx[3];
    float m_mom[3];
 
    ClassDef(TPCG4TrackInfo, 1)
  };
 
} // end namespace Belle2
 
#endif