DiTauBuilder.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#include "DiTauRec/DiTauBuilder.h"
#include "DiTauRec/DiTauToolBase.h"
#include "DiTauRec/DiTauCandidateData.h"
 
#include "xAODJet/JetContainer.h"
#include "xAODTau/DiTauJetContainer.h"
#include "xAODTau/DiTauJetAuxContainer.h"
 
#include "StoreGate/WriteHandle.h"
#include "StoreGate/ReadHandle.h"
 
 
DiTauBuilder::DiTauBuilder( const std::string& name, ISvcLocator* pSvcLocator ) : 
  AthReentrantAlgorithm( name, pSvcLocator ),
  m_minPt(10000),
  m_maxEta(2.5),
  m_Rjet(1.0),
  m_Rsubjet(0.2),
  m_Rcore(0.1),
  m_tools(this)
{
  declareProperty("minPt", m_minPt);
  declareProperty("maxEta", m_maxEta);
  declareProperty("Rjet", m_Rjet);
  declareProperty("Rsubjet", m_Rsubjet);
  declareProperty("Rcore", m_Rcore);
  declareProperty("Tools", m_tools ); 
 
}
 
 
DiTauBuilder::~DiTauBuilder() = default;
 
 
StatusCode DiTauBuilder::initialize() {
  ATH_MSG_INFO ("Initializing " << name() << "...");
 
  // no tools allocated
  if (m_tools.empty()) {
    ATH_MSG_ERROR("no tools given!");
    return StatusCode::FAILURE;
  }
 
  // list allocated tools 
  ATH_CHECK( m_tools.retrieve() );
  ToolHandleArray<DiTauToolBase> ::iterator itT = m_tools.begin();
  ToolHandleArray<DiTauToolBase> ::iterator itTE = m_tools.end();
  ATH_MSG_INFO("List of tools in execution sequence:");
  ATH_MSG_INFO("------------------------------------");
 
  unsigned int tool_count = 0;
 
  for (; itT != itTE; ++itT) {
    if (itT->retrieve().isFailure()) {
      ATH_MSG_WARNING("Cannot find tool named <" << *itT << ">");
    } else {
      ++tool_count;
      ATH_MSG_INFO((*itT)->type() << " - " << (*itT)->name());
    }
  }
  ATH_MSG_INFO(" ");
  ATH_MSG_INFO("------------------------------------");
 
  if (tool_count == 0) {
    ATH_MSG_ERROR("could not allocate any tool!");
    return StatusCode::FAILURE;
  }
 
  ATH_CHECK( m_diTauContainerName.initialize() );
  ATH_CHECK( m_seedJetName.initialize() );
 
  return StatusCode::SUCCESS;
}
 
 
StatusCode DiTauBuilder::finalize() {
  ATH_MSG_INFO ("Finalizing " << name() << "...");
 
  return StatusCode::SUCCESS;
}
 
 
StatusCode DiTauBuilder::execute(const EventContext& ctx) const {
  ATH_MSG_DEBUG ("Executing " << name() << "...");
    
  // preparing DiTau Candidate Container and storage in DiTauData
 
  DiTauCandidateData rDiTauData;
 
  auto pContainer = std::make_unique<xAOD::DiTauJetContainer>();
  auto pAuxContainer = std::make_unique<xAOD::DiTauJetAuxContainer>();
  pContainer->setStore(pAuxContainer.get());
 
  // set properties of DiTau Candidate 
  rDiTauData.xAODDiTau = nullptr;
  rDiTauData.xAODDiTauContainer = pContainer.get();
  rDiTauData.diTauAuxContainer = pAuxContainer.get();
  rDiTauData.seed = nullptr;
  rDiTauData.seedContainer = nullptr;
 
  rDiTauData.Rjet = m_Rjet;
  rDiTauData.Rsubjet = m_Rsubjet;
  rDiTauData.Rcore = m_Rcore;
 
  SG::WriteHandle<xAOD::DiTauJetContainer> diTauContainerH (m_diTauContainerName, ctx);
  ATH_CHECK( diTauContainerH.record (std::move (pContainer),
                     std::move (pAuxContainer)) );
 
  // retrieve di-tau seed jets and loop over seeds
 
  SG::ReadHandle<xAOD::JetContainer> pSeedContainer (m_seedJetName, ctx);
 
  rDiTauData.seedContainer = pSeedContainer.get();
 
  for (const auto* seed: *pSeedContainer) {
    ATH_MSG_DEBUG("Seed pt: "<< seed->pt() << "  eta: "<< seed->eta());
 
    // seed cuts
    if (std::abs(seed->pt()) < m_minPt) continue;
    if (std::abs(seed->eta()) > m_maxEta) continue;
 
    // cuts passed
    rDiTauData.seed = seed;
 
    // create new di-tau candidate
    rDiTauData.xAODDiTau = new xAOD::DiTauJet();
    rDiTauData.xAODDiTauContainer->push_back(rDiTauData.xAODDiTau);
 
    // handle di-tau candidate
    StatusCode sc = StatusCode::SUCCESS;
    for (const auto& tool: m_tools) {
      sc = tool->execute(&rDiTauData, ctx);
      if (sc.isFailure()) break;
    }
 
    if (sc.isSuccess()) {
      ATH_MSG_DEBUG("all tools executed successfully. Di-Tau candidate registered.");
    }
    else {
      ATH_MSG_DEBUG("seed failed a di-tau criterion. Thrown away.");
      rDiTauData.xAODDiTauContainer->pop_back();
    }
  }
 
  // // TODO: tool finalizers needed here
  // sc = StatusCode::SUCCESS;
 
  rDiTauData.xAODDiTau = nullptr;
 
  ATH_MSG_DEBUG("end execute()");
  return StatusCode::SUCCESS;
}