RPCHitAssociator.cc

Go to the documentation of this file.

#include "SimMuon/MCTruth/interface/RPCHitAssociator.h"

using namespace std;



// Constructor
RPCHitAssociator::RPCHitAssociator( const edm::ParameterSet& conf, 
                    edm::ConsumesCollector && iC):
  RPCdigisimlinkTag(conf.getParameter<edm::InputTag>("RPCdigisimlinkTag")),
  // CrossingFrame used or not ?
  crossingframe(conf.getParameter<bool>("crossingframe")),
  RPCsimhitsTag(conf.getParameter<edm::InputTag>("RPCsimhitsTag")),
  RPCsimhitsXFTag(conf.getParameter<edm::InputTag>("RPCsimhitsXFTag"))
{
  if (crossingframe){
    RPCsimhitsXFToken_=iC.consumes<CrossingFrame<PSimHit> >(RPCsimhitsXFTag);
  } else if (!RPCsimhitsTag.label().empty()) {
    RPCsimhitsToken_=iC.consumes<edm::PSimHitContainer>(RPCsimhitsTag);
  }

  RPCdigisimlinkToken_=iC.consumes< edm::DetSetVector<RPCDigiSimLink> >(RPCdigisimlinkTag); 
}

RPCHitAssociator::RPCHitAssociator(const edm::Event& e, const edm::EventSetup& eventSetup, const edm::ParameterSet& conf ):
  RPCdigisimlinkTag(conf.getParameter<edm::InputTag>("RPCdigisimlinkTag")),
  // CrossingFrame used or not ?
  crossingframe(conf.getParameter<bool>("crossingframe")),
  RPCsimhitsTag(conf.getParameter<edm::InputTag>("RPCsimhitsTag")),
  RPCsimhitsXFTag(conf.getParameter<edm::InputTag>("RPCsimhitsXFTag"))
{
  initEvent(e,eventSetup);
}


void RPCHitAssociator::initEvent(const edm::Event& e, const edm::EventSetup& eventSetup)


{
  if (crossingframe) {
    
    edm::Handle<CrossingFrame<PSimHit> > cf;
    LogTrace("RPCHitAssociator") <<"getting CrossingFrame<PSimHit> collection - "<<RPCsimhitsXFTag;
    e.getByLabel(RPCsimhitsXFTag, cf);
    
    std::unique_ptr<MixCollection<PSimHit> > 
      RPCsimhits( new MixCollection<PSimHit>(cf.product()) );
    LogTrace("RPCHitAssociator") <<"... size = "<<RPCsimhits->size();

    //   MixCollection<PSimHit> & simHits = *hits;
    
    for(MixCollection<PSimHit>::MixItr hitItr = RPCsimhits->begin();
    hitItr != RPCsimhits->end(); ++hitItr) 
      {
    _SimHitMap[hitItr->detUnitId()].push_back(*hitItr);
      }
    
  } else if (!RPCsimhitsTag.label().empty()) {
    edm::Handle<edm::PSimHitContainer> RPCsimhits;
    LogTrace("RPCHitAssociator") <<"getting PSimHit collection - "<<RPCsimhitsTag;
    e.getByLabel(RPCsimhitsTag, RPCsimhits);    
    LogTrace("RPCHitAssociator") <<"... size = "<<RPCsimhits->size();
    
    // arrange the hits by detUnit
    for(edm::PSimHitContainer::const_iterator hitItr = RPCsimhits->begin();
    hitItr != RPCsimhits->end(); ++hitItr)
      {
    _SimHitMap[hitItr->detUnitId()].push_back(*hitItr);
      }
  }

  edm::Handle< edm::DetSetVector<RPCDigiSimLink> > thelinkDigis;
  LogTrace("RPCHitAssociator") <<"getting RPCDigiSimLink collection - "<<RPCdigisimlinkTag;
  e.getByLabel(RPCdigisimlinkTag, thelinkDigis);
  _thelinkDigis = thelinkDigis;
}
// end of constructor

std::vector<RPCHitAssociator::SimHitIdpr> RPCHitAssociator::associateRecHit(const TrackingRecHit & hit) const {
  
  std::vector<SimHitIdpr> matched;

  const TrackingRecHit * hitp = &hit;
  const RPCRecHit * rpcrechit = dynamic_cast<const RPCRecHit *>(hitp);

  if (rpcrechit) {
    
    RPCDetId rpcDetId = rpcrechit->rpcId();
    int fstrip = rpcrechit->firstClusterStrip();
    int cls = rpcrechit->clusterSize();
    int bx = rpcrechit->BunchX();
    
    for(int i = fstrip; i < fstrip+cls; ++i) {
      std::set<RPCDigiSimLink> links = findRPCDigiSimLink(rpcDetId.rawId(),i,bx);
      
      if (links.empty()) LogTrace("RPCHitAssociator")
    <<"*** WARNING in RPCHitAssociator::associateRecHit, RPCRecHit "<<*rpcrechit<<", strip "<<i<<" has no associated RPCDigiSimLink !"<<endl;
      
      for(std::set<RPCDigiSimLink>::iterator itlink = links.begin(); itlink != links.end(); ++itlink) {
    SimHitIdpr currentId(itlink->getTrackId(),itlink->getEventId());
    if(find(matched.begin(),matched.end(),currentId ) == matched.end())
      matched.push_back(currentId);
      }
    }
    
  } else LogTrace("RPCHitAssociator")<<"*** WARNING in RPCHitAssociator::associateRecHit, null dynamic_cast !";
  
  return  matched;
}
  
std::set<RPCDigiSimLink>  RPCHitAssociator::findRPCDigiSimLink(uint32_t rpcDetId, int strip, int bx) const {

  std::set<RPCDigiSimLink> links;

  for (edm::DetSetVector<RPCDigiSimLink>::const_iterator itlink = _thelinkDigis->begin(); itlink != _thelinkDigis->end(); itlink++){
    for(edm::DetSet<RPCDigiSimLink>::const_iterator digi_iter=itlink->data.begin();digi_iter != itlink->data.end();++digi_iter){

      uint32_t detid = digi_iter->getDetUnitId();
      int str = digi_iter->getStrip();
      int bunchx = digi_iter->getBx();

      if(detid == rpcDetId && str == strip && bunchx == bx){
        links.insert(*digi_iter);
      }

    }
  }

  return links;
}