IdentifierExtractor.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
// IdentifierExtractor.cxx,         (c) ATLAS Detector software  //
 
#include "TrkEventUtils/IdentifierExtractor.h"
#include "TrkCompetingRIOsOnTrack/CompetingRIOsOnTrack.h"
#include "TrkMeasurementBase/MeasurementBase.h"
 
void Trk::IdentifierExtractor::extract(std::vector<Identifier>& ids, const std::vector<const Trk::MeasurementBase*>& measurements)
{
  using namespace std;
  vector<const Trk::MeasurementBase*>::const_iterator it    = measurements.begin();
  vector<const Trk::MeasurementBase*>::const_iterator itEnd = measurements.end();
  ids.clear();
  for (; it!=itEnd ; ++it)
  {
    Identifier id = extract(*it);
    if( id.is_valid() ) { ids.push_back(id);}
  }
}
 
Identifier Trk::IdentifierExtractor::extract(const Trk::MeasurementBase* m)
{
  Identifier id;
  const Trk::RIO_OnTrack *rot = dynamic_cast<const Trk::RIO_OnTrack*>(m);
  if (rot) {
    id = rot->identify();
  } else {
    // identify by CompetingROT:
    const Trk::CompetingRIOsOnTrack* comprot = dynamic_cast<const Trk::CompetingRIOsOnTrack*>(m);
    if (comprot) {
      id = comprot->rioOnTrack(comprot->indexOfMaxAssignProb()).identify();
    } else {
      id.clear();
    }
  }
  return id;
}