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.

{;}

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_, getHLTprescales::index, reco::PFCandidate::trackRef(), reco::Vertex::tracks_begin(), reco::Vertex::tracks_end(), reco::Vertex::trackWeight(), reco::PFCandidate::vertex(), and w().

Referenced by process().

                                                                                                             {

  
  reco::TrackBaseRef trackBaseRef( pfcand.trackRef() );
  
  size_t  iVertex = 0;
  unsigned index=0;
  unsigned nFoundVertex = 0;
  typedef reco::VertexCollection::const_iterator IV;
  typedef reco::Vertex::trackRef_iterator IT;
  float bestweight=0;
  for(IV iv=vertices.begin(); iv!=vertices.end(); ++iv, ++index) {

    const reco::Vertex& vtx = *iv;
    
    // loop on tracks in vertices
    for(IT iTrack=vtx.tracks_begin(); 
    iTrack!=vtx.tracks_end(); ++iTrack) {
     
      const reco::TrackBaseRef& baseRef = *iTrack;

      // one of the tracks in the vertex is the same as 
      // the track considered in the function
      if(baseRef == trackBaseRef ) {
    float w = vtx.trackWeight(baseRef);
    //select the vertex for which the track has the highest weight
    if (w > bestweight){
      bestweight=w;
      iVertex=index;
      nFoundVertex++;
    }       
      }
    }
  }

  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(IV 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 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, i, reco::PFCandidate::particleId(), pfCandidatesFromPU_, pfCandidatesFromVtx_, and verbose_.

Referenced by ConfigBuilder.ConfigBuilder::addExtraStream(), ConfigBuilder.ConfigBuilder::completeInputCommand(), ConfigBuilder.ConfigBuilder::doNotInlineEventContent(), Vispa.Plugins.ConfigEditor.ConfigDataAccessor.ConfigDataAccessor::dumpPython(), ConfigBuilder.ConfigBuilder.PrintAllModules::leave(), Vispa.Plugins.ConfigEditor.ConfigDataAccessor.ConfigDataAccessor::open(), Vispa.Plugins.ConfigEditor.ConfigDataAccessor.ConfigDataAccessor::outputEventContent(), ConfigBuilder.ConfigBuilder::prepare_FASTSIM(), ConfigBuilder.ConfigBuilder::prepare_HLT(), ConfigBuilder.ConfigBuilder::prepare_LHE(), ConfigBuilder.ConfigBuilder::prepare_VALIDATION(), ConfigBuilder.ConfigBuilder::renameHLTprocessInSequence(), ConfigBuilder.ConfigBuilder::renameInputTagsInSequence(), ConfigBuilder.ConfigBuilder::scheduleSequence(), Vispa.Plugins.ConfigEditor.ConfigDataAccessor.ConfigDataAccessor::setProcess(), and Vispa.Plugins.ConfigEditor.ConfigDataAccessor.ConfigDataAccessor::setProperty().

                                                       {

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

{ checkClosestZVertex_ = val;}

void PFPileUpAlgo::setVerbose ( bool  verbose )

inline

Definition at line 29 of file PFPileUpAlgo.h.

References validate_alignment_devdb10_cfg::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().