/data/refman/pasoursint/CMSSW_4_1_8_patch9/src/Validation/RecoHI/plugins/HitPixelLayersTPSelector.h

Go to the documentation of this file.

#ifndef HitPixelLayersTrackSelection_h
#define HitPixelLayersTrackSelection_h

#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticle.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticleFwd.h"

#include "DataFormats/SiPixelDetId/interface/PXBDetId.h"
#include "DataFormats/SiPixelDetId/interface/PXFDetId.h"

class HitPixelLayersTPSelector
{
  
 public:
  // input collection type
  typedef TrackingParticleCollection collection;
  
  
  // output collection type
  typedef std::vector<const TrackingParticle*> container;
  
  // iterator over result collection type.
  typedef container::const_iterator const_iterator;
  
  // constructor from parameter set configurability
  HitPixelLayersTPSelector(const edm::ParameterSet & iConfig) : 
    tripletSeedOnly_(iConfig.getParameter<bool>("tripletSeedOnly")),
    ptMin_(iConfig.getParameter<double>("ptMin")), 
    minRapidity_(iConfig.getParameter<double>("minRapidity")), 
    maxRapidity_(iConfig.getParameter<double>("maxRapidity")),
    tip_(iConfig.getParameter<double>("tip")), 
    lip_(iConfig.getParameter<double>("lip")), 
    minHit_(iConfig.getParameter<int>("minHit")), 
    signalOnly_(iConfig.getParameter<bool>("signalOnly")), 
    chargedOnly_(iConfig.getParameter<bool>("chargedOnly")), 
    primaryOnly_(iConfig.getParameter<bool>("primaryOnly")),
    tpStatusBased_(iConfig.getParameter<bool>("tpStatusBased")),
    pdgId_(iConfig.getParameter< std::vector<int> >("pdgId"))                              
      {};
    
    // select object from a collection and
    // possibly event content
    void select( const edm::Handle<collection> & TPCH, const edm::Event & iEvent, const edm::EventSetup & iSetup)
    {
      selected_.clear();
      

      const collection & tpc = *(TPCH.product());
      
      for (TrackingParticleCollection::size_type i=0; i<tpc.size(); i++)
        {
          TrackingParticleRef tpr(TPCH, i);
          
          // quickly reject if it is from pile-up
          if (signalOnly_ && !(tpr->eventId().bunchCrossing()==0 && tpr->eventId().event()==0) ) continue;
          if (chargedOnly_ && tpr->charge()==0) continue; //select only if charge!=0
          if (tpStatusBased_ && primaryOnly_ && tpr->status()!=1 ) continue; // TP status based sel primary
          if ((!tpStatusBased_) && primaryOnly_ && tpr->parentVertex()->nSourceTracks()!=0 ) continue; // vertex based sel for primary

          // loop over specified PID values 
          bool testId = false;
          unsigned int idSize = pdgId_.size();
          if (idSize==0) testId = true;
          else for (unsigned int it=0;it!=idSize;++it){
            if (tpr->pdgId()==pdgId_[it]) testId = true;
          }
          
          // selection criteria
          if ( tpr->matchedHit() >= minHit_ &&   
               sqrt(tpr->momentum().perp2()) >= ptMin_ && 
               tpr->momentum().eta() >= minRapidity_ && tpr->momentum().eta() <= maxRapidity_ && 
               sqrt(tpr->vertex().perp2()) <= tip_ &&
               fabs(tpr->vertex().z()) <= lip_ &&
               testId)
            {
              if (tripletSeedOnly_ && !goodHitPattern(pixelHitPattern(tpr)) ) continue; //findable triplet seed
              const TrackingParticle * trap = &(tpc[i]);
              selected_.push_back(trap);
            } 
          
        }
    }
    
    // return pixel layer hit pattern
    std::vector<bool> pixelHitPattern( const TrackingParticleRef& simTrack )
      {
        std::vector<bool> hitpattern(5,false); // PXB 0,1,2  PXF 0,1
        
        for(std::vector<PSimHit>::const_iterator simHit = simTrack->pSimHit_begin();simHit!= simTrack->pSimHit_end();simHit++){
          
          DetId id = DetId(simHit->detUnitId());
          uint32_t detid = id.det();
          uint32_t subdet = id.subdetId();
          
          if (detid == DetId::Tracker) {
            if (subdet == PixelSubdetector::PixelBarrel) 
              hitpattern[PXBDetId(id).layer()-1]=true;
            else if (subdet == PixelSubdetector::PixelEndcap) 
              hitpattern[PXFDetId(id).disk()+2]=true;
          }
          
        }// end simhit loop
        
        return hitpattern;
      }
    
    // test whether hit pattern would give a pixel triplet seed
    bool goodHitPattern( const std::vector<bool> hitpattern )
    {
      if( (hitpattern[0] && hitpattern[1] && hitpattern[2]) ||
          (hitpattern[0] && hitpattern[1] && hitpattern[3]) ||
          (hitpattern[0] && hitpattern[3] && hitpattern[4]) ) 
        return true;
      else 
        return false;
    }
    
    // iterators over selected objects: collection begin
    const_iterator begin() const
    {
      return selected_.begin();
    }
    
    // iterators over selected objects: collection end
    const_iterator end() const
    {
      return selected_.end();
    }
    
    // true if no object has been selected
    size_t size() const
    {
      return selected_.size();
    }
    
    //private:
    
    container selected_;
    bool tripletSeedOnly_;
    double ptMin_;
    double minRapidity_;
    double maxRapidity_;
    double tip_;
    double lip_;
    int    minHit_;
    bool signalOnly_;
    bool chargedOnly_;
    bool primaryOnly_;
    bool tpStatusBased_;
    std::vector<int> pdgId_;
    
    
};


#endif