testEgChargeCorr.cxx

Go to the documentation of this file.

/*
   Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
 */
 
// System include(s):
#include <cstdlib>
#include <memory>
#include <string>
 
// ROOT include(s):
#include <TError.h>
#include <TFile.h>
#include <TString.h>
 
// Infrastructure include(s):
#ifdef XAOD_STANDALONE
#include "xAODRootAccess/Init.h"
#include "xAODRootAccess/TEvent.h"
#endif // XAOD_STANDALONE
 
// EDM include(s):
#include "ElectronEfficiencyCorrection/ElectronChargeEfficiencyCorrectionTool.h"
#include "ElectronEfficiencyCorrection/ElectronEfficiencyHelpers.h"
#include "xAODCore/ShallowCopy.h"
#include "xAODEgamma/Egamma.h"
#include "xAODEgamma/ElectronContainer.h"
#include "xAODEventInfo/EventInfo.h"
 
// Asg includes
#include "AsgMessaging/AsgMessaging.h"
#include "PATCore/AcceptData.h"
#include "PATInterfaces/SystematicsUtil.h"
 
#include <ElectronPhotonSelectorTools/AsgElectronLikelihoodTool.h>
 
#include "AsgMessaging/MessageCheck.h"
#include "AsgMessaging/MsgStream.h"
 
 
namespace asg{
  ANA_MSG_HEADER (testEgChargeCorr)
  ANA_MSG_SOURCE (testEgChargeCorr, "") 
}
 
 
int
main(int argc, char* argv[])
{
  const char* APP_NAME = argv[0];
  using namespace asg::testEgChargeCorr;
  ANA_CHECK_SET_TYPE (int);
  MSG::Level mylevel=MSG::INFO;
  setMsgLevel(mylevel);
  msg().setName(APP_NAME);
 
  // Check if we received a file name:
  if (argc < 3) {
    Error(APP_NAME, "No file name received!");
    Error(APP_NAME,
          "  Usage: %s [xAOD file name] [Correction File name]",
          APP_NAME);
    return 1;
  }
 
  double SF_chargeID = 0;
  int SF_nevents = 0;
 
  double n_chargeID = 0;
  double n_chargeMisID = 0;
 
  // Initialise the application:
  ANA_CHECK(xAOD::Init(APP_NAME));
 
  // Open the input file:
  const TString fileName = argv[1];
  const TString corrFileName = argv[2];
 
  Info(APP_NAME, "Opening file: %s", fileName.Data());
  std::unique_ptr<TFile> ifile(TFile::Open(fileName, "READ"));
  ANA_CHECK(ifile.get());
 
  // Create a TEvent object:
  // xAOD::TEvent event( xAOD::TEvent::kBranchAccess );
  xAOD::TEvent event(xAOD::TEvent::kClassAccess);
  ANA_CHECK(event.readFrom(ifile.get()));
  Info(APP_NAME,
       "Number of events in the file: %i",
       static_cast<int>(event.getEntries()));
 
  std::cout << "==" << std::endl;
 
  // Decide how many events to run over:
  Long64_t entries = event.getEntries();
  if (argc > 2) {
    const Long64_t e = atoll(argv[2]);
    if (e < entries) {
      entries = e;
    }
  }
 
  // Likelihood
  CP::ElectronChargeEfficiencyCorrectionTool myEgCorrections("myEgCorrections");
  ANA_CHECK(myEgCorrections.setProperty("OutputLevel", mylevel));
 
  std::string inputFile =
    corrFileName.Data(); //"data/ChMisIDSF_TightLL_FixedCutTight.root" ;
  ANA_CHECK(myEgCorrections.setProperty("CorrectionFileName", inputFile));
  ANA_CHECK(myEgCorrections.setProperty("ForceDataType", 1));
 
  ANA_CHECK(myEgCorrections.setProperty("DefaultRandomRunNumber",
                                    (unsigned int)311481));
  ANA_CHECK(myEgCorrections.setProperty("UseRandomRunNumber", false));
  ANA_CHECK(myEgCorrections.initialize());
 
  AsgElectronLikelihoodTool* m_LHToolTight =
    new AsgElectronLikelihoodTool("m_LHToolTight");
  ANA_CHECK(
    m_LHToolTight->setProperty("primaryVertexContainer", "PrimaryVertices"));
  m_LHToolTight
    ->setProperty("ConfigFile",
                  "ElectronPhotonSelectorTools/offline/mc15_20160512/"
                  "ElectronLikelihoodLooseOfflineConfig2016_CutBL_Smooth.conf")
    .ignore();
  ANA_CHECK(m_LHToolTight->initialize());
 
  // Get a list of systematics
  CP::SystematicSet recSysts = myEgCorrections.recommendedSystematics();
  // Convert into a simple list
  // std::vector<CP::SystematicSet> sysList =
  // CP::make_systematics_vector(recSysts);
 
  std::cout << "==" << std::endl;
 
  // Loop over the events:
  entries = 600;
 
  Long64_t entry = 0;
  for (; entry < entries; ++entry) {
 
    // Tell the object which entry to look at:
    event.getEntry(entry);
 
    if (entry < 10)
      std::cout << "=================NEXT EVENT=========================="
                << std::endl;
    Info(APP_NAME, "Electron");
 
    const xAOD::EventInfo* event_info = nullptr;
    ANA_CHECK(event.retrieve(event_info, "EventInfo"));
 
    const xAOD::ElectronContainer* electrons = nullptr;
    ANA_CHECK(event.retrieve(electrons, "Electrons"));
 
    // Create shallow copy for this systematic
    std::pair<xAOD::ElectronContainer*, xAOD::ShallowAuxContainer*>
      electrons_shallowCopy = xAOD::shallowCopyContainer(*electrons);
 
    // Iterate over the shallow copy
    xAOD::ElectronContainer* elsCorr = electrons_shallowCopy.first;
    xAOD::ElectronContainer::iterator el_it = elsCorr->begin();
    xAOD::ElectronContainer::iterator el_it_last = elsCorr->end();
 
    static const SG::ConstAccessor<float> sfAcc("chargeIDEffiSF");
 
    unsigned int i = 0;
    double SF = 0;
    for (; el_it != el_it_last; ++el_it, ++i) {
 
      xAOD::Electron* el = *el_it;
      if (el->pt() < 20000)
        continue; // skip electrons outside of recommendations
 
      bool LHacc{ m_LHToolTight->accept(el) };
      std::cout << "acc:  " << LHacc << std::endl;
      if (!m_LHToolTight->accept(el))
        continue;
      double clusAbsEta = std::abs(el->caloCluster()->etaBE(2));
      if ( clusAbsEta > 1.37 && (clusAbsEta < 1.52 || clusAbsEta > 2.47) )
        continue; // skip electrons outside of recommendations (in the crack we fill 1 into the SFs)
 
      SF_nevents++;
 
      Info(APP_NAME, "Electron #%d", i);
 
      if (myEgCorrections.getEfficiencyScaleFactor(*el, SF) !=
          CP::CorrectionCode::Ok) {
        Error(APP_NAME, "Problem in getEfficiencyScaleFactor");
        return EXIT_FAILURE;
      }
 
      if (myEgCorrections.applyEfficiencyScaleFactor(*el) !=
          CP::CorrectionCode::Ok) {
        Error(APP_NAME, "Problem in applyEfficiencyScaleFactor");
        return EXIT_FAILURE;
      }
 
      Info( APP_NAME, "===>>> Resulting SF (from get function) %f, (from apply function) %f",
        SF, sfAcc(*el));
 
      SF_chargeID = SF_chargeID + SF;
 
      for (const auto& sys : recSysts) { // recSysts){
        double systematic = 0;
 
        // Configure the tool for this systematic
        ANA_CHECK(myEgCorrections.applySystematicVariation({ sys }));
        //
        if (myEgCorrections.getEfficiencyScaleFactor(*el, systematic) ==
            CP::CorrectionCode::Ok) {
          Info(APP_NAME,
               "%f Result %f Systematic value %f ",
               SF,
               systematic,
               systematic - SF);
        }
      }
      ANA_CHECK(myEgCorrections.applySystematicVariation({})); // Empty variation is the nominal
 
 
      int truthcharge = false;
      ANA_CHECK(ElectronEfficiencyHelpers::getEleTruthCharge(*el,truthcharge));
      if (el->charge() * truthcharge < 0) {
        n_chargeMisID+=SF;
      } else
        n_chargeID+=SF;
    }
 
    // }
  }
  Info(APP_NAME, "===>>>  done processing event #%lld ", entry);
  Info(APP_NAME, "===>>>  processed #%d electrons", SF_nevents);
  Info(APP_NAME, "===>>>  compared to #%f (from Charge MisId SF)", SF_chargeID);
  Info(APP_NAME, "===>>>  compared to #%f and #%f ", n_chargeID, n_chargeMisID);
 
  // Return gracefully:
  return 0;
}