// This file is part of Eigen, a lightweight C++ template library
// for linear algebra.
//
// Copyright (C) 2022, The Eigen authors.
//
// 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/.

#ifndef EIGEN_CORE_UTIL_ASSERT_H
#define EIGEN_CORE_UTIL_ASSERT_H

// Eigen custom assert function.
//
// The combination of Eigen's relative includes and cassert's `assert` function
// (or any usage of the __FILE__ macro) can lead to ODR issues:
// a header included using different relative paths in two different TUs will
// have two different token-for-token definitions, since __FILE__ is expanded
// as an in-line string with different values.  Normally this would be
// harmless - the linker would just choose one definition. However, it breaks
// with C++20 modules when functions in different modules have different
// definitions.
//
// To get around this, we need to use __builtin_FILE() when available, which is
// considered a single token, and thus satisfies the ODR.

// Only define eigen_plain_assert if we are debugging, and either
//  - we are not compiling for GPU, or
//  - gpu debugging is enabled.
#if !defined(EIGEN_NO_DEBUG) && (!defined(EIGEN_GPU_COMPILE_PHASE) || !defined(EIGEN_NO_DEBUG_GPU))

#include <cassert>

#ifndef EIGEN_USE_CUSTOM_PLAIN_ASSERT
// Disable new custom asserts by default for now.
#define EIGEN_USE_CUSTOM_PLAIN_ASSERT  0
#endif

#if EIGEN_USE_CUSTOM_PLAIN_ASSERT

#ifndef EIGEN_HAS_BUILTIN_FILE
// Clang can check if __builtin_FILE() is supported.
// GCC > 5, MSVC 2019 14.26 (1926) all have __builtin_FILE().
// 
// For NVCC, it's more complicated.  Through trial-and-error:
//   - nvcc+gcc supports __builtin_FILE() on host, and on device after CUDA 11.
//   - nvcc+msvc supports __builtin_FILE() only after CUDA 11.
#if (EIGEN_HAS_BUILTIN(__builtin_FILE) && (EIGEN_COMP_CLANG || !defined(EIGEN_CUDA_ARCH))) || \
    (EIGEN_GNUC_STRICT_AT_LEAST(5, 0, 0) && (EIGEN_COMP_NVCC >= 110000 || !defined(EIGEN_CUDA_ARCH))) || \
    (EIGEN_COMP_MSVC >= 1926 && (!EIGEN_COMP_NVCC || EIGEN_COMP_NVCC >= 110000))
#define EIGEN_HAS_BUILTIN_FILE 1
#else
#define EIGEN_HAS_BUILTIN_FILE 0
#endif
#endif  // EIGEN_HAS_BUILTIN_FILE

#if EIGEN_HAS_BUILTIN_FILE
#  define EIGEN_BUILTIN_FILE __builtin_FILE()
#  define EIGEN_BUILTIN_LINE __builtin_LINE()
#else
// Default (potentially unsafe) values.
#  define EIGEN_BUILTIN_FILE __FILE__
#  define EIGEN_BUILTIN_LINE __LINE__
#endif

// Use __PRETTY_FUNCTION__ when available, since it is more descriptive, as
// __builtin_FUNCTION() only returns the undecorated function name.
// This should still be okay ODR-wise since it is a compiler-specific fixed
// value.  Mixing compilers will likely lead to ODR violations anyways.
#if EIGEN_COMP_MSVC
#  define EIGEN_BUILTIN_FUNCTION __FUNCSIG__
#elif EIGEN_COMP_GNUC
#  define EIGEN_BUILTIN_FUNCTION __PRETTY_FUNCTION__
#else
#  define EIGEN_BUILTIN_FUNCTION __func__
#endif

namespace Eigen {
namespace internal {

// Generic default assert handler.
template<typename EnableIf = void, typename... EmptyArgs>
struct assert_handler_impl {
  EIGEN_DEVICE_FUNC EIGEN_DONT_INLINE
  static inline void run(const char* expression, const char* file, unsigned line, const char* function) {
#ifdef EIGEN_GPU_COMPILE_PHASE
    // GPU device code doesn't allow stderr or abort, so use printf and raise an
    // illegal instruction exception to trigger a kernel failure.
#ifndef EIGEN_NO_IO
    printf("Assertion failed at %s:%u in %s: %s\n",
      file == nullptr ? "<file>" : file,
      line,
      function == nullptr ? "<function>" : function,
      expression);
#endif
    __trap();
      
#else  // EIGEN_GPU_COMPILE_PHASE

    // Print to stderr and abort, as specified in <cassert>.
#ifndef EIGEN_NO_IO
    fprintf(stderr, "Assertion failed at %s:%u in %s: %s\n",
      file == nullptr ? "<file>" : file,
      line,
      function == nullptr ? "<function>" : function,
      expression);
#endif
    std::abort();
    
#endif  // EIGEN_GPU_COMPILE_PHASE
  }
};

// Use POSIX __assert_fail handler when available.
//
// This allows us to integrate with systems that have custom handlers.
//
// NOTE: this handler is not always available on all POSIX systems (otherwise
// we could simply test for __unix__ or similar).  The handler function name
// seems to depend on the specific toolchain implementation, and differs between
// compilers, platforms, OSes, etc.  Hence, we detect support via SFINAE.
template<typename... EmptyArgs>
struct assert_handler_impl<
    void_t<decltype(__assert_fail(
        (const char*)nullptr,         // expression
        (const char*)nullptr,         // file
        0,                            // line
        (const char*)nullptr,         // function
        std::declval<EmptyArgs>()...  // Empty substitution required for SFINAE.
    ))>, EmptyArgs... > {
  EIGEN_DEVICE_FUNC EIGEN_DONT_INLINE
  static inline void run(const char* expression, const char* file, unsigned line, const char* function) {
    // GCC requires this call to be dependent on the template parameters.
    __assert_fail(expression, file, line, function, std::declval<EmptyArgs>()...);
  }
};

EIGEN_DEVICE_FUNC EIGEN_DONT_INLINE
inline void __assert_handler(const char* expression, const char* file, unsigned line, const char* function) {
  assert_handler_impl<>::run(expression, file, line, function);
}

}  // namespace internal
}  // namespace Eigen

#define eigen_plain_assert(expression)                     \
  (EIGEN_PREDICT_FALSE(!(expression)) ?                    \
    Eigen::internal::__assert_handler(#expression,         \
                                      EIGEN_BUILTIN_FILE,  \
                                      EIGEN_BUILTIN_LINE,  \
                                      EIGEN_BUILTIN_FUNCTION) : (void)0)

#else  // EIGEN_USE_CUSTOM_PLAIN_ASSERT

// Use regular assert.
#define eigen_plain_assert(condition) assert(condition)

#endif  // EIGEN_USE_CUSTOM_PLAIN_ASSERT

#else   // EIGEN_NO_DEBUG

#define eigen_plain_assert(condition) ((void)0)

#endif  // EIGEN_NO_DEBUG

#endif  // EIGEN_CORE_UTIL_ASSERT_H