TrdTravMatAction.h

Go to the documentation of this file.

/*
 * TrdTravMatAction.h
 *
 *  Created on: 13 Jul 2020
 *      Author: Nicola Mori
 */

#ifndef TRDTRAVMATACTION_H_
#define TRDTRAVMATACTION_H_

// HerdSoftware headers
#include "simulation/dataobjs/TrdTravMat.h"

// GGS headers
#include "montecarlo/services/GGSRootFileService.h"
#include "montecarlo/useractions/GGSUserAction.h"

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

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

class TrdTravMatAction : public GGSUserAction {

public:
  TrdTravMatAction();

  void PreUserTrackingAction(const G4Track *track) override;

  void UserSteppingAction(const G4Step *step) override;

  void BeginOfEventAction(const G4Event *event) override;

  void EndOfEventAction(const G4Event *event) override;

  void BeginOfRunAction(const G4Run *run) override;

  void EndOfRunAction(const G4Run *run) override;

private:
  std::vector<TrdTravMat> _travMat;

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

  G4GenericMessenger _messenger;
};

#endif /* TRDTRAVMATACTION_H_ */