ReclusterJets.h

Go to the documentation of this file.

#ifndef PhysicsTools_Heppy_ReclusterJets_h
#define PhysicsTools_Heppy_ReclusterJets_h
 
#include <vector>
#include <iostream>
#include <cmath>
#include <TLorentzVector.h>
#include <TMath.h>
#include "DataFormats/Math/interface/LorentzVector.h"
 
#include <fastjet/internal/base.hh>
#include "fastjet/PseudoJet.hh"
#include "fastjet/JetDefinition.hh"
#include "fastjet/ClusterSequence.hh"
#include "fastjet/Selector.hh"
#include "fastjet/PseudoJet.hh"
 
namespace heppy {
  class ReclusterJets {
  public:
    typedef math::XYZTLorentzVector LorentzVector;
 
    ReclusterJets(const std::vector<LorentzVector> &objects, double ktpower, double rparam);
 
    std::vector<LorentzVector> getGrouping(double ptMin = 0.0);
 
    std::vector<LorentzVector> getGroupingExclusive(int njets);
 
    std::vector<LorentzVector> getGroupingExclusive(double dcut);
 
    LorentzVector getPruned(double zcut, double rcutFactor);
 
    LorentzVector getPrunedSubjetExclusive(unsigned int isubjet, double zcut, double rcutFactor);
 
    LorentzVector getPrunedSubjetInclusive(unsigned int isubjet, double zcut, double rcutFactor);
 
  private:
    // pack the returns in a fwlite-friendly way
    std::vector<LorentzVector> makeP4s(const std::vector<fastjet::PseudoJet> &jets);
 
    // prune and return in fa fwlite-friendly way
    LorentzVector getPruned(const fastjet::PseudoJet &jet, double zcut, double rcutFactor);
 
    // used to handle the inputs
    std::vector<fastjet::PseudoJet> fjInputs_;  // fastjet inputs
 
    double ktpower_;
    double rparam_;
 
    typedef std::shared_ptr<fastjet::ClusterSequence> ClusterSequencePtr;
    ClusterSequencePtr fjClusterSeq_;
    std::vector<fastjet::PseudoJet> inclusiveJets_;
    std::vector<fastjet::PseudoJet> exclusiveJets_;
  };
}  // namespace heppy
#endif