TrackParticlePerigeePairAtOOAssociationTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
// $Id$
#include "TrackParticlePerigeePairAtOOAssociationTool.h"
#include "D3PDMakerInterfaces/IObjGetterTool.h"
#include "xAODTracking/TrackParticleContainer.h"
#include "Particle/TrackParticle.h"
#include "ParticleEvent/TrackParticleAssocs.h"
#include "TrkParameters/TrackParameters.h"
#include "AthenaKernel/errorcheck.h"
 
 
namespace D3PD {
 
 
TrackParticlePerigeePairAtOOAssociationTool::TrackParticlePerigeePairAtOOAssociationTool
  (const std::string& type,
   const std::string& name,
   const IInterface* parent)
    : Base (type, name, parent),
      m_assocGetter (this)
{
  declareProperty ("AssocGetter", m_assocGetter,
                   "Getter for the association object.");
}
 
 
StatusCode TrackParticlePerigeePairAtOOAssociationTool::initialize()
{
  CHECK( Base::initialize() );
  if (!m_assocGetter.empty()) {
    CHECK( m_assocGetter.retrieve() );
    CHECK( m_assocGetter->configureD3PD<TrackParticleAssocs> () );
  }
  return StatusCode::SUCCESS;
}
 
 
const D3PD::PerigeePair*
TrackParticlePerigeePairAtOOAssociationTool::get
  (const Rec::TrackParticle& track)
{
  const Rec::TrackParticle& oldtrack = findOldtrack (track);
  return new D3PD::PerigeePair (track.perigee(), oldtrack.perigee());
}
 
 
const D3PD::PerigeePair*
TrackParticlePerigeePairAtOOAssociationTool::get
  (const xAOD::TrackParticle& track)
{
  static const SG::AuxElement::Accessor<ElementLink<xAOD::TrackParticleContainer> >
    orig ("originalTrackParticle");
  if (!orig.isAvailable(track) || !orig(track).isValid())
    return 0;
 
  return new D3PD::PerigeePair (&track.perigeeParameters(),
                                &(*orig(track))->perigeeParameters());
}
 
 
void TrackParticlePerigeePairAtOOAssociationTool::releaseObject
 (const D3PD::PerigeePair* p)
{
  delete p;
}
 
 
const Rec::TrackParticle&
TrackParticlePerigeePairAtOOAssociationTool::findOldtrack
 (const Rec::TrackParticle& track)
{
  if (track.info().trackFitter() != Trk::TrackInfo::GaussianSumFilter)
    return track;
 
  const TrackParticleAssocs* assocs = m_assocGetter->get<TrackParticleAssocs>();
  if (!assocs) {
    REPORT_MESSAGE (MSG::WARNING) << "Can't find GSF track association.";
    return track;
  }
 
  TrackParticleAssocs::asso_iterator beg = assocs->beginAssociation (&track);
  TrackParticleAssocs::asso_iterator end = assocs->endAssociation (&track);
  const Rec::TrackParticle* ret = 0;
  if (beg != end && beg.isValid())
    ret = dynamic_cast<const Rec::TrackParticle*> (*beg);
  m_assocGetter->releaseObject (assocs);
  if (!ret) {
    REPORT_MESSAGE (MSG::WARNING)
      << "Can't find original track for GSF refitted track.";
    return track;
  }
 
  return *ret;
}
 
 
} // namespace D3PD