TpcMCParticle.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 TPCMCPARTICLE_H
#define TPCMCPARTICLE_H
 
#include <simulation/dataobjects/SimHitBase.h>
 
// ROOT
#include <Math/Vector3D.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; }
    ROOT::Math::XYZVector getProductionVertex()  const { return ROOT::Math::XYZVector(m_vtx[0], m_vtx[1], m_vtx[2]); }
    ROOT::Math::XYZVector getMomentum() const { return ROOT::Math::XYZVector(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