GGSPrimaryDisAction.h

Go to the documentation of this file.

/*
 * GGSPrimaryDisAction.h
 *
 *  Created on: 1 Jun 2020
 *      Author: Nicola Mori
 */

#ifndef GGSPRIMARYDISACTION_H_
#define GGSPRIMARYDISACTION_H_

// Standard headers
#include <map>
#include <vector>

// GGS headers
#include "montecarlo/services/GGSRootFileService.h"
#include "montecarlo/useractions/GGSUserAction.h"
// messenger
class GGSHadrIntMessenger;

// GEANT4 headers
#include "G4GenericMessenger.hh"
#include "G4PrimaryParticle.hh"
#include "G4Track.hh"

// ROOT headers
#include "TClonesArray.h"
#include "TFile.h"
#include "TTree.h"

class GGSPrimaryDisAction : public GGSUserAction {

public:
  GGSPrimaryDisAction();

  void PreUserTrackingAction(const G4Track *track);

  void PostUserTrackingAction(const G4Track *track);

  void BeginOfEventAction(const G4Event *event);

  void EndOfEventAction(const G4Event *event);

  void BeginOfRunAction(const G4Run *run);

  void EndOfRunAction(const G4Run *run);

private:
  TClonesArray _disInfo;
  std::map<const G4Track *, int> _tracks; // first: followed track, second: originalTrackID
  bool _outputProducts;

  std::string _outBase;
  std::string _outTreeName;
  TFile *_outRootFile;
  TTree *_outTree;

  G4GenericMessenger _messenger;
};

#endif /* GGSPRIMARYDISACTION_H_ */