SimpleAmbiguityProcessorTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef SIMPLEAMBIGUITYPROCESSORTOOL_H
#define SIMPLEAMBIGUITYPROCESSORTOOL_H
 
 
#include "AmbiguityProcessorBase.h"
#include "TrkEventPrimitives/TrackScore.h"
#include "TrkFitterInterfaces/ITrackFitter.h"
#include "GaudiKernel/ToolHandle.h"
#include "TrkToolInterfaces/IAmbiTrackSelectionTool.h"
 
#include "TrkToolInterfaces/IPRDtoTrackMapTool.h"
#include "TrkEventUtils/PRDtoTrackMap.h"
#include "TrkToolInterfaces/IExtendedTrackSummaryTool.h"
 
//need to include the following, since its a typedef and can't be forward declared.
#include "TrkTrack/TrackCollection.h"
#include "TrkTrack/TrackSeedMap.h"
 
#include <set>
#include <vector>
#include <mutex>
#include <array>
 
 
 
class AtlasDetectorID;
class PixelID;
 
namespace Trk {
 
  class ITrackScoringTool;
  class IPRD_AssociationTool;
  class ITruthToTrack;
 
  class SimpleAmbiguityProcessorTool final: public AmbiguityProcessorBase {
    public:
      // public types
      typedef std::multimap< TrackScore, TrackPtr > TrackScoreMap;
      typedef std::vector<const PrepRawData*> PrdSignature;
      typedef std::set<PrdSignature> PrdSignatureSet;
 
      // default methods
      SimpleAmbiguityProcessorTool(const std::string&,const std::string&,const IInterface*);
      virtual ~SimpleAmbiguityProcessorTool ();
      virtual StatusCode initialize() override final;
      virtual StatusCode finalize  () override final;
 
      virtual const TrackCollection*  process(const TrackCollection*, Trk::PRDtoTrackMap *prdToTrackMap) const override;
      virtual const TrackCollection*  process(const TracksScores* scoredTracks) const override;
 
      virtual void statistics() override;
    private:
 
      const TrackCollection*
      processVector(const TrackCollection &tracks, Trk::PRDtoTrackMap *prdToTrackMap) const;
 
      void
      addNewTracks(const TrackCollection &tracks,
                  TrackScoreMap& trackScoreTrackMap,
                  Trk::PRDtoTrackMap &prdToTrackMap) const;
 
      const TrackCollection *
      solveTracks(TrackScoreMap &trackScoreTrackMap,
                 Trk::PRDtoTrackMap &prdToTrackMap,
                 std::vector<std::unique_ptr<const Trk::Track> > &trackDustbin,
                 Counter &stat) const;
 
      void
      addSubTrack( const std::vector<const TrackStateOnSurface*>& tsos) const;
 
      virtual std::unique_ptr<Trk::Track>
      doBremRefit(const Trk::Track & track) const override final;
 
      Track*
      refitPrds( const Track* track, Trk::PRDtoTrackMap &prdToTrackMap, Counter &stat) const override final;
 
      void
      dumpTracks(const TrackCollection& tracks) const;
 
      void
      dumpStat(MsgStream &out) const;
 
      std::unique_ptr<Trk::Track>
      fit(const Track &track, bool flag, Trk::ParticleHypothesis hypo) const override final;
 
      // private data members
      bool m_dropDouble;
 
      ToolHandle<ITrackFitter> m_fitterTool{this, "Fitter", "Trk::KalmanFitter/InDetTrackFitter"};
      ToolHandle<Trk::IPRDtoTrackMapTool>         m_assoTool{this, "AssociationTool", "Trk::PRDtoTrackMapTool" };
      ToolHandle<IAmbiTrackSelectionTool> m_selectionTool{this, "SelectionTool", "InDet::InDetAmbiTrackSelectionTool/InDetAmbiTrackSelectionTool"};
 
    };
} //end ns
 
#endif // TrackAmbiguityProcessorTool_H