InDetAmbiTrackSelectionTool.h

Go to the documentation of this file.

// -*- C++ -*-
 
/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/

// InDetAmbiTrackSelectionTool.h, (c) ATLAS Detector software
// Markus Elsing
 
#ifndef INDETINDETAMBITRACKSELECTIONTOOL_H
#define INDETINDETAMBITRACKSELECTIONTOOL_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "TrkToolInterfaces/IAmbiTrackSelectionTool.h"
 
#include "InDetRecToolInterfaces/ITrtDriftCircleCutTool.h"
#include "TrkToolInterfaces/IPRDtoTrackMapTool.h"
#include "TrkTrack/TrackStateOnSurface.h"
 
#include "GaudiKernel/ToolHandle.h"
 
#include <map>
#include <vector>
 
class SiliconID;
class Identifier;
namespace Trk {
  class Track;
}
 
namespace InDet 
{
  
 
  class InDetAmbiTrackSelectionTool : public extends<AthAlgTool, Trk::IAmbiTrackSelectionTool>
  {
  public:
    InDetAmbiTrackSelectionTool(const std::string&,const std::string&,const IInterface*);
      
    virtual ~InDetAmbiTrackSelectionTool () = default;
      
    virtual StatusCode initialize() override;
    virtual StatusCode finalize() override;
 
    virtual std::tuple<Trk::Track*,bool> getCleanedOutTrack(const Trk::Track *track,
                                                            const Trk::TrackScore score,
                                                            Trk::ClusterSplitProbabilityContainer &splitProbContainer,
                                                            Trk::PRDtoTrackMap &prd_to_track_map,
                                                            int trackId = -1, int subtrackId = -1) const override;
 
  private:
      
    Trk::Track* createSubTrack( const std::vector<const Trk::TrackStateOnSurface*>& tsos, const Trk::Track* track ) const ;

    PublicToolHandle<ITrtDriftCircleCutTool> m_selectortool
    {this, "DriftCircleCutTool", "InDet::InDetTrtDriftCircleCutTool"};
 
    ToolHandle<Trk::IPRDtoTrackMapTool>         m_assoTool
         {this, "AssociationTool", "Trk::PRDtoTrackMapTool" };

    const SiliconID* m_detID{nullptr};
      
    IntegerProperty m_minHits{this, "minHits", 5};
    IntegerProperty m_minTRT_Hits{this, "minTRTHits", 0};
    int m_maxShared{0};
    IntegerProperty m_maxSharedModules{this, "maxShared", 1};
    IntegerProperty m_maxTracksPerPRD{this, "maxTracksPerSharedPRD", 2};
    IntegerProperty m_minNotShared{this, "minNotShared", 6};
    FloatProperty m_minScoreShareTracks{this, "minScoreShareTracks", 0.0};
    BooleanProperty m_cosmics{this, "Cosmics", false};
    BooleanProperty m_parameterization{this, "UseParameterization", true}; // Use table of min number DCs
    BooleanProperty m_doPixelClusterSplitting{this, "doPixelSplitting", false};
    FloatProperty m_sharedProbCut{this, "sharedProbCut", 0.02};
    IntegerProperty m_maxSplitSize{this, "MaximalSplitSize", 49, "A.S.: remove that when solved properly by updating the SplitProb info with isExcluded. A.S.: to be removed once EDM is updated"};
  }; 
} // end of namespace
 
 
#endif