TRTTrackHoleSearchTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
// TRTTrackHoleSearchTool.h
// author: Ryan D. Reece <ryan.reece@cern.ch>
// created: Jan 2010
 
 
#ifndef TRT_TrackHoleSearch_TRTTrackHoleSearchTool_h
#define TRT_TrackHoleSearch_TRTTrackHoleSearchTool_h
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
#include "GaudiKernel/ServiceHandle.h"
#include "TrkToolInterfaces/ITrackHoleSearchTool.h"
#include "TrkParameters/TrackParameters.h"
#include "TrkExInterfaces/IExtrapolator.h"
#include "InDetConditionsSummaryService/IInDetConditionsSvc.h"
 
#include <atomic>
#include <string>
#include <vector>
#include <fstream>
 
namespace Trk
{
    class CylinderSurface;
    class Track;
    class Surface;
}
 
class TRT_ID;
 
class TRTTrackHoleSearchTool : public Trk::ITrackHoleSearchTool, public AthAlgTool
{
 public:
    TRTTrackHoleSearchTool(const std::string& type, const std::string& name, const IInterface* parent);
 
    StatusCode initialize();
    StatusCode finalize();
 
    void countHoles(const Trk::Track& track, 
                    std::vector<int>& information ,
                    const Trk::ParticleHypothesis partHyp = Trk::pion) const;
    
    const Trk::TrackStates* getHolesOnTrack(const Trk::Track& track, 
                                                                      const Trk::ParticleHypothesis partHyp = Trk::pion) const;
    
    const Trk::Track* getTrackWithHoles(const Trk::Track& track, 
                                        const Trk::ParticleHypothesis partHyp = Trk::pion) const;
        
    
    const Trk::Track* getTrackWithHolesAndOutliers(const Trk::Track& track, 
                                                   const Trk::ParticleHypothesis partHyp = Trk::pion) const;
    
          
 private:
    // configurables
    //----------------------------------
    ToolHandle<Trk::IExtrapolator>  m_extrapolator;
    ServiceHandle<IInDetConditionsSvc> m_conditions_svc;
    bool m_use_conditions_svc;
    float m_outer_radius;
    float m_max_z;
    int m_max_trailing_holes;
    bool m_begin_at_first_trt_hit;
    bool m_end_at_last_trt_hit;
    bool m_bcheck;
    bool m_do_dump_bad_straw_log;
    float m_locR_cut;
    float m_locR_sigma_cut;
    float m_locZ_cut;
       
    // private data
    //----------------------------------
    mutable std::atomic_bool m_has_been_called;
    const TRT_ID* m_TRT_ID;
    Trk::CylinderSurface* m_trt_outer_surf;
 
    // private methods
    //----------------------------------
    int extrapolateBetweenHits(const Trk::TrackParameters* start_parameters,
                               const Trk::Surface& end_surf,
                               Trk::TrackStates* holes,
                               const Trk::ParticleHypothesis partHyp = Trk::pion) const;
 
    void dump_bad_straw_log() const;
        
    Trk::TrackStates::const_iterator
        find_first_trt_hit(const Trk::TrackStates& track_states) const;
 
    Trk::TrackStates::const_iterator
        find_last_hit_before_trt(const Trk::TrackStates& track_states) const;
 
    const Trk::Track* addHolesToTrack(
                                      const Trk::Track& track,
                                      const Trk::TrackStates* holes) const;
};
 
#endif // TRT_TrackHoleSearch_TRTTrackHoleSearchTool_h