testElectronPhotonVariableCorrectionTool.cxx

Go to the documentation of this file.

/*
    Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/

 
// ROOT includes
#include "TFile.h"
#include "TString.h"
 
//EDM includes
#include "xAODEgamma/ElectronContainer.h" 
#include "xAODEgamma/PhotonContainer.h" 
#include "xAODEgamma/PhotonAuxContainer.h"
#include "xAODEgamma/Electron.h" 
#include "xAODEgamma/Photon.h"
 
#include "AsgTools/StandaloneToolHandle.h"
#include "EgammaAnalysisInterfaces/IElectronPhotonShowerShapeFudgeTool.h"
 
#include "AsgMessaging/AsgMessaging.h"

#ifdef ROOTCORE
#include "xAODRootAccess/TEvent.h"
#include "xAODRootAccess/TStore.h"
#endif //ROOTCORE
 
// main test code
 
int main(int argc, char* argv[])
{
    using namespace asg::msgUserCode;
 
    // The application's name:
    const char* APP_NAME = argv[0];
 
    // Check if we received a file name:
    if (argc < 2)
    {
        ANA_MSG_ERROR("No file name received!" );
        ANA_MSG_ERROR( "  Usage: %s [xAOD file name]  %d Num of events to process %d (0 photons , 1 electrons)");
        return EXIT_FAILURE;
    }
 
    //open input file
    const TString fileName = argv[1];
    ANA_MSG_INFO(APP_NAME << " Opening file: " << fileName.Data());
    std::unique_ptr<TFile> inputFile(TFile::Open(fileName, "READ")); //probably rather "READ"?
    ANA_CHECK(inputFile.get());
 
    // Create a TEvent object (persistent store)
    xAOD::TEvent pers(xAOD::TEvent::kClassAccess);
    // Create a TStore object (transient store)
    xAOD::TStore trans;
    ANA_CHECK(pers.readFrom(inputFile.get()));
    //
    ANA_MSG_INFO("Number of events in the file: " << pers.getEntries());
 
    // Decide how many events to run over:
    Long64_t entries = pers.getEntries();
    if (argc > 2)
    {
        const Long64_t userInputEntries = atoll(argv[2]);
        if (userInputEntries < entries)
        {
            entries = userInputEntries;
        }
    }
    ANA_MSG_INFO("Running over " << entries << " events.");
 
    // initialize the tool
    asg::StandaloneToolHandle<IElectronPhotonShowerShapeFudgeTool> myTool("ElectronPhotonVariableCorrectionTool/myTestTool");
    std::string configFile = "EGammaVariableCorrection/EGammaVariableCorrectionTool_ExampleConf.conf";
    ANA_CHECK(myTool.setProperty("ConfigFile", configFile));
    ANA_CHECK(myTool.initialize());
 
    // loop over the events
    for (Long64_t entry = 0; entry < entries; entry++)
    {
        //get entry
        pers.getEntry(entry);
 
        // ====================================
        // Photons
        // ====================================
 
        //get photon container
        const xAOD::PhotonContainer* photons;
        ANA_CHECK(pers.retrieve(photons, "Photons"));
 
        // Make a deep copy of the photon container
        // Create the new container and its auxiliary store.
        auto photons_copy = std::make_unique<xAOD::PhotonContainer>();
        auto photons_copy_aux = std::make_unique<xAOD::AuxContainerBase>();
        photons_copy->setStore (photons_copy_aux.get()); //< Connect the two
        
        //copy photons over
        for (auto photon : *photons) {
            // Copy this photon to the output container:
            xAOD::Photon* photon_copy = new xAOD::Photon();
            photons_copy->push_back (photon_copy); // photon acquires the photon_copy auxstore
            *photon_copy = *photon; // copies auxdata from one auxstore to the other
        }
        // end make deep copy
 
        //loop over deep copy of photon container
        for (unsigned int photon_itr = 0; photon_itr < photons_copy->size(); photon_itr++)
        {
            xAOD::Photon* photon = photons_copy->at(photon_itr);
 
            //apply correction
            ANA_CHECK(myTool->applyCorrection(*photon));
 
        } // loop over deep copy of photon container
 
        // ====================================
        // Electrons
        // ====================================
 
        //get electron container
        const xAOD::ElectronContainer* electrons;
        ANA_CHECK(pers.retrieve(electrons, "Electrons"));
 
        // Make a deep copy of the electron container
        // Create the new container and its auxiliary store.
        auto electrons_copy = std::make_unique<xAOD::ElectronContainer>();
        auto electrons_copy_aux = std::make_unique<xAOD::AuxContainerBase>();
        electrons_copy->setStore (electrons_copy_aux.get()); //< Connect the two
        
        //copy electrons over
        for (auto electron : *electrons) {
            // Copy this electron to the output container:
            xAOD::Electron* electron_copy = new xAOD::Electron();
            electrons_copy->push_back (electron_copy); // electron acquires the electron_copy auxstore
            *electron_copy = *electron; // copies auxdata from one auxstore to the other
        }
        // end make deep copy
 
        //loop over deep copy of electron container
        for (unsigned int electron_itr = 0; electron_itr < electrons_copy->size(); electron_itr++)
        {
            xAOD::Electron* electron = electrons_copy->at(electron_itr);
 
            //apply correction
            ANA_CHECK(myTool->applyCorrection(*electron));
 
        } // loop over deep copy of electrone container
 
    } // loop over events
 
    return 0;
}