Geant4TestActions.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_GEANT4TESTACTIONS_H
#define DDG4_GEANT4TESTACTIONS_H
 
// Framework include files
#include <DDG4/Geant4Handle.h>
#include <DDG4/Geant4Kernel.h>
#include <DDG4/Geant4GeneratorAction.h>
#include <DDG4/Geant4RunAction.h>
#include <DDG4/Geant4EventAction.h>
#include <DDG4/Geant4TrackingAction.h>
#include <DDG4/Geant4SteppingAction.h>
#include <DDG4/Geant4StackingAction.h>
#include <DDG4/Geant4ActionPhase.h>
#include <DDG4/Geant4SensDetAction.h>
 
namespace dd4hep {
 
  namespace sim {
 
    namespace Test {
 
 
      class Geant4TestBase {
      protected:
        int m_value1;
        double m_value2;
        std::string m_value3;
        std::string m_type;
      public:
        Geant4TestBase(Geant4Action* action, const std::string& typ);
        virtual ~Geant4TestBase();
      };
 
 
      class Geant4TestGeneratorAction: public Geant4GeneratorAction, public Geant4TestBase {
      public:
        Geant4TestGeneratorAction(Geant4Context* c, const std::string& n);
        virtual ~Geant4TestGeneratorAction();
        virtual void operator()(G4Event*)  override;
      };
 
 
      class Geant4TestRunAction: public Geant4RunAction, public Geant4TestBase {
      public:
        Geant4TestRunAction(Geant4Context* c, const std::string& n);
        virtual ~Geant4TestRunAction();
        void begin(const G4Run*)  override;
        void end(const G4Run*)  override;
        void beginEvent(const G4Event*);
        void endEvent(const G4Event*);
      };
 
 
      class Geant4TestEventAction: public Geant4EventAction, public Geant4TestBase {
      public:
        Geant4TestEventAction(Geant4Context* c, const std::string& n);
        virtual ~Geant4TestEventAction();
        virtual void begin(const G4Event*)  override;
        virtual void end(const G4Event*)  override;
        void beginRun(const G4Run*);
        void endRun(const G4Run*);
      };
 
 
      class Geant4TestTrackAction: public Geant4TrackingAction, public Geant4TestBase {
      public:
        Geant4TestTrackAction(Geant4Context* c, const std::string& n);
        virtual ~Geant4TestTrackAction();
        virtual void begin(const G4Track*)  override;
        virtual void end(const G4Track*)  override;
      };
 
 
      class Geant4TestStepAction: public Geant4SteppingAction, public Geant4TestBase {
      public:
        Geant4TestStepAction(Geant4Context* c, const std::string& n);
        virtual ~Geant4TestStepAction();
        void operator()(const G4Step*, G4SteppingManager*)  override;
      };
 
 
      class Geant4TestStackAction: public Geant4StackingAction, public Geant4TestBase {
      public:
        Geant4TestStackAction(Geant4Context* c, const std::string& n);
        virtual ~Geant4TestStackAction();
        virtual void newStage(G4StackManager*)  override;
        virtual void prepare(G4StackManager*)  override;
        virtual TrackClassification classifyNewTrack(G4StackManager*, const G4Track*)  override;
      };
 
 
      class Geant4TestSensitive: public Geant4Sensitive, public Geant4TestBase {
      protected:
        size_t m_collectionID;
      public:
        Geant4TestSensitive(Geant4Context* c, const std::string& n, DetElement det, Detector& description);
        virtual ~Geant4TestSensitive();
        virtual void begin(G4HCofThisEvent*)  override;
        virtual void end(G4HCofThisEvent*)  override;
        virtual bool process(const G4Step*, G4TouchableHistory*)  override;
      };
    }   // End namespace Test
  }    // End namespace sim
}      // End namespace dd4hep
 
#endif // DDG4_GEANT4TESTACTIONS_H