TPCG4TrackInfo.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2016  Belle II Collaboration                              *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Igal Jaegle                                              *
 *                                                                        *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#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