JetMatchingMGFastJet.h

Go to the documentation of this file.

#ifndef gen_JetMatchingMGFastJet_h
#define gen_JetMatchingMGFastJet_h
 
//
//  Julia V. Yarba, Feb.10, 2013
//
//  This code takes inspirations in the original implemetation
//  by Steve Mrenna of FNAL (example main32), but is structured
//  somewhat differently, and is also using FastJet package
//  instead of Pythia8's native SlowJet
//
 
#include "GeneratorInterface/PartonShowerVeto/interface/JetMatching.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
 
#include "SimDataFormats/GeneratorProducts/interface/LHECommonBlocks.h"
 
#include "GeneratorInterface/LHEInterface/interface/LHERunInfo.h"
#include "GeneratorInterface/LHEInterface/interface/LHEEvent.h"
 
//
// FastJet package/tools
// Also gives PseudoJet & JetDefinition
//
#include "fastjet/ClusterSequence.hh"
 
#include <iostream>
#include <fstream>
 
namespace gen {
  class JetMatchingMGFastJet : public JetMatching {
  public:
    JetMatchingMGFastJet(const edm::ParameterSet &params);
 
    ~JetMatchingMGFastJet() override {
      if (fJetFinder)
        delete fJetFinder;
    }
 
    const std::vector<int> *getPartonList() override { return typeIdx; }
 
  protected:
    void init(const lhef::LHERunInfo *runInfo) override;
 
    bool initAfterBeams() override;
    void beforeHadronisation(const lhef::LHEEvent *) override;
    void beforeHadronisationExec() override { return; }
 
    int match(const lhef::LHEEvent *partonLevel, const std::vector<fastjet::PseudoJet> *jetInput) override;
 
    double getJetEtaMax() const override;
 
  private:
    //parameters staff from Madgraph
 
    template <typename T>
    static T parseParameter(const std::string &value);
    template <typename T>
    static T getParameter(const std::map<std::string, std::string> &params,
                          const std::string &var,
                          const T &defValue = T());
    template <typename T>
    T getParameter(const std::string &var, const T &defValue = T()) const;
 
    template <typename T>
    static void updateOrDie(const std::map<std::string, std::string> &params, T &param, const std::string &name);
 
    std::map<std::string, std::string> mgParams;
 
    // ----------------------------
 
    enum vetoStatus { NONE, LESS_JETS, MORE_JETS, HARD_JET, UNMATCHED_PARTON };
    enum partonTypes { ID_TOP = 6, ID_GLUON = 21, ID_PHOTON = 22 };
 
    double qCut, qCutSq;
    double clFact;
    int nQmatch;
 
    // Master switch for merging
    bool doMerge;
 
    // Maximum and current number of jets
    int nJetMax, nJetMin;
 
    // Jet algorithm parameters
    int jetAlgoPower;  //  similar to memain_.mektsc ?
    double coneRadius, etaJetMax;
 
    // Merging procedure control flag(s)
    // (there're also inclusive, exclusive, and soup/auto in JetMatchingMGFastJet)
    //
    bool fExcLocal;  // this is similar to memaev_.iexc
 
    // Sort final-state of incoming process into light/heavy jets and 'other'
    std::vector<int> typeIdx[3];
 
    bool runInitialized;
    bool soup;
    bool exclusive;
 
    //
    // FastJets tool(s)
    //
    fastjet::JetDefinition *fJetFinder;
    std::vector<fastjet::PseudoJet> fClusJets, fPtSortedJets;
 
    // output for DJR analysis
    //
    std::ofstream fDJROutput;
    int fDJROutFlag;
 
    bool fIsInit;
  };
 
}  // namespace gen
 
#endif