FastHFShowerLibrary

Macros

Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68
#ifndef FastHFShowerLibrary_H
#define FastHFShowerLibrary_H
///////////////////////////////////////////////////////////////////////////////
// File: FastHFShowerLibrary.h
// Description: Gets information from a shower library
///////////////////////////////////////////////////////////////////////////////

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/FrameworkfwdMostUsed.h"
#include "FWCore/Utilities/interface/ESGetToken.h"

#include "FastSimulation/Utilities/interface/FamosDebug.h"

#include "DataFormats/Math/interface/Vector3D.h"

#include "SimG4CMS/Calo/interface/HFShowerLibrary.h"
#include "SimG4CMS/Calo/interface/CaloHitID.h"
#include "Geometry/HcalCommonData/interface/HcalNumberingFromDDD.h"
#include "Geometry/HcalCommonData/interface/HcalDDDSimConstants.h"
#include "Geometry/HcalCommonData/interface/HcalSimulationConstants.h"
#include "Geometry/Records/interface/HcalSimNumberingRecord.h"
#include "SimG4CMS/Calo/interface/HcalNumberingScheme.h"
#include "G4ThreeVector.hh"

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

#include <string>
#include <memory>
#include <map>

class FSimEvent;
class FSimTrack;
class HFShowerLibrary;
class RandomEngineAndDistribution;

class FastHFShowerLibrary {
public:
  // Constructor and Destructor
  FastHFShowerLibrary(edm::ParameterSet const&, edm::ConsumesCollector&&);
  ~FastHFShowerLibrary() {}

public:
  void const initHFShowerLibrary(const edm::EventSetup&);
  void recoHFShowerLibrary(const FSimTrack& myTrack);
  void modifyDepth(HcalNumberingFromDDD::HcalID& id);
  const std::map<CaloHitID, float>& getHitsMap() { return hitMap; };

  void SetRandom(const RandomEngineAndDistribution*);

private:
  const edm::ParameterSet fast;
  std::unique_ptr<HFShowerLibrary> hfshower;
  std::unique_ptr<HcalNumberingFromDDD> numberingFromDDD;
  const HcalDDDSimConstants* hcalConstants;
  HcalNumberingScheme numberingScheme;

  std::map<CaloHitID, float> hitMap;

  bool applyFidCut;
  std::string name;

  const edm::ESGetToken<HcalDDDSimConstants, HcalSimNumberingRecord> hcalDDDSimConstantsESToken_;
  const edm::ESGetToken<HcalSimulationConstants, HcalSimNumberingRecord> hcalSimulationConstantsESToken_;
};
#endif