InDetTrackHoleSearchTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
// InDetTrackHoleSearchTool.h, (c) ATLAS Detector software
 
#ifndef INDETINDETTRACKHOLESEARCHTOOL_H
#define INDETINDETTRACKHOLESEARCHTOOL_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
#include "GaudiKernel/EventContext.h"
#include "TrkExInterfaces/IExtrapolator.h"
#include "TrkToolInterfaces/ITrackHoleSearchTool.h"
#include "TrkToolInterfaces/IBoundaryCheckTool.h"
#include "TrkEventPrimitives/ParticleHypothesis.h"
#include "TrkParameters/TrackParameters.h"
#include <atomic>
#include <vector>
#include <map>
 
class AtlasDetectorID;
class Identifier;
class AtlasID;
namespace InDet {class IInDetTestPixelLayerTool; }
class IGeoModelSvc;
 
namespace Trk { class RIO_OnTrack; class TrackStateOnSurface; class Track; }
 
namespace InDetDD { class SiDetectorElement; }
 
namespace InDet 
{
 
  class InDetTrackHoleSearchTool : virtual public Trk::ITrackHoleSearchTool, public AthAlgTool
    {
    public:
      InDetTrackHoleSearchTool(const std::string&,const std::string&,const IInterface*);
      
      virtual ~InDetTrackHoleSearchTool ();
      
      virtual StatusCode initialize();
      virtual StatusCode finalize  ();
      
      virtual void countHoles(const Trk::Track& track, 
                  std::vector<int>& information ,
                  const Trk::ParticleHypothesis partHyp = Trk::pion) const ;
      
      virtual const Trk::TrackStates* getHolesOnTrack(const Trk::Track& track, 
                                        const Trk::ParticleHypothesis partHyp = Trk::pion) const ;
      
      virtual const Trk::Track*  getTrackWithHoles(const Trk::Track& track, 
                           const Trk::ParticleHypothesis partHyp = Trk::pion) const ;
      
      
      virtual const Trk::Track* getTrackWithHolesAndOutliers(const Trk::Track& track, 
                                 const Trk::ParticleHypothesis partHyp = Trk::pion) const;
 
    private:
      
      void searchForHoles(const Trk::Track& track, 
              std::vector<int>* information ,
              std::vector<const Trk::TrackStateOnSurface*>* listOfHoles,
              const Trk::ParticleHypothesis partHyp = Trk::pion) const;
      
      const AtlasDetectorID* m_atlasId;
      
      ToolHandle<Trk::IExtrapolator> m_extrapolator{
    this,
    "Extrapolator",
    "Trk::Extrapolator/InDetExtrapolator",
    "Extrapolator used to extrapolate to layers"
      };
      ToolHandle<Trk::IBoundaryCheckTool> m_boundaryCheckTool {
         this,
         "BoundaryCheckTool",
         "InDet::InDetBoundaryCheckTool",
         "Boundary checking tool for detector sensitivities"
      };
 
      BooleanProperty m_extendedListOfHoles{this, "ExtendedListOfHoles", false};
      BooleanProperty m_cosmic{this, "Cosmics", false};
 
      IntegerProperty m_minSiHits{this, "minSiHits", 3};
 
      /* searching for dead modules after the last measurement (needed for robustness of SW)*/
      BooleanProperty m_countDeadModulesAfterLastHit
    {this, "CountDeadModulesAfterLastHit", true};
 
      mutable std::atomic_int m_warning;
 
      bool getMapOfHits(
        const EventContext& ctx,
        const Trk::Track& track,
        const Trk::ParticleHypothesis partHyp,
        std::map<const Identifier, const Trk::TrackStateOnSurface*>& mapOfHits,
        std::map<const Identifier,std::pair<const Trk::TrackParameters*, const bool>>&
          mapOfPredictions) const;
 
      void performHoleSearchStepWise(std::map<const Identifier, const Trk::TrackStateOnSurface*>& mapOfHits,
                     std::map<const Identifier, std::pair<const Trk::TrackParameters*,const bool> >& mapOfPredictions,
                     std::vector<int>* information,
                     std::vector<const Trk::TrackStateOnSurface*>* listOfHoles) const;
      
      static const Trk::TrackStateOnSurface* createHoleTSOS(const Trk::TrackParameters* trackPar) ;
 
      const Trk::Track*  addHolesToTrack(const Trk::Track& oldTrack, 
                     std::vector<const Trk::TrackStateOnSurface*>* listOfHoles) const;
    };
 
} // end of namespace
 
#endif