TrigMuonEFTrackIsolationHypoAlg.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#include "TrigMuonEFTrackIsolationHypoAlg.h"
#include "AthViews/ViewHelper.h"
#include "StoreGate/ReadDecorHandle.h"
 
using namespace TrigCompositeUtils; 
 
// --------------------------------------------------------------------------------
// --------------------------------------------------------------------------------
 
TrigMuonEFTrackIsolationHypoAlg::TrigMuonEFTrackIsolationHypoAlg( const std::string& name,
                      ISvcLocator* pSvcLocator ) :
   ::HypoBase( name, pSvcLocator )
{}
 
 
// --------------------------------------------------------------------------------
// --------------------------------------------------------------------------------
 
StatusCode TrigMuonEFTrackIsolationHypoAlg::initialize()
{
  ATH_CHECK( m_hypoTools.retrieve() );
 
  renounce( m_muonKey );
  renounce( m_muonIso20Key );
  renounce( m_muonIso30Key );
  ATH_CHECK( m_muonKey.initialize() );
  ATH_CHECK( m_muonIso20Key.initialize() );
  ATH_CHECK( m_muonIso30Key.initialize() );
 
  return StatusCode::SUCCESS;
}
 
// --------------------------------------------------------------------------------
// --------------------------------------------------------------------------------
 
StatusCode TrigMuonEFTrackIsolationHypoAlg::execute( const EventContext& context ) const
{
 
  ATH_MSG_DEBUG("Executing ...");
 
  auto previousDecisionsHandle = SG::makeHandle( decisionInput(), context );
  ATH_CHECK( previousDecisionsHandle.isValid() );
  ATH_MSG_DEBUG( "Running with "<< previousDecisionsHandle->size() <<" previous decisions");
 
  // prepare output decisions
  SG::WriteHandle<DecisionContainer> outputHandle = createAndStore(decisionOutput(), context ); 
  auto decisions = outputHandle.ptr();
 
  std::vector<TrigMuonEFTrackIsolationHypoTool::EFIsolationMuonInfo> toolInput;
  size_t counter = 0;
  for ( const auto previousDecision: *previousDecisionsHandle ) {
 
    // get View
    auto viewEL = previousDecision->objectLink<ViewContainer>(viewString());
    ATH_CHECK( viewEL.isValid() );
 
    // get Muon and isolation
    auto muonHandle = ViewHelper::makeHandle( *viewEL, m_muonKey, context );
    SG::ReadDecorHandle<xAOD::MuonContainer, double> muonIso20 ( m_muonIso20Key, context );
    SG::ReadDecorHandle<xAOD::MuonContainer, double> muonIso30 ( m_muonIso30Key, context );
    ATH_CHECK( muonHandle.isValid() );
    ATH_MSG_DEBUG( "Muinfo handle size: " << muonHandle->size() << " ..." );
 
    // It is posisble that no muons are found, in this case we go to the next decision
    if ( muonHandle->size()==0 ) continue;
 
    // this code only gets muon 0, since we build the views centered on a single muon
    auto muonEL = ViewHelper::makeLink( *viewEL, muonHandle, 0 );
    ATH_CHECK( muonEL.isValid() );
    const xAOD::Muon* muon = *muonEL;
 
    // create new dicions
    auto newd = newDecisionIn( decisions, hypoAlgNodeName() );
 
    toolInput.emplace_back( newd, muon, previousDecision, muonIso20(*muon), muonIso30(*muon) );
 
    newd -> setObjectLink( featureString(), muonEL );
    TrigCompositeUtils::linkToPrevious( newd, previousDecision, context );
 
 
    ATH_MSG_DEBUG("REGTEST: " << m_muonKey.key() << " pT = " << (*muonEL)->pt()/Gaudi::Units::GeV << " GeV");
    ATH_MSG_DEBUG("REGTEST: " << m_muonKey.key() << " eta/phi = " << (*muonEL)->eta() << "/" << (*muonEL)->phi());
    ATH_MSG_DEBUG("Added view, feature, previous decision to new decision "<<counter <<" for view "<<(*viewEL)->name()  );
 
    counter++;
  }
 
  ATH_MSG_DEBUG("Found "<<toolInput.size()<<" inputs to tools");
 
  StatusCode sc = StatusCode::SUCCESS;
  for ( auto& tool: m_hypoTools ) {
    ATH_MSG_DEBUG("Go to " << tool );
    sc = tool->decide(toolInput);
    if (!sc.isSuccess()) {
      ATH_MSG_ERROR("MuonHypoTool is failed");
      return StatusCode::FAILURE;
    }
  } // End of tool algorithm
 
  ATH_CHECK(hypoBaseOutputProcessing(outputHandle));
 
  return StatusCode::SUCCESS;
}
 
// --------------------------------------------------------------------------------
// --------------------------------------------------------------------------------