MCParticleInfo.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 <TVector3.h>
#include <mdst/dataobjects/MCParticle.h>
 
namespace Belle2 {
  class MCParticleInfo {
 
  public :
 
    MCParticleInfo(const MCParticle& the_mcParticle, const TVector3& the_magField);
 
    double getPx() { return m_mcParticle.getMomentum().Px(); };
    double getPy() { return m_mcParticle.getMomentum().Py(); };
    double getPz() { return m_mcParticle.getMomentum().Pz(); };
    double getPt() { return m_mcParticle.getMomentum().Pt(); };
    double getP() { return m_mcParticle.getMomentum().Mag(); };
    double getEnergy() { return m_mcParticle.getEnergy(); };
 
    double getX() { return m_mcParticle.getVertex().X(); };
    double getY() { return m_mcParticle.getVertex().Y(); };
    double getZ() { return m_mcParticle.getVertex().Z(); };
 
    double getPtheta() { return m_mcParticle.getMomentum().Theta(); };
    double getPphi() { return m_mcParticle.getMomentum().Phi(); };
 
    double getCharge() { return m_mcParticle.getCharge(); };
 
    double getD0();
    double getZ0();
    double getPhi();
    double getOmega();
    double getCotTheta();
    double getLambda();
 
    double getChi();
 
  private:
 
    const MCParticle& m_mcParticle;
    double m_myBz;
    double m_charge;
  };
}