TrackerHitProducer.h

Go to the documentation of this file.

#ifndef TrackerHitProducer_h
#define TrackerHitProducer_h
 
// framework & common header files
#include "FWCore/Framework/interface/EDProducer.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
 
#include "SimDataFormats/Track/interface/SimTrackContainer.h"
#include "SimDataFormats/TrackingHit/interface/PSimHitContainer.h"
#include "SimDataFormats/Vertex/interface/SimVertexContainer.h"
 
#include <string>
#include <vector>
 
namespace edm {
  class HepMCProduct;
}
class PTrackerSimHit;
 
class TrackerHitProducer : public edm::EDProducer {
public:
  typedef std::vector<float> FloatVector;
  typedef std::vector<int> IntegerVector;
 
  explicit TrackerHitProducer(const edm::ParameterSet &);
  ~TrackerHitProducer() override;
  void beginJob() override;
  void endJob() override;
  void produce(edm::Event &, const edm::EventSetup &) override;
 
private:
  // production related methods
  void fillG4MC(edm::Event &);
  void storeG4MC(PTrackerSimHit &);
  void fillTrk(edm::Event &, const edm::EventSetup &);
  void storeTrk(PTrackerSimHit &);
 
  void clear();
 
private:
  //  parameter information
  bool getAllProvenances;
  bool printProvenanceInfo;
  int verbosity;
 
  // private statistics information
  unsigned int count;
 
  int nRawGenPart;
 
  edm::ParameterSet config_;
 
  edm::EDGetTokenT<edm::HepMCProduct> edmHepMCProductToken_;
  edm::EDGetTokenT<edm::SimVertexContainer> edmSimVertexContainerToken_;
  edm::EDGetTokenT<edm::SimTrackContainer> edmSimTrackContainerToken_;
  edm::EDGetTokenT<edm::PSimHitContainer> edmPSimHitContainer_pxlBrlLow_Token_, edmPSimHitContainer_pxlBrlHigh_Token_;
  edm::EDGetTokenT<edm::PSimHitContainer> edmPSimHitContainer_pxlFwdLow_Token_, edmPSimHitContainer_pxlFwdHigh_Token_;
  edm::EDGetTokenT<edm::PSimHitContainer> edmPSimHitContainer_siTIBLow_Token_, edmPSimHitContainer_siTIBHigh_Token_;
  edm::EDGetTokenT<edm::PSimHitContainer> edmPSimHitContainer_siTOBLow_Token_, edmPSimHitContainer_siTOBHigh_Token_;
  edm::EDGetTokenT<edm::PSimHitContainer> edmPSimHitContainer_siTIDLow_Token_, edmPSimHitContainer_siTIDHigh_Token_;
  edm::EDGetTokenT<edm::PSimHitContainer> edmPSimHitContainer_siTECLow_Token_, edmPSimHitContainer_siTECHigh_Token_;
 
  // G4MC info
  FloatVector G4VtxX;
  FloatVector G4VtxY;
  FloatVector G4VtxZ;
  FloatVector G4TrkPt;
  FloatVector G4TrkE;
  FloatVector G4TrkEta;
  FloatVector G4TrkPhi;
 
  // Tracker info
 
  // Hit info
  IntegerVector HitsSysID;
  FloatVector HitsDuID;
  FloatVector HitsTkID;
  FloatVector HitsProT;
  FloatVector HitsParT;
  FloatVector HitsP;
  FloatVector HitsLpX;
  FloatVector HitsLpY;
  FloatVector HitsLpZ;
  FloatVector HitsLdX;
  FloatVector HitsLdY;
  FloatVector HitsLdZ;
  FloatVector HitsLdTheta;
  FloatVector HitsLdPhi;
  FloatVector HitsExPx;
  FloatVector HitsExPy;
  FloatVector HitsExPz;
  FloatVector HitsEnPx;
  FloatVector HitsEnPy;
  FloatVector HitsEnPz;
  FloatVector HitsEloss;
  FloatVector HitsToF;
 
  std::string fName;
  std::string label;
 
};  // end class declaration
 
#endif