AmbiguityResolutionML.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file AmbiguityResolutionML.hpp

// This file is part of the Acts project.
//
// Copyright (C) 2023 CERN for the benefit of the Acts project
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#pragma once

#include "ActsExamples/EventData/Track.hpp"
#include "ActsExamples/Framework/IAlgorithm.hpp"

#include <map>
#include <string>
#include <vector>

namespace ActsExamples {

class AmbiguityResolutionML : public IAlgorithm {
 public:
  AmbiguityResolutionML(std::string name, Acts::Logging::Level lvl);

 protected:
  std::multimap<int, std::pair<int, std::vector<int>>> mapTrackHits(
      const ConstTrackContainer& tracks, int nMeasurementsMin) const;

  ConstTrackContainer prepareOutputTrack(const ConstTrackContainer& tracks,
                                         std::vector<int>& goodTracks) const;
};

}  // namespace ActsExamples