UserAction.h

 
#ifndef SIMCORE_USERACTION_H
#define SIMCORE_USERACTION_H
 
/*~~~~~~~~~~~~~~~~*/
/*   C++ StdLib   */
/*~~~~~~~~~~~~~~~~*/
#include <iostream>
#include <map>
#include <string>
#include <vector>
 
/*~~~~~~~~~~~~*/
/*   Geant4   */
/*~~~~~~~~~~~~*/
#include "G4EventManager.hh"
#include "G4UserStackingAction.hh"
 
/*~~~~~~~~~~~~~~~*/
/*   Framework   */
/*~~~~~~~~~~~~~~~*/
#include "Framework/Configure/Parameters.h"
#include "SimCore/Event/SimParticle.h"
#include "SimCore/Factory.h"
#include "SimCore/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);
 
  using Factory =
      ::simcore::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;
 
 protected:
  std::string name_{""};
 
  framework::config::Parameters parameters_;
 
};  // UserAction
 
}  // namespace simcore
 
#define DECLARE_ACTION(NS, CLASS)                                             \
  namespace {                                                                 \
  auto v##CLASS = ::simcore::UserAction::Factory::get().declare<NS::CLASS>(); \
  }
 
#endif  // SIMCORE_USERACTION_H