TruthQualitySelectionTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#include "TruthQualitySelectionTool.h"
#include <GaudiKernel/StatusCode.h>
#include "TrackAnalysisCollections.h"
#include "TrackParametersHelper.h"
 
 
IDTPM::TruthQualitySelectionTool::TruthQualitySelectionTool( const std::string& name )
  : asg::AsgTool( name ) { }
 
StatusCode IDTPM::TruthQualitySelectionTool::initialize() {
  ATH_CHECK(not m_truthTool.empty());
  ATH_CHECK(m_truthTool.retrieve());
  return StatusCode::SUCCESS;
}
 
 
StatusCode IDTPM::TruthQualitySelectionTool::selectTracks( TrackAnalysisCollections& trkAnaColls ) {
  std::vector< const xAOD::TruthParticle* > selected;
  for ( auto trk: trkAnaColls.truthPartVec(IDTPM::TrackAnalysisCollections::FS)) {
    if ( m_truthTool->accept(trk) and this->accept(trk)) {
      selected.push_back(trk);
    }
  }
  ATH_MSG_DEBUG("Size before selection: " << trkAnaColls.truthPartVec(IDTPM::TrackAnalysisCollections::FS).size() << "\t Size after selection: " << selected.size());
  ATH_CHECK(trkAnaColls.fillTruthPartVec(selected, IDTPM::TrackAnalysisCollections::FS));
  return StatusCode::SUCCESS;
}
 
 
bool IDTPM::TruthQualitySelectionTool::accept(const xAOD::TruthParticle* truth) {
  if (m_maxEta!=-9999.  and (eta(*truth)) > m_maxEta )              return false;
  if (m_minEta!=-9999.  and (eta(*truth)) < m_minEta )              return false; 
  if (m_minPhi!=-9999.  and (phi(*truth)) < m_minPhi )              return false; 
  if (m_maxPhi!=-9999.  and (phi(*truth)) > m_maxPhi )              return false;  
  if (m_minD0!=-9999.   and (d0(*truth)) < m_minD0 )                return false; 
  if (m_maxD0!=-9999.   and (d0(*truth)) > m_maxD0 )                return false; 
  if (m_minZ0!=-9999.   and (z0(*truth)) < m_minZ0 )                return false; 
  if (m_maxZ0!=-9999.   and (z0(*truth)) > m_maxZ0 )                return false; 
  if (m_minQoPT!=-9999. and (qOverPT(*truth)) < m_minQoPT )         return false; 
  if (m_maxQoPT!=-9999. and (qOverPT(*truth)) > m_maxQoPT )         return false; 
  if (m_minAbsEta!=-9999.  and std::fabs(eta(*truth)) < m_minAbsEta )       return false; 
  if (m_minAbsPhi!=-9999.  and std::fabs(phi(*truth)) < m_minAbsPhi )       return false; 
  if (m_maxAbsPhi!=-9999.  and std::fabs(phi(*truth)) > m_maxAbsPhi )       return false;  
  if (m_minAbsD0!=-9999.   and std::fabs(d0(*truth)) < m_minAbsD0 )         return false; 
  if (m_maxAbsD0!=-9999.   and std::fabs(d0(*truth)) > m_maxAbsD0 )         return false; 
  if (m_minAbsZ0!=-9999.   and std::fabs(z0(*truth)) < m_minAbsZ0 )         return false; 
  if (m_maxAbsZ0!=-9999.   and std::fabs(z0(*truth)) > m_maxAbsZ0 )         return false; 
  if (m_minAbsQoPT!=-9999. and std::fabs(qOverPT(*truth)) < m_minAbsQoPT )  return false; 
  if (m_maxAbsQoPT!=-9999. and std::fabs(qOverPT(*truth)) > m_maxAbsQoPT )  return false; 
  if (m_isHadron    and not isHadron(*truth) )                    return false; 
  return true;
}