xAODMuonRetriever.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#include "xAODJiveXML/xAODMuonRetriever.h"
#include "xAODMuon/MuonContainer.h"
#include "AthenaKernel/Units.h"
 
using Athena::Units::GeV;
 
namespace JiveXML {

  xAODMuonRetriever::xAODMuonRetriever(const std::string& type,const std::string& name,const IInterface* parent):
    AthAlgTool(type,name,parent){}
 
 
  StatusCode xAODMuonRetriever::initialize(){
    ATH_CHECK(m_keys.initialize());
    return StatusCode::SUCCESS;
  }
 

  StatusCode xAODMuonRetriever::retrieve(ToolHandle<IFormatTool> &FormatTool) {
 
    ATH_MSG_DEBUG("In retrieve()");
 
    // Loop through the keys and retrieve the corresponding data
    for (const auto& key : m_keys) {
      SG::ReadHandle<xAOD::MuonContainer> cont(key);
      if (cont.isValid()) {
    DataMap data = getData(&(*cont));
    if (FormatTool->AddToEvent(dataTypeName(), key.key() + "_xAOD", &data).isFailure()) {
      ATH_MSG_WARNING("Failed to add collection " << key.key());
    } else {
      ATH_MSG_DEBUG(" (" << key.key() << ") retrieved");
    }
      } else {
    ATH_MSG_WARNING("Collection " << key.key() << " not found in SG");
      }
    }
    return StatusCode::SUCCESS;
  }
 

  const DataMap xAODMuonRetriever::getData(const xAOD::MuonContainer* muCont) {
 
    ATH_MSG_DEBUG("in getData()");
 
    DataMap DataMap;
 
    DataVect pt; pt.reserve(muCont->size());
    DataVect phi; phi.reserve(muCont->size());
    DataVect eta; eta.reserve(muCont->size());
    DataVect mass; mass.reserve(muCont->size());
    DataVect energy; energy.reserve(muCont->size());
    DataVect chi2; chi2.reserve(muCont->size());
    DataVect pdgId; pdgId.reserve(muCont->size());
 
    xAOD::MuonContainer::const_iterator muItr  = muCont->begin();
    xAOD::MuonContainer::const_iterator muItrE = muCont->end();
 
    int counter = 0;
 
    for (; muItr != muItrE; ++muItr) {
 
      ATH_MSG_DEBUG("  Muon #" << counter++ << " : eta = "  << (*muItr)->eta()
            << ", phi = "  << (*muItr)->phi() << ", pt = " <<  (*muItr)->pt()
            << ", pdgId = " << -13.*(*muItr)->trackParticle(xAOD::Muon::TrackParticleType::Primary)->charge());
 
      phi.emplace_back(DataType((*muItr)->phi()));
      eta.emplace_back(DataType((*muItr)->eta()));
      pt.emplace_back(DataType((*muItr)->pt()/GeV));
 
      mass.emplace_back(DataType((*muItr)->m()/GeV));
      energy.emplace_back( DataType((*muItr)->e()/GeV ) );
      chi2.emplace_back( 1.0 ); //placeholder
      pdgId.emplace_back(DataType( -13.*(*muItr)->trackParticle(xAOD::Muon::TrackParticleType::Primary)->charge() )); // pdgId not available anymore in xAOD
    } // end MuonIterator
 
    // four-vectors
    DataMap["phi"] = phi;
    DataMap["eta"] = eta;
    DataMap["pt"] = pt;
    DataMap["energy"] = energy;
    DataMap["mass"] = mass;
    DataMap["chi2"] = chi2;
    DataMap["pdgId"] = pdgId;
 
    ATH_MSG_DEBUG(" retrieved with " << phi.size() << " entries");
    return DataMap;
  }
} // JiveXML namespace