Geant4TrackingAction.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_GEANT4TRACKINGACTION_H
#define DD4HEP_DDG4_GEANT4TRACKINGACTION_H

// Framework include files
#include "DDG4/Geant4Action.h"
#include "G4VUserTrackInformation.hh"

class G4TrackingManager;
class G4Track;

namespace DD4hep {

  namespace Simulation {

    // Forward declarations
    class Geant4TrackInformation;
    class Geant4TrackingAction;
    class Geant4SharedTrackingAction;
    class Geant4TrackingActionSequence;


    class Geant4TrackingAction: public Geant4Action {
    public:
      typedef Geant4SharedTrackingAction shared_type;
    public:
      Geant4TrackingAction(const Geant4TrackingAction& copy) = delete;
      Geant4TrackingAction(Geant4Context* context, const std::string& name = "");
      virtual ~Geant4TrackingAction();
      G4TrackingManager* trackMgr() const {
        return m_context ? m_context->trackMgr() : 0;
      }
      void mark(const G4Track* track) const;
      virtual void begin(const G4Track* track);
      virtual void end(const G4Track* track);
    };


    class Geant4SharedTrackingAction : public Geant4TrackingAction {
    protected:
      Geant4TrackingAction* m_action;
    public:
      Geant4SharedTrackingAction(const Geant4TrackingAction& copy) = delete;
      Geant4SharedTrackingAction(Geant4Context* context, const std::string& nam);
      virtual ~Geant4SharedTrackingAction();
      virtual void configureFiber(Geant4Context* thread_context);
      virtual void use(Geant4TrackingAction* action);
      virtual void begin(const G4Track* track);
      virtual void end(const G4Track* track);
    };


    class Geant4TrackingActionSequence: public Geant4Action {
    protected:
      CallbackSequence             m_front;
      CallbackSequence             m_begin;
      CallbackSequence             m_end;
      CallbackSequence             m_final;
      Actors<Geant4TrackingAction> m_actors;
    public:
      Geant4TrackingActionSequence(const Geant4TrackingAction& copy) = delete;
      Geant4TrackingActionSequence(Geant4Context* context, const std::string& name);
      virtual ~Geant4TrackingActionSequence();
      virtual void updateContext(Geant4Context* ctxt);
      virtual void configureFiber(Geant4Context* thread_context);
      Geant4TrackingAction* get(const std::string& name) const;
      template <typename Q, typename T>
      void callUpFront(Q* p, void (T::*f)(const G4Track*),
                       CallbackSequence::Location where=CallbackSequence::END) {
        m_front.add(p, f, where);
      }
      template <typename Q, typename T>
      void callAtBegin(Q* p, void (T::*f)(const G4Track*),
                       CallbackSequence::Location where=CallbackSequence::END) {
        m_begin.add(p, f, where);
      }
      template <typename Q, typename T>
      void callAtEnd(Q* p, void (T::*f)(const G4Track*),
                     CallbackSequence::Location where=CallbackSequence::END) {
        m_end.add(p, f, where);
      }
      template <typename Q, typename T>
      void callAtFinal(Q* p, void (T::*f)(const G4Track*),
                       CallbackSequence::Location where=CallbackSequence::END) {
        m_final.add(p, f, where);
      }
      void adopt(Geant4TrackingAction* action);
      virtual void begin(const G4Track* track);
      virtual void end(const G4Track* track);
    };

  }    // End namespace Simulation
}      // End namespace DD4hep
#endif // DD4HEP_DDG4_GEANT4TRACKINGACTION_H