184 lines
10 KiB
C++
184 lines
10 KiB
C++
/*
|
|
Copyright (c) 2013 Microsoft Corporation. All rights reserved.
|
|
Released under Apache 2.0 license as described in the file LICENSE.
|
|
|
|
Author: Leonardo de Moura
|
|
*/
|
|
#pragma once
|
|
#include "mpz.h"
|
|
|
|
namespace lean {
|
|
|
|
// Wrapper for GMP rationals
|
|
class mpq {
|
|
mpq_t m_val;
|
|
static mpz_t const & zval(mpz const & v) { return v.m_val; }
|
|
static mpz_t & zval(mpz & v) { return v.m_val; }
|
|
public:
|
|
void swap(mpq & v) { mpq_swap(m_val, v.m_val); }
|
|
void swap_numerator(mpz & v) { mpz_swap(mpq_numref(m_val), v.m_val); mpq_canonicalize(m_val); }
|
|
void swap_denominator(mpz & v) { mpz_swap(mpq_denref(m_val), v.m_val); mpq_canonicalize(m_val); }
|
|
|
|
mpq & operator=(mpz const & v) { mpq_set_z(m_val, v.m_val); return *this; }
|
|
mpq & operator=(mpq const & v) { mpq_set(m_val, v.m_val); return *this; }
|
|
mpq & operator=(mpq && v) { swap(v); return *this; }
|
|
mpq & operator=(char const * v) { mpq_set_str(m_val, v, 10); return *this; }
|
|
mpq & operator=(unsigned long int v) { mpq_set_ui(m_val, v, 1u); return *this; }
|
|
mpq & operator=(long int v) { mpq_set_si(m_val, v, 1); return *this; }
|
|
mpq & operator=(unsigned int v) { return operator=(static_cast<unsigned long int>(v)); }
|
|
mpq & operator=(int v) { return operator=(static_cast<long int>(v)); }
|
|
|
|
mpq() { mpq_init(m_val); }
|
|
mpq(mpq const & v):mpq() { operator=(v); }
|
|
mpq(mpz const & v):mpq() { operator=(v); }
|
|
mpq(mpq && s):mpq() { mpq_swap(m_val, s.m_val); }
|
|
mpq(char const * v):mpq() { operator=(v); }
|
|
mpq(unsigned long int v):mpq() { operator=(v); }
|
|
mpq(long int v):mpq() { operator=(v); }
|
|
mpq(unsigned int v):mpq() { operator=(v); }
|
|
mpq(int v):mpq() { operator=(v); }
|
|
mpq(unsigned long int n, unsigned long int d):mpq() { mpq_set_ui(m_val, n, d); mpq_canonicalize(m_val); }
|
|
mpq(long int n, long int d):mpq() { mpq_set_si(m_val, n, d); mpq_canonicalize(m_val); }
|
|
mpq(unsigned int n, unsigned int d):mpq() { mpq_set_ui(m_val, n, d); mpq_canonicalize(m_val); }
|
|
mpq(int n, int d):mpq() { mpq_set_si(m_val, n, d); mpq_canonicalize(m_val); }
|
|
mpq(double v):mpq() { mpq_set_d(m_val, v); }
|
|
~mpq() { mpq_clear(m_val); }
|
|
|
|
unsigned hash() const { return static_cast<unsigned>(mpz_get_si(mpq_numref(m_val))); }
|
|
|
|
int sgn() const { return mpq_sgn(m_val); }
|
|
friend int sgn(mpq const & a) { return a.sgn(); }
|
|
bool is_pos() const { return sgn() > 0; }
|
|
bool is_neg() const { return sgn() < 0; }
|
|
bool is_zero() const { return sgn() == 0; }
|
|
bool is_nonpos() const { return !is_pos(); }
|
|
bool is_nonneg() const { return !is_neg(); }
|
|
|
|
void neg() { mpq_neg(m_val, m_val); }
|
|
friend mpq neg(mpq a) { a.neg(); return a; }
|
|
|
|
void abs() { mpq_abs(m_val, m_val); }
|
|
friend mpq abs(mpq a) { a.abs(); return a; }
|
|
|
|
void inv() { mpq_inv(m_val, m_val); }
|
|
friend mpq inv(mpq a) { a.inv(); return a; }
|
|
|
|
double get_double() const { return mpq_get_d(m_val); }
|
|
|
|
bool is_integer() const { return mpz_cmp_ui(mpq_denref(m_val), 1u) == 0; }
|
|
|
|
friend int cmp(mpq const & a, mpq const & b) { return mpq_cmp(a.m_val, b.m_val); }
|
|
friend int cmp(mpq const & a, unsigned b) { return mpq_cmp_ui(a.m_val, b, 1); }
|
|
friend int cmp(mpq const & a, int b) { return mpq_cmp_si(a.m_val, b, 1); }
|
|
|
|
friend bool operator<(mpq const & a, mpq const & b) { return cmp(a, b) < 0; }
|
|
friend bool operator<(mpq const & a, unsigned b) { return cmp(a, b) < 0; }
|
|
friend bool operator<(mpq const & a, int b) { return cmp(a, b) < 0; }
|
|
friend bool operator<(unsigned a, mpq const & b) { return cmp(b, a) > 0; }
|
|
friend bool operator<(int a, mpq const & b) { return cmp(b, a) > 0; }
|
|
|
|
friend bool operator>(mpq const & a, mpq const & b) { return cmp(a, b) > 0; }
|
|
friend bool operator>(mpq const & a, unsigned b) { return cmp(a, b) > 0; }
|
|
friend bool operator>(mpq const & a, int b) { return cmp(a, b) > 0; }
|
|
friend bool operator>(unsigned a, mpq const & b) { return cmp(b, a) < 0; }
|
|
friend bool operator>(int a, mpq const & b) { return cmp(b, a) < 0; }
|
|
|
|
friend bool operator<=(mpq const & a, mpq const & b) { return cmp(a, b) <= 0; }
|
|
friend bool operator<=(mpq const & a, unsigned b) { return cmp(a, b) <= 0; }
|
|
friend bool operator<=(mpq const & a, int b) { return cmp(a, b) <= 0; }
|
|
friend bool operator<=(unsigned a, mpq const & b) { return cmp(b, a) >= 0; }
|
|
friend bool operator<=(int a, mpq const & b) { return cmp(b, a) >= 0; }
|
|
|
|
friend bool operator>=(mpq const & a, mpq const & b) { return cmp(a, b) >= 0; }
|
|
friend bool operator>=(mpq const & a, unsigned b) { return cmp(a, b) >= 0; }
|
|
friend bool operator>=(mpq const & a, int b) { return cmp(a, b) >= 0; }
|
|
friend bool operator>=(unsigned a, mpq const & b) { return cmp(b, a) <= 0; }
|
|
friend bool operator>=(int a, mpq const & b) { return cmp(b, a) <= 0; }
|
|
|
|
friend bool operator==(mpq const & a, mpq const & b) { return mpq_equal(a.m_val, b.m_val) != 0; }
|
|
friend bool operator==(mpq const & a, mpz const & b) { return a.is_integer() && mpz_cmp(mpq_numref(a.m_val), zval(b)) == 0; }
|
|
friend bool operator==(mpq const & a, unsigned int b) { return a.is_integer() && mpz_cmp_ui(mpq_numref(a.m_val), b) == 0; }
|
|
friend bool operator==(mpq const & a, int b) { return a.is_integer() && mpz_cmp_si(mpq_numref(a.m_val), b) == 0; }
|
|
friend bool operator==(mpz const & a, mpq const & b) { return operator==(b, a); }
|
|
friend bool operator==(unsigned int a, mpq const & b) { return operator==(b, a); }
|
|
friend bool operator==(int a, mpq const & b) { return operator==(b, a); }
|
|
|
|
friend bool operator!=(mpq const & a, mpq const & b) { return !operator==(a,b); }
|
|
friend bool operator!=(mpq const & a, mpz const & b) { return !operator==(a,b); }
|
|
friend bool operator!=(mpz const & a, mpq const & b) { return !operator==(a,b); }
|
|
friend bool operator!=(mpq const & a, unsigned int b) { return !operator==(a,b); }
|
|
friend bool operator!=(mpq const & a, int b) { return !operator==(a,b); }
|
|
friend bool operator!=(unsigned int a, mpq const & b) { return !operator==(a,b); }
|
|
friend bool operator!=(int a, mpq const & b) { return !operator==(a,b); }
|
|
|
|
mpq & operator+=(mpq const & o) { mpq_add(m_val, m_val, o.m_val); return *this; }
|
|
mpq & operator+=(mpz const & o) { mpz_addmul(mpq_numref(m_val), mpq_denref(m_val), o.m_val); mpq_canonicalize(m_val); return *this; }
|
|
mpq & operator+=(unsigned int k) { mpz_addmul_ui(mpq_numref(m_val), mpq_denref(m_val), k); mpq_canonicalize(m_val); return *this; }
|
|
mpq & operator+=(int k) { if (k >= 0) return operator+=(static_cast<unsigned int>(k)); else return operator-=(static_cast<unsigned int>(-k)); }
|
|
|
|
mpq & operator-=(mpq const & o) { mpq_sub(m_val, m_val, o.m_val); return *this; }
|
|
mpq & operator-=(mpz const & o) { mpz_submul(mpq_numref(m_val), mpq_denref(m_val), o.m_val); mpq_canonicalize(m_val); return *this; }
|
|
mpq & operator-=(unsigned int k) { mpz_submul_ui(mpq_numref(m_val), mpq_denref(m_val), k); mpq_canonicalize(m_val); return *this; }
|
|
mpq & operator-=(int k) { if (k >= 0) return operator-=(static_cast<unsigned int>(k)); else return operator+=(static_cast<unsigned int>(-k)); }
|
|
|
|
mpq & operator*=(mpq const & o) { mpq_mul(m_val, m_val, o.m_val); return *this; }
|
|
mpq & operator*=(mpz const & o) { mpz_mul(mpq_numref(m_val), mpq_numref(m_val), o.m_val); mpq_canonicalize(m_val); return *this; }
|
|
mpq & operator*=(unsigned int k) { mpz_mul_ui(mpq_numref(m_val), mpq_numref(m_val), k); mpq_canonicalize(m_val); return *this; }
|
|
mpq & operator*=(int k) { mpz_mul_si(mpq_numref(m_val), mpq_numref(m_val), k); mpq_canonicalize(m_val); return *this; }
|
|
|
|
mpq & operator/=(mpq const & o) { mpq_div(m_val, m_val, o.m_val); return *this; }
|
|
mpq & operator/=(mpz const & o) { mpz_mul(mpq_denref(m_val), mpq_denref(m_val), o.m_val); mpq_canonicalize(m_val); return *this; }
|
|
mpq & operator/=(unsigned int k) { mpz_mul_ui(mpq_denref(m_val), mpq_denref(m_val), k); mpq_canonicalize(m_val); return *this; }
|
|
mpq & operator/=(int k) { mpz_mul_si(mpq_denref(m_val), mpq_denref(m_val), k); mpq_canonicalize(m_val); return *this; }
|
|
|
|
friend mpq operator+(mpq a, mpq const & b) { return a += b; }
|
|
friend mpq operator+(mpq a, mpz const & b) { return a += b; }
|
|
friend mpq operator+(mpq a, unsigned b) { return a += b; }
|
|
friend mpq operator+(mpq a, int b) { return a += b; }
|
|
friend mpq operator+(mpz const & a, mpq b) { return b += a; }
|
|
friend mpq operator+(unsigned a, mpq b) { return b += a; }
|
|
friend mpq operator+(int a, mpq b) { return b += a; }
|
|
|
|
friend mpq operator-(mpq a, mpq const & b) { return a -= b; }
|
|
friend mpq operator-(mpq a, mpz const & b) { return a -= b; }
|
|
friend mpq operator-(mpq a, unsigned b) { return a -= b; }
|
|
friend mpq operator-(mpq a, int b) { return a -= b; }
|
|
friend mpq operator-(mpz const & a, mpq b) { b.neg(); return b += a; }
|
|
friend mpq operator-(unsigned a, mpq b) { b.neg(); return b += a; }
|
|
friend mpq operator-(int a, mpq b) { b.neg(); return b += a; }
|
|
|
|
friend mpq operator*(mpq a, mpq const & b) { return a *= b; }
|
|
friend mpq operator*(mpq a, mpz const & b) { return a *= b; }
|
|
friend mpq operator*(mpq a, unsigned b) { return a *= b; }
|
|
friend mpq operator*(mpq a, int b) { return a *= b; }
|
|
friend mpq operator*(mpz const & a, mpq b) { return b *= a; }
|
|
friend mpq operator*(unsigned a, mpq b) { return b *= a; }
|
|
friend mpq operator*(int a, mpq b) { return b *= a; }
|
|
|
|
friend mpq operator/(mpq a, mpq const & b) { return a /= b; }
|
|
friend mpq operator/(mpq a, mpz const & b) { return a /= b; }
|
|
friend mpq operator/(mpq a, unsigned b) { return a /= b; }
|
|
friend mpq operator/(mpq a, int b) { return a /= b; }
|
|
friend mpq operator/(mpz const & a, mpq b) { b.inv(); return b *= a; }
|
|
friend mpq operator/(unsigned a, mpq b) { b.inv(); return b *= a; }
|
|
friend mpq operator/(int a, mpq b) { b.inv(); return b *= a; }
|
|
|
|
mpq & operator++() { return operator+=(1); }
|
|
mpq operator++(int) { mpq r(*this); ++(*this); return r; }
|
|
|
|
mpq & operator--() { return operator-=(1); }
|
|
mpq operator--(int) { mpq r(*this); --(*this); return r; }
|
|
|
|
mpz get_numerator() const { return mpz(mpq_numref(m_val)); }
|
|
mpz get_denominator() const { return mpz(mpq_denref(m_val)); }
|
|
|
|
friend std::ostream & operator<<(std::ostream & out, mpq const & v);
|
|
|
|
void floor();
|
|
friend mpz floor(mpq const & a);
|
|
|
|
void ceil();
|
|
friend mpz ceil(mpq const & a);
|
|
};
|
|
|
|
}
|