GeneratorInterface/PartonShowerVeto/interface/JetMatchingMGFastJet.h - master - githubmirror/cmssw

GeneratorInterface/PartonShowerVeto/interface/JetMatchingMGFastJet.h
117 строк3 KB
Cms Build
Clang-Format
16 май 2019, 15:48
16 май 2019, 15:480847e19
#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