PrimaryVertexSorter< ParticlesCollection > Class Template Reference

#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 Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

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
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
std::vector< edm::ProductResolverIndex > const &	indiciesForPutProducts (BranchType iBranchType) const
	ProducerBase ()
void	registerProducts (ProducerBase *, ProductRegistry *, ModuleDescription const &)
std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...
void	resolvePutIndicies (BranchType iBranchType, std::unordered_multimap< std::string, edm::ProductResolverIndex > const &iIndicies, std::string const &moduleLabel)
virtual	~ProducerBase () noexcept(false)
Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const
	EDConsumerBase ()
	EDConsumerBase (EDConsumerBase const &)=delete
	EDConsumerBase (EDConsumerBase &&)=default
ProductResolverIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const
void	itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
void	itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
std::vector< ProductResolverIndexAndSkipBit > const &	itemsToGetFrom (BranchType iType) const
void	labelsForToken (EDGetToken iToken, Labels &oLabels) const
void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const
EDConsumerBase const &	operator= (EDConsumerBase const &)=delete
EDConsumerBase &	operator= (EDConsumerBase &&)=default
bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const
bool	registeredToConsumeMany (TypeID const &, BranchType) const
void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)
virtual	~EDConsumerBase () noexcept(false)

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::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 81 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<edm::InputTag>("particles"))),
  tokenVertices_(consumes<reco::VertexCollection>(iConfig.getParameter<edm::InputTag>("vertices"))),
  tokenJets_(consumes<edm::View<reco::Candidate> > (iConfig.getParameter<edm::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"))
{

using namespace std;
using namespace edm;
using namespace reco;

  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.

References PrimaryVertexSorter< ParticlesCollection >::produce().

{}

Member Function Documentation

template<class ParticlesCollection >

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

overridevirtual

Definition at line 133 of file PrimaryVertexSorter.h.

References PrimaryVertexSorter< ParticlesCollection >::assignmentAlgo_, PrimaryVertexAssignment::chargedHadronVertex(), edm::helper::Filler< Map >::fill(), edm::EventSetup::get(), edm::Event::getByToken(), mps_fire::i, training_settings::idx, edm::helper::Filler< Map >::insert(), fwrapper::jets, eostools::move(), lumidbDDL::newToOld(), lumidbDDL::oldToNew(), HadronAndPartonSelector_cfi::particles, packedPFCandidateRefMixer_cfi::pf, PrimaryVertexSorter< ParticlesCollection >::produceOriginalMapping_, PrimaryVertexSorter< ParticlesCollection >::producePFNoPileUp_, PrimaryVertexSorter< ParticlesCollection >::producePFPileUp_, PrimaryVertexSorter< ParticlesCollection >::produceSortedVertices_, edm::Handle< T >::product(), edm::Event::put(), MetAnalyzer::pv(), PrimaryVertexSorter< ParticlesCollection >::qualityCut_, alignCSCRings::s, PrimaryVertexSorting::score(), PrimaryVertexSorter< ParticlesCollection >::sortingAlgo_, PrimaryVertexSorter< ParticlesCollection >::tokenCandidates_, PrimaryVertexSorter< ParticlesCollection >::tokenJets_, PrimaryVertexSorter< ParticlesCollection >::tokenVertices_, PrimaryVertexSorter< ParticlesCollection >::useMET_, and ecalDrivenElectronSeedsParameters_cff::vertices.

Referenced by JSONExport.JsonExport::export(), HTMLExport.HTMLExport::export(), HTMLExport.HTMLExportStatic::export(), and PrimaryVertexSorter< ParticlesCollection >::~PrimaryVertexSorter().

                                                                                                     {

using namespace std;
using namespace edm;
using namespace reco;

  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_){
    unique_ptr< CandToVertex>  pfCandToOriginalVertexOutput( new CandToVertex(vertices) );
    unique_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(std::move(pfCandToOriginalVertexOutput) ,"original");
    iEvent.put(std::move(pfCandToOriginalVertexQualityOutput) ,"original");

    unique_ptr< VertexScore>  vertexScoreOriginalOutput( new VertexScore );
    VertexScore::Filler vertexScoreOriginalFiller(*vertexScoreOriginalOutput);
    vertexScoreOriginalFiller.insert(vertices,vertexScoreOriginal.begin(),vertexScoreOriginal.end());
    vertexScoreOriginalFiller.fill();
    iEvent.put(std::move(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!
      }

      unique_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(std::move(sortedVerticesOutput));
    unique_ptr< CandToVertex>  pfCandToVertexOutput( new CandToVertex(oh) );
    unique_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(std::move(pfCandToVertexOutput ));
    iEvent.put(std::move(pfCandToVertexQualityOutput ));

    unique_ptr< VertexScore>  vertexScoreOutput( new VertexScore );
    VertexScore::Filler vertexScoreFiller(*vertexScoreOutput);
    vertexScoreFiller.insert(oh,vertexScore.begin(),vertexScore.end());
    vertexScoreFiller.fill();
    iEvent.put(std::move(vertexScoreOutput));


  }


  unique_ptr< PFCollection >  pfCollectionNOPUOriginalOutput( new PFCollection );
  unique_ptr< PFCollection >  pfCollectionNOPUOutput( new PFCollection );
  unique_ptr< PFCollection >  pfCollectionPUOriginalOutput( new PFCollection );
  unique_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(std::move(pfCollectionNOPUOutput),"NoPileUp" );
  if(producePFPileUp_ && produceSortedVertices_) iEvent.put(std::move(pfCollectionPUOutput), "PileUp");
  if(producePFNoPileUp_ && produceOriginalMapping_) iEvent.put(std::move(pfCollectionNOPUOriginalOutput),"originalNoPileUp" );
  if(producePFPileUp_ && produceOriginalMapping_) iEvent.put(std::move(pfCollectionPUOriginalOutput),"originalPileUp" );
  

} 

template<class ParticlesCollection>

VertexCollection PrimaryVertexSorter< ParticlesCollection >::sortedList

(

const reco::VertexCollection &

primaryVertex

)

const

Definition at line 8 of file PrimaryVertexSorter.cc.

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.

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

template<class ParticlesCollection>

bool PrimaryVertexSorter< ParticlesCollection >::produceOriginalMapping_

private

Definition at line 59 of file PrimaryVertexSorter.h.

Referenced by PrimaryVertexSorter< ParticlesCollection >::PrimaryVertexSorter(), and PrimaryVertexSorter< ParticlesCollection >::produce().

template<class ParticlesCollection>

bool PrimaryVertexSorter< ParticlesCollection >::producePFNoPileUp_

private

Definition at line 62 of file PrimaryVertexSorter.h.

Referenced by PrimaryVertexSorter< ParticlesCollection >::PrimaryVertexSorter(), and PrimaryVertexSorter< ParticlesCollection >::produce().

template<class ParticlesCollection>

bool PrimaryVertexSorter< ParticlesCollection >::producePFPileUp_

private

Definition at line 61 of file PrimaryVertexSorter.h.

Referenced by PrimaryVertexSorter< ParticlesCollection >::PrimaryVertexSorter(), and PrimaryVertexSorter< ParticlesCollection >::produce().

template<class ParticlesCollection>

bool PrimaryVertexSorter< ParticlesCollection >::produceSortedVertices_

private

Definition at line 60 of file PrimaryVertexSorter.h.

Referenced by PrimaryVertexSorter< ParticlesCollection >::PrimaryVertexSorter(), and PrimaryVertexSorter< ParticlesCollection >::produce().

template<class ParticlesCollection>

int PrimaryVertexSorter< ParticlesCollection >::qualityCut_

private

Definition at line 63 of file PrimaryVertexSorter.h.

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

template<class ParticlesCollection>

PrimaryVertexSorting PrimaryVertexSorter< ParticlesCollection >::sortingAlgo_

private

Definition at line 50 of file PrimaryVertexSorter.h.

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

template<class ParticlesCollection>

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

private

Candidates to be analyzed.

Definition at line 53 of file PrimaryVertexSorter.h.

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

template<class ParticlesCollection>

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

private

Definition at line 57 of file PrimaryVertexSorter.h.

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

template<class ParticlesCollection>

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

private

vertices

Definition at line 56 of file PrimaryVertexSorter.h.

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

template<class ParticlesCollection>

bool PrimaryVertexSorter< ParticlesCollection >::useMET_

private

Definition at line 64 of file PrimaryVertexSorter.h.

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