alex/agb
1
0
Fork 0
mirror of https://github.com/italicsjenga/agb.git synced 2025-01-11 01:21:34 +11:00
Code Issues Packages Projects Releases Wiki Activity

assign impls

Browse source
This commit is contained in:
Corwin Kuiper 2021-03-20 01:47:46 +00:00 committed by Corwin
parent 1ec8cbaa20
commit e52c32ccaa
1 changed files with 41 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
src/number.rs
View file

@ -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> {