1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
macro_rules! ops {
(neg) => {
impl<T: num_traits::One + num_traits::CheckedNeg> std::ops::Neg for SCell<T> {
type Output = Self;
fn neg(mut self) -> Self::Output {
self.data = self.data.checked_neg().unwrap_or_else(|| {self.error_tag = true; T::one()});
SCell {
error_tag: self.error_tag,
data: self.data,
}
}
}
};
($op: tt, $fn: ident, $trait: ident) => {
paste::item! {
impl<T: num_traits::One + num_traits::[<Checked $trait>]> std::ops::[<$trait Assign>] for SCell<T> {
fn [<$fn _assign>](&mut self, rhs: Self) {
self.data = self.data.[<checked_ $fn>](&rhs.data).unwrap_or_else(|| {self.error_tag = true; T::one()});
self.error_tag |= rhs.error_tag;
}
}
}
paste::item! {
impl<T: One + num_traits::[<Checked $trait>]> $trait for SCell<T> {
type Output = Self;
fn $fn(mut self, rhs: Self) -> Self::Output {
self $op rhs;
self
}
}
}
}
}