TopEgammaCPTools.h

Go to the documentation of this file.

/*
   Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
 */
 
#ifndef TOPCPTOOLS_TOPEGAMMACPTOOLS_H_
#define TOPCPTOOLS_TOPEGAMMACPTOOLS_H_
 
// Include what you use
#include <vector>
#include <string>
 
// Framework include(s):
#include "AsgTools/AsgTool.h"
#include "AsgTools/ToolHandle.h"
#include "AsgTools/ToolHandleArray.h"
#include "AsgTools/AnaToolHandle.h"
 
// Egamma include(s):
#include "EgammaAnalysisInterfaces/IEgammaCalibrationAndSmearingTool.h"
#include "EgammaAnalysisInterfaces/IAsgElectronEfficiencyCorrectionTool.h"
#include "EgammaAnalysisInterfaces/IElectronPhotonShowerShapeFudgeTool.h"
#include "EgammaAnalysisInterfaces/IAsgPhotonEfficiencyCorrectionTool.h"
#include "EgammaAnalysisInterfaces/IAsgPhotonIsEMSelector.h"
//#include "ElectronPhotonSelectorTools/AsgForwardElectronLikelihoodTool.h"
 
#include "TopConfiguration/Tokenize.h"
 
namespace top {
  class TopConfig;
 
  class EgammaCPTools final: public asg::AsgTool {
  public:
    explicit EgammaCPTools(const std::string& name);
    virtual ~EgammaCPTools() {}
 
    StatusCode initialize();
  private:
    std::shared_ptr<top::TopConfig> m_config;
 
    std::string m_egammaCalibrationModel;
    std::string m_electronEffTriggerFile;
    std::string m_electronEffTriggerLooseFile;
    std::string m_electronEffSFTriggerFile;
    std::string m_electronEffSFTriggerLooseFile;
    std::string m_electronEffSFRecoFile;
    std::string m_electronEffSFIDFile;
    std::string m_electronEffSFIDLooseFile;
    std::string m_electronEffSFIsoFile;
    std::string m_electronEffSFIsoLooseFile;
    std::string m_electronEffSFChargeIDFile;
    std::string m_electronEffSFChargeIDLooseFile;
    std::string m_electronEffSFChargeMisIDFile;
    std::string m_electronEffSFChargeMisIDLooseFile;
 
//    std::string m_fwdElectronEffSFIDFile;
//    std::string m_fwdElectronEffSFIDLooseFile;
 
    ToolHandle<CP::IEgammaCalibrationAndSmearingTool> m_egammaCalibrationAndSmearingTool;
 
    ToolHandle<IAsgElectronEfficiencyCorrectionTool> m_electronEffSFTrigger;
    ToolHandle<IAsgElectronEfficiencyCorrectionTool> m_electronEffSFTriggerLoose;
    ToolHandle<IAsgElectronEfficiencyCorrectionTool> m_electronEffTrigger;
    ToolHandle<IAsgElectronEfficiencyCorrectionTool> m_electronEffTriggerLoose;
    ToolHandle<IAsgElectronEfficiencyCorrectionTool> m_electronEffSFReco;
    ToolHandle<IAsgElectronEfficiencyCorrectionTool> m_electronEffSFID;
    ToolHandle<IAsgElectronEfficiencyCorrectionTool> m_electronEffSFIDLoose;
    ToolHandle<IAsgElectronEfficiencyCorrectionTool> m_electronEffSFIso;
    ToolHandle<IAsgElectronEfficiencyCorrectionTool> m_electronEffSFIsoLoose;
    ToolHandle<IAsgElectronEfficiencyCorrectionTool> m_electronEffSFChargeID;
    ToolHandle<IAsgElectronEfficiencyCorrectionTool> m_electronEffSFChargeIDLoose;
 
    ToolHandle<IAsgElectronEfficiencyCorrectionTool> m_electronEffSFTriggerCorrModel;
    ToolHandle<IAsgElectronEfficiencyCorrectionTool> m_electronEffSFTriggerLooseCorrModel;
    ToolHandle<IAsgElectronEfficiencyCorrectionTool> m_electronEffTriggerCorrModel;
    ToolHandle<IAsgElectronEfficiencyCorrectionTool> m_electronEffTriggerLooseCorrModel;
    ToolHandle<IAsgElectronEfficiencyCorrectionTool> m_electronEffSFRecoCorrModel;
    ToolHandle<IAsgElectronEfficiencyCorrectionTool> m_electronEffSFIDCorrModel;
    ToolHandle<IAsgElectronEfficiencyCorrectionTool> m_electronEffSFIDLooseCorrModel;
    ToolHandle<IAsgElectronEfficiencyCorrectionTool> m_electronEffSFIsoCorrModel;
    ToolHandle<IAsgElectronEfficiencyCorrectionTool> m_electronEffSFIsoLooseCorrModel;
 
    asg::AnaToolHandle<IElectronPhotonShowerShapeFudgeTool> m_photonFudgeTool;
    ToolHandle<IAsgPhotonEfficiencyCorrectionTool> m_photonEffSF;
    ToolHandleArray<IAsgPhotonEfficiencyCorrectionTool> m_photonIsoSFTools;
 
    ToolHandle<IAsgPhotonIsEMSelector> m_photonTightIsEMSelector;
    ToolHandle<IAsgPhotonIsEMSelector> m_photonMediumIsEMSelector;
    ToolHandle<IAsgPhotonIsEMSelector> m_photonLooseIsEMSelector;
 
//    ToolHandle<AsgForwardElectronLikelihoodTool> m_fwdElectronSelector;
//    ToolHandle<AsgForwardElectronLikelihoodTool> m_fwdElectronSelectorLoose;
//    ToolHandle<IAsgElectronEfficiencyCorrectionTool> m_fwdElectronEffSFID;
//    ToolHandle<IAsgElectronEfficiencyCorrectionTool> m_fwdElectronEffSFIDLoose;
 
    StatusCode setupSelectors();
    StatusCode setupCalibration();
    StatusCode setupScaleFactors();
 
    IAsgElectronEfficiencyCorrectionTool* setupElectronSFTool(const std::string& name,
                                  const std::vector<std::string>& file_list,
                                  const int& data_type,
                                  const std::string& correlation_model="TOTAL",
                                  const std::string& correlationModelEtaBinning="",
                                  const std::string& correlationModelEtBinning="");
 
 
 
 
    // Helper function to deal with path resolving the
    // egamma groups very long file names for SFs and efficiencies.
    std::string electronSFFilePath(const std::string& type, const std::string& ID, const std::string& ISO);
 
    // Helper for using maps from egamma groups
    IAsgElectronEfficiencyCorrectionTool* setupElectronSFToolWithMap(const std::string& name,
                                                                     const std::string& map_path,
                                                                     const std::string& reco_key,
                                                                     const std::string& ID_key,
                                                                     const std::string& iso_key,
                                                                     const std::string& trigger_key,
                                                                     const int& data_type,
                                                                     const std::string& correlationModel,
                                                                     const std::string& correlationModelEtaBinning,
                                                                     const std::string& correlationModelEtBinning);
 
 
 
 
    IAsgElectronEfficiencyCorrectionTool* setupElectronSFToolWithMap(const std::string& name,
                                                                     const std::string& map_path,
                                                                     const std::string& reco_key,
                                                                     const std::string& ID_key,
                                                                     const std::string& iso_key,
                                                                     const std::string& trigger_key,
                                                                     const int& data_type,
                                                                     const int& correlationModelNtoys,
                                                                     const int& correlationToySeed,  
                                                                     const std::string& correlationModel,
                                                                     const std::string& correlationModelEtaBinning,
                                                                     const std::string& correlationModelEtBinning);
 
 
 
    std::string electronSFMapFilePath(const std::string& type);
    void setCorrelationModelBinning(IAsgElectronEfficiencyCorrectionTool* tool, const std::string& binningName,
                                    const std::string& binning);
 
 
 
    void setCorrelationModelToys(IAsgElectronEfficiencyCorrectionTool* tool, const std::string& ToysName,
                                    const int& number);
 
 
 
    std::string mapWorkingPoints(const std::string& type);
  };
}  // namespace top
 
#endif  // TOPCPTOOLS_TOPEGAMMACPTOOLS_H_