TrackerHitProducer.h

Go to the documentation of this file.

#ifndef TrackerHitProducer_h
#define TrackerHitProducer_h

// framework & common header files
#include "FWCore/Framework/interface/one/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::one::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