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