ElectronEfficiencyHelpers.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#include "ElectronEfficiencyCorrection/ElectronEfficiencyHelpers.h"
#include "AsgMessaging/AsgMessaging.h"
#include "AthContainers/ConstAccessor.h"

// Get type and charge of the truth electron
// made them static so they can be used by analyses to do truth matches
 
CP::CorrectionCode ElectronEfficiencyHelpers::getEleTruthCharge( const xAOD::Electron& ele,  int& truthcharge )
{
 
  // Define an AsgMessaging instance
  static const asg::AsgMessaging msg("ElectronEfficiencyHelpers");
 
  static const SG::ConstAccessor<int> firstEgMotherPdgIdAcc("firstEgMotherPdgId");
  if ( !(firstEgMotherPdgIdAcc.isAvailable(ele)) ) {
    msg.msg(MSG::ERROR) << "Link not available for firstEgMotherPdgId...BAD!!!" << endmsg;
    msg.msg(MSG::ERROR) << "Need to have present: ( !(ele.isAvailable<int>('firstEgMotherPdgId')) )" << endmsg;
    return CP::CorrectionCode::OutOfValidityRange;
  }
 
  truthcharge = (-1)*firstEgMotherPdgIdAcc(ele);
  // Make truthcharge -1, 0, +1
  truthcharge = (0 < truthcharge) - (truthcharge < 0);
 
  return CP::CorrectionCode::Ok;
 
}
 
 
CP::CorrectionCode ElectronEfficiencyHelpers::isGoodEle( const xAOD::Electron& ele, bool& goodEle)
{
 
  // good ele => (firstEgMotherPdgId) == 11 
  // this is the case for type 2,3 and 4 electrons, so is not a real truth match!!!
  // but we believe that it doesn't hurt to apply scale factors for all these cases
  goodEle = false;
  int firstEgPdgId = -9999;
 
  // Define an AsgMessaging instance
  static const asg::AsgMessaging msg("ElectronEfficiencyHelpers");
 
  static const SG::ConstAccessor<int> firstEgMotherPdgIdAcc("firstEgMotherPdgId");
  if ( !(firstEgMotherPdgIdAcc.isAvailable(ele)) ) {
    msg.msg(MSG::ERROR) << "firstEgMotherPdgId IS NOT AVAILABLE!!" << endmsg;
    return CP::CorrectionCode::OutOfValidityRange;
  }
  else {
 
    firstEgPdgId = firstEgMotherPdgIdAcc(ele);
 
    if ( std::abs(firstEgPdgId) != 11) {
 
      goodEle = false;
      return CP::CorrectionCode::Ok;
 
    }
    else {
 
      goodEle = true;
      return CP::CorrectionCode::Ok;
 
    }
  }
 
  return CP::CorrectionCode::OutOfValidityRange;
}