v8-memory-span.h

Go to the documentation of this file.

// Copyright 2021 the V8 project authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
 
#ifndef INCLUDE_V8_MEMORY_SPAN_H_
#define INCLUDE_V8_MEMORY_SPAN_H_
 
#include <stddef.h>
 
#include <array>
#include <compare>
#include <cstddef>
#include <iterator>
#include <ranges>
#include <type_traits>
 
#include "v8config.h"  // NOLINT(build/include_directory)
 
namespace v8 {
 
template <typename T>
class V8_EXPORT MemorySpan;
 
}  // namespace v8
 
// Mark `MemorySpan` as satisfying the `view` and `borrowed_range` concepts.
// This should be done before the definition of `MemorySpan`, so that any
// inlined calls to range functionality use the correct specializations.
template <typename T>
inline constexpr bool std::ranges::enable_view<v8::MemorySpan<T>> = true;
template <typename T>
inline constexpr bool std::ranges::enable_borrowed_range<v8::MemorySpan<T>> =
    true;
 
namespace v8 {
 
template <typename T>
class V8_EXPORT MemorySpan {
 private:
  template <typename From, typename To>
  using is_array_convertible = std::is_convertible<From (*)[], To (*)[]>;
  template <typename From, typename To>
  static constexpr bool is_array_convertible_v =
      is_array_convertible<From, To>::value;
 
  template <typename It>
  using iter_reference_t = decltype(*std::declval<It&>());
 
  template <typename It, typename = void>
  struct is_compatible_iterator : std::false_type {};
  template <typename It>
  struct is_compatible_iterator<
      It,
      std::void_t<
          std::is_base_of<std::random_access_iterator_tag,
                          typename std::iterator_traits<It>::iterator_category>,
          is_array_convertible<std::remove_reference_t<iter_reference_t<It>>,
                               T>>> : std::true_type {};
  template <typename It>
  static constexpr bool is_compatible_iterator_v =
      is_compatible_iterator<It>::value;
 
  // SFINAE-compatible wrapper for `std::to_address()`.
  // Adapted from "base/types/to_address.h" in chromium.
  template <typename U>
    requires(!std::is_function_v<U>)
  [[nodiscard]] static constexpr U* to_address(U* p) noexcept {
    return p;
  }
 
  template <typename It>
    requires(
        requires(const It& it) { std::pointer_traits<It>::to_address(it); } ||
        requires(const It& it) { it.operator->(); })
  [[nodiscard]] static constexpr auto to_address(const It& it) noexcept {
    return std::to_address(it);
  }
 
 public:
  constexpr MemorySpan() = default;
 
  constexpr MemorySpan(std::nullptr_t, size_t) {}
 
  template <typename Iterator,
            std::enable_if_t<is_compatible_iterator_v<Iterator>, bool> = true>
  constexpr MemorySpan(Iterator first,
                       size_t count)  // NOLINT(runtime/explicit)
      : data_(to_address(first)), size_(count) {}
 
  template <typename Iterator,
            std::enable_if_t<is_compatible_iterator_v<Iterator> &&
                                 !std::is_convertible_v<Iterator, size_t>,
                             bool> = true>
  constexpr MemorySpan(Iterator first,
                       Iterator last)  // NOLINT(runtime/explicit)
      : data_(to_address(first)), size_(last - first) {}
 
  template <size_t N>
  constexpr MemorySpan(T (&a)[N]) noexcept  // NOLINT(runtime/explicit)
      : data_(a), size_(N) {}
 
  template <typename U, size_t N,
            std::enable_if_t<is_array_convertible_v<U, T>, bool> = true>
  constexpr MemorySpan(
      std::array<U, N>& a) noexcept  // NOLINT(runtime/explicit)
      : data_(a.data()), size_{N} {}
 
  template <typename U, size_t N,
            std::enable_if_t<is_array_convertible_v<const U, T>, bool> = true>
  constexpr MemorySpan(
      const std::array<U, N>& a) noexcept  // NOLINT(runtime/explicit)
      : data_(a.data()), size_{N} {}
 
  [[nodiscard]] constexpr T* data() const { return data_; }
  [[nodiscard]] constexpr size_t size() const { return size_; }
 
  [[nodiscard]] constexpr T& operator[](size_t i) const { return data_[i]; }
 
  [[nodiscard]] constexpr bool empty() const { return size() == 0; }
 
  class Iterator {
   public:
    using difference_type = std::ptrdiff_t;
    using value_type = T;
    using pointer = value_type*;
    using reference = value_type&;
    using iterator_category = std::random_access_iterator_tag;
    using iterator_concept = std::contiguous_iterator_tag;
 
    // Required to satisfy `std::semiregular<>`.
    constexpr Iterator() = default;
 
    [[nodiscard]] constexpr bool operator==(const Iterator&) const = default;
    [[nodiscard]] constexpr auto operator<=>(const Iterator&) const = default;
 
    constexpr Iterator& operator++() {
      ++ptr_;
      return *this;
    }
 
    constexpr Iterator operator++(int) {
      Iterator temp = *this;
      ++*this;
      return temp;
    }
 
    constexpr Iterator& operator--() {
      --ptr_;
      return *this;
    }
 
    constexpr Iterator operator--(int) {
      Iterator temp = *this;
      --*this;
      return temp;
    }
 
    constexpr Iterator& operator+=(difference_type rhs) {
      ptr_ += rhs;
      return *this;
    }
 
    [[nodiscard]] friend constexpr Iterator operator+(Iterator lhs,
                                                      difference_type rhs) {
      lhs += rhs;
      return lhs;
    }
 
    [[nodiscard]] friend constexpr Iterator operator+(difference_type lhs,
                                                      const Iterator& rhs) {
      return rhs + lhs;
    }
 
    constexpr Iterator& operator-=(difference_type rhs) {
      ptr_ -= rhs;
      return *this;
    }
 
    [[nodiscard]] friend constexpr Iterator operator-(Iterator lhs,
                                                      difference_type rhs) {
      lhs -= rhs;
      return lhs;
    }
 
    [[nodiscard]] friend constexpr difference_type operator-(
        const Iterator& lhs, const Iterator& rhs) {
      return lhs.ptr_ - rhs.ptr_;
    }
 
    [[nodiscard]] constexpr reference operator*() const { return *ptr_; }
    [[nodiscard]] constexpr pointer operator->() const { return ptr_; }
    [[nodiscard]] constexpr reference operator[](size_t offset) const {
      return ptr_[offset];
    }
 
   private:
    friend class MemorySpan<T>;
 
    constexpr explicit Iterator(T* ptr) : ptr_(ptr) {}
 
    T* ptr_ = nullptr;
  };
 
  [[nodiscard]] Iterator begin() const { return Iterator(data_); }
  [[nodiscard]] Iterator end() const { return Iterator(data_ + size_); }
 
 private:
  T* data_ = nullptr;
  size_t size_ = 0;
};
 
namespace detail {
template <class T, std::size_t N, std::size_t... I>
[[nodiscard]] constexpr std::array<std::remove_cv_t<T>, N> to_array_lvalue_impl(
    T (&a)[N], std::index_sequence<I...>) {
  return {{a[I]...}};
}
 
template <class T, std::size_t N, std::size_t... I>
[[nodiscard]] constexpr std::array<std::remove_cv_t<T>, N> to_array_rvalue_impl(
    T (&&a)[N], std::index_sequence<I...>) {
  return {{std::move(a[I])...}};
}
}  // namespace detail
 
template <class T, std::size_t N>
[[nodiscard]] constexpr std::array<std::remove_cv_t<T>, N> to_array(T (&a)[N]) {
  return detail::to_array_lvalue_impl(a, std::make_index_sequence<N>{});
}
 
template <class T, std::size_t N>
[[nodiscard]] constexpr std::array<std::remove_cv_t<T>, N> to_array(
    T (&&a)[N]) {
  return detail::to_array_rvalue_impl(std::move(a),
                                      std::make_index_sequence<N>{});
}
 
}  // namespace v8
#endif  // INCLUDE_V8_MEMORY_SPAN_H_