HoleSearchValidation.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
// HoleSearchValidation.h, (c) ATLAS Detector software
 
#ifndef TRK_HOLESEARCHVALIDATION_H
#define TRK_HOLESEARCHVALIDATION_H
 
// Gaudi includes
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "CxxUtils/checker_macros.h"
#include "GaudiKernel/ToolHandle.h"
#include "GaudiKernel/ServiceHandle.h"
#include "AthenaKernel/IAthRNGSvc.h"
 
#include "TrkTrack/TrackCollection.h"
#include <mutex>
 
// forward declarations
class AtlasDetectorID;
class PixelID;
class SCT_ID;
class TRT_ID; 
class SiliconID;
 
namespace CLHEP {
  class HepRandomEngine; 
}
 
namespace Trk {
    class ITrackHoleSearchTool;
    class TrackStateOnSurface;
    class Track;
}
 
namespace InDet {
 
  class HoleSearchValidation : public AthReentrantAlgorithm {
  public:
    HoleSearchValidation(const std::string& name, ISvcLocator* pSvcLocator);
    ~HoleSearchValidation();
 
    StatusCode          initialize();
    StatusCode          execute(const EventContext& ctx) const;
    StatusCode          finalize();
 
  private:
 
    void printInfoTSoS( const Trk::TrackStateOnSurface* tsos) const;
    unsigned int doHoleSearch( const Trk::Track* track) const;
 
    const AtlasDetectorID*                m_idHelper;
    const PixelID*                        m_pixelID;
    const SCT_ID*                         m_sctID; 
    const TRT_ID*                         m_trtID; 
    const SiliconID*                      m_siliconID; 
 
    ToolHandle<Trk::ITrackHoleSearchTool>    m_holeSearchTool;
    SG::ReadHandleKey<TrackCollection> m_trackCollectionKey;     
    SG::WriteHandleKey<TrackCollection> m_trackCollectionOutputKey;     
    //unsigned int   m_numberHitsToRemove;  //!< jobOption: number of hits to remove from tracks
    //unsigned int   m_numberHitsToKeep;    //!< jobOption: number of hits to remove 
    bool           m_saveNewTracksInSG;   
   
 
    struct Parts {
      enum EParts {
        kPix0,
        kPix1,
        kPix2,
 
        kSct0,
        kSct1,
        kSct2,
        kSct3,
        kSct4,
        kSct5,
        kSct6,
        kSct7,
        kSct8,
        kSctSide0,
        kSctSide1,
        kNParts
      };
    };
 
    bool m_removeParts[Parts::kNParts];
    bool           m_removeOverlapHitsOnly;
    bool           m_ignoreTrackEnds;
 
    bool           m_randomRemovalMode;
    unsigned int   m_maxNumberOfHoles;   
 
    ServiceHandle<IAthRNGSvc>    m_rndmGenSvc;       
    std::string                  m_randomEngineName; 
 
    // some information for statistics -> printout in finalize
    mutable std::mutex                                  m_trackStatsMutex;
    mutable std::vector< std::vector< unsigned int > >  m_trackStats ATLAS_THREAD_SAFE; 
 
  }; // end of class
 
} // close of namespace
 
#endif // TRK_HOLESEARCHVALIDATION_H