JetSoftDrop.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
// JetSoftDrop.cxx
 
#include "JetRec/JetSoftDrop.h"
#include <iomanip>
#include "fastjet/PseudoJet.hh"
#include "fastjet/JetDefinition.hh"
#include "fastjet/Selector.hh"
#include "fastjet/tools/Filter.hh"
#include "JetEDM/PseudoJetVector.h"
 
 
using std::setw;
using fastjet::PseudoJet;
using xAOD::JetContainer;
 
//**********************************************************************
 
JetSoftDrop::JetSoftDrop(const std::string& name)
  : AsgTool(name), m_bld("",this) {
  //ZCut = 0.1, Beta = 0 are "CMS-like" parameters for SoftDrop: 
  //https://indico.cern.ch/event/439039/contributions/2223279/attachments/1311773/1963161/BOOST16_toptagging_CMS.pdf
  declareProperty("ZCut", m_zcut =0.1);
  declareProperty("Beta", m_beta =0.0);
  declareProperty("R0",   m_R0   =1.0);
  declareProperty("JetBuilder", m_bld);
}
 
//**********************************************************************
 
JetSoftDrop::~JetSoftDrop() = default;
 
//**********************************************************************
 
StatusCode JetSoftDrop::initialize() {
  if ( m_zcut < 0.0 || m_zcut > 10.0 ) {
    ATH_MSG_ERROR("Invalid value for ZCut " << m_zcut);
    return StatusCode::FAILURE;
  }
  if ( m_beta < 0.0 || m_beta > 10.0 ) {
    ATH_MSG_ERROR("Invalid value for Beta " << m_beta);
    return StatusCode::FAILURE;
  }
  if ( m_R0 < 0.0 || m_R0 > 10.0 ) {
    ATH_MSG_ERROR("Invalid value for R0 " << m_R0);
    return StatusCode::FAILURE;
  }
  if ( m_bld.empty() ) {
    ATH_MSG_ERROR("Unable to retrieve jet builder.");
    return StatusCode::FAILURE;
  }
  return StatusCode::SUCCESS;
}
 
//**********************************************************************
 
int JetSoftDrop::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;
  }
 
  //configure soft drop tool
  //http://fastjet.hepforge.org/svn/contrib/contribs/RecursiveTools/tags/1.0.0/SoftDrop.hh
  fastjet::contrib::SoftDrop softdropper(m_beta, m_zcut, m_R0);
  PseudoJet pjsoftdrop = softdropper(*ppjin);
  int npsoftdrop = pjsoftdrop.pieces().size();
  xAOD::Jet* pjet = m_bld->add(pjsoftdrop, pjContainer, jets, &jin);
  if ( pjet == nullptr ) {
    ATH_MSG_ERROR("Unable to add jet to container");
    return 1;
  }
  //pjet->SetAttribute<int>("TransformType", xAOD::JetTransform::SoftDrop); //xAOD::JetTransform::SoftDrop probably doesn't exist yet
  pjet->setAttribute<float>("ZCut", m_zcut);
  pjet->setAttribute<float>("SoftDropBeta", m_beta);
  pjet->setAttribute<float>("SoftDropR0", m_R0);
  pjet->setAttribute<int>("NSoftDropSubjets", npsoftdrop);
 
  ATH_MSG_DEBUG("Properties after softdrop:");
  ATH_MSG_DEBUG("   ncon: " << pjsoftdrop.constituents().size() << "/"
                            << ppjin->constituents().size());
  ATH_MSG_DEBUG("   nsub: " << npsoftdrop);
  ATH_MSG_DEBUG("Added jet to container.");
  
  return 0;
}
 
//**********************************************************************
 
void JetSoftDrop::print() const {
  ATH_MSG_INFO("  Asymmetry measure min: " << m_zcut);
  ATH_MSG_INFO("  Angular exponent: " << m_beta);
  ATH_MSG_INFO("  Characteristic jet radius: " << m_R0);
  ATH_MSG_INFO("  Jet builder: " << m_bld.name());
}
 
//**********************************************************************