EBGeometry_Array.hpp

Go to the documentation of this file.

// SPDX-FileCopyrightText: 2026 Robert Marskar <robert.marskar@sintef.no>
//
// SPDX-License-Identifier: GPL-3.0-or-later
 
#ifndef EBGEOMETRY_ARRAY_HPP
#define EBGEOMETRY_ARRAY_HPP
 
#include <cstddef>
 
#include "EBGeometry_GPU.hpp"
 
namespace EBGeometry {
 
template <class T, std::size_t N>
struct Array
{
  static_assert(N > 0, "Array<T, N> requires N > 0");
 
  T m_data[N];
 
  [[nodiscard]] EBGEOMETRY_HOST_DEVICE
  constexpr T&
  operator[](const std::size_t a_i) noexcept
  {
    return m_data[a_i];
  }
 
  [[nodiscard]] EBGEOMETRY_HOST_DEVICE
  constexpr const T&
  operator[](const std::size_t a_i) const noexcept
  {
    return m_data[a_i];
  }
 
  [[nodiscard]] EBGEOMETRY_HOST_DEVICE
  static constexpr std::size_t
  size() noexcept
  {
    return N;
  }
 
  [[nodiscard]] EBGEOMETRY_HOST_DEVICE
  constexpr T*
  data() noexcept
  {
    return m_data;
  }
 
  [[nodiscard]] EBGEOMETRY_HOST_DEVICE
  constexpr const T*
  data() const noexcept
  {
    return m_data;
  }
 
  [[nodiscard]] EBGEOMETRY_HOST_DEVICE
  constexpr T*
  begin() noexcept
  {
    return m_data;
  }
 
  [[nodiscard]] EBGEOMETRY_HOST_DEVICE
  constexpr const T*
  begin() const noexcept
  {
    return m_data;
  }
 
  [[nodiscard]] EBGEOMETRY_HOST_DEVICE
  constexpr T*
  end() noexcept
  {
    return m_data + N;
  }
 
  [[nodiscard]] EBGEOMETRY_HOST_DEVICE
  constexpr const T*
  end() const noexcept
  {
    return m_data + N;
  }
 
  EBGEOMETRY_HOST_DEVICE
  constexpr void
  fill(const T& a_value) noexcept
  {
    for (std::size_t i = 0; i < N; i++) {
      m_data[i] = a_value;
    }
  }
 
  [[nodiscard]] EBGEOMETRY_HOST_DEVICE
  constexpr bool
  operator==(const Array& a_other) const noexcept
  {
    for (std::size_t i = 0; i < N; i++) {
      if (!(m_data[i] == a_other.m_data[i])) {
        return false;
      }
    }
 
    return true;
  }
 
  [[nodiscard]] EBGEOMETRY_HOST_DEVICE
  constexpr bool
  operator!=(const Array& a_other) const noexcept
  {
    return !(*this == a_other);
  }
};
 
} // namespace EBGeometry
 
#endif