ParticleLevelJetsAlg.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
 
#include "TruthParticleLevelAnalysisAlgorithms/ParticleLevelJetsAlg.h"
 
namespace CP {
 
StatusCode ParticleLevelJetsAlg::initialize() {
 
  ANA_CHECK(m_jetsKey.initialize());
  ANA_CHECK(m_eventInfoKey.initialize());
 
  return StatusCode::SUCCESS;
}
 
StatusCode ParticleLevelJetsAlg::execute() {
 
  SG::ReadHandle<xAOD::JetContainer> jets(m_jetsKey);
  SG::ReadHandle<xAOD::EventInfo> eventInfo(m_eventInfoKey);
 
  // accessors
  static const SG::AuxElement::ConstAccessor<int> acc_flav(
      "HadronConeExclTruthLabelID");
 
  // decorators
  static const SG::AuxElement::Decorator<int> dec_nBJets(
      "num_truth_bjets_nocuts");
  static const SG::AuxElement::Decorator<int> dec_nCJets(
      "num_truth_cjets_nocuts");
 
  // the number of b- and c-jets without any event cuts applied
  int num_bjets(0), num_cjets(0);
 
  for (const auto* jet : *jets) {
 
    // check the flavour label of the jet
    if (acc_flav.isAvailable(*jet)) {
      int flavourLabel = acc_flav(*jet);
      if (flavourLabel == 5)
        num_bjets++;
      if (flavourLabel == 4)
        num_cjets++;
    } else {
      ANA_MSG_WARNING(
          "Truth jet is missing the decoration: HadronConeExclTruthLabelID.");
      num_bjets = num_cjets = -999;
    }
  }
 
  // decorate the EventInfo with the number of b- and c-jets
  dec_nBJets(*eventInfo) = num_bjets;
  dec_nCJets(*eventInfo) = num_cjets;
 
  return StatusCode::SUCCESS;
}
 
}  // namespace CP