GGSHitsAction.h

Go to the documentation of this file.

/*
 * GGSHitsAction.h
 *
 *  Created on: Oct 8, 2010
 *      Author: Elena Vanuccini and Nicola Mori
 *  Reworked on: 17 Aug 2011
 *      Author: Nicola mori
 *      Content: MC truth (GGSTSimEvent) moved to another action. Name changed.
 *               Only hits are handled by this action.
 */

#ifndef GGSHITSACTION_H_
#define GGSHITSACTION_H_

// GEANT4 headers
#include "G4Event.hh"
#include "G4GenericMessenger.hh"
#include "G4HCofThisEvent.hh"
#include "G4LogicalVolumeStore.hh"
#include "G4PhysicalVolumeStore.hh"
#include "G4Run.hh"
#include "G4SDManager.hh"
#include "G4VHitsCollection.hh"

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

// GGS headers
#include "montecarlo/dataobjs/GGSTHits.h"
#include "montecarlo/scoring/GGSIntHit.h"
#include "montecarlo/services/GGSTClonesArrayService.h"
#include "montecarlo/useractions/GGSUserAction.h"

// messenger
class GGSHitsMessenger;

#include <iostream>
#include <unordered_map>

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......

class GGSHitsAction : public GGSUserAction {

public:
  GGSHitsAction();

  ~GGSHitsAction();

  void BeginOfRunAction(const G4Run *run);

  void EndOfRunAction(const G4Run *run);

  void BeginOfEventAction(const G4Event *event);

  void EndOfEventAction(const G4Event *event);

  void SetOutputFileBase(const std::string &outFileBase) { _outFileBase = outFileBase; }

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

  void SetOutputIntHitClass(std::string detectorName, const std::string &className);

  void SetOutputPartHitClass(std::string detectorName, const std::string &className);

  void SetOutputPosHitClass(std::string detectorName, const std::string &className);

  void SetHitThreshold(const std::string &detectorName, const std::string &hitType, const std::string &valueStr,
                       const std::string &unit);

private:
  G4GenericMessenger _messenger;
  std::string _outFileBase; 

  TFile *_outFile;      
  TTree *_outTree;      
  TString _outTreeName; 

  // Struct which groups all the output tools for a given detector.
  struct OutputHandle {
    std::string intHitClassName, partHitClassName, posHitClassName;
    std::unique_ptr<GGSTClonesArrayService> partHitCAService, posHitCAService;
    std::unique_ptr<TClonesArray> intHitArray;
    std::unique_ptr<TObjArray> partHitArray, posHitArray;
    float intHitThreshold, partHitThreshold, posHitThreshold; // In internal G4 energy units
    OutputHandle()
        : intHitClassName("GGSTIntHit"), partHitClassName("GGSTPartHit"), posHitClassName("GGSTPosHit"),
          intHitThreshold(0.), partHitThreshold(0.), posHitThreshold(0.) {}
  };

  std::unordered_map<std::string, OutputHandle> _outputHandles;

  struct ThresholdStruct {
    std::string detectorName;
    int hitType;
    float value;
  };

  std::vector<ThresholdStruct> _thresholds;

  void _GiveBackAllArrays(); // Give back to GGSTClonesArrayServices all the TClonesArrays.

  void _Convert(const GGSPosHitsCollection &posHits, TClonesArray &tPosHits, OutputHandle &handle);

  void _Convert(const GGSPartHitsCollection &partHits, TClonesArray &tPartHits, OutputHandle &handle);

  void _Convert(const GGSIntHitsCollection &intHits, TClonesArray &tIntHits, OutputHandle &handle,
                const std::string &detectorName);

  class GGSHitsActionPrivateMessenger : public G4UImessenger {
  public:
    GGSHitsActionPrivateMessenger(GGSHitsAction *hitsAction);

    void SetNewValue(G4UIcommand *command, G4String newValue);

  private:
    std::unique_ptr<G4UIcommand> _setThreshCmd;
    GGSHitsAction *_hitsAction;
  };

  friend class GGSHitsActionPrivateMessenger;
  GGSHitsActionPrivateMessenger _privateMessenger;
};

#endif /* GGSHITSACTION_H_ */