IndexSourceLink.h

// This file is part of the Acts project.
//
// Copyright (C) 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 <cassert>
 
#include "Acts/EventData/SourceLink.hpp"
#include "Acts/Surfaces/Surface.hpp"
#include "Tracking/Sim/GeometryContainers.h"
#include "Tracking/Sim/Index.h"
 
namespace acts_examples {
 
class IndexSourceLink final {
 public:
  constexpr IndexSourceLink(Acts::GeometryIdentifier gid, Index idx)
      : m_geometry_id_(gid), m_index_(idx) {}
 
  // Construct an invalid source link. Must be default constructible to
  IndexSourceLink() = default;
  IndexSourceLink(const IndexSourceLink&) = default;
  IndexSourceLink(IndexSourceLink&&) = default;
  IndexSourceLink& operator=(const IndexSourceLink&) = default;
  IndexSourceLink& operator=(IndexSourceLink&&) = default;
 
  constexpr Index index() const { return m_index_; }
 
  Acts::GeometryIdentifier geometryId() const { return m_geometry_id_; }
 
 private:
  Acts::GeometryIdentifier m_geometry_id_;
  Index m_index_ = 0;
 
  friend bool operator==(const IndexSourceLink& lhs,
                         const IndexSourceLink& rhs) {
    return (lhs.geometryId() == rhs.geometryId()) and
           (lhs.m_index_ == rhs.m_index_);
  }
  friend bool operator!=(const IndexSourceLink& lhs,
                         const IndexSourceLink& rhs) {
    return not(lhs == rhs);
  }
};
 
using IndexSourceLinkContainer =
    acts_examples::GeometryIdMultiset<IndexSourceLink>;
struct IndexSourceLinkAccessor
    : acts_examples::GeometryIdMultisetAccessor<IndexSourceLink> {
  using BaseIterator = GeometryIdMultisetAccessor<IndexSourceLink>::Iterator;
 
  using Iterator = Acts::SourceLinkAdapterIterator<BaseIterator>;
 
  // get the range of elements with requested geoId
  std::pair<Iterator, Iterator> range(const Acts::Surface& surface) const {
    assert(container_ != nullptr);
    auto [begin, end] = container_->equal_range(surface.geometryId());
    return {Iterator{begin}, Iterator{end}};
  }
};
}  // namespace acts_examples