#include <RecoPixelVertexing/PixelVertexFinding/src/PixelVertexCollectionTrimmer.cc>

Inheritance diagram for PixelVertexCollectionTrimmer:

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

	~PixelVertexCollectionTrimmer ()

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
	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

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)

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

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

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

Private Attributes
double	fractionSumPt2_

unsigned int	maxVtx_

double	minSumPt2_

PVClusterComparer *	pvComparer_

edm::EDGetTokenT < reco::VertexCollection >	vtxToken_

Additional Inherited Members
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

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 40 of file PixelVertexCollectionTrimmer.cc.

Constructor & Destructor Documentation

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

explicit

Definition at line 58 of file PixelVertexCollectionTrimmer.cc.

References fractionSumPt2_, edm::ParameterSet::getParameter(), maxVtx_, minSumPt2_, pvComparer_, and vtxToken_.

 {
 
   edm::InputTag vtxInputTag = iConfig.getParameter<edm::InputTag>("src" );
   vtxToken_ = consumes<reco::VertexCollection>(vtxInputTag);
   maxVtx_         = iConfig.getParameter<unsigned int> ("maxVtx" );
   fractionSumPt2_ = iConfig.getParameter<double> ("fractionSumPt2" );
   minSumPt2_      = iConfig.getParameter<double> ("minSumPt2" );
 
   edm::ParameterSet PVcomparerPSet = iConfig.getParameter<edm::ParameterSet>("PVcomparer");
   double track_pt_min   = PVcomparerPSet.getParameter<double>("track_pt_min");
   double track_pt_max   = PVcomparerPSet.getParameter<double>("track_pt_max");
   double track_chi2_max = PVcomparerPSet.getParameter<double>("track_chi2_max");
   double track_prob_min = PVcomparerPSet.getParameter<double>("track_prob_min");
 
   pvComparer_ = new PVClusterComparer(track_pt_min, track_pt_max, track_chi2_max, track_prob_min);
 
   produces<reco::VertexCollection>();
 }

PixelVertexCollectionTrimmer::~PixelVertexCollectionTrimmer ( )

Definition at line 78 of file PixelVertexCollectionTrimmer.cc.

79 {

80 }

Member Function Documentation

void PixelVertexCollectionTrimmer::fillDescriptions ( edm::ConfigurationDescriptions & descriptions )

static

Definition at line 123 of file PixelVertexCollectionTrimmer.cc.

References edm::ConfigurationDescriptions::add(), edm::ParameterSetDescription::add(), and edm::ParameterDescriptionNode::setComment().

                                                                                          {
   //The following says we do not know what parameters are allowed so do no validation
   // Please change this to state exactly what you do use, even if it is no parameters
   edm::ParameterSetDescription desc;
   desc.add<edm::InputTag>    ("src",           edm::InputTag(""))
     ->setComment("input (pixel) vertex collection");
   desc.add<unsigned int>     ("maxVtx",        100)
     ->setComment("max output collection size (number of accepted vertices)");
   desc.add<double>           ("fractionSumPt2",  0.3)
     ->setComment("threshold on sumPt2 fraction of the leading vertex");
   desc.add<double>           ("minSumPt2",       0. )
     ->setComment("min sumPt2");
   edm::ParameterSetDescription PVcomparerPSet;
   PVcomparerPSet.add<double>("track_pt_min",1.0)
     ->setComment("min track p_T");
   PVcomparerPSet.add<double>("track_pt_max",10.0)
     ->setComment("max track p_T");
   PVcomparerPSet.add<double>("track_chi2_max",99999.)
     ->setComment("max track chi2");
   PVcomparerPSet.add<double>("track_prob_min",-1.)
     ->setComment("min track prob");
   desc.add<edm::ParameterSetDescription>("PVcomparer", PVcomparerPSet)
     ->setComment("from RecoPixelVertexing/PixelVertexFinding/python/PVClusterComparer_cfi.py");
   descriptions.add("hltPixelVertexCollectionTrimmer",desc);
 }

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

overrideprivatevirtual

Implements edm::stream::EDProducerBase.

Definition at line 84 of file PixelVertexCollectionTrimmer.cc.

References fractionSumPt2_, edm::Event::getByToken(), maxVtx_, minSumPt2_, PVClusterComparer::pTSquaredSum(), edm::Event::put(), pvComparer_, and vtxToken_.

 {
   using namespace edm;
    
   std::auto_ptr<reco::VertexCollection> vtxs_trim(new reco::VertexCollection);
 
   edm::Handle<reco::VertexCollection> vtxs;
   iEvent.getByToken(vtxToken_,vtxs);
 
   double sumpt2 ;
   //double sumpt2previous = -99. ;
 
   // this is not the logic we want, at least for now
   // if requires the sumpt2 for vtx_n to be > threshold * sumpt2 vtx_n-1
   // for (reco::VertexCollection::const_iterator vtx = vtxs->begin(); vtx != vtxs->end(); ++vtx, ++counter){
   // if (counter > maxVtx_) break ;
   // sumpt2 = PVCluster.pTSquaredSum(*vtx) ;
   // if (sumpt2 > sumpt2previous*fractionSumPt2_ && sumpt2 > minSumPt2_ ) vtxs_trim->push_back(*vtx) ;
   // else if (counter == 0 ) vtxs_trim->push_back(*vtx) ;
   // sumpt2previous = sumpt2 ;
   // }
 
   double sumpt2first = pvComparer_->pTSquaredSum(*(vtxs->begin())) ;
 
   for (reco::VertexCollection::const_iterator vtx = vtxs->begin(), evtx=vtxs->end(); 
        vtx != evtx; ++vtx) {
     if (vtxs_trim->size() >= maxVtx_) break ;
     sumpt2 = pvComparer_->pTSquaredSum(*vtx) ;
     //    std::cout << "sumpt2: " << sumpt2 << "[" << sumpt2first << "]" << std::endl;
     //    if (sumpt2 >= sumpt2first*fractionSumPt2_ && sumpt2 > minSumPt2_ ) vtxs_trim->push_back(*vtx) ;
     if (sumpt2 >= sumpt2first*fractionSumPt2_ && sumpt2 > minSumPt2_ ) vtxs_trim->push_back(*vtx) ;
   }
   //  std::cout << " ==> # vertices: " << vtxs_trim->size() << std::endl;
   iEvent.put(vtxs_trim);
     
 }