Geant4ParticleHandler.h

Go to the documentation of this file.

// $Id: $
//==========================================================================
//  AIDA Detector description implementation for LCD
//--------------------------------------------------------------------------
// 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 DD4HEP_DDG4_GEANT4PARTICLEHANDLER_H
#define DD4HEP_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 Simulation {

    // 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;
      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);
      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);
      virtual void mark(const G4Track* track, int reason);
      virtual void mark(const G4Step* step);
      virtual void mark(const G4Step* step, int reason);

      static bool defaultKeepParticle(Particle& particle);

    };
  }    // End namespace Simulation
}      // End namespace DD4hep

#endif // DD4HEP_DDG4_GEANT4PARTICLEHANDLER_H