TpcMCParticle.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 TPCMCPARTICLE_H
#define TPCMCPARTICLE_H
 
#include <simulation/dataobjects/SimHitBase.h>
 
// ROOT
#include <TVector3.h>
 
namespace Belle2 {
  class TpcMCParticle : public SimHitBase {
  public:
 
    TpcMCParticle(): m_PDG(0), m_Mass(0), m_Energy(0), m_vtx(), m_mom() {}
 
 
    TpcMCParticle(int PDG, float Mass, float Energy, const float* vtx, const float* mom):
      SimHitBase(), m_PDG(PDG), m_Mass(Mass), m_Energy(Energy)
    {
      std::copy(vtx, vtx + 3, m_vtx);
      std::copy(mom, mom + 3, m_mom);
    }
 
    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_PDG;
    float m_Mass;
    float m_Energy;
    float m_vtx[3];
    float m_mom[3];
 
    ClassDef(TpcMCParticle, 1)
  };
 
} // end namespace Belle2
 
#endif