GGSMCTruthAction.h

Go to the documentation of this file.

/*
 * GGSMCTruthAction.h
 *
 *  Created on: 16 Aug 2011
 *      Author: Nicola Mori
 */

#ifndef GGSMCTRUTHACTION_H_
#define GGSMCTRUTHACTION_H_

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

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

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

class GGSMCTruthAction : public GGSUserAction {

public:
  GGSMCTruthAction();

  ~GGSMCTruthAction();

  void EndOfEventAction(const G4Event *event);

  void BeginOfRunAction(const G4Run *run);

  void EndOfRunAction(const G4Run *run);

  void SetOutputFileBase(const std::string &outFileBase) { _outBase = outFileBase; }
  void SetOutputTreeName(const std::string &outTreeName) { _outTreeName = outTreeName; }

private:
  void _Convert(const G4Event *event, GGSTMCTruthInfo *&mcTruth);

  GGSTMCTruthInfo *_mcTruthInfo;
  GGSTParticle *_primaryParticle;

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

  GGSMCTruthMessenger *_messenger;
};

#endif /* GGSMCTRUTHACTION_H_ */