TrigEgammaPrecisionPhotonCaloIsoHypoAlg.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#include "Gaudi/Property.h"
#include "TrigEgammaPrecisionPhotonCaloIsoHypoAlg.h"
#include "TrigCompositeUtils/HLTIdentifier.h"
#include "TrigCompositeUtils/TrigCompositeUtils.h"
#include "TrigSteeringEvent/TrigRoiDescriptor.h"
#include "AthViews/ViewHelper.h"
 
namespace TCU = TrigCompositeUtils;
 
TrigEgammaPrecisionPhotonCaloIsoHypoAlg::TrigEgammaPrecisionPhotonCaloIsoHypoAlg( const std::string& name,  
                                                                        ISvcLocator* pSvcLocator ) :
  ::HypoBase( name, pSvcLocator ) {}
 
 
StatusCode TrigEgammaPrecisionPhotonCaloIsoHypoAlg::initialize() 
{
  ATH_MSG_DEBUG ( "Initializing " << name() << "..." );
  ATH_CHECK( m_hypoTools.retrieve() );
  ATH_CHECK( m_photonsKey.initialize() );
  ATH_CHECK( m_IsophotonsKey.initialize() );
  renounce( m_photonsKey );// photons are made in views, so they are not in the EvtStore: hide them
  renounce( m_IsophotonsKey );// photons are made in views, so they are not in the EvtStore: hide them
 
  return StatusCode::SUCCESS;
}
 
 
StatusCode TrigEgammaPrecisionPhotonCaloIsoHypoAlg::execute( const EventContext& context ) const 
{  
  ATH_MSG_DEBUG ( "Executing " << name() << "..." );
  auto previousDecisionsHandle = SG::makeHandle( decisionInput(), context );
  ATH_CHECK( previousDecisionsHandle.isValid() );  
  ATH_MSG_DEBUG( "Running with "<< previousDecisionsHandle->size() <<" previous decisions");
 
 
  // create handles for isolated photons
  SG::WriteHandle<xAOD::PhotonContainer> h_isoPhotons = SG::makeHandle(m_IsophotonsKey, context);
  //make the output photon container
  ATH_CHECK(h_isoPhotons.record(std::make_unique<xAOD::PhotonContainer>(),
                              std::make_unique<xAOD::PhotonAuxContainer>()));
 
  // new decisions
 
  // new output decisions
  SG::WriteHandle<TCU::DecisionContainer> outputHandle = TCU::createAndStore(decisionOutput(), context );
  auto decisions = outputHandle.ptr();
 
  // input for decision
  std::vector<ITrigEgammaPrecisionPhotonCaloIsoHypoTool::PhotonInfo> toolInput;
 
  // loop over previous decisions
  size_t counter=0;
  for ( auto previousDecision: *previousDecisionsHandle ) {
 
    //get updated RoI  
    auto roiELInfo = TCU::findLink<TrigRoiDescriptorCollection>( previousDecision, TCU::roiString() );
 
    ATH_CHECK( roiELInfo.isValid() );
    const TrigRoiDescriptor* roi = *(roiELInfo.link);
 
    // not using View so commenting out following lines
    const auto viewEL = previousDecision->objectLink<ViewContainer>( TCU::viewString() );
    ATH_CHECK( viewEL.isValid() );
    auto photonHandle = ViewHelper::makeHandle( *viewEL, m_photonsKey, context);
    ATH_CHECK( photonHandle.isValid() );
    ATH_MSG_DEBUG ( "Photon handle size: " << photonHandle->size() << "..." );
    // Loop over the photonHandles
    size_t validphotons=0;
    for (size_t cl=0; cl< photonHandle->size(); cl++){
    
      {
        auto ph = ViewHelper::makeLink( *viewEL, photonHandle, cl );
        ATH_MSG_DEBUG ( "Checking ph.isValid()...");
        if( !ph.isValid() ) {
          ATH_MSG_DEBUG ( "PhotonHandle in position " << cl << " -> invalid ElemntLink!. Skipping...");
        }
      
        ATH_CHECK(ph.isValid());
 
        ATH_MSG_DEBUG ( "PhotonHandle in position " << cl << " processing...");
        ATH_MSG_DEBUG ( "Copying xAOD photon to new isolated container " << m_IsophotonsKey << " to run isolation tool decoration later...");
        // Copy the photon in non-iso container to the To-Be-Isolation-Evaluated IsoPhoton container
        xAOD::Photon *isoPhoton = new xAOD::Photon();
        h_isoPhotons->push_back(isoPhoton);
        const xAOD::Photon *originalPhoton = *ph;
        *isoPhoton = *originalPhoton; // Deep copy
 
        auto d = TCU::newDecisionIn( decisions, TCU::hypoAlgNodeName() );
        d->setObjectLink( TCU::featureString(), ElementLink<xAOD::PhotonContainer>(*h_isoPhotons, isoPhoton->index(), context) );
        TrigCompositeUtils::linkToPrevious( d, decisionInput().key(), counter );
 
        ITrigEgammaPrecisionPhotonCaloIsoHypoTool::PhotonInfo info(d, roi, isoPhoton, previousDecision);
 
 
        toolInput.push_back(info);
        validphotons++;
      }
    }
 
    ATH_MSG_DEBUG( "Photons with valid links: " << validphotons );
    
    ATH_MSG_DEBUG( "roi, photon, previous decision to new decision " << counter << " for roi " );
    counter++;
  }
 
  ATH_MSG_DEBUG( "Found "<<toolInput.size()<<" inputs to tools");
 
  for ( auto& tool: m_hypoTools ) {
    ATH_CHECK( tool->decide( toolInput ) );
  }
 
  ATH_CHECK( hypoBaseOutputProcessing(outputHandle) );
  return StatusCode::SUCCESS;
}