xeos
/
num-traits
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Implement BitAndAssign and BitOrAssign for BigUint

This commit is contained in:
Alice Ryhl 2017-09-02 23:37:54 +02:00
parent 23085800e0
commit 8c3b2de11c
1 changed files with 25 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
bigint/src/biguint.rs
View File

@ -1,7 +1,7 @@
use std::borrow::Cow; use std::borrow::Cow;
use std::default::Default; use std::default::Default;
use std::iter::repeat; use std::iter::repeat;
use std::ops::{Add, BitAnd, BitOr, BitXor, Div, Mul, Neg, Rem, Shl, Shr, Sub, AddAssign, SubAssign, MulAssign, DivAssign, RemAssign}; use std::ops::{Add, BitAnd, BitOr, BitXor, Div, Mul, Neg, Rem, Shl, Shr, Sub, AddAssign, SubAssign, MulAssign, DivAssign, RemAssign, BitAndAssign, BitOrAssign};
use std::str::{self, FromStr}; use std::str::{self, FromStr};
use std::fmt; use std::fmt;
use std::cmp; use std::cmp;
@ -268,36 +268,48 @@ impl Num for BigUint {
} }
forward_all_binop_to_val_ref_commutative!(impl BitAnd for BigUint, bitand); forward_all_binop_to_val_ref_commutative!(impl BitAnd for BigUint, bitand);
forward_val_assign!(impl BitAndAssign for BigUint, bitand_assign);
impl<'a> BitAnd<&'a BigUint> for BigUint { impl<'a> BitAnd<&'a BigUint> for BigUint {
type Output = BigUint; type Output = BigUint;
#[inline] #[inline]
fn bitand(self, other: &BigUint) -> BigUint { fn bitand(mut self, other: &BigUint) -> BigUint {
let mut data = self.data; self &= other;
for (ai, &bi) in data.iter_mut().zip(other.data.iter()) { self
}
}
impl<'a> BitAndAssign<&'a BigUint> for BigUint {
#[inline]
fn bitand_assign(&mut self, other: &BigUint) {
for (ai, &bi) in self.data.iter_mut().zip(other.data.iter()) {
*ai &= bi; *ai &= bi;
} }
data.truncate(other.data.len()); self.data.truncate(other.data.len());
BigUint::new(data)
} }
} }
forward_all_binop_to_val_ref_commutative!(impl BitOr for BigUint, bitor); forward_all_binop_to_val_ref_commutative!(impl BitOr for BigUint, bitor);
forward_val_assign!(impl BitOrAssign for BigUint, bitor_assign);
impl<'a> BitOr<&'a BigUint> for BigUint { impl<'a> BitOr<&'a BigUint> for BigUint {
type Output = BigUint; type Output = BigUint;
fn bitor(self, other: &BigUint) -> BigUint { fn bitor(mut self, other: &BigUint) -> BigUint {
let mut data = self.data; self |= other;
for (ai, &bi) in data.iter_mut().zip(other.data.iter()) { self
}
}
impl<'a> BitOrAssign<&'a BigUint> for BigUint {
#[inline]
fn bitor_assign(&mut self, other: &BigUint) {
for (ai, &bi) in self.data.iter_mut().zip(other.data.iter()) {
*ai |= bi; *ai |= bi;
} }
if other.data.len() > data.len() { if other.data.len() > self.data.len() {
let extra = &other.data[data.len()..]; let extra = &other.data[self.data.len()..];
data.extend(extra.iter().cloned()); self.data.extend(extra.iter().cloned());
} }
BigUint::new(data)
} }
} }