DTHitAssociator.h

Go to the documentation of this file.

#ifndef MCTruth_DTHitAssociator_h
#define MCTruth_DTHitAssociator_h
 
#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/DTDigi/interface/DTDigiCollection.h"
#include "DataFormats/DTRecHit/interface/DTRecHit1D.h"
#include "DataFormats/DTRecHit/interface/DTRecHitCollection.h"
#include "DataFormats/MuonDetId/interface/DTWireId.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 "Geometry/DTGeometry/interface/DTGeometry.h"
#include "SimDataFormats/DigiSimLinks/interface/DTDigiSimLinkCollection.h"
#include "SimDataFormats/EncodedEventId/interface/EncodedEventId.h"
#include "SimDataFormats/TrackingHit/interface/PSimHitContainer.h"
 
#include <map>
#include <vector>
 
class DTHitAssociator {
public:
  typedef std::pair<uint32_t, EncodedEventId> SimHitIdpr;
  typedef std::pair<PSimHit, bool> PSimHit_withFlag;
  typedef std::map<DTWireId, std::vector<PSimHit_withFlag>> SimHitMap;
  typedef std::map<DTWireId, std::vector<DTRecHit1DPair>> RecHitMap;
  typedef std::map<DTWireId, std::vector<DTDigi>> DigiMap;
  typedef std::map<DTWireId, std::vector<DTDigiSimLink>> LinksMap;
 
  DTHitAssociator(const edm::Event &, const edm::EventSetup &, const edm::ParameterSet &, bool printRtS);
  DTHitAssociator(const edm::ParameterSet &, edm::ConsumesCollector &&iC);
 
  void initEvent(const edm::Event &, const edm::EventSetup &);
 
  virtual ~DTHitAssociator() {}
 
  std::vector<SimHitIdpr> associateHitId(const TrackingRecHit &hit) const;
  std::vector<SimHitIdpr> associateDTHitId(const DTRecHit1D *dtrechit) const;
 
  std::vector<PSimHit> associateHit(const TrackingRecHit &hit) const;
 
  SimHitMap mapOfSimHit;
  RecHitMap mapOfRecHit;
  DigiMap mapOfDigi;
  LinksMap mapOfLinks;
 
private:
  edm::InputTag DTsimhitsTag;
  edm::InputTag DTsimhitsXFTag;
  edm::InputTag DTdigiTag;
  edm::InputTag DTdigisimlinkTag;
  edm::InputTag DTrechitTag;
 
  bool dumpDT;
  bool crossingframe;
  bool links_exist;
  bool associatorByWire;
 
  bool SimHitOK(const edm::ESHandle<DTGeometry> &, const PSimHit &);
  bool printRtS;
};
 
#endif