TrackParticlePerigeeAtPVAssociationTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
#include "TrackParticlePerigeeAtPVAssociationTool.h"
 
#include "Particle/TrackParticle.h"
#include "TrkParameters/TrackParameters.h"
#include "xAODTracking/VertexContainer.h"
#include "AthenaKernel/errorcheck.h"
#include "VxVertex/PrimaryVertexSelector.h"
 
namespace D3PD {
 
TrackParticlePerigeeAtPVAssociationTool::TrackParticlePerigeeAtPVAssociationTool
  (const std::string& type,
   const std::string& name,
   const IInterface* parent)
    : Base (type, name, parent),
      m_resolver ("TrackParticlePerigeeAtPVAssociationTool",
                  evtStore(),
                  m_vxCandidate)
{
  declareProperty ("SGKey", m_vxCandidate = "PrimaryVertices,VxPrimaryCandidate");
}
 
StatusCode TrackParticlePerigeeAtPVAssociationTool::initialize(){
 
  CHECK( Base::initialize() );
  CHECK( m_trackToVertexTool.retrieve() );
 
  return StatusCode::SUCCESS;
}
 
 
const Trk::TrackParameters*
TrackParticlePerigeeAtPVAssociationTool::get (const Rec::TrackParticle& track)
{
  m_resolver.initialize<VxContainer>().ignore();
  const VxContainer* vxContainer = 0;
  StatusCode sc = evtStore()->retrieve(vxContainer, m_resolver.key());
  if (sc.isFailure() || !vxContainer) {
    REPORT_MESSAGE (MSG::WARNING) << "Could not retrieve primary vertex container: " << m_vxCandidate;
    return 0;
  }
 
  if(vxContainer->size()<1){
    REPORT_MESSAGE (MSG::WARNING) << "No primary vertices reconstructed";
    return 0;
  }
 
  if (track.measuredPerigee()->covariance() && track.measuredPerigee()->covariance()->rows() == 0)
  {
    REPORT_MESSAGE (MSG::WARNING) << "Bad track; can't find perigee at vertex.";
    return 0;
  }
 
  const Trk::VxCandidate* vxI = PrimaryVertexSelector(*vxContainer);
  if (!vxI){
      REPORT_MESSAGE (MSG::WARNING) << "No primary vertices reconstructed";
      return 0;
  }
  
  return m_trackToVertexTool->perigeeAtVertex(Gaudi::Hive::currentContext(), track, vxI->recVertex().position()).release();
}
 
 
const Trk::TrackParameters*
TrackParticlePerigeeAtPVAssociationTool::get (const xAOD::TrackParticle& track)
{
  m_resolver.initialize<xAOD::VertexContainer>().ignore();
  const xAOD::VertexContainer* vxContainer = 0;
  StatusCode sc = evtStore()->retrieve(vxContainer, m_resolver.key());
  if (sc.isFailure() || !vxContainer) {
    REPORT_MESSAGE (MSG::WARNING) << "Could not retrieve primary vertex container: " << m_vxCandidate;
    return 0;
  }
 
  if(vxContainer->size()<1){
    REPORT_MESSAGE (MSG::WARNING) << "No primary vertices reconstructed";
    return 0;
  }
 
  for (const xAOD::Vertex* vx : *vxContainer) {
    if (vx->vertexType() == xAOD::VxType::PriVtx)
      return m_trackToVertexTool->perigeeAtVertex(Gaudi::Hive::currentContext(), track, vx->position()).release();
  }
 
  REPORT_MESSAGE (MSG::WARNING) << "No primary vertices reconstructed";
  return 0;
}
 
 
void TrackParticlePerigeeAtPVAssociationTool::releaseObject (const Trk::TrackParameters* p)
{
  if(p) delete p;
}
 
} // namespace D3PD