AmbiguityProcessorUtility.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#include "AmbiguityProcessorUtility.h"
#include "AthContainers/DataVector.h"
#include "TrkTrack/TrackStateOnSurface.h"
#include "TrkEventPrimitives/FitQuality.h"
#include "TrkTrack/TrackInfo.h"
#include "TrkTrack/Track.h"
 
 
namespace AmbiguityProcessor{
  TrackFilterCategory
  categoriseTrack(const Trk::Track & track, const Trk::TrackScore & score, const bool dropDuplicates, const AssociationTool & associate, AssociationMap & map, DuplicationCheckSet & set){
    if (score == 0) return  TrackFilterCategory::ScoreIsZero;
    if (dropDuplicates){
      if(const auto & [p, uniqueTrack] = set.insert(associate->getPrdsOnTrack( map, track)); not uniqueTrack) return TrackFilterCategory::TrackIsDuplicate;
    }
    return TrackFilterCategory::TrackAccepted;
  }
  //
  float 
  calculateFitQuality(const Trk::Track & track){
    float result{0.0};
    if  (const auto *const quality=track.fitQuality(); quality and quality->numberDoF()>0 ){ 
      result = quality->chiSquared()/quality->numberDoF();
    }
    return result; 
  }
  //
  std::unique_ptr<Trk::Track>
  createNewFitQualityTrack(const Trk::Track & track){
    double reXi2 = 0.; 
    int nDF = 0;
    const Trk::TrackStates* tsos = track.trackStateOnSurfaces();
    auto vecTsos = std::make_unique<Trk::TrackStates>();
    // loop over TSOS, copy TSOS and push into vector
    Trk::TrackStates::const_iterator iTsos    = tsos->begin();
    Trk::TrackStates::const_iterator iTsosEnd = tsos->end(); 
    for ( ; iTsos != iTsosEnd ; ++iTsos) {
      const Trk::TrackStateOnSurface* newTsos = new Trk::TrackStateOnSurface(**iTsos);
      vecTsos->push_back(newTsos);
      if((*iTsos)->type(Trk::TrackStateOnSurface::Measurement)){  //Get the chi2 and number of hits
        if ((*iTsos)->fitQualityOnSurface()) {
          reXi2 += (*iTsos)->fitQualityOnSurface().chiSquared();
          nDF   += (*iTsos)->fitQualityOnSurface().numberDoF();
        }
      }
    }
    auto fq = std::make_unique<Trk::FitQuality>(reXi2,nDF-5);
    Trk::TrackInfo info;
    info.addPatternRecoAndProperties(track.info());
    Trk::TrackInfo newInfo;
    newInfo.setPatternRecognitionInfo(Trk::TrackInfo::SimpleAmbiguityProcessorTool);
    info.addPatternReco(newInfo); 
    return std::make_unique<Trk::Track>(info, std::move(vecTsos), std::move(fq));
  }
  //
  int getUid() {
    static std::atomic<std::uint32_t> uid { 0 };
    return ++uid;
  }
 
}