GEMHitAssociator.h

Go to the documentation of this file.

#ifndef MCTruth_GEMHitAssociator_h
#define MCTruth_GEMHitAssociator_h
 
#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/MuonDetId/interface/GEMDetId.h"
#include "DataFormats/TrackingRecHit/interface/TrackingRecHit.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "SimDataFormats/EncodedEventId/interface/EncodedEventId.h"
#include "SimDataFormats/GEMDigiSimLink/interface/GEMDigiSimLink.h"
#include "SimDataFormats/TrackingHit/interface/PSimHit.h"
#include "SimDataFormats/TrackingHit/interface/PSimHitContainer.h"
 
#include <map>
#include <set>
#include <string>
#include <vector>
 
#include "DataFormats/Common/interface/DetSetVector.h"
#include "DataFormats/DetId/interface/DetId.h"
#include "DataFormats/GEMDigi/interface/GEMDigiCollection.h"
#include "DataFormats/GEMRecHit/interface/GEMRecHitCollection.h"
#include "DataFormats/GEMRecHit/interface/GEMSegmentCollection.h"
#include "DataFormats/MuonDetId/interface/MuonSubdetId.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "Geometry/GEMGeometry/interface/GEMGeometry.h"
#include "Geometry/Records/interface/MuonGeometryRecord.h"
#include "SimDataFormats/CrossingFrame/interface/CrossingFrame.h"
#include "SimDataFormats/CrossingFrame/interface/MixCollection.h"
 
class GEMHitAssociator {
public:
  typedef edm::DetSetVector<GEMDigiSimLink> DigiSimLinks;
  typedef edm::DetSet<GEMDigiSimLink> LayerLinks;
  typedef std::pair<uint32_t, EncodedEventId> SimHitIdpr;
 
  // Constructor with configurable parameters
  GEMHitAssociator(const edm::ParameterSet &, edm::ConsumesCollector &&ic);
  GEMHitAssociator(const edm::Event &e, const edm::EventSetup &eventSetup, const edm::ParameterSet &conf);
 
  void initEvent(const edm::Event &, const edm::EventSetup &);
 
  // Destructor
  ~GEMHitAssociator() {}
 
  std::vector<SimHitIdpr> associateRecHit(const GEMRecHit *gemrechit) const;
 
private:
  const DigiSimLinks *theDigiSimLinks;
  edm::InputTag GEMdigisimlinkTag;
 
  bool crossingframe;
  bool useGEMs_;
  edm::InputTag GEMsimhitsTag;
  edm::InputTag GEMsimhitsXFTag;
 
  edm::EDGetTokenT<CrossingFrame<PSimHit>> GEMsimhitsXFToken_;
  edm::EDGetTokenT<edm::PSimHitContainer> GEMsimhitsToken_;
  edm::EDGetTokenT<edm::DetSetVector<GEMDigiSimLink>> GEMdigisimlinkToken_;
 
  std::map<unsigned int, edm::PSimHitContainer> _SimHitMap;
};
 
#endif