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

Inheritance diagram for PixelVertexCollectionTrimmer:

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

	~PixelVertexCollectionTrimmer () override

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

bool	hasAbilityToProduceInLumis () const final

bool	hasAbilityToProduceInRuns () const final

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

Private Member Functions
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

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_, PVClusterComparer_cfi::track_chi2_max, PVClusterComparer_cfi::track_prob_min, PVClusterComparer_cfi::track_pt_max, PVClusterComparer_cfi::track_pt_min, 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 ( )

override

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(), DEFINE_FWK_MODULE, 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
	)

overrideprivate

Definition at line 84 of file PixelVertexCollectionTrimmer.cc.

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

 {
   using namespace edm;
    
   auto vtxs_trim = std::make_unique<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(std::move(vtxs_trim));
     
 }