/data/refman/pasoursint/CMSSW_4_1_8_patch9/src/PhysicsTools/PFCandProducer/interface/FastJetAlgo.h

Go to the documentation of this file.

#ifndef PhysicsTools_PFCandProducer_FastJetAlgo
#define PhysicsTools_PFCandProducer_FastJetAlgo

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h"
#include "DataFormats/JetReco/interface/PFJetCollection.h"
#include "DataFormats/JetReco/interface/Jet.h"

#include "fastjet/PseudoJet.hh"
#include "fastjet/JetDefinition.hh"

#include <iostream>

namespace pf2pat {
  

  //COLIN change name to FastJetAlgo

  class FastJetAlgo {

  public:
    typedef std::vector< fastjet::PseudoJet > PseudoJetCollection;
    typedef PseudoJetCollection::const_iterator PJI;
    
    // template it?
    typedef reco::PFCandidate           InputType;
    typedef std::vector<InputType>      InputCollection; 
    typedef edm::Handle< InputCollection > InputHandle;

    // template it?
    typedef reco::PFJet           JetType;
    typedef std::vector<JetType>  JetCollection; 
    typedef JetCollection::const_iterator JI;

    FastJetAlgo( const edm::ParameterSet& ps ); 
    
    void setJetDefinition( const edm::ParameterSet& ps);  

    void setJetDefinition( const fastjet::JetDefinition& jetDef);
    
    const JetCollection& produce( const InputHandle& inputColl); 
    
    void printPseudoJets( std::ostream& out = std::cout) const;

    void printJets( std::ostream& out = std::cout) const;

    
  private:
    void recoToFastJet(const InputCollection& inputColl); 

    void runJetClustering(); 

    const JetCollection& fastJetToReco();

    JetType makeJet( const fastjet::PseudoJet& pseudoJet) const;

    reco::Jet::Constituents makeConstituents(const fastjet::PseudoJet& pseudoJet) const; 

    InputHandle          inputHandle_;

    PseudoJetCollection  input_;
    
    PseudoJetCollection  output_;

    JetCollection jetCollection_; 

    fastjet::JetDefinition  jetDefinition_;

    fastjet::ClusterSequence* clusterSequence_;

  };

}

#endif