KtJets.hh

Go to the documentation of this file.

// -*- C++ -*-
#ifndef RIVET_KtJets_HH
#define RIVET_KtJets_HH

#include "Rivet/Projection.hh"
#include "Rivet/Projections/FinalState.hh"
#include "KtJet/KtJet.h"
#include "KtJet/KtEvent.h"


namespace Rivet {

  /// Project out jets based on configurable kT algorithm.
  class KtJets : public Projection {
 
  public:
 
    /// @name Standard constructors and destructors.
    //@{

    /// Default constructor. Must specify a FinalState projection which is
    //  assumed to live throughout the run.
    KtJets(const FinalState& fsp)
      : _pktev(0), _type(4), _angle(2), _recom(1),
        _rparameter(1.0)
    {
      setName("KtJets");
      addProjection(fsp, "FS");
    }


    /// Argument constructor. Allows the to be run with different parameters.
    /// Must specify a FinalState projection which is assumed to live throughout the run.
    KtJets(const FinalState& fsp, int type, int angle, int recom, double rparameter)
      : _pktev(0), _type(type), _angle(angle), _recom(recom),
        _rparameter(rparameter)
    {
      setName("KtJets");
      addProjection(fsp, "FS");
    }
 

    /// Clone on the heap.
    virtual const Projection* clone() const {
      return new KtJets(*this);
    }


    /// Destructor.
    virtual ~KtJets() {
      delete _pktev;
    }
    //@}

 
  protected:

    /// Perform the projection on the Event.
    void project(const Event& e);

    /// Compare projections.
    int compare(const Projection& p) const;


  public:
 
    /// @name Access the projected NJets.
    //@ {
    int getNJets() const { return _pktev->getNJets(); }
    int getNConstituents() const { return _pktev->getNConstituents(); }
    vector<KtJet::KtLorentzVector> copyConstituents() const { return _pktev->copyConstituents(); }
    double getETot() const { return _pktev->getETot(); } // had trouble building with Ktfloat, used double instead
    int getType() const { return _pktev->getType(); }
    int getAngle() const { return _pktev->getAngle(); }
    int getRecom() const { return _pktev->getRecom(); }
    bool isInclusive() const { return _pktev->isInclusive(); }

    vector<KtJet::KtLorentzVector> getJets() const { return _pktev->getJets(); }
    vector<KtJet::KtLorentzVector> getJetsE() const { return _pktev->getJetsE(); }
    vector<KtJet::KtLorentzVector> getJetsEt() const { return _pktev->getJetsEt(); }
    vector<KtJet::KtLorentzVector> getJetsPt() const { return _pktev->getJetsPt(); }
    vector<KtJet::KtLorentzVector> getJetsRapidity() const { return _pktev->getJetsRapidity(); }
    vector<KtJet::KtLorentzVector> getJetsEta() const { return _pktev->getJetsEta(); }
    //@}

    /// Get the subjet splitting variables for the given jet.
    vector<double> getYSubJet(const KtJet::KtLorentzVector& jet) const;


  private:
 
    /// Internal KtEvent, rebuilt every time an event is projected, but not otherwise.
    KtJet::KtEvent* _pktev;

    int _type, _angle, _recom;
    double _rparameter;

    /// Map of vectors of y scales. This is mutable so we can use caching/lazy evaluation.
    mutable map<int, vector<double> > _yscales;
 
  };

}

#endif