#include <RecoBTag/FastPrimaryVertexProducer/src/FastPrimaryVertexProducer.cc>

Inheritance diagram for FastPrimaryVertexProducer:

Public Member Functions
	FastPrimaryVertexProducer (const edm::ParameterSet &)

Public Member Functions inherited from edm::EDProducer
	EDProducer ()

ModuleDescription const &	moduleDescription () const

virtual	~EDProducer ()

Public Member Functions inherited from edm::ProducerBase
	ProducerBase ()

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...

virtual	~ProducerBase ()

Public Member Functions inherited from edm::EDConsumerBase
	EDConsumerBase ()

ProductHolderIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

Private Member Functions
virtual void	produce (edm::Event &, const edm::EventSetup &) override

Private Attributes
edm::InputTag	m_beamSpot

double	m_clusterLength

edm::InputTag	m_clusters

edm::InputTag	m_jets

double	m_maxDeltaPhi

double	m_maxSizeX

double	m_maxZ

std::string	m_pixelCPE

Additional Inherited Members
Public Types inherited from edm::EDProducer
typedef EDProducer	ModuleType

Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Static Public Member Functions inherited from edm::EDProducer
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Protected Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Detailed Description

Description: [one line class summary]

Implementation: [Notes on implementation]

Definition at line 74 of file FastPrimaryVertexProducer.cc.

Constructor & Destructor Documentation

FastPrimaryVertexProducer::FastPrimaryVertexProducer ( const edm::ParameterSet & iConfig )

explicit

Definition at line 90 of file FastPrimaryVertexProducer.cc.

References edm::ParameterSet::getParameter(), and AlCaHLTBitMon_QueryRunRegistry::string.

 {
   m_clusters          = iConfig.getParameter<edm::InputTag>("clusters");
   m_jets              = iConfig.getParameter<edm::InputTag>("jets");
   m_beamSpot          = iConfig.getParameter<edm::InputTag>("beamSpot");
   m_pixelCPE          = iConfig.getParameter<std::string>("pixelCPE");
   m_maxZ          = iConfig.getParameter<double>("maxZ");   
   m_maxSizeX          = iConfig.getParameter<double>("maxSizeX");   
   m_maxDeltaPhi       = iConfig.getParameter<double>("maxDeltaPhi");    
   m_clusterLength     = iConfig.getParameter<double>("clusterLength");  
   produces<reco::VertexCollection>();
 }

Member Function Documentation

void FastPrimaryVertexProducer::produce	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

overrideprivatevirtual

Implements edm::EDProducer.

Definition at line 106 of file FastPrimaryVertexProducer.cc.

References SiPixelRawToDigiRegional_cfi::beamSpot, SiPixelRawToDigiRegional_cfi::deltaPhi, alignCSCRings::e, edm::EventSetup::get(), edm::Event::getByLabel(), j, fwrapper::jets, max(), n, AlCaHLTBitMon_ParallelJobs::p, PV3DBase< T, PVType, FrameType >::phi(), clustersummaryproducer_cfg::pixelClusters, createTree::pp, RecoTauCleanerPlugins::pt, edm::Event::put(), dt_dqm_sourceclient_common_cff::reco, edmNew::DetSet< T >::size(), SiPixelCluster::sizeX(), SiPixelCluster::sizeY(), python.multivaluedict::sort(), patCandidatesForDimuonsSequences_cff::tracker, findQualityFiles::v, PV3DBase< T, PVType, FrameType >::x(), PV3DBase< T, PVType, FrameType >::y(), detailsBasic3DVector::z, and PV3DBase< T, PVType, FrameType >::z().

 {
    using namespace edm;
    using namespace reco;
    using namespace std;
 
    Handle<SiPixelClusterCollectionNew> cH;
    iEvent.getByLabel(m_clusters,cH);
    const SiPixelClusterCollectionNew & pixelClusters = *cH.product();
 
    Handle<edm::View<reco::Jet> > jH;
    iEvent.getByLabel(m_jets,jH);
    const edm::View<reco::Jet> & jets = *jH.product();
 
    CaloJetCollection selectedJets;
    for(edm::View<reco::Jet>::const_iterator it = jets.begin() ; it != jets.end() ; it++)
    {
     if(it->pt() > 40 && fabs(it->eta()) < 1.6)
     {
       const CaloJet * ca = dynamic_cast<const CaloJet *>( &(*it));
       if(ca ==0) abort();
       selectedJets.push_back(*ca);
 //    std::cout << "Jet eta,phi,pt: "<< it->eta() << "," << it->phi() << "," << it->pt()   << std::endl;
     } 
    }
   
    edm::ESHandle<PixelClusterParameterEstimator> pe; 
    const PixelClusterParameterEstimator * pp ;
    iSetup.get<TkPixelCPERecord>().get(m_pixelCPE , pe );  
    pp = pe.product();
 
    edm::Handle<BeamSpot> beamSpot;
    iEvent.getByLabel(m_beamSpot,beamSpot);
  
    edm::ESHandle<TrackerGeometry> tracker;
    iSetup.get<TrackerDigiGeometryRecord>().get(tracker);
    const TrackerGeometry * trackerGeometry = tracker.product();
 
 
    float lengthBmodule=6.66;//cm 
    std::vector<float> zProjections;
    for(CaloJetCollection::const_iterator jit = selectedJets.begin() ; jit != selectedJets.end() ; jit++)
    {
      float px=jit->px();
      float py=jit->py();
      float pz=jit->pz();
      float pt=jit->pt();
 
      float jetZOverRho = jit->momentum().Z()/jit->momentum().Rho();
      int minSizeY = fabs(2.*jetZOverRho)-1; 
      int  maxSizeY = fabs(2.*jetZOverRho)+2;
      if( fabs(jit->eta()) > 1.6)
      {
         minSizeY = 1;
      }
 
      for(SiPixelClusterCollectionNew::const_iterator it = pixelClusters.begin() ; it != pixelClusters.end() ; it++) //Loop on pixel modules with clusters
      {
         DetId id = it->detId();
         const edmNew::DetSet<SiPixelCluster> & detset  = (*it);
         Point3DBase<float, GlobalTag> modulepos=trackerGeometry->idToDet(id)->position();
         float zmodule = modulepos.z() - ((modulepos.x()-beamSpot->x0())*px+(modulepos.y()-beamSpot->y0())*py)/pt * pz/pt;
         if ((fabs(deltaPhi(jit->momentum().Phi(),modulepos.phi()))< m_maxDeltaPhi*2)&&(fabs(zmodule)<(m_maxZ+lengthBmodule/2))){
        
         for(size_t j = 0 ; j < detset.size() ; j ++) // Loop on pixel clusters on this module
         {
       const SiPixelCluster & aCluster =  detset[j];
           if(aCluster.sizeX() < m_maxSizeX && aCluster.sizeY() >= minSizeY && aCluster.sizeY() <= maxSizeY) {
             Point3DBase<float, GlobalTag> v = trackerGeometry->idToDet(id)->surface().toGlobal(pp->localParametersV( aCluster,( *trackerGeometry->idToDetUnit(id)))[0].first) ;
             GlobalPoint v_bs(v.x()-beamSpot->x0(),v.y()-beamSpot->y0(),v.z());
             if(fabs(deltaPhi(jit->momentum().Phi(),v_bs.phi())) < m_maxDeltaPhi)
             {
               float z = v.z() - ((v.x()-beamSpot->x0())*px+(v.y()-beamSpot->y0())*py)/pt * pz/pt;   
               if(fabs(z) < m_maxZ)
               {
             zProjections.push_back(z); 
           } 
         }
           } //if compatible cluster   
     } // loop on module hits
       } // if compatible module
     } // loop on pixel modules
       
    } // loop on selected jets
   std::sort(zProjections.begin(),zProjections.end());
    
   std::vector<float>::iterator itCenter = zProjections.begin();
   std::vector<float>::iterator itLeftSide = zProjections.begin();
   std::vector<float>::iterator itRightSide = zProjections.begin();
   std::vector<int> counts;
   float zCluster = m_clusterLength/2.0; //cm 
   int max=0;
   std::vector<float>::iterator left,right;
   for(;itCenter!=zProjections.end(); itCenter++)
   {
   
     while(itLeftSide != zProjections.end() && (*itCenter - *itLeftSide) > zCluster  ) itLeftSide++;
     while(itRightSide != zProjections.end() && (*itRightSide - *itCenter) < zCluster  ) itRightSide++;
    
     int n= itRightSide-itLeftSide;
    // std::cout << "algo :"<< *itCenter << " " << itCenter-zProjections.begin() << "  dists: " <<  (*itCenter - *itLeftSide) << " " << (*itRightSide - *itCenter) << " count: " <<  n << std::endl; 
     counts.push_back(n);
     if(n > max) {
          max=n; 
          left=itLeftSide;
     }
     if(n >= max) {
       max=n;
           right=itRightSide;
 //          std::cout << "algo :"<< i << " " << j << " " << *itCenter << " " << itCenter-zProjections.begin() << "  dists: " <<  (*itCenter - *itLeftSide) << " " << (*itRightSide - *itCenter) << " count: " <<  n << std::endl;
     }
   }
 
 
  
 
   float res=0; 
   if(zProjections.size() > 0) 
   {
      res=*(left+(right-left)/2);
 //     std::cout << "RES " << res << std::endl;
      Vertex::Error e; 
      e(0, 0) = 0.0015 * 0.0015;
      e(1, 1) = 0.0015 * 0.0015;
      e(2, 2) = 1.5 * 1.5;
      Vertex::Point p(beamSpot->x(res), beamSpot->y(res), res);
      Vertex thePV(p, e, 1, 1, 0);
      std::auto_ptr<reco::VertexCollection> pOut(new reco::VertexCollection());
      pOut->push_back(thePV);
      iEvent.put(pOut);
    } else
    {
   //   std::cout << "DUMMY " << res << std::endl;
 
      Vertex::Error e;
      e(0, 0) = 0.0015 * 0.0015;
      e(1, 1) = 0.0015 * 0.0015;
      e(2, 2) = 1.5 * 1.5;
      Vertex::Point p(beamSpot->x(res), beamSpot->y(res), res);
      Vertex thePV(p, e, 0, 0, 0);
      std::auto_ptr<reco::VertexCollection> pOut(new reco::VertexCollection());
      pOut->push_back(thePV);
      iEvent.put(pOut);
 
    }
 
  
 }