GGSLostEnergyAction.h

Go to the documentation of this file.

/*
 * GGSLostEnergyAction.h
 *
 *  Created on: 16 Jun 2011
 *      Author: Nicola Mori
 */

#ifndef GGSLOSTENERGYACTION_H_
#define GGSLOSTENERGYACTION_H_

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


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

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

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

class GGSLostEnergyAction: public GGSUserAction {

public:

  GGSLostEnergyAction();

  virtual ~GGSLostEnergyAction();


  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);

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

  void SetKineticAsLostEnergy() {
    _kinetic = true;
  }

  void SetTotalAsLostEnergy() {
    _kinetic = false;
  }

private:

  GGSTLostEnergyInfo *_lostEnergyInfo;

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

  bool _kinetic;

  float _worldXMin, _worldXMax, _worldYMin, _worldYMax, _worldZMin, _worldZMax;
  float _tolerance;

  GGSLostEnergyMessenger *_messenger;

};

#endif /* GGSLOSTENERGYACTION_H_ */