UserAction.h

Go to the documentation of this file.

 
#ifndef SIMCORE_USERACTION_H
#define SIMCORE_USERACTION_H
 
/*~~~~~~~~~~~~~~~~*/
/*   C++ StdLib   */
/*~~~~~~~~~~~~~~~~*/
#include <map>
#include <string>
#include <vector>
 
/*~~~~~~~~~~~~*/
/*   Geant4   */
/*~~~~~~~~~~~~*/
#include "G4EventManager.hh"
#include "G4UserStackingAction.hh"
 
/*~~~~~~~~~~~~~~~*/
/*   Framework   */
/*~~~~~~~~~~~~~~~*/
#include "Framework/Configure/Parameters.h"
#include "Framework/Factory.h"
#include "Framework/Logger.h"
#include "SimCore/Event/SimParticle.h"
#include "SimCore/G4User/UserEventInformation.h"
 
// Forward Declarations
class G4Event;
class G4Run;
class G4Step;
class G4Track;
 
namespace simcore {
 
enum TYPE { RUN = 1, EVENT, TRACKING, STEPPING, STACKING, NONE };
 
class UserAction {
 public:
  UserAction(const std::string& name,
             framework::config::Parameters& parameters);
 
  DECLARE_FACTORY(UserAction, std::shared_ptr<UserAction>, const std::string&,
                  framework::config::Parameters&);
 
  virtual ~UserAction() = default;
 
  virtual void beginOfEventAction(const G4Event*) {};
 
  virtual void endOfEventAction(const G4Event*) {};
 
  virtual void beginOfRunAction(const G4Run*) {};
 
  virtual void endOfRunAction(const G4Run*) {};
 
  virtual void preUserTrackingAction(const G4Track*) {};
 
  virtual void postUserTrackingAction(const G4Track*) {};
 
  virtual void stepping(const G4Step*) {};
 
  virtual G4ClassificationOfNewTrack classifyNewTrack(
      const G4Track*, const G4ClassificationOfNewTrack& cl) {
    return cl;
  };
 
  virtual void newStage() {};
 
  virtual void prepareNewEvent() {};
 
  virtual std::vector<TYPE> getTypes() = 0;
 
 protected:
  UserEventInformation* getEventInfo() const;
 
  const std::map<int, ldmx::SimParticle>& getCurrentParticleMap() const;
 
  void abortEvent(const std::string& reason) const;
 
 protected:
  std::string name_{""};
 
  framework::config::Parameters parameters_;
 
  mutable ::framework::logging::logger the_log_;
 
};  // UserAction
 
}  // namespace simcore
 
#define DECLARE_ACTION(CLASS) FACTORY_REGISTRATION(simcore::UserAction, CLASS)
 
#endif  // SIMCORE_USERACTION_H