From e52c32ccaab2e9017f643a3fc25564726898053f Mon Sep 17 00:00:00 2001
From: Corwin Kuiper <corwin@kuiper.dev>
Date: Sat, 20 Mar 2021 01:47:46 +0000
Subject: [PATCH] assign impls

---
 src/number.rs | 42 +++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 41 insertions(+), 1 deletion(-)

diff --git a/src/number.rs b/src/number.rs
index 3e411106..aa8ecc0c 100644
--- a/src/number.rs
+++ b/src/number.rs
@@ -1,6 +1,6 @@
 use core::{
     fmt::Display,
-    ops::{Add, Div, Mul, Neg, Sub},
+    ops::{Add, AddAssign, Div, DivAssign, Mul, MulAssign, Neg, Sub, SubAssign},
 };
 
 #[derive(Clone, Copy, PartialEq, Eq, PartialOrd, Ord)]
@@ -19,6 +19,12 @@ impl<const N: usize> Add for Num<N> {
     }
 }
 
+impl<const N: usize> AddAssign for Num<N> {
+    fn add_assign(&mut self, rhs: Self) {
+        self.0 += rhs.0
+    }
+}
+
 impl<const N: usize> Sub for Num<N> {
     type Output = Self;
     fn sub(self, rhs: Num<N>) -> Self::Output {
@@ -26,6 +32,12 @@ impl<const N: usize> Sub for Num<N> {
     }
 }
 
+impl<const N: usize> SubAssign for Num<N> {
+    fn sub_assign(&mut self, rhs: Self) {
+        self.0 -= rhs.0
+    }
+}
+
 impl<const N: usize> Mul for Num<N> {
     type Output = Self;
     fn mul(self, rhs: Num<N>) -> Self::Output {
@@ -37,6 +49,16 @@ impl<const N: usize> Mul for Num<N> {
     }
 }
 
+impl<const N: usize> MulAssign for Num<N> {
+    fn mul_assign(&mut self, rhs: Self) {
+        if N % 2 == 0 {
+            self.0 = (self.0 >> (N / 2)) * (rhs.0 >> (N / 2))
+        } else {
+            self.0 = (self.0 >> (1 + N / 2)) * (rhs.0 >> (N / 2))
+        }
+    }
+}
+
 impl<const N: usize> Div for Num<N> {
     type Output = Self;
     fn div(self, rhs: Num<N>) -> Self::Output {
@@ -48,6 +70,16 @@ impl<const N: usize> Div for Num<N> {
     }
 }
 
+impl<const N: usize> DivAssign for Num<N> {
+    fn div_assign(&mut self, rhs: Self) {
+        if N % 2 == 0 {
+            self.0 = (self.0 << (N / 2)) / (rhs.0 >> (N / 2))
+        } else {
+            self.0 = (self.0 << (1 + N / 2)) / (rhs.0 >> (N / 2))
+        }
+    }
+}
+
 impl<const N: usize> Neg for Num<N> {
     type Output = Self;
     fn neg(self) -> Self::Output {
@@ -62,6 +94,14 @@ impl<const N: usize> Num<N> {
     pub fn min() -> Self {
         Num(i32::MIN)
     }
+
+    pub fn int(&self) -> i32 {
+        self.0 >> N
+    }
+
+    pub fn new(integral: i32) -> Self {
+        Self(integral << N)
+    }
 }
 
 impl<const N: usize> Display for Num<N> {