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"
// Trk
#include "TrkToolInterfaces/IRIO_OnTrackCreator.h"
#include "TrkParameters/TrackParameters.h"
 
class AtlasDetectorID;
 
namespace Trk {
 
  class PrepRawData;
  class RIO_OnTrack;

 
  class RIO_OnTrackCreator final : public extends<AthAlgTool, IRIO_OnTrackCreator> {
   public:
    // Public methods:

    using base_class::base_class;
    virtual ~RIO_OnTrackCreator();

    virtual StatusCode initialize() override;

    RIO_OnTrack* correct(const PrepRawData& rio,
                         const TrackParameters& pars,
                         const EventContext& ctx) const override final;
 
   private:
    ToolHandle<IRIO_OnTrackCreator> m_pixClusCor{this, "ToolPixelCluster", ""};
    ToolHandle<IRIO_OnTrackCreator> m_sctClusCor{this, "ToolSCT_Cluster", ""};
    ToolHandle<IRIO_OnTrackCreator> m_trt_Cor{this, "ToolTRT_DriftCircle", ""};
    ToolHandle<IRIO_OnTrackCreator> m_muonDriftCircleCor{this, "ToolMuonDriftCircle", ""};
    ToolHandle<IRIO_OnTrackCreator> m_muonClusterCor{this, "ToolMuonCluster",""};
 
    Gaudi::Property<std::string>m_mode{this, "Mode" ,"all" };   
    //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