1 files changed, 57 insertions, 10 deletions

diff --git a/src/utils.rs b/src/utils.rs
index c53b959..0dc91f7 100644
--- a/src/utils.rs
+++ b/src/utils.rs
@@ -1,3 +1,5 @@
+use std::ops::*;
+
 pub struct Rational
 {
     n: i64,
@@ -5,31 +7,76 @@ pub struct Rational
 }
 
 fn gcd(a: i64, b: i64) -> i64
-{if b != 0 { gcd(b, a % b) } else { a }}
+{match b { 0 => a, _ => gcd(b, a % b) }}
 
 impl Default for Rational
 {
     fn default() -> Rational { Rational::from_int(0) }
 }
 
-impl Rational
+impl Add for Rational
 {
-    fn from_int(v: i64) -> Rational {Rational{n: v, d: 1}}
-    fn reduced(self) -> Rational
-    {
-        let c = gcd(self.n, self.d);
-        Rational{n: self.n / c, d: self.d / c}
-    }
-    fn add(self, other: Rational) -> Rational
+    type Output = Self;
+
+    fn add(self, other: Self) -> Self
     {
         let c = gcd(self.d, other.d);
         Rational{n: self.n * (other.d / c) + other.n * (self.d / c),
                  d: self.d / c * other.d}.reduced()
     }
-    fn multiply(self, other: Rational) -> Rational
+}
+
+impl AddAssign for Rational
+{
+    fn add_assign(&mut self, other: Self)
+    {
+        let c = gcd(self.d, other.d);
+        *self = Rational{n: self.n * (other.d / c) + other.n * (self.d / c),
+                         d: self.d / c * other.d}.reduced()
+    }
+}
+
+impl Mul for Rational
+{
+    type Output = Self;
+    fn mul(self, other: Self) -> Self
     {
         Rational{n: self.n * other.n, d: self.d * other.d}.reduced()
     }
+}
+
+impl Div for Rational
+{
+    type Output = Self;
+    fn div(self, other: Self) -> Self
+    {
+        Rational{n: self.n * other.d, d: self.d * other.n}.reduced()
+    }
+}
+
+impl From<u32> for Rational
+{
+    fn from(v: u32) -> Self { Rational::from_int(v as i64) }
+}
+
+impl From<u16> for Rational
+{
+    fn from(v: u16) -> Self { Rational::from_int(v as i64) }
+}
+
+impl From<u8> for Rational
+{
+    fn from(v: u8) -> Self { Rational::from_int(v as i64) }
+}
+
+impl Rational
+{
+    fn from_int(v: i64) -> Rational {Rational{n: v, d: 1}}
+    fn reduced(self) -> Rational
+    {
+        let c = gcd(self.n, self.d);
+        Rational{n: self.n / c, d: self.d / c}
+    }
     fn as_int_trunc(self) -> i64 {self.n / self.d}
     fn as_int_round(self) -> i64 {(self.n as f64 / self.d as f64).round() as i64}
 }