TrigGlobEffCorrExample4.cxx File Reference

#include <TFile.h>
#include <TError.h>
#include "AthAnalysisBaseComps/AthAnalysisHelper.h"
#include "POOLRootAccess/TEvent.h"
#include "AsgMessaging/MessageCheck.h"
#include "xAODEventInfo/EventInfo.h"
#include "xAODEgamma/ElectronContainer.h"
#include "PATCore/PATCoreEnums.h"
#include "AsgTools/AnaToolHandle.h"
#include "EgammaAnalysisInterfaces/IAsgElectronEfficiencyCorrectionTool.h"
#include "TrigGlobalEfficiencyCorrection/TrigGlobalEfficiencyCorrectionTool.h"
#include <random>

Go to the source code of this file.

Macros
#define	MSGSOURCE   "Example 4"

Functions
StatusCode	testElectrons (const char *APP_NAME, bool quiet)
StatusCode	testPhotons (const char *APP_NAME, bool quiet)
int	main (int argc, char *argv[])

Macro Definition Documentation

MSGSOURCE

#define MSGSOURCE   "Example 4"

Definition at line 40 of file TrigGlobEffCorrExample4.cxx.

Function Documentation

main()

int main	(	int	argc,
		char *	argv[] )

Definition at line 48 of file TrigGlobEffCorrExample4.cxx.

{
    bool quiet = false;
    ANA_CHECK_SET_TYPE(bool)
    const std::string flagQuiet("--quiet");
    for(int i=1;i<argc;++i)
    {
        if(argv[i] == flagQuiet) quiet = true;
    }
    const char* APP_NAME = argv[0];
    #ifdef XAOD_STANDALONE
        xAOD::Init(APP_NAME).ignore();
        StatusCode::enableFailure();
    #else
       IAppMgrUI* app = POOL::Init();
    #endif
    
    ANA_CHECK(testElectrons(APP_NAME, quiet));
    ANA_CHECK(testPhotons(APP_NAME, quiet));
    
    #ifndef XAOD_STANDALONE
        ANA_CHECK(app->finalize());
    #endif
    return 0;
}

testElectrons()

StatusCode testElectrons	(	const char *	APP_NAME,
		bool	quiet )

Retrieve the list of electron map keys for the chosen trigger combination

Then create all the needed electron tools and initialize the TrigGlob tool using the information returned by suggestElectronMapKeys()

Trigger efficiency/scale factor CP tools for electrons and muons

one tool instance for efficiencies, another for scale factors

Safer to retrieve the name from the final ToolHandle, it might be prefixed (by the parent tool name) when the handle is copied

Definition at line 75 of file TrigGlobEffCorrExample4.cxx.

{
    std::map<std::string,std::string> triggerCombination;
    triggerCombination["2015"] = "2e12_lhloose_L12EM10VH || e17_lhloose_mu14  || mu18_mu8noL1"
        "|| e24_lhmedium_L1EM20VH_OR_e60_lhmedium_OR_e120_lhloose || mu20_iloose_L1MU15";
    triggerCombination["2016"] = "2e17_lhvloose_nod0 || e17_lhloose_nod0_mu14  || mu22_mu8noL1 "
        "|| e26_lhtight_nod0_ivarloose_OR_e60_lhmedium_nod0_OR_e140_lhloose_nod0 || mu26_ivarmedium";
    triggerCombination["2017"] = "2e17_lhvloose_nod0_L12EM15VHI || 2e24_lhvloose_nod0 || e17_lhloose_nod0_mu14  || mu22_mu8noL1 "
        "|| e26_lhtight_nod0_ivarloose_OR_e60_lhmedium_nod0_OR_e140_lhloose_nod0 || mu26_ivarmedium";
    triggerCombination["2018"] = triggerCombination["2017"];
    std::map<std::string,std::string> legsPerKey, legsAlone;
    auto cc = TrigGlobalEfficiencyCorrectionTool::suggestElectronMapKeys(triggerCombination, "", legsAlone);
    if(cc == CP::CorrectionCode::Ok)
    {
        if(!quiet)
        {
            std::string msg = "List of trigger legs for this combination of triggers:\n";
            for(auto& kv : legsAlone)
            {
                msg += "   - " + kv.second + '\n';
            }
            Info(APP_NAME, "%s", msg.c_str());
        }
    }
    else
    {
        Error(APP_NAME, "Unable to find list of trigger legs!");
        return StatusCode::FAILURE;
    }
    
    cc = TrigGlobalEfficiencyCorrectionTool::suggestElectronMapKeys(triggerCombination, "2015_2018/rel21.2/Precision_Summer2020_v1", legsPerKey);
    if(cc == CP::CorrectionCode::Ok)
    {
        if(!quiet)
        {
            std::string msg = "List of map keys necessary for this combination of triggers:\n";
            for(auto& kv : legsPerKey)
            {
                msg += "   - tool with key \"" + kv.first + "\" chosen for legs " + kv.second + '\n';
            }
            Info(APP_NAME, "%s", msg.c_str());
        }
    }
    else
    {
        Error(APP_NAME, "Unable to find list of map keys!");
        return StatusCode::FAILURE;
    }
    

    ToolHandleArray<IAsgElectronEfficiencyCorrectionTool> electronEffTools, electronSFTools;
    std::vector<asg::AnaToolHandle<IAsgElectronEfficiencyCorrectionTool> > electronToolsFactory; // for RAII
    std::map<std::string,std::string> legsPerTool;
    int nTools = 0;
    for(auto& kv : legsPerKey)
    {
        const std::string& trigKey = kv.first;
        for(int j=0;j<2;++j) 
        {
            auto t = electronToolsFactory.emplace(electronToolsFactory.end(), "AsgElectronEfficiencyCorrectionTool/ElTrigEff_"+std::to_string(nTools++));
            t->setProperty("MapFilePath", "ElectronEfficiencyCorrection/2015_2018/rel21.2/Precision_Summer2020_v1/map4.txt").ignore();
            t->setProperty("TriggerKey", (j? trigKey : "Eff_"+trigKey)).ignore();
            t->setProperty("IdKey", "Tight").ignore();
            t->setProperty("IsoKey", "FCTight").ignore();
            t->setProperty("CorrelationModel", "TOTAL").ignore();
            t->setProperty("ForceDataType", (int)PATCore::ParticleDataType::Full).ignore();
            t->setProperty("OutputLevel", MSG::ERROR).ignore();
            ANA_CHECK(t->initialize());
            auto& handles = j? electronSFTools : electronEffTools;
            handles.push_back(t->getHandle());
            std::string name = handles[handles.size()-1].name();
            legsPerTool[name] = legsPerKey[trigKey];
        }
    }
    
    ToolHandleArray<CP::IMuonTriggerScaleFactors> muonTools;
    asg::AnaToolHandle<CP::IMuonTriggerScaleFactors> muonTool("CP::MuonTriggerScaleFactors/MuonTrigEff");
    muonTool.setProperty("MuonQuality", "Medium").ignore();
    muonTool.setProperty("OutputLevel", MSG::ERROR).ignore();
    if(muonTool.initialize() != StatusCode::SUCCESS)
    {
        Error(APP_NAME, "Unable to initialize the muon CP tool!");
        return StatusCode::FAILURE;
    }
    muonTools.push_back(muonTool.getHandle());
    
    asg::AnaToolHandle<ITrigGlobalEfficiencyCorrectionTool> myTool("TrigGlobalEfficiencyCorrectionTool/MyTool1");
    myTool.setProperty("ElectronEfficiencyTools", electronEffTools).ignore();
    myTool.setProperty("ElectronScaleFactorTools", electronSFTools).ignore();
    myTool.setProperty("MuonTools", muonTools).ignore();
    myTool.setProperty("TriggerCombination", triggerCombination).ignore();
    myTool.setProperty("ListOfLegsPerTool", legsPerTool).ignore();
    myTool.setProperty("OutputLevel", quiet? MSG::WARNING : MSG::INFO).ignore();
    ANA_CHECK(myTool.initialize());
    return StatusCode::SUCCESS;
}

testPhotons()

StatusCode testPhotons	(	const char *	APP_NAME,
		bool	quiet )

Retrieve the list of electron map keys for the chosen trigger combination

Then create all the needed photon tools and initialize the TrigGlob tool using the information returned by suggestPhotonMapKeys()

Trigger efficiency/scale factor CP tools for photons

one tool instance for efficiencies, another for scale factors

Definition at line 178 of file TrigGlobEffCorrExample4.cxx.

{
    std::map<std::string,std::string> triggerCombination;
    triggerCombination["2015"] = "g35_loose_g25_loose || 2g20_tight";
    triggerCombination["2016"] = " g35_loose_g25_loose || 2g22_tight";
    triggerCombination["2017"] = "g35_medium_g25_medium_L12EM20VH || 2g22_tight_L12EM15VHI";
    triggerCombination["2018"] = triggerCombination["2017"];
    std::map<std::string,std::string> legsPerKey;
    auto cc = TrigGlobalEfficiencyCorrectionTool::suggestPhotonMapKeys(triggerCombination, "2015_2018/rel21.2/Summer2020_Rec_v1", legsPerKey);
    if(cc == CP::CorrectionCode::Ok)
    {
        if(!quiet)
        {
            std::string msg = "List of map keys necessary for this combination of triggers:\n";
            for(auto& kv : legsPerKey)
            {
                msg += "   - tool with key \"" + kv.first + "\" chosen for legs " + kv.second + '\n';
            }
            Info(APP_NAME, "%s", msg.c_str());
        }
    }
    else
    {
        Error(APP_NAME, "Unable to find list of map keys!");
        return StatusCode::FAILURE;
    }
    

    ToolHandleArray<IAsgPhotonEfficiencyCorrectionTool> photonEffTools, photonSFTools;
    std::vector<asg::AnaToolHandle<IAsgPhotonEfficiencyCorrectionTool> > photonToolsFactory; // for RAII
    std::map<std::string,std::string> legsPerTool;
    int nTools = 0;
    for(auto& kv : legsPerKey)
    {
        const std::string& trigKey = kv.first;
        for(int j=0;j<2;++j) 
        {
            auto t = photonToolsFactory.emplace(photonToolsFactory.end(), "AsgPhotonEfficiencyCorrectionTool/PhTrigEff_"+std::to_string(nTools++));
            t->setProperty("MapFilePath", "PhotonEfficiencyCorrection/2015_2018/rel21.2/Summer2020_Rec_v1/map3.txt").ignore();
            t->setProperty("TriggerKey", (j? trigKey : "Eff_"+trigKey)).ignore();
            t->setProperty("IsoKey", "TightCaloOnly").ignore();
            t->setProperty("ForceDataType", (int)PATCore::ParticleDataType::Full).ignore();
            t->setProperty("OutputLevel", MSG::ERROR).ignore();
            ANA_CHECK(t->initialize());
            auto& handles = j? photonSFTools : photonEffTools;
            handles.push_back(t->getHandle());
            std::string name = handles[handles.size()-1].name();
            legsPerTool[name] = legsPerKey[trigKey];
        }
    }
    asg::AnaToolHandle<ITrigGlobalEfficiencyCorrectionTool> myTool("TrigGlobalEfficiencyCorrectionTool/MyTool2");
    myTool.setProperty("PhotonEfficiencyTools", photonEffTools).ignore();
    myTool.setProperty("PhotonScaleFactorTools", photonSFTools).ignore();
    myTool.setProperty("TriggerCombination", triggerCombination).ignore();
    myTool.setProperty("ListOfLegsPerTool", legsPerTool).ignore();
    myTool.setProperty("OutputLevel", quiet? MSG::WARNING : MSG::INFO).ignore();
    ANA_CHECK(myTool.initialize());
    return StatusCode::SUCCESS;
}