TrackingAction.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 <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 setRegionZBackward(double z) { m_regionZBackward = z; }
      void setRegionZForward(double z) { m_regionZForward = z; }
      void setRegionRho(double rho) { m_regionRho = rho; }
      void setKineticEnergyThreshold(double threshold) { m_kineticEnergyThreshold = threshold; }
      void setDistanceThreshold(double threshold) { m_distanceThreshold = threshold; }
      void setDoNotStoreEMParticles(bool value) { m_doNotStoreEMParticles = value; }
      void setDoNotStoreNuclei(bool value) { m_doNotStoreNuclei = value; }
      void setUseSeenInECL(bool use) { m_useSeenInECL = use; }
      void setUseDetailedParticleMatching(bool use) { m_useDetailedParticleMatching = use; }

      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;        
 
      double m_regionZBackward; 
      double m_regionZForward; 
      double m_regionRho; 
      double m_kineticEnergyThreshold; 
      double m_distanceThreshold; 
      bool m_doNotStoreEMParticles; 
      bool m_doNotStoreNuclei; 
      bool m_useSeenInECL; 
      bool m_useDetailedParticleMatching; 
 
 
      int m_storeTrajectories;    
      double m_distanceTolerance;  

      StoreArray<MCParticleTrajectory> m_storeMCTrajectories;
      RelationArray m_relMCTrajectories;
    };
 
  }  // end namespace Simulation
} // end namespace Belle2