RIO_OnTrackCreator.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
// RIO_OnTrackCreator.h
//   Header file for class RIO_OnTrackCreator
// (c) ATLAS Detector software
// Wolfgang.Liebig@cern.ch, Andreas.Salzburger@cern.ch
 
 
#ifndef TRKTOOLS_RIOONTRACKCREATOR_H
#define TRKTOOLS_RIOONTRACKCREATOR_H
 
// Athena
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
// Trk
#include "TrkToolInterfaces/IRIO_OnTrackCreator.h"
#include "TrkParameters/TrackParameters.h"
 
class AtlasDetectorID;
 
namespace Trk {
 
  class PrepRawData;
  class RIO_OnTrack;
 
  class RIO_OnTrackCreator final : public AthAlgTool,
                                   virtual public IRIO_OnTrackCreator {
   public:
    // Public methods:
 
    RIO_OnTrackCreator(const std::string&,const std::string&,
               const IInterface*);
    virtual ~RIO_OnTrackCreator();
 
    virtual StatusCode initialize() override;
 
    RIO_OnTrack* correct(const PrepRawData&,
                         const TrackParameters&) const override;
 
   private:
    // Private data:
 
    const AtlasDetectorID*           m_idHelper{nullptr};
    ToolHandle<IRIO_OnTrackCreator> m_pixClusCor{
        this, "ToolPixelCluster",
        "InDet::PixelClusterOnTrackTool/PixelClusterOnTrackTool"};
    ToolHandle<IRIO_OnTrackCreator> m_sctClusCor{
        this, "ToolSCT_Cluster",
        "InDet::SCT_ClusterOnTrackTool/SCT_ClusterOnTrackTool"};
    ToolHandle<IRIO_OnTrackCreator> m_trt_Cor{
        this, "ToolTRT_DriftCircle",
        "InDet::TRT_DriftCircleOnTrackTool/TRT_DriftCircleOnTrackTool"};
    ToolHandle<IRIO_OnTrackCreator> m_muonDriftCircleCor{
        this, "ToolMuonDriftCircle",
        "Muon::MdtDriftCircleOnTrackCreator/MdtDriftCircleOnTrackTool"};
    ToolHandle<IRIO_OnTrackCreator> m_muonClusterCor{
        this, "ToolMuonCluster",
        "Muon::MuonClusterOnTrackCreator/MuonClusterOnTrackTool"};
 
    Gaudi::Property<std::string>m_mode{this, "Mode" ,"all" };   
    bool                             m_doPixel{true}; 
    bool                             m_doSCT{true};   
    bool                             m_doTRT{true};   
    //emum for the flag
    enum struct Mode {
      all = 0,
      indet = 1,
      muon = 2,
      invalid = 3
    };
    Mode m_enumMode = Mode::all;
  };
 
} // end of namespace
 
#endif // TRKTOOLS_RIOONTRACKCREATOR_H