ViewCreatorCentredOnIParticleROITool.cxx

Go to the documentation of this file.

 
/*
Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
#include "TrigSteeringEvent/TrigRoiDescriptorCollection.h"
#include "AthViews/ViewHelper.h"
#include "ViewCreatorCentredOnIParticleROITool.h"
#include "xAODMuon/Muon.h"
#include "xAODMuon/MuonContainer.h"
 
using namespace TrigCompositeUtils;
 
ViewCreatorCentredOnIParticleROITool::ViewCreatorCentredOnIParticleROITool(const std::string& type, const std::string& name, const IInterface* parent)
  : base_class(type, name, parent) 
  {}
 
 
StatusCode ViewCreatorCentredOnIParticleROITool::initialize()  {
  ATH_CHECK(m_roisWriteHandleKey.initialize());
  return StatusCode::SUCCESS;
}
 
 
StatusCode ViewCreatorCentredOnIParticleROITool::attachROILinks(TrigCompositeUtils::DecisionContainer& decisions, const EventContext& ctx) const {
  SG::WriteHandle<TrigRoiDescriptorCollection> roisWriteHandle = createAndStoreNoAux(m_roisWriteHandleKey, ctx);
  
  for ( Decision* outputDecision : decisions ) { 
    const std::vector<LinkInfo<xAOD::IParticleContainer>> myFeature = findLinks<xAOD::IParticleContainer>(outputDecision, m_iParticleLinkName, TrigDefs::lastFeatureOfType);
    
    if (myFeature.size() != 1) {
      ATH_MSG_ERROR("Did not find exactly one most-recent xAOD::IParticle '" << m_iParticleLinkName << "' for Decision object index " << outputDecision->index()
        << ", found " << myFeature.size());
      return StatusCode::FAILURE;
    }
    ATH_CHECK(myFeature.at(0).isValid());
 
    const ElementLink<xAOD::IParticleContainer> p4EL = myFeature.at(0).link;
 
    const double reta  = (*p4EL)->eta();
    const double retap = reta + m_roiEtaWidth;
    const double retam = reta - m_roiEtaWidth;
    const double rphi  = (*p4EL)->phi();
    const double rphip = rphi + m_roiPhiWidth;
    const double rphim = rphi - m_roiPhiWidth;
 
    TrigRoiDescriptor *newROI = nullptr;
 
    if ( m_roiZedWidth >= 0  ) {
 
         const xAOD::Muon* muon = dynamic_cast< const xAOD::Muon*>(*p4EL); //get muon of this found object
         double zed0  = 0.0; //initialization
      
         bool update_z_width = true;
      
         if ( m_useZedPosition ) { 
        if ( muon && muon->primaryTrackParticle() ) { 
           zed0 = muon->primaryTrackParticle()->z0();
           if ( m_useBeamspot ) zed0 += muon->primaryTrackParticle()->vz();
        }
        else update_z_width = false;
         }
    
     if ( update_z_width ) { 
 
         double zed0p   = zed0 + m_roiZedWidth; // in mm
         double zed0m   = zed0 - m_roiZedWidth; // in mm
         
         if ( m_roiZedSinThetaFlag ) { 
           double cosheta = std::cosh( (*p4EL)->eta() );
           zed0p   = zed0 + m_roiZedWidth*cosheta; // in mm
           zed0m   = zed0 - m_roiZedWidth*cosheta; // in mm
         }
         
         ATH_MSG_DEBUG( "New ROI for xAOD::Particle ET="<< (*p4EL)->p4().Et()
                << " eta="<< (*p4EL)->eta() << " +- " << m_roiEtaWidth
                << " phi="<< (*p4EL)->phi() << " +- " << m_roiPhiWidth
                << " zed0="<<  zed0 << " +- " << m_roiZedWidth );
       
         newROI = new TrigRoiDescriptor( reta, retam, retap,
                         rphi, rphim, rphip,
                         zed0, zed0m, zed0p );
     }
     else {
       newROI = new TrigRoiDescriptor( reta, retam, retap,
                       rphi, rphim, rphip);
     }   
    }
    else {
      newROI = new TrigRoiDescriptor( reta, retam, retap,
                      rphi, rphim, rphip);
    }
    
    roisWriteHandle->push_back( newROI );
    
    const ElementLink<TrigRoiDescriptorCollection> roiEL = ElementLink<TrigRoiDescriptorCollection>(*roisWriteHandle, roisWriteHandle->size() - 1, ctx);
    
    outputDecision->setObjectLink(roiString(), roiEL);
  }
  
  return StatusCode::SUCCESS;
}