IndexSourceLink.hpp

Go to the documentation of this file.

// This file is part of the ACTS project.
//
// Copyright (C) 2016 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 https://mozilla.org/MPL/2.0/.
 
#pragma once
 
#include "Acts/EventData/SourceLink.hpp"
#include "Acts/Geometry/TrackingGeometry.hpp"
#include "Acts/Surfaces/Surface.hpp"
#include "GeometryContainers.hpp"
#include "Index.hpp"
 
#include <cassert>
 
namespace SHiP::trout {
 
struct IndexSourceLinkSurfaceAccessor;

class IndexSourceLink final {
   public:
    using SurfaceAccessor = IndexSourceLinkSurfaceAccessor;

    constexpr IndexSourceLink(Acts::GeometryIdentifier gid, ActsExamples::Index idx)
        : m_geometryId(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 ActsExamples::Index index() const { return m_index; }
 
    Acts::GeometryIdentifier geometryId() const { return m_geometryId; }
 
   private:
    Acts::GeometryIdentifier m_geometryId;
    ActsExamples::Index m_index = 0;
 
    friend bool operator==(const IndexSourceLink& lhs, const IndexSourceLink& rhs) {
        return (lhs.geometryId() == rhs.geometryId()) && (lhs.m_index == rhs.m_index);
    }
};
 
struct IndexSourceLinkSurfaceAccessor {
    const Acts::TrackingGeometry& geometry;
 
    const Acts::Surface* operator()(const Acts::SourceLink& sourceLink) const {
        const auto& indexSourceLink = sourceLink.get<IndexSourceLink>();
        return geometry.findSurface(indexSourceLink.geometryId());
    }
};

struct IndexSourceLinkAccessor : ActsExamples::GeometryIdMultisetAccessor<IndexSourceLink> {
    using BaseIterator = ActsExamples::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 SHiP::trout