TrkrClusterMultiTrajectory.hpp

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

// This file is part of the Acts project.
//
// Copyright (C) 2019 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 <unordered_map>
#include <utility>

#include "ActsExamples/EventData/TrkrClusterSourceLink.hpp"
#include "ActsExamples/Validation/ProtoTrackClassification.hpp"
#include "Acts/EventData/MultiTrajectory.hpp"
#include "Acts/EventData/TrackParameters.hpp"

namespace ActsExamples {
  using IndexedParams = std::unordered_map<size_t, Acts::BoundTrackParameters>;
  using SourceLink = ActsExamples::TrkrClusterSourceLink;
  
struct TrkrClusterMultiTrajectory {
 public:
  TrkrClusterMultiTrajectory() = default;

  TrkrClusterMultiTrajectory(const Acts::MultiTrajectory<SourceLink>& multiTraj,
           const std::vector<size_t>& tTips,
           const IndexedParams& parameters,
           const int& vertexId)
      : m_multiTrajectory(multiTraj),
        m_trackTips(tTips),
        m_trackParameters(parameters),
  m_vertexId(vertexId){}

  TrkrClusterMultiTrajectory(const TrkrClusterMultiTrajectory& rhs)
      : m_multiTrajectory(rhs.m_multiTrajectory),
        m_trackTips(rhs.m_trackTips),
        m_trackParameters(rhs.m_trackParameters),
  m_vertexId(rhs.m_vertexId){}

  TrkrClusterMultiTrajectory(TrkrClusterMultiTrajectory&& rhs)
      : m_multiTrajectory(std::move(rhs.m_multiTrajectory)),
        m_trackTips(std::move(rhs.m_trackTips)),
        m_trackParameters(std::move(rhs.m_trackParameters)),
  m_vertexId(std::move(rhs.m_vertexId)){}
  
  ~TrkrClusterMultiTrajectory() = default;

  TrkrClusterMultiTrajectory& operator=(const TrkrClusterMultiTrajectory& rhs) {
    m_multiTrajectory = rhs.m_multiTrajectory;
    m_trackTips = rhs.m_trackTips;
    m_trackParameters = rhs.m_trackParameters;
    m_vertexId = rhs.m_vertexId;
    return *this;
  }

  TrkrClusterMultiTrajectory& operator=(TrkrClusterMultiTrajectory&& rhs) {
    m_multiTrajectory = std::move(rhs.m_multiTrajectory);
    m_trackTips = std::move(rhs.m_trackTips);
    m_trackParameters = std::move(rhs.m_trackParameters);
    m_vertexId = std::move(rhs.m_vertexId);
    return *this;
  }

  bool hasTrajectory(const size_t& entryIndex) const {
    return std::count(m_trackTips.begin(), m_trackTips.end(), entryIndex) > 0;
  }

  bool hasTrackParameters(const size_t& entryIndex) const {
    return m_trackParameters.count(entryIndex) > 0;
  }

  std::pair<std::vector<size_t>, Acts::MultiTrajectory<SourceLink>>
  trajectory() const {
    return std::make_pair(m_trackTips, m_multiTrajectory);
  }

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

 private:
  // The multiTrajectory
  Acts::MultiTrajectory<SourceLink> m_multiTrajectory;

  // The entry indices of trajectories stored in multiTrajectory
  std::vector<size_t> m_trackTips = {};

  // The fitted parameters at the provided surface for individual trajectories
  IndexedParams m_trackParameters = {};
  
  unsigned int m_vertexId = 9999;
};

}  // namespace ActsExamples