Geant4ParticleHandler.h

Go to the documentation of this file.

//==========================================================================
//  AIDA Detector description implementation 
//--------------------------------------------------------------------------
// Copyright (C) Organisation europeenne pour la Recherche nucleaire (CERN)
// All rights reserved.
//
// For the licensing terms see $DD4hepINSTALL/LICENSE.
// For the list of contributors see $DD4hepINSTALL/doc/CREDITS.
//
// Author     : M.Frank
//
//==========================================================================
 
#ifndef DDG4_GEANT4PARTICLEHANDLER_H
#define DDG4_GEANT4PARTICLEHANDLER_H
 
// Framework include files
#include <DDG4/Geant4Primary.h>
#include <DDG4/Geant4GeneratorAction.h>
#include <DDG4/Geant4MonteCarloTruth.h>
 
// Forward declarations
class G4Step;
class G4Track;
class G4Event;
class G4SteppingManager;
 
namespace dd4hep {
 
  namespace sim {
 
    // Forward declarations
    class Geant4UserParticleHandler;
 
 
    class Geant4ParticleHandler : public Geant4GeneratorAction, public Geant4MonteCarloTruth  {
    public:
      typedef Geant4ParticleMap::Particle         Particle;
      typedef Geant4ParticleMap::ParticleMap      ParticleMap;
      typedef Geant4ParticleMap::TrackEquivalents TrackEquivalents;
#if defined(__CINT__) || defined(__MAKECINT__) || defined(G__DICTIONARY)
      // Need to force to public for the ROOT dictionary
    public:
#else
    private:
#endif
      Geant4ParticleHandler();
      Geant4ParticleHandler& operator=(const Geant4ParticleHandler& c);
 
    public:
      typedef std::vector<std::string> Processes;
 
    protected:
 
      bool m_printStartTracking;
      bool m_printEndTracking;
      bool m_keepAll;
      bool m_ownsParticles;
      double m_kinEnergyCut;
      double m_minDistToParentVertex;
      Processes                  m_processNames;
 
      Geant4UserParticleHandler* m_userHandler;
 
      int               m_globalParticleID;
      Geant4PrimaryMap* m_primaryMap;
      Particle          m_currTrack;
      ParticleMap       m_particleMap;
      ParticleMap       m_suspendedPM;
      bool              m_haveSuspended = false;
      TrackEquivalents  m_equivalentTracks;
 
      int recombineParents();
      void clear();
      void checkConsistency()  const;
 
      void rebaseSimulatedTracks(int base);
      void dumpMap(const char* tag) const;
 
      void setVertexEndpointBit();
 
    public:
      Geant4ParticleHandler(Geant4Context* context, const std::string& nam);
      virtual ~Geant4ParticleHandler();
      bool adopt(Geant4Action* action);
      virtual void operator()(G4Event* event)  override;
      virtual void step(const G4Step* step, G4SteppingManager* mgr);
      virtual void beginEvent(const G4Event* event);
      virtual void endEvent(const G4Event* event);
      virtual void begin(const G4Track* track);
      virtual void end(const G4Track* track);
 
      virtual void mark(const G4Track* track)  override;
      virtual void mark(const G4Track* track, int reason)  override;
      virtual void mark(const G4Step* step)  override;
      virtual void mark(const G4Step* step, int reason)  override;
 
      static bool defaultKeepParticle(Particle& particle);
 
    };
  }    // End namespace sim
}      // End namespace dd4hep
 
#endif // DDG4_GEANT4PARTICLEHANDLER_H