InDetTrackSummaryHelperTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef INDETTRACKSUMMARYHELPERTOOL_H
#define INDETTRACKSUMMARYHELPERTOOL_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "TrkToolInterfaces/IExtendedTrackSummaryHelperTool.h"
 
#include "TRT_ConditionsServices/ITRT_StrawStatusSummaryTool.h"
#include "TrkEventPrimitives/ParticleHypothesis.h"
#include "TrkToolInterfaces/IPixelToTPIDTool.h"
#include "TrkToolInterfaces/ITrackHoleSearchTool.h"
#include "TrkTrackSummary/TrackSummary.h" // defines the Trk::numberOfDetectorTypes enum
 
#include "InDetPrepRawData/PixelCluster.h"
#include "TrkEventUtils/ClusterSplitProbabilityContainer.h"
 
#include "GaudiKernel/ToolHandle.h"
 
#include <bitset>
#include <vector>
 
class ITRT_StrawSummaryTool;
class PixelID;
class SCT_ID;
class TRT_ID;
 
namespace Trk {
class CompetingRIOsOnTrack;
class RIO_OnTrack;
class Track;
class TrackStateOnSurface;
}
 
namespace InDet {
 
class InDetTrackSummaryHelperTool final
  : public extends<AthAlgTool, Trk::IExtendedTrackSummaryHelperTool>
{
public:
  InDetTrackSummaryHelperTool(const std::string&,
                              const std::string&,
                              const IInterface*);

  virtual ~InDetTrackSummaryHelperTool() = default;

  virtual StatusCode initialize() override;
  virtual StatusCode finalize() override;

  using IExtendedTrackSummaryHelperTool::addDetailedTrackSummary;
  using IExtendedTrackSummaryHelperTool::analyse;
 
  virtual void analyse(
    const EventContext& ctx,
    const Trk::Track& track,
    const Trk::RIO_OnTrack* rot,
    const Trk::TrackStateOnSurface* tsos,
    std::vector<int>& information,
    std::bitset<Trk::numberOfDetectorTypes>& hitPattern) const override final;
 
  virtual void analyse(
    const EventContext& ctx,
    const Trk::Track& track,
    const Trk::CompetingRIOsOnTrack* crot,
    const Trk::TrackStateOnSurface* tsos,
    std::vector<int>& information,
    std::bitset<Trk::numberOfDetectorTypes>& hitPattern) const override final;
 
  virtual void addDetailedTrackSummary(const EventContext& ctx,
                                       const Trk::Track&,
                                       Trk::TrackSummary&) const override final;

  virtual void searchForHoles(
    const Trk::Track& track,
    std::vector<int>& information,
    const Trk::ParticleHypothesis partHyp = Trk::pion) const override final;
 
 
private:

  const PixelID* m_pixelId{ nullptr };

  const SCT_ID* m_sctId{ nullptr };

  const TRT_ID* m_trtId{ nullptr };
 
  ToolHandle<Trk::ITrackHoleSearchTool> m_holeSearchTool{
    this,
    "HoleSearch",
    "InDet::InDetTrackHoleSearchTool"
  };
 
  ToolHandle<ITRT_StrawStatusSummaryTool> m_TRTStrawSummaryTool{
    this,
    "TRTStrawSummarySvc",
    "TRT_StrawStatusSummaryTool",
    "The ConditionsSummaryTool"
  };
 
  BooleanProperty m_usePixel{ this, "usePixel", true };
  BooleanProperty m_useSCT{ this, "useSCT", true };
  BooleanProperty m_useTRT{ this, "useTRT", true };
};
 
}
#endif