EnergyLossRecorder.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/

// EnergyLossRecorder.h, (c) ATLAS Detector software
 
#ifndef EnergyLossRecorder_H
#define EnergyLossRecorder_H
 
#include <string>
#include <vector>
 
#include "G4UserRunAction.hh"
#include "G4UserEventAction.hh"
#include "G4UserSteppingAction.hh"

 
namespace Trk
{
  class IPositionMomentumWriter;
}
 
namespace G4UA
{
 
  class EnergyLossRecorder: public G4UserRunAction, public G4UserEventAction, public G4UserSteppingAction
  {
 
  public:
 
    struct Config
    {
      Trk::IPositionMomentumWriter *pmWriter=nullptr;
    };
 
    EnergyLossRecorder(const Config& config);
    virtual void BeginOfRunAction(const G4Run*) override;
    virtual void EndOfRunAction(const G4Run*) override;
    virtual void BeginOfEventAction(const G4Event*) override;
    virtual void EndOfEventAction(const G4Event*) override;
    virtual void UserSteppingAction(const G4Step*) override;
  private:
    Config m_config;
    unsigned int                             m_entries;
  }; // class EnergyLossRecorder
 
} // namespace G4UA
 
#endif