UserInfo.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 USERINFO_H_
#define USERINFO_H_
 
#include <framework/core/FrameworkExceptions.h>
#include <mdst/dataobjects/MCParticleGraph.h>
#include <simulation/dataobjects/MCParticleTrajectory.h>
 
#include "G4VUserTrackInformation.hh"
#include "G4VUserPrimaryParticleInformation.hh"
 
 
namespace Belle2 {
 
  namespace Simulation {
 
    //Define exceptions
    BELLE2_DEFINE_EXCEPTION(CouldNotFindUserInfo, "No User Information was found !");

    template<class Info, class Payload> class UserInfo: public Info {
 
    public:

      explicit UserInfo(Payload data) : Info(), m_data(data), m_status(0), m_fraction(1.0), m_trajectory(nullptr) {}

      Payload getData() { return m_data; }

      int getStatus() {return m_status;}

      double getFraction() {return m_fraction;}

      MCParticleTrajectory* getTrajectory() { return m_trajectory; }

      void setStatus(int status) { m_status = status;}

      void setFraction(double fraction) { m_fraction = fraction;}

      void setTrajectory(MCParticleTrajectory* trajectory) { m_trajectory = trajectory; }

      virtual void Print() const {;}

      template<class Carrier> static Payload getInfo(Carrier& obj)
      {
        UserInfo<Info, Payload>* userinfo = dynamic_cast<UserInfo<Info, Payload>*>(obj.GetUserInformation());
        if (!userinfo) throw(CouldNotFindUserInfo());
        return userinfo->getData();
      }
 
    protected:
 
      Payload m_data; 
 
      int m_status;      
      double m_fraction; 
      MCParticleTrajectory* m_trajectory; 
 
    };
 
    typedef UserInfo< G4VUserTrackInformation, MCParticleGraph::GraphParticle& > TrackInfo;
    typedef UserInfo< G4VUserPrimaryParticleInformation, MCParticleGraph::GraphParticle& > ParticleInfo;
 
  } // end namespace Simulation
} // end namespace Belle2
 
#endif /* USERINFO_H_ */