From 17030ea412638ceda3cef6bd92e9b00268cd4a72 Mon Sep 17 00:00:00 2001 From: Alice Ryhl Date: Sat, 26 Aug 2017 14:30:25 +0200 Subject: [PATCH] Implement SubAssign for BigUint --- bigint/src/biguint.rs | 26 ++++++++++++++++++++++---- 1 file changed, 22 insertions(+), 4 deletions(-) diff --git a/bigint/src/biguint.rs b/bigint/src/biguint.rs index c62cc64..a21b5cd 100644 --- a/bigint/src/biguint.rs +++ b/bigint/src/biguint.rs @@ -1,7 +1,7 @@ use std::borrow::Cow; use std::default::Default; use std::iter::repeat; -use std::ops::{Add, BitAnd, BitOr, BitXor, Div, Mul, Neg, Rem, Shl, Shr, Sub, AddAssign, MulAssign}; +use std::ops::{Add, BitAnd, BitOr, BitXor, Div, Mul, Neg, Rem, Shl, Shr, Sub, AddAssign, SubAssign, MulAssign}; use std::str::{self, FromStr}; use std::fmt; use std::cmp; @@ -466,8 +466,14 @@ impl<'a> Sub<&'a BigUint> for BigUint { type Output = BigUint; fn sub(mut self, other: &BigUint) -> BigUint { + self -= other; + self + } +} +impl<'a> SubAssign<&'a BigUint> for BigUint { + fn sub_assign(&mut self, other: &'a BigUint) { sub2(&mut self.data[..], &other.data[..]); - self.normalized() + self.normalize(); } } @@ -494,8 +500,14 @@ impl Sub for BigUint { #[inline] fn sub(mut self, other: BigDigit) -> BigUint { + self -= other; + self + } +} +impl SubAssign for BigUint { + fn sub_assign(&mut self, other: BigDigit) { sub2(&mut self.data[..], &[other]); - self.normalized() + self.normalize(); } } @@ -518,9 +530,15 @@ impl Sub for BigUint { #[inline] fn sub(mut self, other: DoubleBigDigit) -> BigUint { + self -= other; + self + } +} +impl SubAssign for BigUint { + fn sub_assign(&mut self, other: DoubleBigDigit) { let (hi, lo) = big_digit::from_doublebigdigit(other); sub2(&mut self.data[..], &[lo, hi]); - self.normalized() + self.normalize(); } }