Trajectories.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file Trajectories.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 "Acts/EventData/MultiTrajectory.hpp"
#include "Acts/EventData/VectorMultiTrajectory.hpp"
#include "Acts/Utilities/ThrowAssert.hpp"
#include "ActsExamples/EventData/IndexSourceLink.hpp"
#include "ActsExamples/EventData/Track.hpp"

#include <algorithm>
#include <unordered_map>
#include <vector>

namespace ActsExamples {

struct Trajectories final {
 public:
  using MultiTrajectory = Acts::ConstVectorMultiTrajectory;
  using IndexedParameters =
      std::unordered_map<Acts::MultiTrajectoryTraits::IndexType,
                         TrackParameters>;

  Trajectories() = default;
  Trajectories(const MultiTrajectory& multiTraj,
               const std::vector<Acts::MultiTrajectoryTraits::IndexType>& tTips,
               const IndexedParameters& parameters)
      : m_multiTrajectory(&multiTraj),
        m_trackTips(tTips),
        m_trackParameters(parameters) {}

  bool empty() const { return m_trackTips.empty(); }

  const MultiTrajectory& multiTrajectory() const {
    throw_assert(m_multiTrajectory != nullptr, "MultiTrajectory is null");
    return *m_multiTrajectory;
  }

  const std::vector<Acts::MultiTrajectoryTraits::IndexType>& tips() const {
    return m_trackTips;
  }

  bool hasTrajectory(Acts::MultiTrajectoryTraits::IndexType entryIndex) const {
    return (0 < std::count(m_trackTips.begin(), m_trackTips.end(), entryIndex));
  }

  bool hasTrackParameters(
      Acts::MultiTrajectoryTraits::IndexType entryIndex) const {
    return (0 < m_trackParameters.count(entryIndex));
  }

  const TrackParameters& trackParameters(
      Acts::MultiTrajectoryTraits::IndexType entryIndex) const {
    auto it = m_trackParameters.find(entryIndex);
    if (it == m_trackParameters.end()) {
      throw std::runtime_error(
          "No fitted track parameters for trajectory with entry index = " +
          std::to_string(entryIndex));
    }
    return it->second;
  }

 private:
  // The track container
  const MultiTrajectory* m_multiTrajectory{nullptr};
  // The entry indices of trajectories stored in multiTrajectory
  std::vector<Acts::MultiTrajectoryTraits::IndexType> m_trackTips = {};
  // The fitted parameters at the provided surface for individual trajectories
  IndexedParameters m_trackParameters = {};
};

using TrajectoriesContainer = std::vector<Trajectories>;

}  // namespace ActsExamples