TrackClassification.hpp

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

// This file is part of the Acts project.
//
// Copyright (C) 2019-2020 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/Index.hpp"
#include "ActsExamples/EventData/ProtoTrack.hpp"
#include "ActsExamples/EventData/Trajectories.hpp"
#include "ActsFatras/EventData/Barcode.hpp"

#include <cstddef>
#include <utility>
#include <vector>

namespace ActsExamples {
struct Trajectories;

struct ParticleHitCount {
  ActsFatras::Barcode particleId;
  size_t hitCount;
};

void identifyContributingParticles(
    const IndexMultimap<ActsFatras::Barcode>& hitParticlesMap,
    const ProtoTrack& protoTrack,
    std::vector<ParticleHitCount>& particleHitCounts);

void identifyContributingParticles(
    const IndexMultimap<ActsFatras::Barcode>& hitParticlesMap,
    const Trajectories& trajectories, size_t trajectoryTip,
    std::vector<ParticleHitCount>& particleHitCounts);

}  // namespace ActsExamples