number.hh

#pragma once
 
#include <clingo/util/print.hh>
 
#include <cstdint>
#include <optional>
#include <ostream>
#include <span>
#include <string>
 
namespace CppClingo {
 
 
enum class Base : uint8_t {
    dec = 10, 
    hex = 16, 
    oct = 8,  
    bin = 2,  
};
 
class Number {
  public:
    Number(int32_t value) noexcept;
    Number(char const *str, Base base = Base::dec);
    Number(std::string_view str, Base base = Base::dec);
 
    Number(Number const &other);
    Number(Number &&other) noexcept;
    auto operator=(Number const &other) -> Number &;
    auto operator=(Number &&other) noexcept -> Number &;
 
    ~Number() noexcept;
 
    [[nodiscard]] auto as_int() const -> std::optional<int32_t>;
 
    [[nodiscard]] auto as_string() const -> std::string;
 
    void swap(Number &other) noexcept;
 
    // comparison
 
    [[nodiscard]] auto hash() const noexcept -> size_t;
 
    friend auto compare(Number const &a, Number const &b) -> int;
    friend auto compare(Number const &a, int32_t b) -> int;
    friend auto compare(int32_t a, Number const &b) -> int;
 
    friend auto operator==(Number const &a, Number const &b) { return compare(a, b) == 0; }
    friend auto operator==(int32_t a, Number const &b) -> bool { return compare(a, b) == 0; }
    friend auto operator==(Number const &a, int32_t b) -> bool { return compare(a, b) == 0; }
 
    friend auto operator<=>(Number const &a, Number const &b) { return compare(a, b) <=> 0; }
    friend auto operator<=>(int32_t a, Number const &b) { return compare(a, b) <=> 0; }
    friend auto operator<=>(Number const &a, int32_t b) { return compare(a, b) <=> 0; }
 
    // addition
 
    friend auto operator+(Number const &a, Number const &b) -> Number;
    friend auto operator+(Number &&a, Number const &b) -> Number;
    friend auto operator+(Number const &a, Number &&b) -> Number;
    friend auto operator+(Number &&a, Number &&b) -> Number;
 
    friend auto operator+=(Number &a, Number const &b) -> Number &;
    friend auto operator+=(Number &a, Number &&b) -> Number &;
 
    // subtraction
 
    friend auto operator-(Number const &a, Number const &b) -> Number;
    friend auto operator-(Number &&a, Number const &b) -> Number;
    friend auto operator-(Number const &a, Number &&b) -> Number;
    friend auto operator-(Number &&a, Number &&b) -> Number;
 
    friend auto operator-=(Number &a, Number const &b) -> Number &;
    friend auto operator-=(Number &a, Number &&b) -> Number &;
 
    // multiplication
 
    friend auto operator*(Number const &a, Number const &b) -> Number;
    friend auto operator*(Number &&a, Number const &b) -> Number;
    friend auto operator*(Number const &a, Number &&b) -> Number;
    friend auto operator*(Number &&a, Number &&b) -> Number;
 
    friend auto operator*=(Number &a, Number const &b) -> Number &;
    friend auto operator*=(Number &a, Number &&b) -> Number &;
 
    // division
 
    friend auto operator/(Number const &a, Number const &b) -> Number;
    friend auto operator/(Number &&a, Number const &b) -> Number;
    friend auto operator/(Number const &a, Number &&b) -> Number;
    friend auto operator/(Number &&a, Number &&b) -> Number;
 
    friend auto operator/=(Number &a, Number const &b) -> Number &;
    friend auto operator/=(Number &a, Number &&b) -> Number &;
 
    // modulus
 
    friend auto operator%(Number const &a, Number const &b) -> Number;
    friend auto operator%(Number &&a, Number const &b) -> Number;
    friend auto operator%(Number const &a, Number &&b) -> Number;
    friend auto operator%(Number &&a, Number &&b) -> Number;
 
    friend auto operator%=(Number &a, Number const &b) -> Number &;
    friend auto operator%=(Number &a, Number &&b) -> Number &;
 
    // unary minus
 
    friend auto operator-(Number const &a) -> Number;
    friend auto operator-(Number &&a) -> Number;
 
    // complement
 
    friend auto operator~(Number const &a) -> Number;
    friend auto operator~(Number &&a) -> Number;
 
    // binary and
 
    friend auto operator&(Number const &a, Number const &b) -> Number;
    friend auto operator&(Number &&a, Number const &b) -> Number;
    friend auto operator&(Number const &a, Number &&b) -> Number;
    friend auto operator&(Number &&a, Number &&b) -> Number;
 
    friend auto operator&=(Number &a, Number const &b) -> Number &;
    friend auto operator&=(Number &a, Number &&b) -> Number &;
 
    // binary or
 
    friend auto operator|(Number const &a, Number const &b) -> Number;
    friend auto operator|(Number &&a, Number const &b) -> Number;
    friend auto operator|(Number const &a, Number &&b) -> Number;
    friend auto operator|(Number &&a, Number &&b) -> Number;
 
    friend auto operator|=(Number &a, Number const &b) -> Number &;
    friend auto operator|=(Number &a, Number &&b) -> Number &;
 
    // binary xor
 
    friend auto operator^(Number const &a, Number const &b) -> Number;
    friend auto operator^(Number &&a, Number const &b) -> Number;
    friend auto operator^(Number const &a, Number &&b) -> Number;
    friend auto operator^(Number &&a, Number &&b) -> Number;
 
    friend auto operator^=(Number &a, Number const &b) -> Number &;
    friend auto operator^=(Number &a, Number &&b) -> Number &;
 
    // exponentiation
 
    friend auto pow(Number const &a, Number const &b) -> Number;
    friend auto pow(Number &&a, Number const &b) -> Number;
    friend auto pow(Number const &a, Number &&b) -> Number;
    friend auto pow(Number &&a, Number &&b) -> Number;
 
    // absolute
 
    friend auto abs(Number const &a) -> Number;
    friend auto abs(Number &&a) -> Number;
 
    // get the sign of the number
 
    friend auto get_sign(Number const &a) -> int;
 
    // output
 
    friend auto operator<<(std::ostream &out, Number const &num) -> std::ostream &;
    friend auto operator<<(Util::OutputBuffer &out, Number const &num) -> Util::OutputBuffer &;
 
    // conversion between numbers and their representations
 
    static auto from_repr(uint64_t repr) -> Number { return {repr}; }
 
    static auto to_repr(Number const &num) -> uint64_t { return num.repr_; }
 
    static auto release(Number &num) -> uint64_t {
        auto repr = num.repr_;
        num.repr_ = 0;
        return repr;
    }
 
  private:
    class Impl;
    friend class Impl;
    Number(uint64_t repr);
 
    uint64_t repr_;
};
 
using NumberSpan = std::span<Number const>;
 
 
} // namespace CppClingo