PFPileUpAlgo Class Reference

#include <PFPileUpAlgo.h>

Public Types
typedef std::vector< edm::FwdPtr< reco::PFCandidate > >	PFCollection

Public Member Functions
int	chargedHadronVertex (const reco::VertexCollection &vertices, const reco::PFCandidate &pfcand) const
const PFCollection &	getPFCandidatesFromPU () const
const PFCollection &	getPFCandidatesFromVtx () const
	PFPileUpAlgo ()
	PFPileUpAlgo (bool checkClosestZVertex, bool verbose=false)
void	process (const PFCollection &pfCandidates, const reco::VertexCollection &vertices)
void	setCheckClosestZVertex (bool val)
void	setVerbose (bool verbose)
	~PFPileUpAlgo ()

Private Attributes
bool	checkClosestZVertex_
	use the closest z vertex if a track is not in a vertex More...
PFCollection	pfCandidatesFromPU_
PFCollection	pfCandidatesFromVtx_
bool	verbose_
	verbose ? More...

Detailed Description

Definition at line 12 of file PFPileUpAlgo.h.

Member Typedef Documentation

typedef std::vector< edm::FwdPtr<reco::PFCandidate> > PFPileUpAlgo::PFCollection

Definition at line 16 of file PFPileUpAlgo.h.

Constructor & Destructor Documentation

PFPileUpAlgo::PFPileUpAlgo ( )

inline

Definition at line 18 of file PFPileUpAlgo.h.

:checkClosestZVertex_(true), verbose_(false) {;}

PFPileUpAlgo::PFPileUpAlgo ( bool  checkClosestZVertex, bool  verbose = false  )

inline

Definition at line 20 of file PFPileUpAlgo.h.

                                                             :
    checkClosestZVertex_(checkClosestZVertex), verbose_(verbose) {;}

PFPileUpAlgo::~PFPileUpAlgo ( )

inline

Definition at line 23 of file PFPileUpAlgo.h.

References slimmedMuons_cfi::pfCandidates, process(), and electrons_cff::vertices.

{;}

Member Function Documentation

int PFPileUpAlgo::chargedHadronVertex	(	const reco::VertexCollection &	vertices,
		const reco::PFCandidate &	pfcand
	)		const

Definition at line 43 of file PFPileUpAlgo.cc.

References checkClosestZVertex_, PVValHelper::dz, HiIsolationCommonParameters_cff::track, reco::PFCandidate::trackRef(), reco::PFCandidate::vertex(), badGlobalMuonTaggersAOD_cff::vtx, and w.

Referenced by getPFCandidatesFromVtx(), and process().

                                                                                                             {

  auto const & track = pfcand.trackRef();  
  size_t  iVertex = 0;
  unsigned int index=0;
  unsigned int nFoundVertex = 0;
  float bestweight=0;
  for( auto const & vtx : vertices) {
      float w = vtx.trackWeight(track);
     //select the vertex for which the track has the highest weight
    if (w > bestweight){
      bestweight=w;
      iVertex=index;
      nFoundVertex++;
    }
     ++index;
  }

  if (nFoundVertex>0){
    if (nFoundVertex!=1)
      edm::LogWarning("TrackOnTwoVertex")<<"a track is shared by at least two verteces. Used to be an assert";
    return iVertex;
  }
  // no vertex found with this track. 

  // optional: as a secondary solution, associate the closest vertex in z
  if ( checkClosestZVertex_ ) {

    double dzmin = 10000;
    double ztrack = pfcand.vertex().z();
    bool foundVertex = false;
    index = 0;
    for(auto iv=vertices.begin(); iv!=vertices.end(); ++iv, ++index) {

      double dz = fabs(ztrack - iv->z());
      if(dz<dzmin) {
    dzmin = dz; 
    iVertex = index;
    foundVertex = true;
      }
    }

    if( foundVertex ) 
      return iVertex;  

  }


  return -1 ;
}

const PFCollection& PFPileUpAlgo::getPFCandidatesFromPU ( ) const

inline

Definition at line 33 of file PFPileUpAlgo.h.

References pfCandidatesFromPU_.

{return pfCandidatesFromPU_;}

const PFCollection& PFPileUpAlgo::getPFCandidatesFromVtx ( ) const

inline

Definition at line 35 of file PFPileUpAlgo.h.

References chargedHadronVertex(), pfDeepBoostedJetPreprocessParams_cfi::pfcand, and pfCandidatesFromVtx_.

{return pfCandidatesFromVtx_;}

void PFPileUpAlgo::process	(	const PFCollection &	pfCandidates,
		const reco::VertexCollection &	vertices
	)

Definition at line 5 of file PFPileUpAlgo.cc.

References chargedHadronVertex(), gather_cfg::cout, reco::PFCandidate::h, mps_fire::i, reco::PFCandidate::particleId(), pfCandidatesFromPU_, pfCandidatesFromVtx_, and verbose_.

Referenced by Vispa.Plugins.ConfigEditor.ConfigDataAccessor.ConfigDataAccessor::dumpPython(), Vispa.Plugins.ConfigEditor.ConfigDataAccessor.ConfigDataAccessor::open(), Vispa.Plugins.ConfigEditor.ConfigDataAccessor.ConfigDataAccessor::outputEventContent(), Vispa.Plugins.ConfigEditor.ConfigDataAccessor.ConfigDataAccessor::setProcess(), Vispa.Plugins.ConfigEditor.ConfigDataAccessor.ConfigDataAccessor::setProperty(), and ~PFPileUpAlgo().

                                                       {

  pfCandidatesFromVtx_.clear();
  pfCandidatesFromPU_.clear();

  for( unsigned i=0; i<pfCandidates.size(); i++ ) {
    
    const reco::PFCandidate& cand = * ( pfCandidates[i] );
    
    int ivertex;

    switch( cand.particleId() ) {
    case reco::PFCandidate::h:
      ivertex = chargedHadronVertex( vertices, cand );
      break;
    default:
      continue;
    } 
    
    // no associated vertex, or primary vertex
    // not pile-up
    if( ivertex == -1  || 
    ivertex == 0 ) {
      if(verbose_)
    std::cout<<"VTX "<<i<<" "<< *(pfCandidates[i])<<std::endl;
      pfCandidatesFromVtx_.push_back( pfCandidates[i] );
    } else {
      if(verbose_)
    std::cout<<"PU  "<<i<<" "<< *(pfCandidates[i])<<std::endl;
      // associated to a vertex
      pfCandidatesFromPU_.push_back( pfCandidates[i] );
    }
  }
}

void PFPileUpAlgo::setCheckClosestZVertex ( bool  val )

inline

Definition at line 31 of file PFPileUpAlgo.h.

References checkClosestZVertex_, and heppy_batch::val.

{ checkClosestZVertex_ = val;}

void PFPileUpAlgo::setVerbose ( bool  verbose )

inline

Definition at line 29 of file PFPileUpAlgo.h.

References verbose, and verbose_.

{ verbose_ = verbose; }

Member Data Documentation

bool PFPileUpAlgo::checkClosestZVertex_

private

use the closest z vertex if a track is not in a vertex

Definition at line 44 of file PFPileUpAlgo.h.

Referenced by chargedHadronVertex(), and setCheckClosestZVertex().

PFCollection PFPileUpAlgo::pfCandidatesFromPU_

private

Definition at line 51 of file PFPileUpAlgo.h.

Referenced by getPFCandidatesFromPU(), and process().

PFCollection PFPileUpAlgo::pfCandidatesFromVtx_

private

Definition at line 50 of file PFPileUpAlgo.h.

Referenced by getPFCandidatesFromVtx(), and process().

bool PFPileUpAlgo::verbose_

private

verbose ?

Definition at line 48 of file PFPileUpAlgo.h.

Referenced by process(), and setVerbose().