interval.hpp

// Copyright Take Vos 2020-2022.
// Distributed under the Boost Software License, Version 1.0.
// (See accompanying file LICENSE_1_0.txt or copy at https://www.boost.org/LICENSE_1_0.txt)
 
#pragma once
 
#include "utility.hpp"
#include "assert.hpp"
#include "concepts.hpp"
#include "rapid/numeric_array.hpp"
#include <type_traits>
#include <limits>
#include <concepts>
#include <algorithm>
#include <compare>
 
namespace hi::inline v1 {
 
template<numeric_limited T>
struct interval {
public:
    using value_type = T;
    using bound_type = numeric_array<value_type, 2>;
 
    bound_type v;
 
    constexpr interval(interval const &rhs) noexcept = default;
    constexpr interval(interval &&rhs) noexcept = default;
    constexpr interval &operator=(interval const &rhs) noexcept = default;
    constexpr interval &operator=(interval &&rhs) noexcept = default;
 
    constexpr interval() noexcept
    {
        if constexpr (std::is_floating_point_v<value_type>) {
            v[0] = -std::numeric_limits<value_type>::infinite();
            v[1] = -std::numeric_limits<value_type>::infinite();
        } else {
            v[0] = std::numeric_limits<value_type>::min();
            v[1] = -std::numeric_limits<value_type>::max();
        }
        hi_axiom(holds_invariant());
    }
 
    [[nodiscard]] constexpr interval(value_type lower, value_type upper) noexcept : v(lower, -upper)
    {
        hi_axiom(holds_invariant());
    }
 
    [[nodiscard]] static constexpr interval raw(bound_type bounds) noexcept
    {
        interval r;
        r.v = bounds;
        hi_axiom(r.holds_invariant());
        return r;
    }
 
    [[nodiscard]] constexpr bool holds_invariant() const noexcept
    {
        return v[0] <= -v[1];
    }
 
    [[nodiscard]] constexpr value_type lower() const noexcept
    {
        return v[0];
    }
 
    [[nodiscard]] constexpr value_type upper() const noexcept
    {
        return -v[1];
    }
 
    [[nodiscard]] constexpr value_type delta() const noexcept
    {
        return upper() - lower();
    }
 
    [[nodiscard]] constexpr bool is_value() const noexcept
    {
        return delta() == 0;
    }
 
    [[nodiscard]] constexpr bool is_range() const noexcept
    {
        return delta() > 0;
    }
 
    template<numeric_limited T>
    [[nodiscard]] constexpr bool type_contains_range() const noexcept
    {
        return std::numeric_limits<T>::lower() <= lower() and upper() <= std::numeric_limits<T>::max();
    }
 
    template<numeric_limited T>
    [[nodiscard]] constexpr bool range_contains_type() const noexcept
    {
        return lower() <= std::numeric_limits<T>::lower() and std::numeric_limits<T>::max() <= upper();
    }
 
    operator bool() const noexcept
    {
        return v[0] != 0 or v[1] != 0;
    }
 
    [[nodiscard]] constexpr interval operator-() const noexcept
    {
        return interval::raw(v.yx());
    }
 
    [[nodiscard]] constexpr interval operator+(interval const &rhs) const noexcept
    {
        return interval::raw(v + rhs.v);
    }
 
    [[nodiscard]] constexpr interval operator-(interval const &rhs) const noexcept
    {
        return *this + (-rhs);
    }
 
    [[nodiscard]] constexpr interval operator*(interval const &rhs) const noexcept
    {
        hilet ge_zero = ge(v, bound_type{});
        hilet lt_zero = ~ge_zero;
 
        hilet ac = (ge_zero & rhs.v.xx()) | (lt_zero & -rhs.v.yy());
        hilet db = (ge_zero & rhs.v.yy()) | (lt_zero & -rhs.v.xx());
 
        hilet ac_mul = ac * v;
        hilet db_mul = db * v;
 
        return raw(min(ac_mul, db_mul.yx()));
    }
 
    [[nodiscard]] constexpr interval positive_mul(interval const &rhs) const noexcept
    {
        hi_axiom(v[0] >= 0 and rhs.v[0] >= 0);
 
        return raw(v * neg<0b10>(rhs.v));
    }
 
    [[nodiscard]] constexpr interval operator/(interval const &rhs) const noexcept
    {
        if (rhs.v[0] <= 0 and 0 <= rhs.v[1]) {
            // Return an infinite interval when it is possible to divide by zero.
            return interval{};
        }
 
        hilet rhs_ge_zero = ge(rhs.v, bound_type{});
        hilet rhs_lt_zero = ~rhs_ge_zero;
 
        hilet b_ma = (rhs_ge_zero & v.yy()) | (rhs_lt_zero & -v.xx());
        hilet a_mb = -b_ma.yx();
 
        hilet a_mb_mul = a_mb / rhs.v;
        hilet b_ma_mul = b_ma / rhs.v;
 
        return raw(min(a_mb, b_ma));
    }
 
    [[nodiscard]] constexpr interval operator%(interval const &rhs) const noexcept
    {
        if (rhs.v[0] <= 0 and 0 <= rhs.v[1]) {
            // Return an infinite interval when it is possible to divide by zero.
            return interval{};
        }
 
        // In C++ the sign of the modulo operator's result is the same as the left operand.
        hilet rhs_abs = abs(rhs);
        if (v[0] > 0) {
            return rhs_abs;
        } else if (v[1] > 0) {
            return -rhs_abs;
        } else {
            return raw(rhs_abs.yy());
        }
    }
 
    [[nodiscard]] friend constexpr interval reciprocal(interval const &rhs) noexcept
    {
        if (rhs.v[0] <= 0 and 0 <= rhs.v[1]) {
            // Return an infinite interval when it is possible to divide by zero.
            return interval{};
        }
 
        return raw(bound_type::broadcast(value_type{-1}) / rhs.yx());
    }
 
    [[nodiscard]] friend constexpr interval abs(interval const &rhs) noexcept
    {
        bound_type r;
        r[0] = std::max({value_type{0}, rhs.v[0], rhs.v[1]});
        r[1] = std::min(rhs.v[0], rhs.v[1]);
        return raw(r);
    }
 
    [[nodiscard]] friend constexpr interval square(interval const &rhs) noexcept
    {
        hilet abs_rhs = abs(rhs);
        return abs_rhs.positive_mul(abs_rhs);
    }
 
    interval &operator+=(interval const &rhs) noexcept
    {
        return *this = *this + rhs;
    }
 
    interval &operator-=(interval const &rhs) noexcept
    {
        return *this = *this - rhs;
    }
 
    interval &operator*=(interval const &rhs) noexcept
    {
        return *this = *this * rhs;
    }
 
    [[nodiscard]] friend constexpr bool operator==(value_type const &lhs, interval const &rhs) noexcept
    {
        return rhs.lower() <= lhs and lhs <= rhs.upper();
    }
 
    [[nodiscard]] friend constexpr auto operator<=>(value_type const &lhs, interval const &rhs) noexcept
    {
        if constexpr (std::is_floating_point_v<value_type>) {
            if (lhs < rhs.lower()) {
                return std::partial_ordering::less;
            } else if (lhs > rhs.upper()) {
                return std::partial_ordering::greater;
            } else if (lhs >= rhs.lower() and lhs <= rhs.upper()) {
                return std::partial_ordering::equivalent;
            } else {
                return std::partial_ordering::unordered;
            }
        } else {
            if (lhs < rhs.lower()) {
                return std::weak_ordering::less;
            } else if (lhs > rhs.upper()) {
                return std::weak_ordering::greater;
            } else {
                return std::weak_ordering::equivalent;
            }
        }
    }
 
    [[nodiscard]] constexpr bool is_fully_inside(interval const &other) const noexcept
    {
        return ge(v, other.v) == 0b11;
    }
};
 
using finterval = interval<float>;
using dinterval = interval<double>;
 
} // namespace hi::inline v1