CMSSW_4_4_3_patch1/src/TrackingTools/RoadSearchHitAccess/src/DetHitAccess.cc

Go to the documentation of this file.

#include "TrackingTools/RoadSearchHitAccess/interface/DetHitAccess.h"

template<typename T>
inline edmNew::DetSet<T> detSetOrEmpty(const edmNew::DetSetVector<T> &dsv, DetId detid) {
    typename edmNew::DetSetVector<T>::const_iterator iter = dsv.find(detid.rawId());
    if (iter == dsv.end()) {
        return typename edmNew::DetSet<T>(detid.rawId(), static_cast<const T *>(0), size_t(0) );
    } else {
        return *iter;
    }
}


DetHitAccess::DetHitAccess() {
  // default for access mode
  accessMode_ = standard;

  use_rphiRecHits_ = true;
  use_rphiRecHits_ = true;
}

DetHitAccess::~DetHitAccess(){
}

DetHitAccess::DetHitAccess(const SiStripRecHit2DCollection* rphiRecHits,
                           const SiStripRecHit2DCollection* stereoRecHits,
                           const SiStripMatchedRecHit2DCollection* matchedRecHits,
                           const SiPixelRecHitCollection* pixelRecHits) {
  
  // default for access mode
  accessMode_ = standard;

  use_rphiRecHits_   = true;
  use_stereoRecHits_ = true;

  // set collections
  setCollections(rphiRecHits,stereoRecHits,matchedRecHits,pixelRecHits);
  
}
                
void DetHitAccess::setCollections(const SiStripRecHit2DCollection* rphiRecHits,
                                  const SiStripRecHit2DCollection* stereoRecHits,
                                  const SiStripMatchedRecHit2DCollection* matchedRecHits,
                                  const SiPixelRecHitCollection* pixelRecHits) {
  
  rphiHits_ = rphiRecHits;
  stereoHits_ = stereoRecHits;
  matchedHits_ = matchedRecHits;
  pixelHits_ = pixelRecHits;

}


std::vector<TrackingRecHit*> DetHitAccess::getHitVector(const DetId* detid) {
  //
  //DetId that is given to getHitVector is *always* rphi
  //
  
  std::vector<TrackingRecHit*> RecHitVec;
  
  if ( (unsigned int)detid->subdetId() == StripSubdetector::TIB  
       || (unsigned int)detid->subdetId() == StripSubdetector::TOB  
       || (unsigned int)detid->subdetId() == StripSubdetector::TID  
       || (unsigned int)detid->subdetId() == StripSubdetector::TEC )  { 
    
    StripSubdetector StripDetId(*detid);


    if (accessMode_ == rphi ) {
      //
      //return only r-phi RecHits; in case of double modules eliminate recurring r-phi RecHits
      //
      if( !StripDetId.glued() ) {
        if ( rphiHits_ != 0 ) {
          SiStripRecHit2DCollection::DetSet rphiDetHits = detSetOrEmpty(*rphiHits_, *detid);
          for ( SiStripRecHit2DCollection::DetSet::const_iterator rphiDetHit = rphiDetHits.begin();
                rphiDetHit != rphiDetHits.end(); 
                ++rphiDetHit ) {
            RecHitVec.push_back((TrackingRecHit*)(&(*rphiDetHit)));
          }
        } else {
          edm::LogWarning("RoadSearch") << "rphi RecHit collection not set properly";
        }
      } else {
        if ( matchedHits_ != 0 ) {
          DetId useDetId(StripDetId.glued());
          SiStripMatchedRecHit2DCollection::DetSet matchedDetHits = detSetOrEmpty(*matchedHits_, useDetId);
          for ( SiStripMatchedRecHit2DCollection::DetSet::const_iterator matchedDetHit = matchedDetHits.begin();
                matchedDetHit != matchedDetHits.end(); ++matchedDetHit ) {
            bool add = true;
            TrackingRecHit *rphi = (TrackingRecHit*)matchedDetHit->monoHit();
            for ( std::vector<TrackingRecHit*>::iterator hit = RecHitVec.begin();
                  hit != RecHitVec.end();
                  ++hit ) {
              if ((*hit)->localPosition().x() == rphi->localPosition().x()) {
                if ((*hit)->localPosition().y() == rphi->localPosition().y()) {
                  add = false;
                  break;
                }
              }
            }
            if ( add ) {
              RecHitVec.push_back(rphi);
            }
          }
        } else {
          edm::LogWarning("RoadSearch") << "matched RecHit collection not set properly";
        }
      }
    } else if (accessMode_ == rphi_stereo ) {
      //
      //return only r-phi and stereo RecHits
      //
      if( !StripDetId.glued() ) {
        if ( rphiHits_ != 0 ) {
          SiStripRecHit2DCollection::DetSet rphiDetHits = detSetOrEmpty(*rphiHits_, *detid);
          for ( SiStripRecHit2DCollection::DetSet::const_iterator rphiDetHit = rphiDetHits.begin();
                rphiDetHit != rphiDetHits.end(); 
                ++rphiDetHit ) {
            RecHitVec.push_back((TrackingRecHit*)(&(*rphiDetHit)));
          }
        } else {
          edm::LogWarning("RoadSearch") << "rphi RecHit collection not set properly";
        }
        
      } else {
        DetId rphiDetId(StripDetId.glued()+2);
        if ( rphiHits_ != 0 ) {
          SiStripRecHit2DCollection::DetSet rphiDetHits = detSetOrEmpty(*rphiHits_, rphiDetId);
          for ( SiStripRecHit2DCollection::DetSet::const_iterator rphiDetHit = rphiDetHits.begin();
                rphiDetHit != rphiDetHits.end(); 
                ++rphiDetHit ) {
            RecHitVec.push_back((TrackingRecHit*)(&(*rphiDetHit)));
          }
        } else {
          edm::LogWarning("RoadSearch") << "rphi RecHit collection not set properly";
        }

        DetId stereoDetId(StripDetId.glued()+1);
        if ( stereoHits_ != 0 ) {
          SiStripRecHit2DCollection::DetSet stereoDetHits = detSetOrEmpty(*stereoHits_, stereoDetId);
          for ( SiStripRecHit2DCollection::DetSet::const_iterator stereoDetHit = stereoDetHits.begin();
                stereoDetHit != stereoDetHits.end(); 
                ++stereoDetHit ) {
            RecHitVec.push_back((TrackingRecHit*)(&(*stereoDetHit)));
          }
        } else {
          edm::LogWarning("RoadSearch") << "stereo RecHit collection not set properly";
        }
      }
    } else if (accessMode_ == standard ) {
      //
      //single modules: return r-phi RecHits
      //double modules: return matched RecHits + r-phi RecHits that are not used by matched RecHits
      //
      if( !StripDetId.glued() ) {
        if ( rphiHits_ != 0 ) {
          SiStripRecHit2DCollection::DetSet rphiDetHits = detSetOrEmpty(*rphiHits_, *detid);
          for ( SiStripRecHit2DCollection::DetSet::const_iterator rphiDetHit = rphiDetHits.begin();
                rphiDetHit != rphiDetHits.end(); 
                ++rphiDetHit ) {
            RecHitVec.push_back((TrackingRecHit*)(&(*rphiDetHit)));
          }
        } else {
          edm::LogWarning("RoadSearch") << "rphi RecHit collection not set properly";
        }
      } else {
        if ( matchedHits_ != 0 ) {
          DetId useDetId(StripDetId.glued());
          SiStripMatchedRecHit2DCollection::DetSet matchedDetHits = detSetOrEmpty(*matchedHits_, useDetId);
          for ( SiStripMatchedRecHit2DCollection::DetSet::const_iterator matchedDetHit = matchedDetHits.begin();
                matchedDetHit != matchedDetHits.end(); ++matchedDetHit ) {
            RecHitVec.push_back((TrackingRecHit*)(&(*matchedDetHit)));
          }
        } else {
          edm::LogWarning("RoadSearch") << "matched RecHit collection not set properly";
        }
        
        //check for additional r-phi RecHits (not used by matched RecHits)
        if(use_rphiRecHits_) {
          DetId rphiDetId(StripDetId.glued()+2);
          if ( rphiHits_ != 0 ) {
            SiStripRecHit2DCollection::DetSet rphiDetHits = detSetOrEmpty(*rphiHits_, rphiDetId);
            for ( SiStripRecHit2DCollection::DetSet::const_iterator rphiDetHit = rphiDetHits.begin();
                  rphiDetHit != rphiDetHits.end(); ++rphiDetHit ) {
              bool use_rphi=true;
              DetId useDetId(StripDetId.glued());
              SiStripMatchedRecHit2DCollection::DetSet matchedDetHits = detSetOrEmpty(*matchedHits_, useDetId);
              //SiStripMatchedRecHit2DCollection::DetSet matchedDetHits = detSetOrEmpty(*matchedHits_, *detid);
              for ( SiStripMatchedRecHit2DCollection::DetSet::const_iterator matchedDetHit = matchedDetHits.begin();
                    matchedDetHit != matchedDetHits.end(); ++matchedDetHit ) { 
                if (rphiDetHit->localPosition().x()==matchedDetHit->monoHit()->localPosition().x() 
                    && rphiDetHit->localPosition().y()==matchedDetHit->monoHit()->localPosition().y() ) {
                  use_rphi=false;
                  break;
                }
              }
              if(use_rphi) RecHitVec.push_back((TrackingRecHit*)(&(*rphiDetHit)));
            }
          } else {
            edm::LogWarning("RoadSearch") << "rphi RecHit collection not set properly";
          }
        }
          
        //check for additional stereo RecHits (not used by matched RecHits)
        if(use_stereoRecHits_) {
          DetId stereoDetId(StripDetId.glued()+1);
          if ( stereoHits_ != 0 ) {
            SiStripRecHit2DCollection::DetSet stereoDetHits = detSetOrEmpty(*stereoHits_, stereoDetId);
            for ( SiStripRecHit2DCollection::DetSet::const_iterator stereoDetHit = stereoDetHits.begin();
                  stereoDetHit != stereoDetHits.end(); ++stereoDetHit ) {
              bool use_stereo=true;
              DetId useDetId(StripDetId.glued());
              SiStripMatchedRecHit2DCollection::DetSet matchedDetHits = detSetOrEmpty(*matchedHits_, useDetId);
              //SiStripMatchedRecHit2DCollection::DetSet matchedDetHits = detSetOrEmpty(*matchedHits_, *detid);
              for ( SiStripMatchedRecHit2DCollection::DetSet::const_iterator matchedDetHit = matchedDetHits.begin();
                    matchedDetHit != matchedDetHits.end(); ++matchedDetHit ) { 
                if (stereoDetHit->localPosition().x()==matchedDetHit->stereoHit()->localPosition().x() 
                    && stereoDetHit->localPosition().y()==matchedDetHit->stereoHit()->localPosition().y() ) {
                  use_stereo=false;
                  break;
                }
              }
              if(use_stereo) RecHitVec.push_back((TrackingRecHit*)(&(*stereoDetHit)));
            }
          } else {
            edm::LogWarning("RoadSearch") << "stereo RecHit collection not set properly";
          }  
        }
      }
    }
    
  } else if (    (unsigned int)detid->subdetId() == PixelSubdetector::PixelBarrel 
                 || (unsigned int)detid->subdetId() == PixelSubdetector::PixelEndcap) {
    
    if ( pixelHits_ != 0 ) {
      SiPixelRecHitCollection::DetSet pixelDetHits = detSetOrEmpty(*pixelHits_, *detid);
      for ( SiPixelRecHitCollection::DetSet::const_iterator pixelDetHit = pixelDetHits.begin(); 
            pixelDetHit!= pixelDetHits.end(); ++pixelDetHit) {
        RecHitVec.push_back((TrackingRecHit*)(&(*pixelDetHit)));
      }
    } else {
      edm::LogWarning("RoadSearch") << "pixel RecHit collection not set properly";
    } 
  } else {
    
    edm::LogError("RoadSearch") << "NEITHER PIXEL NOR STRIP DETECTOR ID";

  }
  
  
  return RecHitVec;
}