HFShowerLibrary.h

Go to the documentation of this file.

#ifndef SimG4CMS_HFShowerLibrary_h
#define SimG4CMS_HFShowerLibrary_h 1
// File: HFShowerLibrary.h
// Description: Gets information from a shower library

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "SimG4CMS/Calo/interface/HFFibre.h"
#include "SimDataFormats/CaloHit/interface/HFShowerPhoton.h"
#include "DetectorDescription/Core/interface/DDsvalues.h"

#include "G4ParticleTable.hh"
#include "G4ThreeVector.hh"
 
//ROOT
#include "TROOT.h"
#include "TFile.h"
#include "TTree.h"

#include <string>
#include <memory>

class DDCompactView;    
class G4Step;

class HFShowerLibrary {
  
public:
  
  //Constructor and Destructor
  HFShowerLibrary(std::string & name, const DDCompactView & cpv,
          edm::ParameterSet const & p);
  ~HFShowerLibrary();

public:

  struct Hit {
    Hit() {}
    G4ThreeVector             position;
    int                       depth;
    double                    time;
  };

  void                initRun(G4ParticleTable * theParticleTable);
  std::vector<Hit>    getHits(G4Step * aStep, bool &ok, double weight, 
                  bool onlyLong=false);
  std::vector<Hit>    fillHits(G4ThreeVector & p, G4ThreeVector & v,
                               int parCode, double parEnergy, bool & ok,
                               double weight, double time, bool onlyLong=false);
protected:

  bool                rInside(double r);
  void                getRecord(int, int);
  void                loadEventInfo(TBranch *);
  void                interpolate(int, double);
  void                extrapolate(int, double);
  void                storePhoton(int j);
  std::vector<double> getDDDArray(const std::string&, const DDsvalues_type&,
                  int&);

private:

  HFFibre *           fibre;
  TFile *             hf;
  TBranch             *emBranch, *hadBranch;

  bool                verbose, applyFidCut, newForm;
  int                 nMomBin, totEvents, evtPerBin;
  float               libVers, listVersion; 
  std::vector<double> pmom;

  double              probMax, backProb;
  double              dphi, rMin, rMax;
  std::vector<double> gpar;

  int                 emPDG, epPDG, gammaPDG;
  int                 pi0PDG, etaPDG, nuePDG, numuPDG, nutauPDG;
  int                 anuePDG, anumuPDG, anutauPDG, geantinoPDG;

  int                 npe;
  HFShowerPhotonCollection pe;
  HFShowerPhotonCollection* photo;
  HFShowerPhotonCollection photon;

};
#endif