PFJetAlgorithm Class Reference

#include <PFJetAlgorithm.h>

Classes
class	Jet

Public Types
typedef std::map< double, PFJetAlgorithm::Jet, std::greater< double > >::iterator	IJ
typedef std::map< double, int, std::greater< double > >::const_iterator	IV

Public Member Functions
void	CleanUp ()
void	Clear ()
const std::vector < PFJetAlgorithm::Jet > &	FindJets (const std::vector< TLorentzVector > *vecs)
double	GetConeAngle () const
const std::vector < PFJetAlgorithm::Jet > &	GetJets () const
void	MergeJets (std::map< double, PFJetAlgorithm::Jet, std::greater< double > > &etjets)
	PFJetAlgorithm ()
	PFJetAlgorithm (double cone, double et, double conemerge)
void	SetConeAngle (double coneAngle)
void	SetConeMerge (double coneMerge)
void	SetSeedEt (double et)
void	Update ()
virtual	~PFJetAlgorithm ()

Static Public Member Functions
static double	DeltaR (double eta1, double phi1, double eta2, double phi2)

Private Attributes
const std::vector < TLorentzVector > *	fAllVecs
std::vector< int >	fAssignedVecs
double	fConeAngle
double	fConeMerge
std::map< double, int, std::greater< double > >	fEtOrderedSeeds
std::vector< PFJetAlgorithm::Jet >	fJets
double	fSeedEt

Detailed Description

Definition at line 11 of file PFJetAlgorithm.h.

Member Typedef Documentation

typedef std::map< double, PFJetAlgorithm::Jet, std::greater<double> >::iterator PFJetAlgorithm::IJ

Definition at line 66 of file PFJetAlgorithm.h.

typedef std::map<double, int, std::greater<double> >::const_iterator PFJetAlgorithm::IV

Definition at line 67 of file PFJetAlgorithm.h.

Constructor & Destructor Documentation

PFJetAlgorithm::PFJetAlgorithm ( )

inline

Definition at line 69 of file PFJetAlgorithm.h.

: fConeAngle(0.5), fSeedEt(2),  fConeMerge(0.5) {}

PFJetAlgorithm::PFJetAlgorithm ( double  cone, double  et, double  conemerge  )

inline

Definition at line 71 of file PFJetAlgorithm.h.

                                                           : 
    fConeAngle(cone), 
    fSeedEt(et),
    fConeMerge(conemerge)
    {}

virtual PFJetAlgorithm::~PFJetAlgorithm ( )

inlinevirtual

Definition at line 77 of file PFJetAlgorithm.h.

{}

Member Function Documentation

void PFJetAlgorithm::CleanUp

(

)

Definition at line 141 of file PFJetAlgorithm.cc.

References i, and tmp.

                             {
  
  //   cout<<"CleanUp : -----------------------------------------"<<endl;
  
  //   for(unsigned i=0; i<fJets.size(); i++) {
  //     cout<<fJets[i]<<endl;
  //   }

  vector< PFJetAlgorithm::Jet >  tmp = fJets;

  map< double, PFJetAlgorithm::Jet, greater<double> > etjets;
  // typedef map< double, PFJetAlgorithm::Jet, greater<double> >::iterator IJ;
  
  vector< PFJetAlgorithm::Jet >  tmp2;
  fJets.clear();

  for(unsigned i=0; i<tmp.size(); i++) {
    if( tmp[i].GetMomentum().Et() > 0 ) 
      // tmp2.push_back( tmp[i]); 
      etjets.insert( make_pair(tmp[i].GetMomentum().Et(), tmp[i]) );
  }

  //   cout<<"et map : "<<endl;
  //   for(IJ ij = etjets.begin(); ij != etjets.end(); ij++) {
  //     cout<<ij->second<<endl;
  //   }  

  MergeJets( etjets );

  //   for(IJ ij = etjets.begin(); ij != etjets.end(); ij++) {

  //     const TLorentzVector& mom1 = ij->second.GetMomentum();

  //     double eta1 = mom1.Eta();
  //     double phi1 = mom1.Phi();

  //     IJ closest = etjets.end();
  //     double drmin = 99999;
  //     for(IJ jj = etjets.begin(); jj != etjets.end(); jj++) {
   
  //       if( jj == ij ) continue;  
   
  //       const TLorentzVector& mom2 = jj->second.GetMomentum();
      
  //       double eta2 = mom2.Eta();
  //       double phi2 = mom2.Phi();

  //       double dr = DeltaR(eta1, phi1, eta2, phi2);

  //       if(dr<drmin) {
  //    drmin = dr; 
  //    closest = jj;
  //       }

  //       if(closest != etjets.end() ) {
  //    if ( dr < fConeMerge ) {
  //      ij->second += jj->second;
  //    }               
  //       } 
  //     }
  //   }

  //   cout<<"et map 2: "<<endl;
  for(IJ ij = etjets.begin(); ij != etjets.end(); ij++) {
    //     cout<<ij->second<<endl;
    fJets.push_back(ij->second);
  }  

  //   set<int> used;
  //   for(unsigned i=0; i<tmp2.size(); i++) {


  //     set<int>::iterator isused = used.find(i);
  //     if( isused != used.end() ) continue;
    
  //     Jet& jet1 = tmp2[i];
    
  //     // cout<<"\t jet "<<jet1<<endl;


  //     TLorentzVector mom1 = jet1.GetMomentum();

  //     double eta1 = mom1.Eta();
  //     double phi1 = mom1.Phi();
    
  //     // merge close jets
  //     for(unsigned j=0; j<tmp2.size(); j++) {
  //       if(i==j) continue;
  //       Jet& jet2 = tmp2[j];

      
  //       TLorentzVector mom2 = jet2.GetMomentum();
      
  //       double eta2 = mom2.Eta();
  //       double phi2 = mom2.Phi();
      
  //       double dr = DeltaR(eta1, phi1, eta2, phi2);

  //       // cout<<"\t\t test merge with "<<jet2<<", dr = "<<dr<<endl;
      
  //       if ( dr < fConeMerge ) {
  //    jet1 += jet2;
  //    used.insert( j );
  //    // cout<<"\t\t  yes "<<endl;
  //       }    
  //       else {
  //    // cout<<"\t\t  no "<<endl;
  //       }
  //     }
    
  //     used.insert( i );
  //     fJets.push_back(jet1);
  //   }
}

void PFJetAlgorithm::Clear ( )

inline

Definition at line 92 of file PFJetAlgorithm.h.

References fAssignedVecs, fEtOrderedSeeds, and fJets.

Referenced by PFRootEventManager::tauBenchmark().

               { 
    fJets.clear(); 
    fAssignedVecs.clear(); 
    fEtOrderedSeeds.clear();
  } 

double PFJetAlgorithm::DeltaR ( double  eta1, double  phi1, double  eta2, double  phi2  )

static

Definition at line 78 of file PFJetAlgorithm.cc.

References Phi_mpi_pi(), and mathSSE::sqrt().

                                                                                {
  double deta = eta1 - eta2;
  double dphi = TVector2::Phi_mpi_pi(phi1 - phi2);
  return sqrt(deta*deta + dphi*dphi);
}

const vector< PFJetAlgorithm::Jet > & PFJetAlgorithm::FindJets

(

const std::vector< TLorentzVector > *

vecs

)

Definition at line 18 of file PFJetAlgorithm.cc.

References PFJetAlgorithm::Jet::Add(), reco::tau::disc::Eta(), PFJetAlgorithm::Jet::GetMomentum(), i, and colinearityKinematic::Phi.

Referenced by PFRootEventManager::tauBenchmark().

                                                            {


  fAllVecs = vecs;

  fAssignedVecs.clear(); 
  fAssignedVecs.reserve( vecs->size() );
  
  fEtOrderedSeeds.clear();
  
  for(unsigned i = 0; i<vecs->size(); i++) {
    // cout<<"i = "<<i<<endl;
    double et = (*vecs)[i].Et();
    
    int assigned = -1;
    if( et >= fSeedEt) {
      fJets.push_back( Jet(i, fAllVecs) );
      fEtOrderedSeeds.insert( make_pair(et, fJets.size()-1) );
      assigned = i;
    }
    fAssignedVecs.push_back(assigned);
  }

  
  // loop on seeds 
  for(IV iv = fEtOrderedSeeds.begin(); iv != fEtOrderedSeeds.end(); iv++ ) {
    
    Jet& currentjet = fJets[iv->second];

    double etaseed = currentjet.GetMomentum().Eta();
    double phiseed = currentjet.GetMomentum().Phi();
    
    
    // adding particles
    for(unsigned i = 0; i<fAllVecs->size(); i++) {
      
      if( fAssignedVecs[i] > -1 ) continue;

      double dr = DeltaR(etaseed, phiseed, 
                         (*fAllVecs)[i].Eta(), (*fAllVecs)[i].Phi());
      // cout<<"\t\tparticle "<<i<<" "<<dr<<" "<<(*fAllVecs)[i].Et()<<endl; 
                         
      if( dr < fConeAngle) {
        // cout<<"\t\tadding"<<endl;
        currentjet.Add(i);
        fAssignedVecs[i] = iv->second;
      }
    }
    
    // cout<<"\t seed processed"<<endl;
  }
  // cout<<"end find jets"<<endl;

  Update();
  CleanUp();

  return fJets;
}

double PFJetAlgorithm::GetConeAngle ( ) const

inline

Definition at line 103 of file PFJetAlgorithm.h.

References fConeAngle.

{ return fConeAngle;}

const std::vector< PFJetAlgorithm::Jet >& PFJetAlgorithm::GetJets ( ) const

inline

Definition at line 104 of file PFJetAlgorithm.h.

References fJets.

{ return fJets;}

void PFJetAlgorithm::MergeJets

(

std::map< double, PFJetAlgorithm::Jet, std::greater< double > > &

etjets

)

Definition at line 257 of file PFJetAlgorithm.cc.

                                                                                        {

  // look for smallest distance between 2 jets : 
  
  IJ j1 = etjets.end();
  IJ j2 = etjets.end();
  double smallestdistance = 999999;

  for(IJ ij = etjets.begin(); ij != etjets.end(); ij++) {
    
    const TLorentzVector& mom1 = ij->second.GetMomentum();
    
    double eta1 = mom1.Eta();
    double phi1 = mom1.Phi();
    
    for(IJ jj = etjets.begin(); jj != etjets.end(); jj++) {
   
      if( jj == ij ) continue;  
      
      const TLorentzVector& mom2 = jj->second.GetMomentum();
      
      double eta2 = mom2.Eta();
      double phi2 = mom2.Phi();

      double dr = DeltaR(eta1, phi1, eta2, phi2);
      
      if(dr<smallestdistance) {
        smallestdistance = dr; 
        j1 = ij;
        j2 = jj;
      }
    }
  } 

  //   cout<<"smallest distance is between : "<<endl;
  //   cout<<j1->second<<endl;
  //   cout<<j2->second<<endl;

  if( smallestdistance < fConeMerge ) {
    j1->second += j2->second;
    etjets.erase(j2);
    
    MergeJets( etjets ); 
  }
}

void PFJetAlgorithm::SetConeAngle ( double  coneAngle )

inline

Definition at line 83 of file PFJetAlgorithm.h.

References fConeAngle.

Referenced by PFRootEventManager::readOptions().

{fConeAngle = coneAngle;}

void PFJetAlgorithm::SetConeMerge ( double  coneMerge )

inline

Definition at line 85 of file PFJetAlgorithm.h.

References fConeMerge.

Referenced by PFRootEventManager::readOptions().

{fConeMerge = coneMerge;}

void PFJetAlgorithm::SetSeedEt ( double  et )

inline

Definition at line 84 of file PFJetAlgorithm.h.

References fSeedEt.

Referenced by PFRootEventManager::readOptions().

{fSeedEt = et;}

void PFJetAlgorithm::Update

(

)

Definition at line 85 of file PFJetAlgorithm.cc.

References PFJetAlgorithm::Jet::Add(), PFJetAlgorithm::Jet::Clear(), reco::tau::disc::Eta(), PFJetAlgorithm::Jet::GetMomentum(), i, and colinearityKinematic::Phi.

                            {
  // use existing jets as seeds 
  //   cout<<"clearing seeds"<<endl;
  fEtOrderedSeeds.clear();
  for(unsigned ij = 0; ij<fJets.size(); ij++ ) {
    double et = fJets[ij].GetMomentum().Et();
    if(et >= fSeedEt)
      fEtOrderedSeeds.insert( make_pair(et, ij) );
  }

  // loop on seeds and add particles 
  //   cout<<"clearing assigned"<<endl;
  for(unsigned i = 0; i<fAssignedVecs.size(); i++) {
    fAssignedVecs[i] = -1;
  }

  //   cout<<"loop on seeds"<<endl;
  bool needupdate = false;
  for(IV iv = fEtOrderedSeeds.begin(); iv != fEtOrderedSeeds.end(); iv++ ) {
    
    Jet& currentjet = fJets[iv->second];

    TLorentzVector seedmom = currentjet.GetMomentum();

    double etaseed = seedmom.Eta();
    double phiseed = seedmom.Phi();
    //     cout<<"SEED\t"<<etaseed<<" "<<phiseed<<endl;

    currentjet.Clear();

    // adding particles
    for(unsigned i = 0; i<fAllVecs->size(); i++) {
      
      if( fAssignedVecs[i] > -1 ) continue;

      double dr = DeltaR(etaseed, phiseed, 
                         (*fAllVecs)[i].Eta(), (*fAllVecs)[i].Phi());
      // cout<<"\t\tparticle "<<i<<" "<<dr<<" "<<(*fAllVecs)[i].Et()<<endl; 
                         
      if( dr < fConeAngle) {
        // cout<<"\t\tadding"<<endl;
        currentjet.Add(i);
        fAssignedVecs[i] = iv->second;
      }
    }
    
    TLorentzVector deltav = currentjet.GetMomentum();
    deltav -= seedmom;
    if(deltav.M() > 0.001) needupdate = true;
  }
  
  if(needupdate) Update();
}

Member Data Documentation

const std::vector<TLorentzVector>* PFJetAlgorithm::fAllVecs

private

Definition at line 52 of file PFJetAlgorithm.h.

std::vector< int > PFJetAlgorithm::fAssignedVecs

private

Definition at line 54 of file PFJetAlgorithm.h.

Referenced by Clear().

double PFJetAlgorithm::fConeAngle

private

Definition at line 58 of file PFJetAlgorithm.h.

Referenced by GetConeAngle(), and SetConeAngle().

double PFJetAlgorithm::fConeMerge

private

Definition at line 60 of file PFJetAlgorithm.h.

Referenced by SetConeMerge().

std::map<double, int, std::greater<double> > PFJetAlgorithm::fEtOrderedSeeds

private

Definition at line 55 of file PFJetAlgorithm.h.

Referenced by Clear().

std::vector< PFJetAlgorithm::Jet > PFJetAlgorithm::fJets

private

Definition at line 53 of file PFJetAlgorithm.h.

Referenced by Clear(), and GetJets().

double PFJetAlgorithm::fSeedEt

private

Definition at line 59 of file PFJetAlgorithm.h.

Referenced by SetSeedEt().