fpenvironment.h

Go to the documentation of this file.

/*
 *
 *    Copyright (c) 2015-2018
 *      SMASH Team
 *
 *    GNU General Public License (GPLv3 or later)
 *
 */

#ifndef SRC_INCLUDE_FPENVIRONMENT_H_
#define SRC_INCLUDE_FPENVIRONMENT_H_

#include <cfenv>

namespace smash {

#if defined _GNU_SOURCE && !defined __clang__

inline bool enable_float_traps(int mask) { return -1 != feenableexcept(mask); }
#elif defined __SSE__ && !defined __clang__
bool enable_float_traps(int mask);
#else
inline bool enable_float_traps(int) { return false; }
#endif

void setup_default_float_traps();

class DisableFloatTraps {
 public:
  explicit DisableFloatTraps(int mask = 0) {
    std::feholdexcept(&environment_);
    if (mask != 0) {
      reenable_traps(mask);
    }
  }

  ~DisableFloatTraps() { std::fesetenv(&environment_); }

 private:
  void reenable_traps(int mask);

  std::fenv_t environment_;
};

template <typename F>
void without_float_traps(F &&f) {
  DisableFloatTraps guard;
  f();
}

}  // namespace smash

#endif  // SRC_INCLUDE_FPENVIRONMENT_H_