AsgSelectionToolWrapper.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
// Author: James Catmore (james.catmore@cern.ch)
 
#include "DerivationFrameworkTools/AsgSelectionToolWrapper.h"
#include "PATCore/AcceptData.h"
#include "StoreGate/ReadHandle.h"
#include "StoreGate/WriteDecorHandle.h"
 
 
namespace DerivationFramework {
 
  StatusCode AsgSelectionToolWrapper::initialize() {
    ATH_CHECK(m_tool.retrieve());
    ATH_CHECK(m_containerKey.initialize());
    ATH_CHECK(m_decorKey.initialize());
    return StatusCode::SUCCESS;
  }
 
 
 
  StatusCode AsgSelectionToolWrapper::addBranches(const EventContext& ctx) const
  {
    // retrieve container
 
    SG::WriteDecorHandle<xAOD::IParticleContainer, char> decorator (m_decorKey, ctx);
    if( ! decorator.isValid() ) {
        ATH_MSG_ERROR ("Couldn't retrieve IParticles with key: " << m_containerKey.fullKey() );
        return StatusCode::FAILURE;
    }
 
    // Write mask for each element and record to SG for subsequent selection
    for ( const xAOD::IParticle* part : *decorator) {
      auto theAccept = m_tool->accept(part);  // asg::AcceptData or TAccept
      if(m_cut.empty()){
        decorator(*part) = true && theAccept;
      } else{
        decorator(*part) = true && theAccept.getCutResult(m_cut);
      }
    }
 
    return StatusCode::SUCCESS;
  }
}