#include <PrimaryVertexSorter.h>

Inheritance diagram for PrimaryVertexSorter< ParticlesCollection >:

Public Types
typedef edm::Association < reco::VertexCollection >	CandToVertex

typedef edm::ValueMap< int >	CandToVertexQuality

typedef ParticlesCollection	PFCollection

typedef edm::ValueMap< float >	VertexScore

Public Types inherited from edm::stream::EDProducer<>
typedef CacheContexts< T...>	CacheTypes

typedef CacheTypes::GlobalCache	GlobalCache

typedef AbilityChecker< T...>	HasAbility

typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache

typedef LuminosityBlockContextT < LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext

typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache

typedef CacheTypes::RunCache	RunCache

typedef RunContextT< RunCache, GlobalCache >	RunContext

typedef CacheTypes::RunSummaryCache	RunSummaryCache

Public Types inherited from edm::stream::EDProducerBase
typedef EDProducerAdaptorBase	ModuleType

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

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

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

std::vector< reco::Vertex >	sortedList (const reco::VertexCollection &primaryVertex) const

	~PrimaryVertexSorter ()

Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default

Public Member Functions inherited from edm::stream::EDProducerBase
	EDProducerBase ()

ModuleDescription const &	moduleDescription () const

virtual	~EDProducerBase ()

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
std::vector< ConsumesInfo >	consumesInfo () const

	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

void	modulesDependentUpon (const std::string &iProcessName, std::vector< const char * > &oModuleLabels) const

void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

Private Attributes
PrimaryVertexAssignment	assignmentAlgo_

bool	produceOriginalMapping_

bool	producePFNoPileUp_

bool	producePFPileUp_

bool	produceSortedVertices_

int	qualityCut_

PrimaryVertexSorting	sortingAlgo_

edm::EDGetTokenT< PFCollection >	tokenCandidates_
	Candidates to be analyzed. More...

edm::EDGetTokenT< edm::View < reco::Candidate > >	tokenJets_

edm::EDGetTokenT < reco::VertexCollection >	tokenVertices_
	vertices More...

bool	useMET_

Additional Inherited Members
Static Public Member Functions inherited from edm::stream::EDProducerBase
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

template<class ParticlesCollection>
class PrimaryVertexSorter< ParticlesCollection >

Author: Andrea Rizzi

class to sort VertexCollection in decreasing order of the sum of the squared track pT's

Definition at line 32 of file PrimaryVertexSorter.h.

Member Typedef Documentation

template<class ParticlesCollection>

typedef edm::Association<reco::VertexCollection> PrimaryVertexSorter< ParticlesCollection >::CandToVertex

Definition at line 35 of file PrimaryVertexSorter.h.

template<class ParticlesCollection>

typedef edm::ValueMap<int> PrimaryVertexSorter< ParticlesCollection >::CandToVertexQuality

Definition at line 36 of file PrimaryVertexSorter.h.

template<class ParticlesCollection>

typedef ParticlesCollection PrimaryVertexSorter< ParticlesCollection >::PFCollection

Definition at line 39 of file PrimaryVertexSorter.h.

template<class ParticlesCollection>

typedef edm::ValueMap<float> PrimaryVertexSorter< ParticlesCollection >::VertexScore

Definition at line 37 of file PrimaryVertexSorter.h.

Constructor & Destructor Documentation

template<class ParticlesCollection >

PrimaryVertexSorter< ParticlesCollection >::PrimaryVertexSorter ( const edm::ParameterSet & iConfig )

explicit

Definition at line 84 of file PrimaryVertexSorter.h.

References PrimaryVertexSorter< ParticlesCollection >::produceOriginalMapping_, PrimaryVertexSorter< ParticlesCollection >::producePFNoPileUp_, PrimaryVertexSorter< ParticlesCollection >::producePFPileUp_, and PrimaryVertexSorter< ParticlesCollection >::produceSortedVertices_.

                                                                                             :
   assignmentAlgo_(iConfig.getParameterSet("assignment")),
   sortingAlgo_(iConfig.getParameterSet("sorting")),
   tokenCandidates_(consumes<ParticlesCollection>(iConfig.getParameter<InputTag>("particles"))),
   tokenVertices_(consumes<VertexCollection>(iConfig.getParameter<InputTag>("vertices"))),
   tokenJets_(consumes<edm::View<reco::Candidate> > (iConfig.getParameter<InputTag>("jets"))),
   produceOriginalMapping_(iConfig.getParameter<bool>("produceAssociationToOriginalVertices")),
   produceSortedVertices_(iConfig.getParameter<bool>("produceSortedVertices")),
   producePFPileUp_(iConfig.getParameter<bool>("producePileUpCollection")),
   producePFNoPileUp_(iConfig.getParameter<bool>("produceNoPileUpCollection")),
   qualityCut_(iConfig.getParameter<int>("qualityForPrimary")),
   useMET_(iConfig.getParameter<bool>("usePVMET"))
 {
 
 
   if(produceOriginalMapping_){
       produces< CandToVertex> ("original");
       produces< CandToVertexQuality> ("original");
       produces< VertexScore> ("original");
   }
   if(produceSortedVertices_){
       produces< reco::VertexCollection> ();
       produces< CandToVertex> ();
       produces< CandToVertexQuality> ();
       produces< VertexScore> ();
   }
 
   if(producePFPileUp_){
       if(produceOriginalMapping_)
             produces< PFCollection> ("originalPileUp");
       if(produceSortedVertices_)
             produces< PFCollection> ("PileUp");
   }
 
   if(producePFNoPileUp_){
       if(produceOriginalMapping_)
             produces< PFCollection> ("originalNoPileUp");
       if(produceSortedVertices_)
             produces< PFCollection> ("NoPileUp");
   }
 
 
 }

template<class ParticlesCollection>

PrimaryVertexSorter< ParticlesCollection >::~PrimaryVertexSorter ( )

inline

Definition at line 43 of file PrimaryVertexSorter.h.

43 {}

Member Function Documentation

template<class ParticlesCollection >

void PrimaryVertexSorter< ParticlesCollection >::produce	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

overridevirtual

Implements edm::stream::EDProducerBase.

Definition at line 133 of file PrimaryVertexSorter.h.

References edm::helper::Filler< Map >::fill(), edm::EventSetup::get(), edm::Event::getByToken(), i, customizeTrackingMonitorSeedNumber::idx, edm::helper::Filler< Map >::insert(), fwrapper::jets, lumidbDDL::newToOld(), lumidbDDL::oldToNew(), edm::Handle< T >::product(), edm::Event::put(), MetAnalyzer::pv(), alignCSCRings::s, and HLT_25ns14e33_v1_cff::vertices.

Referenced by JSONExport.JsonExport::export(), HTMLExport.HTMLExport::export(), and HTMLExport.HTMLExportStatic::export().

                                                                                                {
 
 
   Handle<edm::View<reco::Candidate> > jets;
   iEvent.getByToken( tokenJets_, jets);
 
   edm::ESHandle<TransientTrackBuilder> builder;
   iSetup.get<TransientTrackRecord>().get("TransientTrackBuilder", builder);
 
 
   Handle<VertexCollection> vertices;
   iEvent.getByToken( tokenVertices_, vertices);
 
   Handle<ParticlesCollection> particlesHandle;
   iEvent.getByToken( tokenCandidates_, particlesHandle);
 
   ParticlesCollection particles = *particlesHandle.product();
   std::vector<int> pfToPVVector;
   std::vector<PrimaryVertexAssignment::Quality> pfToPVQualityVector;
   //reverse mapping
   std::vector< std::vector<int> > pvToPFVector(vertices->size());
   std::vector< std::vector<const reco::Candidate *> > pvToCandVector(vertices->size());
   std::vector< std::vector<PrimaryVertexAssignment::Quality> > pvToPFQualityVector(vertices->size());
   std::vector<float> vertexScoreOriginal(vertices->size());
   std::vector<float> vertexScore(vertices->size());
 
     for(auto const & pf : particles) {
     std::pair<int,PrimaryVertexAssignment::Quality> vtxWithQuality=assignmentAlgo_.chargedHadronVertex(*vertices,pf,*jets,*builder);
     pfToPVVector.push_back(vtxWithQuality.first); 
     pfToPVQualityVector.push_back(vtxWithQuality.second); 
   }
 
   //Invert the mapping
   for(size_t i = 0; i < pfToPVVector.size();i++)
   {
     auto pv = pfToPVVector[i];
     auto qual = pfToPVQualityVector[i];
     if(pv >=0 and qual >= qualityCut_){
        pvToPFVector[pv].push_back(i);
 //    std::cout << i << std::endl;
 //     const typename  ParticlesCollection::value_type & cp = particles[i];
 //     std::cout << "CP " << &cp <<  std::endl;
        pvToCandVector[pv].push_back( &particles[i] );
        pvToPFQualityVector[pv].push_back(qual);
     }
   }
 
   //Use multimap for sorting of indices
   std::multimap<float,int> scores;
   for(unsigned int i=0;i<vertices->size();i++){
      float s=sortingAlgo_.score((*vertices)[i],pvToCandVector[i],useMET_);
      vertexScoreOriginal[i]=s;
      scores.insert(std::pair<float,int>(-s,i));    
   }
 
   //create indices
   std::vector<int> oldToNew(vertices->size()),  newToOld(vertices->size());
   size_t newIdx=0;
   for(auto const &  idx :  scores)
   {
 //    std::cout << newIdx << " score: " << idx.first << " oldidx: " << idx.second << " "<< producePFPileUp_ << std::endl;
     vertexScore[newIdx]=-idx.first;
     oldToNew[idx.second]=newIdx;
     newToOld[newIdx]=idx.second;
     newIdx++;
   }
   
 
 
 
   if(produceOriginalMapping_){
     auto_ptr< CandToVertex>  pfCandToOriginalVertexOutput( new CandToVertex(vertices) );
     auto_ptr< CandToVertexQuality>  pfCandToOriginalVertexQualityOutput( new CandToVertexQuality() );
     CandToVertex::Filler cand2VertexFiller(*pfCandToOriginalVertexOutput);
     CandToVertexQuality::Filler cand2VertexQualityFiller(*pfCandToOriginalVertexQualityOutput);
 
     cand2VertexFiller.insert(particlesHandle,pfToPVVector.begin(),pfToPVVector.end());
     cand2VertexQualityFiller.insert(particlesHandle,pfToPVQualityVector.begin(),pfToPVQualityVector.end());
 
     cand2VertexFiller.fill();
     cand2VertexQualityFiller.fill();
     iEvent.put( pfCandToOriginalVertexOutput ,"original");
     iEvent.put( pfCandToOriginalVertexQualityOutput ,"original");
 
     auto_ptr< VertexScore>  vertexScoreOriginalOutput( new VertexScore );
     VertexScore::Filler vertexScoreOriginalFiller(*vertexScoreOriginalOutput);
     vertexScoreOriginalFiller.insert(vertices,vertexScoreOriginal.begin(),vertexScoreOriginal.end());
     vertexScoreOriginalFiller.fill();
     iEvent.put( vertexScoreOriginalOutput ,"original");
  
   }
 
   if(produceSortedVertices_){
       std::vector<int> pfToSortedPVVector;
 //      std::vector<int> pfToSortedPVQualityVector;
       for(size_t i=0;i<pfToPVVector.size();i++) {
         pfToSortedPVVector.push_back(oldToNew[pfToPVVector[i]]);
 //        pfToSortedPVQualityVector.push_back(pfToPVQualityVector[i]); //same as old!
       }
 
       auto_ptr< reco::VertexCollection>  sortedVerticesOutput( new reco::VertexCollection );
       for(size_t i=0;i<vertices->size();i++){
          sortedVerticesOutput->push_back((*vertices)[newToOld[i]]); 
       }
     edm::OrphanHandle<reco::VertexCollection> oh = iEvent.put( sortedVerticesOutput);
     auto_ptr< CandToVertex>  pfCandToVertexOutput( new CandToVertex(oh) );
     auto_ptr< CandToVertexQuality>  pfCandToVertexQualityOutput( new CandToVertexQuality() );
     CandToVertex::Filler cand2VertexFiller(*pfCandToVertexOutput);
     CandToVertexQuality::Filler cand2VertexQualityFiller(*pfCandToVertexQualityOutput);
 
     cand2VertexFiller.insert(particlesHandle,pfToSortedPVVector.begin(),pfToSortedPVVector.end());
     cand2VertexQualityFiller.insert(particlesHandle,pfToPVQualityVector.begin(),pfToPVQualityVector.end());
 
     cand2VertexFiller.fill();
     cand2VertexQualityFiller.fill();
     iEvent.put( pfCandToVertexOutput );
     iEvent.put( pfCandToVertexQualityOutput );
 
     auto_ptr< VertexScore>  vertexScoreOutput( new VertexScore );
     VertexScore::Filler vertexScoreFiller(*vertexScoreOutput);
     vertexScoreFiller.insert(oh,vertexScore.begin(),vertexScore.end());
     vertexScoreFiller.fill();
     iEvent.put( vertexScoreOutput);
 
 
   }
 
 
   auto_ptr< PFCollection >  pfCollectionNOPUOriginalOutput( new PFCollection );
   auto_ptr< PFCollection >  pfCollectionNOPUOutput( new PFCollection );
   auto_ptr< PFCollection >  pfCollectionPUOriginalOutput( new PFCollection );
   auto_ptr< PFCollection >  pfCollectionPUOutput( new PFCollection );
 
   for(size_t i=0;i<particles.size();i++) {
     auto pv = pfToPVVector[i];
     auto qual = pfToPVQualityVector[i];
 
 
      if(producePFNoPileUp_ && produceSortedVertices_) 
          if(pv == newToOld[0] and qual >= qualityCut_) 
                    pfCollectionNOPUOutput->push_back(particles[i]);
 
      if(producePFPileUp_ && produceSortedVertices_) 
          if(pv != newToOld[0] and qual >= qualityCut_) 
                    pfCollectionPUOutput->push_back(particles[i]);
 
      if(producePFNoPileUp_ && produceOriginalMapping_) 
          if(pv == 0 and qual >= qualityCut_) 
                    pfCollectionNOPUOriginalOutput->push_back(particles[i]);
 
      if(producePFPileUp_ && produceOriginalMapping_) 
          if(pv != 0 and qual >= qualityCut_) 
                    pfCollectionPUOriginalOutput->push_back(particles[i]);
 
   }              
   if(producePFNoPileUp_ && produceSortedVertices_) iEvent.put(pfCollectionNOPUOutput,"NoPileUp" );
   if(producePFPileUp_ && produceSortedVertices_) iEvent.put(pfCollectionPUOutput, "PileUp");
   if(producePFNoPileUp_ && produceOriginalMapping_) iEvent.put(pfCollectionNOPUOriginalOutput,"originalNoPileUp" );
   if(producePFPileUp_ && produceOriginalMapping_) iEvent.put(pfCollectionPUOriginalOutput,"originalPileUp" );
   
 
 } 

template<class ParticlesCollection>

VertexCollection PrimaryVertexSorter< ParticlesCollection >::sortedList ( const reco::VertexCollection & primaryVertex ) const

Definition at line 8 of file PrimaryVertexSorter.cc.

References python.multivaluedict::sort().

Referenced by QualityCutsAnalyzer::LoopOverJetTracksAssociation().

 {
   VertexCollection pvs = unsortedPVColl;
   sort(pvs.begin(), pvs.end(), VertexHigherPtSquared());
   return pvs;
 }

Member Data Documentation

template<class ParticlesCollection>

PrimaryVertexAssignment PrimaryVertexSorter< ParticlesCollection >::assignmentAlgo_

private

Definition at line 49 of file PrimaryVertexSorter.h.

template<class ParticlesCollection>

bool PrimaryVertexSorter< ParticlesCollection >::produceOriginalMapping_

private

Definition at line 59 of file PrimaryVertexSorter.h.

Referenced by PrimaryVertexSorter< ParticlesCollection >::PrimaryVertexSorter().

template<class ParticlesCollection>

bool PrimaryVertexSorter< ParticlesCollection >::producePFNoPileUp_

private

Definition at line 62 of file PrimaryVertexSorter.h.

Referenced by PrimaryVertexSorter< ParticlesCollection >::PrimaryVertexSorter().

template<class ParticlesCollection>

bool PrimaryVertexSorter< ParticlesCollection >::producePFPileUp_

private

Definition at line 61 of file PrimaryVertexSorter.h.

Referenced by PrimaryVertexSorter< ParticlesCollection >::PrimaryVertexSorter().

template<class ParticlesCollection>

bool PrimaryVertexSorter< ParticlesCollection >::produceSortedVertices_

private

Definition at line 60 of file PrimaryVertexSorter.h.

Referenced by PrimaryVertexSorter< ParticlesCollection >::PrimaryVertexSorter().

template<class ParticlesCollection>

int PrimaryVertexSorter< ParticlesCollection >::qualityCut_

private

Definition at line 63 of file PrimaryVertexSorter.h.

template<class ParticlesCollection>

PrimaryVertexSorting PrimaryVertexSorter< ParticlesCollection >::sortingAlgo_

private

Definition at line 50 of file PrimaryVertexSorter.h.

template<class ParticlesCollection>

edm::EDGetTokenT<PFCollection> PrimaryVertexSorter< ParticlesCollection >::tokenCandidates_

private

Candidates to be analyzed.

Definition at line 53 of file PrimaryVertexSorter.h.

template<class ParticlesCollection>

edm::EDGetTokenT<edm::View<reco::Candidate> > PrimaryVertexSorter< ParticlesCollection >::tokenJets_

private

Definition at line 57 of file PrimaryVertexSorter.h.

template<class ParticlesCollection>

edm::EDGetTokenT<reco::VertexCollection> PrimaryVertexSorter< ParticlesCollection >::tokenVertices_

private

vertices

Definition at line 56 of file PrimaryVertexSorter.h.

template<class ParticlesCollection>

bool PrimaryVertexSorter< ParticlesCollection >::useMET_

private

Definition at line 64 of file PrimaryVertexSorter.h.

Public Types

Public Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

template<class ParticlesCollection> class PrimaryVertexSorter< ParticlesCollection >

Member Typedef Documentation

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation

template<class ParticlesCollection>
class PrimaryVertexSorter< ParticlesCollection >