TrackingAction.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010-2011  Belle II Collaboration                         *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Andreas Moll, Marko Staric                               *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#ifndef TRACKINGACTION_H_
#define TRACKINGACTION_H_
 
#include <mdst/dataobjects/MCParticleGraph.h>
#include <simulation/dataobjects/MCParticleTrajectory.h>
#include <framework/datastore/StoreArray.h>
#include <framework/datastore/RelationArray.h>
 
#include <G4UserTrackingAction.hh>
 
namespace Belle2 {
  namespace Simulation {
 
    class TrackingAction : public G4UserTrackingAction {
 
    public:
 
      explicit TrackingAction(MCParticleGraph& mcParticleGraph);
 
      virtual ~TrackingAction();
 
      void PreUserTrackingAction(const G4Track* track);
 
      void PostUserTrackingAction(const G4Track* track);
 
      void setIgnoreOpticalPhotons(bool ignore = true) {m_ignoreOpticalPhotons = ignore;}
 
      void setIgnoreSecondaries(bool ignore = true) {m_ignoreSecondaries = ignore;}
 
      void setSecondariesEnergyCut(double cut_MeV) {m_secondariesEnergyCut = cut_MeV;}
 
      void setIgnoreBremsstrahlungPhotons(bool ignore = true) {m_ignoreBremsstrahlungPhotons = ignore;}
 
      void setBremsstrahlungPhotonsEnergyCut(double cut_MeV) {m_bremsstrahlungPhotonsEnergyCut = cut_MeV;}
 
      void setIgnorePairConversions(bool ignore = true) {m_ignorePairConversions = ignore;}
 
      void setPairConversionsEnergyCut(double cut_MeV) {m_pairConversionsEnergyCut = cut_MeV;}
 
      void setStoreTrajectories(int store, double distanceTolerance);
 
 
    protected:
 
      MCParticleGraph& m_mcParticleGraph;
 
      bool m_ignoreOpticalPhotons;              
      bool m_ignoreSecondaries;                 
      double m_secondariesEnergyCut;            
      bool m_ignoreBremsstrahlungPhotons;       
      double m_bremsstrahlungPhotonsEnergyCut;  
      bool m_ignorePairConversions;             
      double m_pairConversionsEnergyCut;        
      int m_storeTrajectories;    
      double m_distanceTolerance;  
      StoreArray<MCParticleTrajectory> m_storeMCTrajectories;
      RelationArray m_relMCTrajectories;
    };
 
  }  // end namespace Simulation
} // end namespace Belle2
 
#endif /* TRACKINGACTION_H_ */