ExtendedFloat.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: The Bio++ Development Group
//
// SPDX-License-Identifier: CECILL-2.1
 
#ifndef BPP_PHYL_LIKELIHOOD_DATAFLOW_EXTENDEDFLOAT_H
#define BPP_PHYL_LIKELIHOOD_DATAFLOW_EXTENDEDFLOAT_H
 
#include <Bpp/Exceptions.h>
#include <Eigen/Core>
#include <cmath>
#include <iostream>
#include <limits>
#include <string>
 
 
namespace bpp
{
template<typename T> constexpr T constexpr_power (T d, int n)
{
  return n == 0 ? 1.0 : (n > 0 ? constexpr_power (d, n - 1) * d : constexpr_power (d, n + 1) / d);
}
 
// Compute power of integers
inline int powi (int base, unsigned int exp)
{
  int res = 1;
  while (exp)
  {
    if (exp & 1)
      res *= base;
    exp >>= 1;
    base *= base;
  }
  return res;
}
 
class ExtendedFloat
{
  // Assumes positive integer
 
public:
  using FloatType = double;
  using ExtType = int;
 
  // Parameter: decide how much product we can do safely before having to normalize (smaller -> less normalizations)
  static constexpr int allowed_product_without_normalization = 2;
 
  // Radix is the float exponent base
  static constexpr int radix = std::numeric_limits<FloatType>::radix;
 
  const static double ln_radix;
 
  // biggest_repr_radix_power = max { n ; radix^n is representable }
  static constexpr int biggest_repr_radix_power = std::numeric_limits<FloatType>::max_exponent - 1;
 
  // biggest_normalized_radix_power = max { n ; (radix^n)^allowed_product_without_normalization is representable }
  static constexpr int biggest_normalized_radix_power =
      biggest_repr_radix_power / allowed_product_without_normalization;
 
 
  // biggest_normalized_value = max { f ; f^allowed_product_without_normalization is representable }
  static constexpr FloatType biggest_normalized_value =
      constexpr_power (FloatType (radix), biggest_normalized_radix_power);
 
  // smallest_repr_radix_power = min { n ; radix^n is representable }
  static constexpr int smallest_repr_radix_power = std::numeric_limits<FloatType>::min_exponent + 1;
 
  // smallest_normalized_radix_power = min { n ; (radix^n)^allowed_product_denorm is representable }
  static constexpr int smallest_normalized_radix_power =
      -((-smallest_repr_radix_power) / allowed_product_without_normalization);
 
  // smallest_normalized_value = min { f ; f^allowed_product_denorm is representable }
  static constexpr FloatType smallest_normalized_value =
      constexpr_power (FloatType (radix), smallest_normalized_radix_power);
 
  // factors to scale f_ to renormalize.
  static constexpr FloatType normalize_big_factor = 1. / biggest_normalized_value;
  static constexpr FloatType normalize_small_factor = 1. / smallest_normalized_value;
 
  // TODO add denorm info for sum
 
  constexpr ExtendedFloat (FloatType f = 0.0, ExtType e = 0) noexcept : f_ (f), exp_ (e)
  {}
 
  ExtendedFloat (const ExtendedFloat& ef) noexcept : f_ (ef.f_), exp_ (ef.exp_)
  {}
 
  const FloatType& float_part () const noexcept { return f_; }
  const ExtType& exponent_part () const noexcept { return exp_; }
 
  bool normalize_big () noexcept
  {
    if (std::isfinite (f_))
    {
      bool normalized = false;
      while (std::abs(f_) > biggest_normalized_value)
      {
        f_ *= (double)normalize_big_factor;
        exp_ += biggest_normalized_radix_power;
        normalized = true;
      }
      return normalized;
    }
    return false;
  }
 
  bool normalize_small ()
  {
    if (f_ != 0)
    {
      bool normalized = false;
      while (std::abs(f_) < smallest_normalized_value)
      {
        f_ *= (double)normalize_small_factor;
        exp_ += smallest_normalized_radix_power;
        normalized = true;
      }
      return normalized;
    }
    return false;
  }
 
  void normalize () noexcept
  {
    if (!normalize_big())
    {
      normalize_small();
    }
  }
 
  // Static methods without normalization
  inline static ExtendedFloat denorm_mul (const ExtendedFloat& lhs, const ExtendedFloat& rhs)
  {
    return {lhs.float_part () * rhs.float_part (), lhs.exponent_part () + rhs.exponent_part ()};
  }
 
  inline static ExtendedFloat denorm_div (const ExtendedFloat& lhs, const ExtendedFloat& rhs)
  {
    return {lhs.float_part () / rhs.float_part (), lhs.exponent_part () - rhs.exponent_part ()};
  }
 
  inline static ExtendedFloat denorm_add (const ExtendedFloat& lhs, const ExtendedFloat& rhs)
  {
    return (lhs.exponent_part () >= rhs.exponent_part ()) ?
           ExtendedFloat(lhs.float_part () + rhs.float_part () * constexpr_power<double>(ExtendedFloat::radix, rhs.exponent_part () - lhs.exponent_part ()), lhs.exponent_part ()) :
           ExtendedFloat(rhs.float_part () + lhs.float_part () * constexpr_power<double>(ExtendedFloat::radix, lhs.exponent_part () - rhs.exponent_part ()), rhs.exponent_part ());
  }
 
  inline static ExtendedFloat denorm_sub (const ExtendedFloat& lhs, const ExtendedFloat& rhs)
  {
    return (lhs.exponent_part () >= rhs.exponent_part ()) ?
           ExtendedFloat(lhs.float_part () - rhs.float_part () * constexpr_power<double>(ExtendedFloat::radix, rhs.exponent_part () - lhs.exponent_part ()), lhs.exponent_part ()) :
           ExtendedFloat(rhs.float_part () - lhs.float_part () * constexpr_power<double>(ExtendedFloat::radix, lhs.exponent_part () - rhs.exponent_part ()), rhs.exponent_part ());
  }
 
  inline static ExtendedFloat denorm_sub (const ExtendedFloat& lhs, const double& rhs)
  {
    return (lhs.exponent_part () >= 0) ?
           ExtendedFloat(lhs.float_part () - rhs * constexpr_power<double>(ExtendedFloat::radix, -lhs.exponent_part ()), lhs.exponent_part ()) :
           ExtendedFloat(rhs - lhs.float_part () * constexpr_power<double>(ExtendedFloat::radix, lhs.exponent_part ()), 0);
  }
 
  inline static ExtendedFloat denorm_pow (const ExtendedFloat& lhs, double exp)
  {
    double b = lhs.exponent_part() * exp;
    ExtendedFloat::ExtType e = ExtendedFloat::ExtType(std::lround(b));
    ExtendedFloat r(std::pow(lhs.float_part(), exp) * std::pow(ExtendedFloat::radix, (b - e)), e);
    return r;
  }
 
  inline static ExtendedFloat denorm_pow (const ExtendedFloat& lhs, int exp)
  {
    if (exp == 0)
      return ExtendedFloat(1.0);
    if (exp & 1)
      return exp > 0 ? denorm_mul(lhs, denorm_pow(lhs, exp - 1)) : denorm_div(denorm_pow(lhs, exp + 1), lhs);
    else
    {
      ExtendedFloat r2(std::pow(lhs.float_part(), 2));
      r2.normalize();
      auto r2k = denorm_pow(r2, exp >> 1);
      r2k.normalize();
      ExtendedFloat r(r2k.float_part(), r2k.exponent_part() + lhs.exponent_part() * exp);
      return r;
    }
  }
 
 
  /*********************************
  ** Utilities
  *********************************/
  inline ExtendedFloat& operator=(const ExtendedFloat& ef)
  {
    f_ = ef.float_part();
    exp_ = ef.exponent_part();
    return *this;
  }
 
  inline ExtendedFloat operator+(const ExtendedFloat& rhs) const
  {
    auto r = denorm_add (*this, rhs);
    r.normalize ();
    return r;
  }
 
  inline ExtendedFloat operator-(const ExtendedFloat& rhs) const
  {
    auto r = denorm_sub (*this, rhs);
    r.normalize ();
    return r;
  }
 
  template<typename F, typename = typename std::enable_if<std::is_arithmetic<F>::value>::type>
  inline ExtendedFloat operator-(const F& rhs) const
  {
    auto r = denorm_sub (*this, rhs);
//      r.normalize ();
    return r;
  }
 
  inline ExtendedFloat operator*(const ExtendedFloat& rhs) const
  {
    auto r = denorm_mul (*this, rhs);
    r.normalize ();
    return r;
  }
 
  template<typename F, typename = typename std::enable_if<std::is_arithmetic<F>::value>::type>
  inline ExtendedFloat operator*(const F& rhs) const
  {
    ExtendedFloat r(float_part () * rhs, exponent_part ());
    r.normalize ();
    return r;
  }
 
  inline ExtendedFloat operator/(const ExtendedFloat& rhs) const
  {
    auto r = denorm_div (*this, rhs);
    r.normalize ();
    return r;
  }
 
  inline ExtendedFloat& operator*=(const ExtendedFloat& rhs)
  {
    float_part () *= rhs.float_part ();
    exponent_part () += rhs.exponent_part ();
    normalize();
    return *this;
  }
 
  inline ExtendedFloat& operator/=(const ExtendedFloat& rhs)
  {
    float_part () /= rhs.float_part ();
    exponent_part () -= rhs.exponent_part ();
    normalize();
    return *this;
  }
 
  inline ExtendedFloat& operator+=(const ExtendedFloat& rhs)
  {
    if (exponent_part () >= rhs.exponent_part ())
    {
      float_part() += rhs.float_part () * constexpr_power<double>(ExtendedFloat::radix, rhs.exponent_part () - exponent_part ());
    }
    else
    {
      float_part() = rhs.float_part () + float_part () * constexpr_power<double>(ExtendedFloat::radix, exponent_part () - rhs.exponent_part ());
      exponent_part() = rhs.exponent_part ();
    }
    normalize ();
    return *this;
  }
 
  inline ExtendedFloat& operator-=(const ExtendedFloat& rhs)
  {
    if (exponent_part () >= rhs.exponent_part ())
    {
      float_part() -= rhs.float_part () * constexpr_power<double>(ExtendedFloat::radix, rhs.exponent_part () - exponent_part ());
    }
    else
    {
      float_part() = rhs.float_part () - float_part () * constexpr_power<double>(ExtendedFloat::radix, exponent_part () - rhs.exponent_part ());
      exponent_part() = rhs.exponent_part ();
    }
    normalize ();
    return *this;
  }
 
  inline ExtendedFloat operator-() const
  {
    return ExtendedFloat(-float_part(), exponent_part());
  }
 
  inline ExtendedFloat pow (double exp) const
  {
    auto r = denorm_pow(*this, exp);
    r.normalize ();
    return r;
  }
 
  inline ExtendedFloat pow (int exp) const
  {
    auto r = denorm_pow(*this, exp);
    r.normalize ();
    return r;
  }
 
  /*
   * Tests
   *
   */
  inline bool operator==(const ExtendedFloat& rhs) const
  {
    return float_part() == rhs.float_part() && exponent_part() == rhs.exponent_part();
  }
 
  inline bool operator!=(const ExtendedFloat& rhs) const
  {
    return float_part() != rhs.float_part() || exponent_part() != rhs.exponent_part();
  }
 
  inline bool operator<(const ExtendedFloat& rhs) const
  {
    return exponent_part() < rhs.exponent_part() || (exponent_part() == rhs.exponent_part() && float_part() < rhs.float_part());
  }
 
  inline bool operator<=(const ExtendedFloat& rhs) const
  {
    return exponent_part() < rhs.exponent_part() || (exponent_part() == rhs.exponent_part() && float_part() <= rhs.float_part());
  }
 
  inline bool operator>=(const ExtendedFloat& rhs) const
  {
    return exponent_part() > rhs.exponent_part() || (exponent_part() == rhs.exponent_part() && float_part() >= rhs.float_part());
  }
 
  inline bool operator>=(const double& rhs) const
  {
    return convert(*this) >= rhs;
  }
 
  inline bool operator>(const ExtendedFloat& rhs) const
  {
    return exponent_part() > rhs.exponent_part() || (exponent_part() == rhs.exponent_part() && float_part() > rhs.float_part());
  }
 
  inline bool operator>(const double& rhs) const
  {
    return convert(*this) > rhs;
  }
 
  inline double log () const
  {
    return std::log (float_part ()) + static_cast<double>(exponent_part ()) * ln_radix;
  }
 
  inline ExtendedFloat abs () const
  {
    return ExtendedFloat(std::abs (float_part ()), exponent_part ());
  }
 
  // Compute lround, and return tuple <lround, remainder>
  inline std::tuple<int, double> lround() const
  {
    throw Exception("ExtendedFloat::lround need to be checked.");
    auto c = float_part ();
    auto b = exponent_part();
    if (b <= 0)
    {
      double t = convert(*this);
      auto d = std::lround(t);
      return std::tuple<int, double>{d, remainder(t, 1)};
    }
    if (b > 0)
    {
      long long res(0);
      while (b > 0)
      {
        auto lrc = std::lround(c);
        res += lrc * powi(radix, uint(b));
        c -= (double)lrc;
        c *= biggest_normalized_value;
        b -= biggest_normalized_radix_power;
      }
      auto t = c * constexpr_power<double>(radix, b);
      auto it = std::lround(t);
      return std::tuple<int, double>{res + it, remainder(t, 1)};
    }
  }
 
 
  // exp(a.r^b)=r^(a/log(r) * r^b) = r^(c * r^b) = r^([c * r^b]) * r^(c * r^b - [c * r^b])
  // with c=a/log(r)
 
  inline ExtendedFloat exp () const
  {
    auto c = float_part () / ln_radix;
    auto ef = ExtendedFloat(c, exponent_part());
    auto u = ef.lround();
    return ExtendedFloat(std::pow(radix, std::get<1>(u)), (int)std::get<0>(u));
  }
 
  /*************/
  /* Dirty trick to allow for Eigen::Nullary_wrapers output only the float part, for ExtendedFloatEigen */
 
private:
  /* Necessary to keep this private to prevent misuse */
  operator double() const
  {
    return float_part();
  }
 
  template<typename Scalar, typename NullaryOp, bool has_nullary, bool has_unary, bool has_binary>
  friend struct Eigen::internal::nullary_wrapper;
 
public:
  // !!! no check on the validation of the conversion
  static inline double convert(const ExtendedFloat& ef)
  {
    return ef.float_part () * constexpr_power<double>(ExtendedFloat::radix, ef.exponent_part ());
  }
 
protected:
  FloatType f_;
  ExtType exp_;
 
  FloatType& float_part () noexcept { return f_; }
  ExtType& exponent_part () noexcept { return exp_; }
};
 
inline std::ostream& operator<<(std::ostream& os, const ExtendedFloat& ef)
{
  os << ef.float_part () << " * 2^" << ef.exponent_part ();
  return os;
}
 
inline std::string to_string (const ExtendedFloat& ef)
{
  using std::to_string;
  return "double(" + to_string (ef.float_part ()) + " * 2^" + to_string (ef.exponent_part ()) + ")";
}
 
inline double log (const ExtendedFloat& ef)
{
  return ef.log();
}
 
inline ExtendedFloat operator*(const double& lhs, const ExtendedFloat& rhs)
{
  return rhs * lhs;
}
 
inline ExtendedFloat operator-(const double& lhs, const ExtendedFloat& rhs)
{
  return rhs.ExtendedFloat::operator+(-lhs);
}
 
inline ExtendedFloat operator+(const double& lhs, const ExtendedFloat& rhs)
{
  return rhs.ExtendedFloat::operator+(lhs);
}
 
inline ExtendedFloat operator/(const double& lhs, const ExtendedFloat& rhs)
{
  return rhs * (1 / lhs);
}
 
 
inline ExtendedFloat pow (const ExtendedFloat& ef,  double exp)
{
  return ef.pow(exp);
}
 
inline ExtendedFloat exp (const ExtendedFloat& ef)
{
  return ef.exp();
}
 
inline ExtendedFloat abs (const ExtendedFloat& ef)
{
  return ef.abs();
}
 
 
// !!! no check on the validation of the conversion
inline double convert(const bpp::ExtendedFloat& ef)
{
  return ef.float_part () * bpp::constexpr_power<double>(bpp::ExtendedFloat::radix, ef.exponent_part ());
}
} // namespace bpp
 
 
/*
 * Storing ExtendedFloat in Eigen objects
 *
 */
 
 
namespace Eigen
{
template<>
struct NumTraits<bpp::ExtendedFloat> :
  NumTraits<double> // permits to get the epsilon, dummy_precision, lowest, highest functions
{
  typedef bpp::ExtendedFloat Real;
  typedef bpp::ExtendedFloat NonInteger;
  typedef bpp::ExtendedFloat Nested;
  enum
  {
    IsComplex = 0,
    IsInteger = 0,
    IsSigned = 1,
    RequireInitialization = 1,
    ReadCost = 1,
    AddCost = 3,
    MulCost = 2
  };
};
template<typename BinaryOp>
struct ScalarBinaryOpTraits<bpp::ExtendedFloat, double, BinaryOp> { typedef bpp::ExtendedFloat ReturnType;  };
 
template<typename BinaryOp>
struct ScalarBinaryOpTraits<double, bpp::ExtendedFloat, BinaryOp> { typedef bpp::ExtendedFloat ReturnType;  };
}
#endif // BPP_PHYL_LIKELIHOOD_DATAFLOW_EXTENDEDFLOAT_H