TrigBSExtraction.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#include "StoreGate/ReadHandle.h"
#include "StoreGate/WriteHandle.h"
 
#include "TrigBSExtraction/TrigBSExtraction.h"
#include "TrigSteeringEvent/HLTResult.h"
#include "TrigSerializeCnvSvc/TrigSerializeConvHelper.h"
 
TrigBSExtraction::TrigBSExtraction(const std::string& name, ISvcLocator* pSvcLocator)
  : AthAlgorithm(name, pSvcLocator)
{
}
 
 
StatusCode TrigBSExtraction::initialize() {
 
  // L2 navigation tool is optional
  if ( !m_l2ResultKeyIn.empty() ) ATH_CHECK( m_navToolL2.retrieve() );
  else m_navToolL2.disable();
 
  ATH_CHECK( m_navTool.retrieve() );
 
  // xAOD converter tool (for Run-1 data)
  if ( !m_xAODTool.empty() ) ATH_CHECK( m_xAODTool.retrieve() );
  else m_xAODTool.disable();
 
  // Initialize handle keys
  ATH_CHECK( m_l2ResultKeyIn.initialize(SG::AllowEmpty) );
  ATH_CHECK( m_l2ResultKeyOut.initialize(SG::AllowEmpty) );
 
  ATH_CHECK( m_hltResultKeyIn.initialize(SG::AllowEmpty) );
  ATH_CHECK( m_hltResultKeyOut.initialize(SG::AllowEmpty) );
 
  if ( m_dataScoutingKeysIn.size() != m_dataScoutingKeysOut.size() ) {
    ATH_MSG_ERROR("Size of DSResultKeysIn/Out does not match: " << m_dataScoutingKeysIn <<
                  " vs " << m_dataScoutingKeysOut );
    return StatusCode::FAILURE;
  }
  ATH_CHECK( m_dataScoutingKeysIn.initialize() );
  ATH_CHECK( m_dataScoutingKeysOut.initialize() );
 
  // Initialize the TrigSerializeConcHelper here so that all the additional streamerinfos are read in.
  // In case of DataScouting this may give problems if the DS ROBs contain containers which
  // were not compiled with navigation (the container type information is then not known
  // for a given CLID).
  ToolHandle<ITrigSerializeConvHelper> trigCnvHelper("TrigSerializeConvHelper",this);
  ATH_CHECK( trigCnvHelper.retrieve() );
 
  return StatusCode::SUCCESS;
}
 
 
StatusCode TrigBSExtraction::execute() {
 
  const bool isRun1 = m_navToolL2.isEnabled();
  if ( isRun1 ) {
    if ( repackFeaturesToSG(*m_navToolL2, m_l2ResultKeyIn, m_l2ResultKeyOut, false, false).isFailure() )
      ATH_MSG_WARNING( "failed unpacking features from BS to SG for: " << m_l2ResultKeyIn  );
  }
  
  if ( !m_hltResultKeyIn.empty() ) {
    // unpack, merge with L2 result and do xAOD conversion
    // xAOD conversion is only done for HLTResult_EF in Run-1
    if ( repackFeaturesToSG(*m_navTool, m_hltResultKeyIn, m_hltResultKeyOut, isRun1, isRun1).isFailure() )
      ATH_MSG_WARNING( "failed unpacking features from BS to SG for: " << m_hltResultKeyIn  );
  }
 
  for (size_t i = 0; i<m_dataScoutingKeysIn.size(); i++ ) {
    if ( repackFeaturesToSG(*m_navTool, m_dataScoutingKeysIn[i], m_dataScoutingKeysOut[i], false, false).isFailure() )
      ATH_MSG_WARNING( "failed unpacking features from BS to SG for: " << m_dataScoutingKeysIn[i] );
  }
 
  if ( isRun1 ) m_navToolL2->reset();
  m_navTool->reset();
 
  return StatusCode::SUCCESS;
}
 
 
StatusCode TrigBSExtraction::repackFeaturesToSG (HLT::Navigation& navTool,
                                                 const SG::ReadHandleKey<HLT::HLTResult>& key,
                                                 SG::WriteHandleKey<HLT::HLTResult>& keyOut,
                                                 bool equalize,
                                                 bool xAODCnv) {
 
  ATH_MSG_DEBUG( "Trying to deserialize content of " << key );
  auto cresult = SG::makeHandle(key);
 
  if ( !cresult.get() ) {
    ATH_MSG_WARNING( "No HLTResult found with key " << key );
    return StatusCode::SUCCESS;
  }
 
  ATH_MSG_DEBUG("HLTResult is level=" << cresult->getHLTLevel()  );
 
  const std::vector<uint32_t>& navData = cresult->getNavigationResult();
  if ( !navData.empty() ) {
    ATH_MSG_DEBUG( "Navigation payload obtained from " << key << " has size " << navData.size()  );
    navTool.deserialize( navData );
  } else {
    ATH_MSG_WARNING( "Navigation payload obtained from " << key << " has size 0" );
  }
 
  if ( equalize && m_navToolL2.isEnabled() ) {
    ATH_MSG_DEBUG( "Merging L2 and EF navigation structures for " << key );
    navTool.merge(*m_navToolL2);
  }
 
  navTool.prepare();
 
  // optional xAOD conversion for Run-1 AOD containers
  if ( xAODCnv && m_xAODTool.isEnabled() ) {
 
    ATH_CHECK( m_xAODTool->convert(&navTool) );
 
    // after AOD TrigPassBitsCollection was converted to xAOD::TrigPassBitsContainer,
    // let's fill new xAOD::TrigPassBits objects with the proper pointers to
    // converted xAOD containers
    ATH_CHECK( m_xAODTool->setTrigPassBits(&navTool) );
 
    // this will redirect all Features to the xAOD converters
    ATH_CHECK( m_xAODTool->rewireNavigation(&navTool) );
  }
 
  // Create a copy of HLTResult and pack navigation back into it
  auto result = std::make_unique<HLT::HLTResult>(*cresult);
 
  result->getNavigationResult().clear();
  bool status = navTool.serialize(result->getNavigationResult(), result->getNavigationResultCuts());
  ATH_MSG_DEBUG( "New serialized navigation for " << keyOut << " has size " << result->getNavigationResult().size() );
 
  ATH_CHECK( SG::makeHandle(keyOut).record(std::move(result)) );
 
  return StatusCode(status);
}