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 "Geometry/HcalCommonData/interface/HcalDDDSimConstants.h"
#include "CondFormats/GeometryObjects/interface/HcalSimulationParameters.h"
#include "SimG4CMS/Calo/interface/HFFibre.h"
#include "SimDataFormats/CaloHit/interface/HFShowerPhoton.h"

#include "G4ThreeVector.hh"

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

#include <string>
#include <memory>

class G4Step;
class G4ParticleTable;

class HFShowerLibrary {
public:
  //Constructor and Destructor
  HFShowerLibrary(const std::string &name,
                  const HcalDDDSimConstants *hcons,
                  const HcalSimulationParameters *hps,
                  edm::ParameterSet const &p);
  ~HFShowerLibrary();

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

  std::vector<Hit> getHits(const G4Step *aStep, bool &ok, double weight, bool onlyLong = false);
  std::vector<Hit> fillHits(const G4ThreeVector &p,
                            const 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);

private:
  const HcalDDDSimConstants *hcalConstant_;
  std::unique_ptr<HFFibre> fibre_;
  TFile *hf;
  TBranch *emBranch, *hadBranch;

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

  int fileVersion_;
  bool equalizeTimeShift_;
  double probMax, backProb;
  double dphi, rMin, rMax;
  std::vector<double> gpar;

  int npe;
  HFShowerPhotonCollection pe;
  std::unique_ptr<HFShowerPhotonCollection> photo;
  HFShowerPhotonCollection photon;
};
#endif