X-Git-Url: https://git.ucc.asn.au/?p=ipdf%2Fcode.git;a=blobdiff_plain;f=src%2Frational.h;h=0fa5d9910f3b7fa8136717bc971c5a7b6f23487a;hp=6bc5aec2edd6ec7edeffef62e64915c8e38b2306;hb=888817a67a9d840be66b52811b01eb77f10ff3e6;hpb=85e08223a33439f9b4911129b48ee45ff65869eb diff --git a/src/rational.h b/src/rational.h index 6bc5aec..0fa5d99 100644 --- a/src/rational.h +++ b/src/rational.h @@ -10,6 +10,8 @@ #include #include "arbint.h" #include "gmpint.h" +#include +#include namespace IPDF { @@ -21,6 +23,7 @@ template T Tabs(const T & a) template <> Arbint Tabs(const Arbint & a); template <> Gmpint Tabs(const Gmpint & a); + /* Recursive version of GCD template T gcd(const T & a, const T & b) @@ -75,6 +78,8 @@ struct Rational { Simplify(); } + + Rational(const T & _P, const T & _Q) : P(_P), Q(_Q) { @@ -162,16 +167,31 @@ struct Rational //Rational operator*(const Rational & r) const {return Rational(ToDouble()*r.ToDouble());} //Rational operator/(const Rational & r) const {return Rational(ToDouble()/r.ToDouble());} - Rational operator-() const {Rational r(*this); r.P = -r.P;} + Rational operator-() const {Rational r(*this); r.P = -r.P; return r;} Rational & operator=(const Rational & r) {P = r.P; Q = r.Q; Simplify(); return *this;} Rational & operator+=(const Rational & r) {this->operator=(*this+r); return *this;} Rational & operator-=(const Rational & r) {this->operator=(*this-r); return *this;} Rational & operator*=(const Rational & r) {this->operator=(*this*r); return *this;} Rational & operator/=(const Rational & r) {this->operator=(*this/r); return *this;} + Rational Sqrt() const + { + return Rational(sqrt(ToDouble())); + } + + int64_t ToInt64() const + { + return (int64_t)ToDouble(); + } double ToDouble() const { - return (double)P/(double)Q; + T num = P, denom = Q; + while (Tabs(num) > T(1e10)) + { + num /= T(16); + denom /= T(16); + } + return ((double)(num))/((double)(denom)); } bool CheckAccuracy(double d, const char * msg, double threshold = 1e-3) const { @@ -196,17 +216,13 @@ struct Rational T Q; }; -inline Rational pow(const Rational & a, const Rational & b) + +template +Rational

Abs(const Rational

& a) { - //TODO:Implement properly - int64_t P = std::pow((double)a.P, b.ToDouble()); - int64_t Q = std::pow((double)a.Q, b.ToDouble()); - return Rational(P, Q); + return Rational

(Tabs(a.P), a.Q); } - - - } #endif //_RATIONAL_H