JetPruner.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
// JetPruner.cxx
 
#include "JetRec/JetPruner.h"
#include <iomanip>
#include "fastjet/PseudoJet.hh"
#include "fastjet/tools/Pruner.hh"
#include "JetEDM/PseudoJetVector.h"
#include "JetEDM/FastJetUtils.h"
#include "JetRec/JetDumper.h"
 
using std::setw;
using fastjet::PseudoJet;
using xAOD::JetContainer;
 
//**********************************************************************
 
JetPruner::JetPruner(const std::string& name)
  : AsgTool(name), m_bld("",this) {
  declareProperty("JetAlgorithm", m_jetalg ="CamKt");
  declareProperty("RCut", m_rcut =0.3);
  declareProperty("ZCut", m_zcut =0.10);
  declareProperty("JetBuilder", m_bld);
}
 
//**********************************************************************
 
JetPruner::~JetPruner() = default;
 
//**********************************************************************
 
StatusCode JetPruner::initialize() {
  xAOD::JetAlgorithmType::ID ialg = xAOD::JetAlgorithmType::algId(m_jetalg);
  m_fjalg = xAOD::JetAlgorithmType::fastJetDef(ialg);
  if ( m_fjalg == fastjet::undefined_jet_algorithm ) {
    ATH_MSG_ERROR("Invalid jet algorithm name: " << m_jetalg);
    ATH_MSG_ERROR("Allowed values are Kt, CamKt, AntiKt, etc.");
    return StatusCode::FAILURE;
  }
  if ( m_rcut < 0.0 || m_rcut > 10.0 ) {
    ATH_MSG_ERROR("Invalid value for RCut " << m_rcut);
    return StatusCode::FAILURE;
  }
  if ( m_zcut < 0.0 || m_zcut > 1.0 ) {
    ATH_MSG_ERROR("Invalid value for ZCut " << m_zcut);
    return StatusCode::FAILURE;
  }
  if ( m_bld.empty() ) {
    ATH_MSG_ERROR("Unable to retrieve jet builder.");
    return StatusCode::FAILURE;
  }
  return StatusCode::SUCCESS;
}
 
//**********************************************************************
 
int JetPruner::groom(const xAOD::Jet& jin,
                     const PseudoJetContainer& pjContainer,
                     xAOD::JetContainer& jets) const {
  if ( pseudojetRetriever() == nullptr ) {
    ATH_MSG_WARNING("Pseudojet retriever is null.");
    return 1;
  }
  const PseudoJet* ppjin = pseudojetRetriever()->pseudojet(jin);
  if ( ppjin == nullptr ) {
    ATH_MSG_WARNING("Jet does not have a pseudojet.");
    return 1;
  }
  // Prune.
  fastjet::Pruner pruner(m_fjalg, m_zcut, m_rcut);
  PseudoJet pjprun = pruner(*ppjin);
  ATH_MSG_VERBOSE("   Input cluster sequence: " << ppjin->associated_cluster_sequence());
  ATH_MSG_VERBOSE(" Pruned cluster sequence: " << pjprun.associated_cluster_sequence());
  // Add jet to collection.
  xAOD::Jet* pjet = m_bld->add(pjprun, pjContainer, jets, &jin);
  pjet->setAttribute<float>("RCut", m_rcut);
  pjet->setAttribute<float>("ZCut", m_zcut);
  pjet->setAttribute<int>("TransformType", xAOD::JetTransform::Prune);
  ATH_MSG_DEBUG("Properties after pruning:");
  ATH_MSG_DEBUG("   ncon: " << pjprun.constituents().size() << "/"
                            << ppjin->constituents().size());
  ATH_MSG_DEBUG("   nsub: " << pjprun.pieces().size());
  return 0;
}
 
//**********************************************************************
 
void JetPruner::print() const {
  ATH_MSG_INFO("  Jet algorithm: " << m_jetalg);
  ATH_MSG_INFO("   R cut factor: " << m_rcut);
  ATH_MSG_INFO("          Z cut: " << m_zcut);
  ATH_MSG_INFO("  Jet builder: " << m_bld.name());
}
 
//**********************************************************************