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