Geant4StackingAction.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_GEANT4STACKINGACTION_H
#define DDG4_GEANT4STACKINGACTION_H
 
#include <DDG4/Geant4Action.h>
 
#include <G4ClassificationOfNewTrack.hh>
 
class G4StackManager;
 
namespace dd4hep {
 
  namespace sim {
 
    // Forward declarations
    class Geant4StackingAction;
    class Geant4SharedStackingAction;
    class Geant4StackingActionSequence;
 
    enum Geant4StackingActionTrackClassification {
      NoTrackClassification = 0xFEED
    };
    union TrackClassification {
      G4ClassificationOfNewTrack value;
      int                        type;
      TrackClassification() { type = NoTrackClassification; }
      TrackClassification(G4ClassificationOfNewTrack val) { value = val; }
    };
 
 
    class Geant4StackingAction: public Geant4Action {
    public:
      friend class Geant4StackingActionSequence;
      typedef Geant4SharedStackingAction shared_type;
    public:
      DDG4_DEFINE_ACTION_CONSTRUCTORS(Geant4StackingAction);
 
    public:
      Geant4StackingAction(Geant4Context* ctxt, const std::string& name);
      virtual ~Geant4StackingAction();
      virtual void newStage(G4StackManager* /* stackManager */)   {
      }
      virtual void prepare(G4StackManager* /* stackManager */)   {
      }
      virtual TrackClassification 
    classifyNewTrack(G4StackManager* /* stackManager */, const G4Track* track);
    };
 
 
    class Geant4SharedStackingAction : public Geant4StackingAction {
    protected:
      Geant4StackingAction* m_action;
      DDG4_DEFINE_ACTION_CONSTRUCTORS(Geant4SharedStackingAction);
    public:
      Geant4SharedStackingAction(Geant4Context* context, const std::string& nam);
      virtual ~Geant4SharedStackingAction();
      virtual void configureFiber(Geant4Context* thread_context)  override;
      virtual void use(Geant4StackingAction* action);
      virtual void newStage(G4StackManager* stackManager)  override;
      virtual void prepare(G4StackManager* stackManager)  override;
      virtual TrackClassification 
    classifyNewTrack(G4StackManager* stackManager, const G4Track* track)  override;
    };
 
 
    class Geant4StackingActionSequence: public Geant4Action {
    protected:
      CallbackSequence m_newStage;
      // Callback sequence for the prepare call
      CallbackSequence m_prepare;
      Actors<Geant4StackingAction> m_actors;
 
      DDG4_DEFINE_ACTION_CONSTRUCTORS(Geant4StackingActionSequence);
    public:
      Geant4StackingActionSequence(Geant4Context* ctxt, const std::string& name);
      virtual ~Geant4StackingActionSequence();
      virtual void updateContext(Geant4Context* ctxt)  override;
      virtual void configureFiber(Geant4Context* thread_context)  override;
      Geant4StackingAction* get(const std::string& name) const;
      template <typename T> void callAtNewStage(T* p, void (T::*f)(G4StackManager*)) {
        m_newStage.add(p, f);
      }
      template <typename T> void callAtPrepare(T* p, void (T::*f)(G4StackManager*)) {
        m_prepare.add(p, f);
      }
      void adopt(Geant4StackingAction* action);
      virtual void newStage(G4StackManager* stackManager);
      virtual void prepare(G4StackManager* stackManager);
      virtual TrackClassification 
    classifyNewTrack(G4StackManager* stackManager, const G4Track* track);
    };
 
  }    // End namespace sim
}      // End namespace dd4hep
#endif // DDG4_GEANT4STACKINGACTION_H