MlMsTrackSeeder.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONTRACKFINDINGALGS_MLMSTRACKSEEDER_H
#define MUONTRACKFINDINGALGS_MLMSTRACKSEEDER_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "MuonRecToolInterfacesR4/ITrackSeedingTool.h"
#include "StoreGate/ReadDecorHandleKey.h"
#include "StoreGate/ReadHandleKey.h"
#include "xAODMuon/MuonSegmentContainer.h"
 
#include <span>
#include <string>
 
namespace MuonR4 {

class MlMsTrackSeeder final :
    public extends<AthAlgTool, ITrackSeedingTool> {
 public:
  using base_class::base_class;
 
  StatusCode initialize() override final;

  StatusCode findTrackSeeds(const EventContext& ctx,
                            std::vector<MsTrackSeed>& outSeeds) const override final;

  Acts::Result<Acts::BoundTrackParameters> estimateStartParameters(
      const EventContext& ctx,
      const MsTrackSeed& seed) const override final;
 
 private:
  ToolHandle<ITrackSeedingTool> m_seedParameterEstimator{
      this, "BaselineSeeder", "",
      "Tool providing initial-parameter estimates for ML-built seeds"};
  SG::ReadHandleKey<xAOD::MuonSegmentContainer> m_segmentKey{
      this, "SegmentContainer", "MuonSegmentsFromR4"};
  Gaudi::Property<std::string> m_candidateDecoration{
      this, "CandidateDecoration", "trackCandidateIds",
      "Segment [componentId, isSeedAnchor] decoration"};
  SG::ReadDecorHandleKey<xAOD::MuonSegmentContainer> m_candidateDecorKey{
      this, "CandidateDecoration", m_segmentKey, "trackCandidateIds",
      "Scheduler dependency on the ML candidate decoration"};
  Gaudi::Property<unsigned int> m_minSegmentsPerCandidate{
      this, "MinSegmentsPerCandidate", 2};
};
 
}  // namespace MuonR4
 
#endif