GEMHitAssociator.cc

Go to the documentation of this file.

#include "SimMuon/MCTruth/interface/GEMHitAssociator.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<StripDigiSimLink> >(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 TrackingRecHit & hit) const {
  
  std::vector<SimHitIdpr> matched;

  if(useGEMs_){
    
      //std::cout<<"gemboo "<<useGEMs_<<std::endl;

      const TrackingRecHit * hitp = &hit;
      const GEMRecHit * gemrechit = dynamic_cast<const GEMRecHit *>(hitp);

      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->channel());
                if(ch != i) continue;

                SimHitIdpr currentId(itlink->SimTrackId(), itlink->eventId());
                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;

}