GeneratorInterface/RivetInterface/interface/RivetAnalysis.h - master - githubmirror/cmssw

GeneratorInterface/RivetInterface/interface/RivetAnalysis.h
225 строк8 KB
Shahzad Malik Muzaffar
[GENERATORS] Changes suggested by new llvm18 clang-format
05 сен 2024, 17:14
05 сен 2024, 17:14f8c46a4
#ifndef GeneratorInterface_RivetInterface_RivetAnalysis_H
#define GeneratorInterface_RivetInterface_RivetAnalysis_H

#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "Rivet/Analysis.hh"
#include "Rivet/Projections/FinalState.hh"
#include "Rivet/Particle.hh"
#include "Rivet/Particle.fhh"
#include "Rivet/Event.hh"
#include "Rivet/Projections/FastJets.hh"
#include "Rivet/Projections/ChargedLeptons.hh"
#include "Rivet/Projections/PromptFinalState.hh"
#include "Rivet/Projections/LeptonFinder.hh"
#include "Rivet/Projections/VetoedFinalState.hh"
#include "Rivet/Projections/IdentifiedFinalState.hh"
#include "Rivet/Projections/MissingMomentum.hh"
#include "Rivet/Tools/RivetHepMC.hh"

namespace Rivet {

  class RivetAnalysis : public Analysis {
  public:
    Particles leptons() const { return _leptons; }
    Particles photons() const { return _photons; }
    Particles neutrinos() const { return _neutrinos; }
    Jets jets() const { return _jets; }
    Jets fatjets() const { return _fatjets; }
    Vector3 met() const { return _met; }

  private:
    bool _usePromptFinalStates;
    bool _excludePromptLeptonsFromJetClustering;
    bool _excludeNeutrinosFromJetClustering;
    bool _doJetClustering;

    double _particleMinPt, _particleMaxEta;
    double _lepConeSize, _lepMinPt, _lepMaxEta;
    double _jetConeSize, _jetMinPt, _jetMaxEta;
    double _fatJetConeSize, _fatJetMinPt, _fatJetMaxEta;

    double _phoMinPt, _phoMaxEta, _phoIsoConeSize, _phoMaxRelIso;

    Particles _leptons, _photons, _neutrinos;
    Jets _jets, _fatjets;
    Vector3 _met;

  public:
    RivetAnalysis(const edm::ParameterSet& pset)
        : Analysis("RivetAnalysis"),
          _usePromptFinalStates(pset.getParameter<bool>("usePromptFinalStates")),
          _excludePromptLeptonsFromJetClustering(pset.getParameter<bool>("excludePromptLeptonsFromJetClustering")),
          _excludeNeutrinosFromJetClustering(pset.getParameter<bool>("excludeNeutrinosFromJetClustering")),
          _doJetClustering(pset.getParameter<bool>("doJetClustering")),

          _particleMinPt(pset.getParameter<double>("particleMinPt")),
          _particleMaxEta(pset.getParameter<double>("particleMaxEta")),

          _lepConeSize(pset.getParameter<double>("lepConeSize")),
          _lepMinPt(pset.getParameter<double>("lepMinPt")),
          _lepMaxEta(pset.getParameter<double>("lepMaxEta")),

          _jetConeSize(pset.getParameter<double>("jetConeSize")),
          _jetMinPt(pset.getParameter<double>("jetMinPt")),
          _jetMaxEta(pset.getParameter<double>("jetMaxEta")),

          _fatJetConeSize(pset.getParameter<double>("fatJetConeSize")),
          _fatJetMinPt(pset.getParameter<double>("fatJetMinPt")),
          _fatJetMaxEta(pset.getParameter<double>("fatJetMaxEta")),

          _phoMinPt(pset.getParameter<double>("phoMinPt")),
          _phoMaxEta(pset.getParameter<double>("phoMaxEta")),
          _phoIsoConeSize(pset.getParameter<double>("phoIsoConeSize")),
          _phoMaxRelIso(pset.getParameter<double>("phoMaxRelIso")) {}

    // Initialize Rivet projections
    void init() override {
      // Cuts
      Cut particle_cut = (Cuts::abseta < _particleMaxEta) and (Cuts::pT > _particleMinPt * GeV);
      Cut lepton_cut = (Cuts::abseta < _lepMaxEta) and (Cuts::pT > _lepMinPt * GeV);

      // Generic final state
      FinalState fs(particle_cut);

      declare(fs, "FS");

      // Dressed leptons
      ChargedLeptons charged_leptons(fs);
      IdentifiedFinalState photons(fs);
      photons.acceptIdPair(PID::PHOTON);

      PromptFinalState prompt_leptons(charged_leptons);
      prompt_leptons.acceptMuonDecays(true);
      prompt_leptons.acceptTauDecays(true);
      declare(prompt_leptons, "PromptLeptons");

      PromptFinalState prompt_photons(photons);
      prompt_photons.acceptMuonDecays(true);
      prompt_photons.acceptTauDecays(true);

      LeptonFinder prompt_dressed_leptons(prompt_leptons, prompt_photons, _lepConeSize, lepton_cut);
      LeptonFinder dressed_leptons(charged_leptons, photons, _lepConeSize, lepton_cut);
      declare(_usePromptFinalStates ? prompt_dressed_leptons : dressed_leptons, "DressedLeptons");

      declare(photons, "Photons");

      // Jets
      VetoedFinalState fsForJets(fs);
      if (_usePromptFinalStates and _excludePromptLeptonsFromJetClustering)
        fsForJets.addVetoOnThisFinalState(prompt_dressed_leptons);
      JetInvisibles invisiblesStrategy = JetInvisibles::DECAY;
      if (_excludeNeutrinosFromJetClustering)
        invisiblesStrategy = JetInvisibles::NONE;
      declare(FastJets(fsForJets, JetAlg::ANTIKT, _jetConeSize, JetMuons::ALL, invisiblesStrategy), "Jets");

      // FatJets
      declare(FastJets(fsForJets, JetAlg::ANTIKT, _fatJetConeSize, JetMuons::ALL, invisiblesStrategy), "FatJets");

      // Neutrinos
      IdentifiedFinalState neutrinos(fs);
      neutrinos.acceptNeutrinos();
      if (_usePromptFinalStates) {
        PromptFinalState prompt_neutrinos(neutrinos);
        prompt_neutrinos.acceptMuonDecays(true);
        prompt_neutrinos.acceptTauDecays(true);
        declare(prompt_neutrinos, "Neutrinos");
      } else
        declare(neutrinos, "Neutrinos");

      // MET
      declare(MissingMomentum(fs), "MET");
    };

    // Apply Rivet projections
    void analyze(const Event& event) override {
      _jets.clear();
      _fatjets.clear();
      _leptons.clear();
      _photons.clear();
      _neutrinos.clear();

      // Get analysis objects from projections
      Cut jet_cut = (Cuts::abseta < _jetMaxEta) and (Cuts::pT > _jetMinPt * GeV);
      Cut fatjet_cut = (Cuts::abseta < _fatJetMaxEta) and (Cuts::pT > _fatJetMinPt * GeV);

      _leptons = apply<LeptonFinder>(event, "DressedLeptons").particlesByPt();

      // search tau ancestors
      Particles promptleptons = apply<PromptFinalState>(event, "PromptLeptons").particles();
      for (auto& lepton : _leptons) {
        const auto& cl = lepton.constituents().front();  // this has no ancestors anymore :(
        for (auto const& pl : promptleptons) {
          if (cl.momentum() == pl.momentum()) {
            for (auto& p : pl.ancestors()) {
              if (p.abspid() == 15) {
                p.setMomentum(p.momentum() * 1e-20);
                lepton.addConstituent(p, false);
              }
            }
            break;
          }
        }
      }

      // Photons
      Particles fsparticles = apply<FinalState>(event, "FS").particles();

      for (auto& photon : apply<FinalState>(event, "Photons").particlesByPt()) {
        if (photon.pt() < _phoMinPt)
          continue;

        if (abs(photon.eta()) > _phoMaxEta)
          continue;

        double photonptsum = 0;

        for (auto& fsparticle : fsparticles) {
          if (deltaR(fsparticle, photon) == 0)
            continue;

          if (deltaR(fsparticle, photon) > _phoIsoConeSize)
            continue;

          photonptsum += fsparticle.pt();
        }

        if (photonptsum / photon.pt() > _phoMaxRelIso)
          continue;

        _photons.push_back(photon);
      }

      if (_doJetClustering) {
        _jets = apply<FastJets>(event, "Jets").jetsByPt(jet_cut);
        _fatjets = apply<FastJets>(event, "FatJets").jetsByPt(fatjet_cut);
      }
      _neutrinos = apply<FinalState>(event, "Neutrinos").particlesByPt();
      _met = apply<MissingMomentum>(event, "MET").missingMomentum().p3();

      // check for leptonic decays of tag particles
      for (auto& jet : _jets) {
        for (auto& pt : jet.tags()) {
          for (auto& p : pt.children(isLepton)) {
            pt.addConstituent(p, false);
          }
        }
      }
      for (auto& jet : _fatjets) {
        for (auto& pt : jet.tags()) {
          for (auto& p : pt.children(isLepton)) {
            pt.addConstituent(p, false);
          }
        }
      }
    };

    // Do nothing here
    void finalize() override {}

    std::string status() const override { return "VALIDATED"; }
  };

}  // namespace Rivet

#endif