JetFtagEffPlots.h

Go to the documentation of this file.

/*
   Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
 */
 
#ifndef TOPEVENTSELECTIONTOOLS_JETFTAGEFFPLOTS_H_
#define TOPEVENTSELECTIONTOOLS_JETFTAGEFFPLOTS_H_
 
#include <string>
 
#include "TopEventSelectionTools/EventSelectorBase.h"
#include "TopEventSelectionTools/PlotManager.h"
#include "PMGTools/PMGTruthWeightTool.h"
#include <AsgTools/AnaToolHandle.h>
#include "FTagAnalysisInterfaces/IBTaggingSelectionTool.h"
#include "TopCorrections/ScaleFactorRetriever.h"
#include "PMGTools/PMGTruthWeightTool.h"
 
 
class TFile;
 
namespace EL {
  class Worker;
}
 
namespace {
  enum class WEIGHTTYPE {
    MURUP, MURDOWN, MUFUP, MUFDOWN, VAR3CUP, VAR3CDOWN, FSRUP, FSRDOWN
  };
}
 
namespace top {
  class TopConfig;
 
  class JetFtagEffPlots: public EventSelectorBase {
  public:
    JetFtagEffPlots(const std::string& name, TFile* outputFile, const std::string& params,
                    std::shared_ptr<top::TopConfig> config,
                    EL::Worker* wk = nullptr);
 
 
    bool apply(const top::Event& event) const override;
 
 
    std::string name() const override;
  private:
    // File units are MeV and normally people like plots in GeV.
    static const double toGeV;
 
    // Easy access to histograms.
    std::shared_ptr<PlotManager> m_hists = nullptr;
    std::shared_ptr<PlotManager> m_hists_muRup = nullptr;
    std::shared_ptr<PlotManager> m_hists_muRdown = nullptr;
    std::shared_ptr<PlotManager> m_hists_muFup = nullptr;
    std::shared_ptr<PlotManager> m_hists_muFdown = nullptr;
    std::shared_ptr<PlotManager> m_hists_Var3cup = nullptr;
    std::shared_ptr<PlotManager> m_hists_Var3cdown = nullptr;
    std::shared_ptr<PlotManager> m_hists_FSRup = nullptr;
    std::shared_ptr<PlotManager> m_hists_FSRdown = nullptr;
 
 
    // Nominal hash value
    std::size_t m_nominalHashValue;
 
    std::string m_CDIfile;
 
    bool m_fill_total_hists;
 
    //optional suffix you can add to your histogram
    std::string m_histogram_suffix;
    bool m_dont_use_event_weight;
    bool m_use_track_jets;
    bool m_doNominal;
    bool m_doMuRup;
    bool m_doMuRdown;
    bool m_doMuFup;
    bool m_doMuFdown;
    bool m_doVar3cup;
    bool m_doVar3cdown;
    bool m_doFSRup;
    bool m_doFSRdown;
 
    // pT and eta bin edges
    std::string m_ptBins;
    std::string m_etaBins;
 
    float m_max_pT;
    float m_min_pT;
    int m_N_pT_bins;
 
    float m_max_Eta;
    float m_min_Eta;
    int m_N_Eta_bins;
 
    // name of the jet collection - this is needed for the names of the histograms
    std::string m_jetCollection;
    std::string m_WP;
    std::string m_tagger;
    // shared pointed to instance of TopConfig
    std::shared_ptr<top::TopConfig> m_config;
 
 
    asg::AnaToolHandle<IBTaggingSelectionTool> m_selection_tool;
 
    PMGTools::PMGTruthWeightTool* m_PMGTruthWeights;
 
    top::ScaleFactorRetriever* m_sfRetriever;
 
    // function to translate the binnings into vector of bin edges
    void formatBinning(const std::string& str, std::vector<double>& binEdges);
 
    //helper function to fill histograms
    void FillHistograms(std::shared_ptr<PlotManager> h_ptr, double w_event, const top::Event& event) const;
 
    float GetPMGTruthWeight(const xAOD::EventInfo* evtInfo, WEIGHTTYPE type) const;
  };
}  // namespace top
 
#endif  // TOPEVENTSELECTIONTOOLS_JETFTAGEFFPLOTS_H_