TestActionTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef G4USERACTIONS_TESTACTIONTOOL_H
#define G4USERACTIONS_TESTACTIONTOOL_H
 
// Framework includes
#include "AthenaBaseComps/AthAlgTool.h"
 
// Geant4 includes
#include "G4UserRunAction.hh"
#include "G4UserEventAction.hh"
#include "G4UserSteppingAction.hh"
 
// G4Atlas includes
#include "G4AtlasInterfaces/IUserActionTool.h"
#include "G4AtlasTools/UserActionToolBase.h"
 
 
namespace G4UA
{

  class TestAction : public G4UserRunAction,
                     public G4UserEventAction,
                     public G4UserSteppingAction
  {
 
    public:

      struct Report
      {
        int numEvent = 0;
        int numStep = 0;
        void merge(const Report& other) {
          numEvent += other.numEvent;
          numStep += other.numStep;
        }
      };

      TestAction();

      virtual void BeginOfRunAction(const G4Run*) override final;

      virtual void EndOfRunAction(const G4Run*) override final;

      virtual void BeginOfEventAction(const G4Event*) override final;

      virtual void EndOfEventAction(const G4Event*) override final;

      virtual void UserSteppingAction(const G4Step* step) override final;

      inline const Report& getReport() const
      { return m_report; }
 
    private:

      Report m_report;
 
  }; // class TestAction
 

  class TestActionTool : public UserActionToolBase<TestAction>
  {
 
    public:

      TestActionTool(const std::string& type, const std::string& name,
                     const IInterface* parent);

      virtual StatusCode initialize() override final;

      virtual StatusCode finalize() override final;
 
    protected:

      virtual std::unique_ptr<TestAction>
      makeAndFillAction(G4AtlasUserActions& actions) override final;
 
  }; // class TestActionTool
 
} // namespace G4UA
 
#endif // G4USERACTIONS_TESTACTIONTOOL_H