#ifndef FPU_H
#define FPU_H

#ifdef __GNU_VISIBLE
#undef __GNU_VISIBLE
#define __GNU_VISIBLE 1
#endif

#include "common.h"

#ifdef HAVE_FEENABLEEXCEPT
#include <fenv.h>

inline int fpu_exceptions() {
  int excepts=0;
#ifdef FE_INVALID
  excepts |= FE_INVALID;
#endif
#ifdef FE_DIVBYZERO
  excepts |= FE_DIVBYZERO;
#endif
#ifdef FE_OVERFLOW
  excepts |= FE_OVERFLOW;
#endif
  return excepts;
}

inline void fpu_trap(bool trap=true)
{
  // Conditionally trap FPU exceptions on NaN, zero divide and overflow.
  if(trap) feenableexcept(fpu_exceptions());
  else fedisableexcept(fpu_exceptions());
}

#elif defined(_WIN32)
inline void fpu_trap(bool trap = true)
{
  unsigned int mask = _EM_INVALID | _EM_ZERODIVIDE | _EM_OVERFLOW;
  unsigned int current_word = 0;
  _controlfp_s(&current_word, trap ? 0 : mask, mask);
}

#else
inline void fpu_trap(bool=true) {}
#endif

#endif