ArrayViewCommon.h

// -*- tab-width: 2; indent-tabs-mode: nil; coding: utf-8-with-signature -*-
//-----------------------------------------------------------------------------
// Copyright 2000-2026 CEA (www.cea.fr) IFPEN (www.ifpenergiesnouvelles.com)
// See the top-level COPYRIGHT file for details.
// SPDX-License-Identifier: Apache-2.0
//-----------------------------------------------------------------------------
/*---------------------------------------------------------------------------*/
/* ArrayViewCommon.h                                           (C) 2000-2025 */
/*                                                                           */
/* Common declarations for the ArrayView, ConstArrayView, and Span classes.  */
/*---------------------------------------------------------------------------*/
#ifndef ARCCORE_BASE_ARRAYVIEWCOMMON_H
#define ARCCORE_BASE_ARRAYVIEWCOMMON_H
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#include "arccore/base/ArrayIterator.h"
 
#include <iostream>
 
// 'assert' is necessary for accelerator code
#include <assert.h>
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace Arcane::impl
{
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/

template <typename ViewType> ARCCORE_HOST_DEVICE auto subViewInterval(ViewType view,
                                                                      typename ViewType::size_type index,
                                                                      typename ViewType::size_type nb_interval) -> ViewType
{
  using size_type = typename ViewType::size_type;
  if (nb_interval <= 0)
    return ViewType();
  if (index < 0 || index >= nb_interval)
    return ViewType();
  size_type n = view.size();
  size_type isize = n / nb_interval;
  size_type ibegin = index * isize;
  // For the last interval, take the remaining elements
  if ((index + 1) == nb_interval)
    isize = n - ibegin;
  return ViewType::create(view.data() + ibegin, isize);
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/

template <typename ViewType> inline void
dumpArray(std::ostream& o, ViewType val, int max_print)
{
  using size_type = typename ViewType::size_type;
  size_type n = val.size();
  if (max_print > 0 && n > max_print) {
    // Only displays the first (max_print/2) and the last (max_print/2)
    // otherwise if the array is very large it can generate enormous
    // output listings.
    size_type z = (max_print / 2);
    size_type z2 = n - z;
    o << "[0]=\"" << val[0] << '"';
    for (size_type i = 1; i < z; ++i)
      o << " [" << i << "]=\"" << val[i] << '"';
    o << " ... ... (skipping indexes " << z << " to " << z2 << " ) ... ... ";
    for (size_type i = (z2 + 1); i < n; ++i)
      o << " [" << i << "]=\"" << val[i] << '"';
  }
  else {
    for (size_type i = 0; i < n; ++i) {
      if (i != 0)
        o << ' ';
      o << "[" << i << "]=\"" << val[i] << '"';
    }
  }
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/

template <typename ViewType> inline bool
areEqual(ViewType rhs, ViewType lhs)
{
  using size_type = typename ViewType::size_type;
  if (rhs.size() != lhs.size())
    return false;
  size_type s = rhs.size();
  for (size_type i = 0; i < s; ++i) {
    if (rhs[i] != lhs[i])
      return false;
  }
  return true;
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/

template <typename View2DType> inline bool
areEqual2D(View2DType rhs, View2DType lhs)
{
  using size_type = typename View2DType::size_type;
  const size_type dim1_size = rhs.dim1Size();
  const size_type dim2_size = rhs.dim2Size();
  if (dim1_size != lhs.dim1Size())
    return false;
  if (dim2_size != lhs.dim2Size())
    return false;
  for (size_type i = 0; i < dim1_size; ++i) {
    for (size_type j = 0; j < dim2_size; ++j) {
      if (rhs(i, j) != lhs(i, j))
        return false;
    }
  }
  return true;
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/

extern "C++" ARCCORE_BASE_EXPORT void
arccoreThrowTooBigInteger [[noreturn]] (std::size_t size);
 
extern "C++" ARCCORE_BASE_EXPORT void
arccoreThrowTooBigInt64 [[noreturn]] (std::size_t size);
 
extern "C++" ARCCORE_BASE_EXPORT void
arccoreThrowNegativeSize [[noreturn]] (Int64 size);
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/

inline constexpr ARCCORE_HOST_DEVICE void
arccoreCheckIsPositive(Int64 size)
{
  if (size < 0) {
#ifdef ARCCORE_DEVICE_CODE
    assert("'size' is negative");
#else
    impl::arccoreThrowNegativeSize(size);
#endif
  }
}

inline constexpr ARCCORE_HOST_DEVICE void
arccoreCheckIsValidInteger(Int64 size)
{
  if (size >= ARCCORE_INTEGER_MAX) {
#ifdef ARCCORE_DEVICE_CODE
    assert("'size' is bigger than ARCCORE_INTEGER_MAX");
#else
    impl::arccoreThrowTooBigInteger(size);
#endif
  }
}

inline constexpr ARCCORE_HOST_DEVICE void
arccoreCheckIsValidInt64(size_t size)
{
  if (size >= ARCCORE_INT64_MAX) {
#ifdef ARCCORE_DEVICE_CODE
    assert("'size' is bigger than ARCCORE_INT64_MAX");
#else
    impl::arccoreThrowTooBigInt64(size);
#endif
  }
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
} // namespace Arcane::impl
 
namespace Arcane
{
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/

inline constexpr ARCCORE_HOST_DEVICE Integer
arccoreCheckArraySize(unsigned long long size)
{
  impl::arccoreCheckIsValidInteger(size);
  return static_cast<Integer>(size);
}

inline constexpr Integer
arccoreCheckArraySize(long long size)
{
  impl::arccoreCheckIsValidInteger(size);
  impl::arccoreCheckIsPositive(size);
  return static_cast<Integer>(size);
}

inline constexpr ARCCORE_BASE_EXPORT Integer
arccoreCheckArraySize(unsigned long size)
{
  impl::arccoreCheckIsValidInteger(size);
  return static_cast<Integer>(size);
}

inline constexpr ARCCORE_HOST_DEVICE Integer
arccoreCheckArraySize(long size)
{
  impl::arccoreCheckIsValidInteger(size);
  impl::arccoreCheckIsPositive(size);
  return static_cast<Integer>(size);
}

inline constexpr ARCCORE_HOST_DEVICE Integer
arccoreCheckArraySize(unsigned int size)
{
  impl::arccoreCheckIsValidInteger(size);
  return static_cast<Integer>(size);
}

inline constexpr ARCCORE_HOST_DEVICE Integer
arccoreCheckArraySize(int size)
{
  impl::arccoreCheckIsValidInteger(size);
  impl::arccoreCheckIsPositive(size);
  return static_cast<Integer>(size);
}

inline constexpr ARCCORE_HOST_DEVICE Int64
arccoreCheckLargeArraySize(size_t size)
{
  impl::arccoreCheckIsValidInt64(size);
  return static_cast<Int64>(size);
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
template <typename IntType> class ArraySizeChecker;

template <>
class ArraySizeChecker<Int32>
{
 public:
 
  template <typename SizeType> ARCCORE_HOST_DEVICE static Int32 check(SizeType size)
  {
    return arccoreCheckArraySize(size);
  }
};

template <>
class ArraySizeChecker<Int64>
{
 public:
 
  static ARCCORE_HOST_DEVICE Int64 check(std::size_t size)
  {
    return arccoreCheckLargeArraySize(size);
  }
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
} // namespace Arcane
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace Arccore::impl
{
using Arcane::impl::arccoreCheckIsPositive;
using Arcane::impl::arccoreCheckIsValidInt64;
using Arcane::impl::arccoreCheckIsValidInteger;
using Arcane::impl::arccoreThrowNegativeSize;
using Arcane::impl::arccoreThrowTooBigInt64;
using Arcane::impl::arccoreThrowTooBigInteger;
using Arcane::impl::areEqual;
using Arcane::impl::areEqual2D;
using Arcane::impl::dumpArray;
using Arcane::impl::subViewInterval;
} // namespace Arccore::impl
 
namespace Arccore
{
using Arcane::arccoreCheckArraySize;
using Arcane::arccoreCheckLargeArraySize;
} // namespace Arccore
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#endif