GEMHitAssociator.cc

Go to the documentation of this file.

#include "SimMuon/MCTruth/interface/GEMHitAssociator.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
 
using namespace std;
 
// Constructor
GEMHitAssociator::GEMHitAssociator(const edm::ParameterSet &conf, edm::ConsumesCollector &&iC)
    : GEMdigisimlinkTag(conf.getParameter<edm::InputTag>("GEMdigisimlinkTag")),
      // CrossingFrame used or not ?
      crossingframe(conf.getParameter<bool>("crossingframe")),
      useGEMs_(conf.getParameter<bool>("useGEMs")),
      GEMsimhitsTag(conf.getParameter<edm::InputTag>("GEMsimhitsTag")),
      GEMsimhitsXFTag(conf.getParameter<edm::InputTag>("GEMsimhitsXFTag")) {
  if (crossingframe) {
    GEMsimhitsXFToken_ = iC.consumes<CrossingFrame<PSimHit>>(GEMsimhitsXFTag);
  } else if (!GEMsimhitsTag.label().empty()) {
    GEMsimhitsToken_ = iC.consumes<edm::PSimHitContainer>(GEMsimhitsTag);
  }
 
  GEMdigisimlinkToken_ = iC.consumes<edm::DetSetVector<GEMDigiSimLink>>(GEMdigisimlinkTag);
}
 
GEMHitAssociator::GEMHitAssociator(const edm::Event &e,
                                   const edm::EventSetup &eventSetup,
                                   const edm::ParameterSet &conf)
    : GEMdigisimlinkTag(conf.getParameter<edm::InputTag>("GEMdigisimlinkTag")),
      // CrossingFrame used or not ?
      crossingframe(conf.getParameter<bool>("crossingframe")),
      useGEMs_(conf.getParameter<bool>("useGEMs")),
      GEMsimhitsTag(conf.getParameter<edm::InputTag>("GEMsimhitsTag")),
      GEMsimhitsXFTag(conf.getParameter<edm::InputTag>("GEMsimhitsXFTag")) {
  initEvent(e, eventSetup);
}
 
void GEMHitAssociator::initEvent(const edm::Event &e, const edm::EventSetup &eventSetup) {
  if (useGEMs_) {
    if (crossingframe) {
      edm::Handle<CrossingFrame<PSimHit>> cf;
      LogTrace("GEMHitAssociator") << "getting CrossingFrame<PSimHit> collection - " << GEMsimhitsXFTag;
      e.getByLabel(GEMsimhitsXFTag, cf);
 
      std::unique_ptr<MixCollection<PSimHit>> GEMsimhits(new MixCollection<PSimHit>(cf.product()));
      LogTrace("GEMHitAssociator") << "... size = " << GEMsimhits->size();
 
      //   MixCollection<PSimHit> & simHits = *hits;
 
      for (MixCollection<PSimHit>::MixItr hitItr = GEMsimhits->begin(); hitItr != GEMsimhits->end(); ++hitItr) {
        _SimHitMap[hitItr->detUnitId()].push_back(*hitItr);
      }
 
    } else if (!GEMsimhitsTag.label().empty()) {
      edm::Handle<edm::PSimHitContainer> GEMsimhits;
      LogTrace("GEMHitAssociator") << "getting PSimHit collection - " << GEMsimhitsTag;
      e.getByLabel(GEMsimhitsTag, GEMsimhits);
      LogTrace("GEMHitAssociator") << "... size = " << GEMsimhits->size();
 
      // arrange the hits by detUnit
      for (edm::PSimHitContainer::const_iterator hitItr = GEMsimhits->begin(); hitItr != GEMsimhits->end(); ++hitItr) {
        _SimHitMap[hitItr->detUnitId()].push_back(*hitItr);
      }
    }
 
    edm::Handle<DigiSimLinks> digiSimLinks;
    LogTrace("GEMHitAssociator") << "getting GEM Strip DigiSimLink collection - " << GEMdigisimlinkTag;
    e.getByLabel(GEMdigisimlinkTag, digiSimLinks);
    theDigiSimLinks = digiSimLinks.product();
  }
}
// end of constructor
 
std::vector<GEMHitAssociator::SimHitIdpr> GEMHitAssociator::associateRecHit(const GEMRecHit *gemrechit) const {
  std::vector<SimHitIdpr> matched;
 
  if (useGEMs_) {
    if (gemrechit) {
      GEMDetId gemDetId = gemrechit->gemId();
      int fstrip = gemrechit->firstClusterStrip();
      int cls = gemrechit->clusterSize();
      // int bx = gemrechit->BunchX();
 
      DigiSimLinks::const_iterator layerLinks = theDigiSimLinks->find(gemDetId);
 
      if (layerLinks != theDigiSimLinks->end()) {
        for (int i = fstrip; i < (fstrip + cls); ++i) {
          for (LayerLinks::const_iterator itlink = layerLinks->begin(); itlink != layerLinks->end(); ++itlink) {
            int ch = static_cast<int>(itlink->getStrip());
            if (ch != i)
              continue;
 
            SimHitIdpr currentId(itlink->getTrackId(), itlink->getEventId());
            if (find(matched.begin(), matched.end(), currentId) == matched.end())
              matched.push_back(currentId);
          }
        }
 
      } else
        edm::LogWarning("GEMHitAssociator")
            << "*** WARNING in GEMHitAssociator: GEM layer " << gemDetId << " has no DigiSimLinks !" << std::endl;
 
    } else
      edm::LogWarning("GEMHitAssociator") << "*** WARNING in GEMHitAssociator::associateRecHit, null "
                                             "dynamic_cast !";
  }
 
  return matched;
}