1 //! Definitions of `Saturating<T>`.
4 use crate::ops::{Add, AddAssign, BitAnd, BitAndAssign, BitOr, BitOrAssign};
5 use crate::ops::{BitXor, BitXorAssign, Div, DivAssign};
6 use crate::ops::{Mul, MulAssign, Neg, Not};
7 use crate::ops::{Sub, SubAssign};
9 /// Provides intentionally-wrapped arithmetic on `T`.
11 /// Operations like `+` on `u32` values are intended to never overflow,
12 /// and in some debug configurations overflow is detected and results
13 /// in a panic. While most arithmetic falls into this category, some
14 /// code explicitly expects and relies upon saturating arithmetic.
16 /// Saturating arithmetic can be achieved either through methods like
17 /// `saturating_add`, or through the `Saturating<T>` type, which says that
18 /// all standard arithmetic operations on the underlying value are
19 /// intended to have saturating semantics.
21 /// The underlying value can be retrieved through the `.0` index of the
22 /// `Saturating` tuple.
27 /// #![feature(saturating_int_impl)]
28 /// use std::num::Saturating;
30 /// let max = Saturating(u32::MAX);
31 /// let one = Saturating(1u32);
33 /// assert_eq!(u32::MAX, (max + one).0);
35 #[unstable(feature = "saturating_int_impl", issue = "87920")]
36 #[derive(PartialEq, Eq, PartialOrd, Ord, Clone, Copy, Default, Hash)]
38 pub struct Saturating<T>(#[unstable(feature = "saturating_int_impl", issue = "87920")] pub T);
40 #[unstable(feature = "saturating_int_impl", issue = "87920")]
41 impl<T: fmt::Debug> fmt::Debug for Saturating<T> {
42 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
47 #[unstable(feature = "saturating_int_impl", issue = "87920")]
48 impl<T: fmt::Display> fmt::Display for Saturating<T> {
49 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
54 #[unstable(feature = "saturating_int_impl", issue = "87920")]
55 impl<T: fmt::Binary> fmt::Binary for Saturating<T> {
56 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
61 #[unstable(feature = "saturating_int_impl", issue = "87920")]
62 impl<T: fmt::Octal> fmt::Octal for Saturating<T> {
63 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
68 #[unstable(feature = "saturating_int_impl", issue = "87920")]
69 impl<T: fmt::LowerHex> fmt::LowerHex for Saturating<T> {
70 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
75 #[unstable(feature = "saturating_int_impl", issue = "87920")]
76 impl<T: fmt::UpperHex> fmt::UpperHex for Saturating<T> {
77 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
82 // FIXME(30524): impl Op<T> for Saturating<T>, impl OpAssign<T> for Saturating<T>
83 macro_rules! saturating_impl {
85 #[unstable(feature = "saturating_int_impl", issue = "87920")]
86 impl Add for Saturating<$t> {
87 type Output = Saturating<$t>;
90 fn add(self, other: Saturating<$t>) -> Saturating<$t> {
91 Saturating(self.0.saturating_add(other.0))
94 forward_ref_binop! { impl Add, add for Saturating<$t>, Saturating<$t>,
95 #[unstable(feature = "saturating_int_impl", issue = "87920")] }
97 #[unstable(feature = "saturating_int_impl", issue = "87920")]
98 impl AddAssign for Saturating<$t> {
100 fn add_assign(&mut self, other: Saturating<$t>) {
101 *self = *self + other;
104 forward_ref_op_assign! { impl AddAssign, add_assign for Saturating<$t>, Saturating<$t> }
106 #[unstable(feature = "saturating_int_impl", issue = "87920")]
107 impl Sub for Saturating<$t> {
108 type Output = Saturating<$t>;
111 fn sub(self, other: Saturating<$t>) -> Saturating<$t> {
112 Saturating(self.0.saturating_sub(other.0))
115 forward_ref_binop! { impl Sub, sub for Saturating<$t>, Saturating<$t>,
116 #[unstable(feature = "saturating_int_impl", issue = "87920")] }
118 #[unstable(feature = "saturating_int_impl", issue = "87920")]
119 impl SubAssign for Saturating<$t> {
121 fn sub_assign(&mut self, other: Saturating<$t>) {
122 *self = *self - other;
125 forward_ref_op_assign! { impl SubAssign, sub_assign for Saturating<$t>, Saturating<$t> }
127 #[unstable(feature = "saturating_int_impl", issue = "87920")]
128 impl Mul for Saturating<$t> {
129 type Output = Saturating<$t>;
132 fn mul(self, other: Saturating<$t>) -> Saturating<$t> {
133 Saturating(self.0.saturating_mul(other.0))
136 forward_ref_binop! { impl Mul, mul for Saturating<$t>, Saturating<$t>,
137 #[unstable(feature = "saturating_int_impl", issue = "87920")] }
139 #[unstable(feature = "saturating_int_impl", issue = "87920")]
140 impl MulAssign for Saturating<$t> {
142 fn mul_assign(&mut self, other: Saturating<$t>) {
143 *self = *self * other;
146 forward_ref_op_assign! { impl MulAssign, mul_assign for Saturating<$t>, Saturating<$t> }
148 #[unstable(feature = "saturating_int_impl", issue = "87920")]
149 impl Div for Saturating<$t> {
150 type Output = Saturating<$t>;
153 fn div(self, other: Saturating<$t>) -> Saturating<$t> {
154 // saturating div is the default behavior?
155 Saturating(self.0.div(other.0))
158 forward_ref_binop! { impl Div, div for Saturating<$t>, Saturating<$t>,
159 #[unstable(feature = "saturating_int_impl", issue = "87920")] }
161 #[unstable(feature = "saturating_int_impl", issue = "87920")]
162 impl DivAssign for Saturating<$t> {
164 fn div_assign(&mut self, other: Saturating<$t>) {
165 *self = *self / other;
168 forward_ref_op_assign! { impl DivAssign, div_assign for Saturating<$t>, Saturating<$t> }
170 #[unstable(feature = "saturating_int_impl", issue = "87920")]
171 impl Not for Saturating<$t> {
172 type Output = Saturating<$t>;
175 fn not(self) -> Saturating<$t> {
179 forward_ref_unop! { impl Not, not for Saturating<$t>,
180 #[unstable(feature = "saturating_int_impl", issue = "87920")] }
182 #[unstable(feature = "saturating_int_impl", issue = "87920")]
183 impl BitXor for Saturating<$t> {
184 type Output = Saturating<$t>;
187 fn bitxor(self, other: Saturating<$t>) -> Saturating<$t> {
188 Saturating(self.0 ^ other.0)
191 forward_ref_binop! { impl BitXor, bitxor for Saturating<$t>, Saturating<$t>,
192 #[unstable(feature = "saturating_int_impl", issue = "87920")] }
194 #[unstable(feature = "saturating_int_impl", issue = "87920")]
195 impl BitXorAssign for Saturating<$t> {
197 fn bitxor_assign(&mut self, other: Saturating<$t>) {
198 *self = *self ^ other;
201 forward_ref_op_assign! { impl BitXorAssign, bitxor_assign for Saturating<$t>, Saturating<$t> }
203 #[unstable(feature = "saturating_int_impl", issue = "87920")]
204 impl BitOr for Saturating<$t> {
205 type Output = Saturating<$t>;
208 fn bitor(self, other: Saturating<$t>) -> Saturating<$t> {
209 Saturating(self.0 | other.0)
212 forward_ref_binop! { impl BitOr, bitor for Saturating<$t>, Saturating<$t>,
213 #[unstable(feature = "saturating_int_impl", issue = "87920")] }
215 #[unstable(feature = "saturating_int_impl", issue = "87920")]
216 impl BitOrAssign for Saturating<$t> {
218 fn bitor_assign(&mut self, other: Saturating<$t>) {
219 *self = *self | other;
222 forward_ref_op_assign! { impl BitOrAssign, bitor_assign for Saturating<$t>, Saturating<$t> }
224 #[unstable(feature = "saturating_int_impl", issue = "87920")]
225 impl BitAnd for Saturating<$t> {
226 type Output = Saturating<$t>;
229 fn bitand(self, other: Saturating<$t>) -> Saturating<$t> {
230 Saturating(self.0 & other.0)
233 forward_ref_binop! { impl BitAnd, bitand for Saturating<$t>, Saturating<$t>,
234 #[unstable(feature = "saturating_int_impl", issue = "87920")] }
236 #[unstable(feature = "saturating_int_impl", issue = "87920")]
237 impl BitAndAssign for Saturating<$t> {
239 fn bitand_assign(&mut self, other: Saturating<$t>) {
240 *self = *self & other;
243 forward_ref_op_assign! { impl BitAndAssign, bitand_assign for Saturating<$t>, Saturating<$t> }
245 #[unstable(feature = "saturating_int_impl", issue = "87920")]
246 impl Neg for Saturating<$t> {
249 fn neg(self) -> Self {
253 forward_ref_unop! { impl Neg, neg for Saturating<$t>,
254 #[unstable(feature = "saturating_int_impl", issue = "87920")] }
259 saturating_impl! { usize u8 u16 u32 u64 u128 isize i8 i16 i32 i64 i128 }
261 macro_rules! saturating_int_impl {
263 impl Saturating<$t> {
264 /// Returns the smallest value that can be represented by this integer type.
271 /// #![feature(saturating_int_impl)]
272 /// use std::num::Saturating;
274 #[doc = concat!("assert_eq!(<Saturating<", stringify!($t), ">>::MIN, Saturating(", stringify!($t), "::MIN));")]
276 #[unstable(feature = "saturating_int_impl", issue = "87920")]
277 pub const MIN: Self = Self(<$t>::MIN);
279 /// Returns the largest value that can be represented by this integer type.
286 /// #![feature(saturating_int_impl)]
287 /// use std::num::Saturating;
289 #[doc = concat!("assert_eq!(<Saturating<", stringify!($t), ">>::MAX, Saturating(", stringify!($t), "::MAX));")]
291 #[unstable(feature = "saturating_int_impl", issue = "87920")]
292 pub const MAX: Self = Self(<$t>::MAX);
294 /// Returns the size of this integer type in bits.
301 /// #![feature(saturating_int_impl)]
302 /// use std::num::Saturating;
304 #[doc = concat!("assert_eq!(<Saturating<", stringify!($t), ">>::BITS, ", stringify!($t), "::BITS);")]
306 #[unstable(feature = "saturating_int_impl", issue = "87920")]
307 pub const BITS: u32 = <$t>::BITS;
309 /// Returns the number of ones in the binary representation of `self`.
316 /// #![feature(saturating_int_impl)]
317 /// use std::num::Saturating;
319 #[doc = concat!("let n = Saturating(0b01001100", stringify!($t), ");")]
321 /// assert_eq!(n.count_ones(), 3);
324 #[doc(alias = "popcount")]
325 #[doc(alias = "popcnt")]
326 #[unstable(feature = "saturating_int_impl", issue = "87920")]
327 pub const fn count_ones(self) -> u32 {
331 /// Returns the number of zeros in the binary representation of `self`.
338 /// #![feature(saturating_int_impl)]
339 /// use std::num::Saturating;
341 #[doc = concat!("assert_eq!(Saturating(!0", stringify!($t), ").count_zeros(), 0);")]
344 #[unstable(feature = "saturating_int_impl", issue = "87920")]
345 pub const fn count_zeros(self) -> u32 {
349 /// Returns the number of trailing zeros in the binary representation of `self`.
356 /// #![feature(saturating_int_impl)]
357 /// use std::num::Saturating;
359 #[doc = concat!("let n = Saturating(0b0101000", stringify!($t), ");")]
361 /// assert_eq!(n.trailing_zeros(), 3);
364 #[unstable(feature = "saturating_int_impl", issue = "87920")]
365 pub const fn trailing_zeros(self) -> u32 {
366 self.0.trailing_zeros()
369 /// Shifts the bits to the left by a specified amount, `n`,
370 /// saturating the truncated bits to the end of the resulting
373 /// Please note this isn't the same operation as the `<<` shifting
381 /// #![feature(saturating_int_impl)]
382 /// use std::num::Saturating;
384 /// let n: Saturating<i64> = Saturating(0x0123456789ABCDEF);
385 /// let m: Saturating<i64> = Saturating(-0x76543210FEDCBA99);
387 /// assert_eq!(n.rotate_left(32), m);
390 #[unstable(feature = "saturating_int_impl", issue = "87920")]
391 pub const fn rotate_left(self, n: u32) -> Self {
392 Saturating(self.0.rotate_left(n))
395 /// Shifts the bits to the right by a specified amount, `n`,
396 /// saturating the truncated bits to the beginning of the resulting
399 /// Please note this isn't the same operation as the `>>` shifting
407 /// #![feature(saturating_int_impl)]
408 /// use std::num::Saturating;
410 /// let n: Saturating<i64> = Saturating(0x0123456789ABCDEF);
411 /// let m: Saturating<i64> = Saturating(-0xFEDCBA987654322);
413 /// assert_eq!(n.rotate_right(4), m);
416 #[unstable(feature = "saturating_int_impl", issue = "87920")]
417 pub const fn rotate_right(self, n: u32) -> Self {
418 Saturating(self.0.rotate_right(n))
421 /// Reverses the byte order of the integer.
428 /// #![feature(saturating_int_impl)]
429 /// use std::num::Saturating;
431 /// let n: Saturating<i16> = Saturating(0b0000000_01010101);
432 /// assert_eq!(n, Saturating(85));
434 /// let m = n.swap_bytes();
436 /// assert_eq!(m, Saturating(0b01010101_00000000));
437 /// assert_eq!(m, Saturating(21760));
440 #[unstable(feature = "saturating_int_impl", issue = "87920")]
441 pub const fn swap_bytes(self) -> Self {
442 Saturating(self.0.swap_bytes())
445 /// Reverses the bit pattern of the integer.
449 /// Please note that this example is shared between integer types.
450 /// Which explains why `i16` is used here.
455 /// #![feature(saturating_int_impl)]
456 /// use std::num::Saturating;
458 /// let n = Saturating(0b0000000_01010101i16);
459 /// assert_eq!(n, Saturating(85));
461 /// let m = n.reverse_bits();
463 /// assert_eq!(m.0 as u16, 0b10101010_00000000);
464 /// assert_eq!(m, Saturating(-22016));
466 #[unstable(feature = "saturating_int_impl", issue = "87920")]
467 #[rustc_const_stable(feature = "const_reverse_bits", since = "1.37.0")]
470 pub const fn reverse_bits(self) -> Self {
471 Saturating(self.0.reverse_bits())
474 /// Converts an integer from big endian to the target's endianness.
476 /// On big endian this is a no-op. On little endian the bytes are
484 /// #![feature(saturating_int_impl)]
485 /// use std::num::Saturating;
487 #[doc = concat!("let n = Saturating(0x1A", stringify!($t), ");")]
489 /// if cfg!(target_endian = "big") {
490 #[doc = concat!(" assert_eq!(<Saturating<", stringify!($t), ">>::from_be(n), n)")]
492 #[doc = concat!(" assert_eq!(<Saturating<", stringify!($t), ">>::from_be(n), n.swap_bytes())")]
496 #[unstable(feature = "saturating_int_impl", issue = "87920")]
497 pub const fn from_be(x: Self) -> Self {
498 Saturating(<$t>::from_be(x.0))
501 /// Converts an integer from little endian to the target's endianness.
503 /// On little endian this is a no-op. On big endian the bytes are
511 /// #![feature(saturating_int_impl)]
512 /// use std::num::Saturating;
514 #[doc = concat!("let n = Saturating(0x1A", stringify!($t), ");")]
516 /// if cfg!(target_endian = "little") {
517 #[doc = concat!(" assert_eq!(<Saturating<", stringify!($t), ">>::from_le(n), n)")]
519 #[doc = concat!(" assert_eq!(<Saturating<", stringify!($t), ">>::from_le(n), n.swap_bytes())")]
523 #[unstable(feature = "saturating_int_impl", issue = "87920")]
524 pub const fn from_le(x: Self) -> Self {
525 Saturating(<$t>::from_le(x.0))
528 /// Converts `self` to big endian from the target's endianness.
530 /// On big endian this is a no-op. On little endian the bytes are
538 /// #![feature(saturating_int_impl)]
539 /// use std::num::Saturating;
541 #[doc = concat!("let n = Saturating(0x1A", stringify!($t), ");")]
543 /// if cfg!(target_endian = "big") {
544 /// assert_eq!(n.to_be(), n)
546 /// assert_eq!(n.to_be(), n.swap_bytes())
550 #[unstable(feature = "saturating_int_impl", issue = "87920")]
551 pub const fn to_be(self) -> Self {
552 Saturating(self.0.to_be())
555 /// Converts `self` to little endian from the target's endianness.
557 /// On little endian this is a no-op. On big endian the bytes are
565 /// #![feature(saturating_int_impl)]
566 /// use std::num::Saturating;
568 #[doc = concat!("let n = Saturating(0x1A", stringify!($t), ");")]
570 /// if cfg!(target_endian = "little") {
571 /// assert_eq!(n.to_le(), n)
573 /// assert_eq!(n.to_le(), n.swap_bytes())
577 #[unstable(feature = "saturating_int_impl", issue = "87920")]
578 pub const fn to_le(self) -> Self {
579 Saturating(self.0.to_le())
582 /// Raises self to the power of `exp`, using exponentiation by squaring.
589 /// #![feature(saturating_int_impl)]
590 /// use std::num::Saturating;
592 #[doc = concat!("assert_eq!(Saturating(3", stringify!($t), ").pow(4), Saturating(81));")]
595 /// Results that are too large are wrapped:
598 /// #![feature(saturating_int_impl)]
599 /// use std::num::Saturating;
601 /// assert_eq!(Saturating(3i8).pow(5), Saturating(127));
602 /// assert_eq!(Saturating(3i8).pow(6), Saturating(127));
605 #[unstable(feature = "saturating_int_impl", issue = "87920")]
606 pub fn pow(self, exp: u32) -> Self {
607 Saturating(self.0.saturating_pow(exp))
613 saturating_int_impl! { usize u8 u16 u32 u64 u128 isize i8 i16 i32 i64 i128 }
615 macro_rules! saturating_int_impl_signed {
617 impl Saturating<$t> {
618 /// Returns the number of leading zeros in the binary representation of `self`.
625 /// #![feature(saturating_int_impl)]
626 /// use std::num::Saturating;
628 #[doc = concat!("let n = Saturating(", stringify!($t), "::MAX >> 2);")]
630 /// assert_eq!(n.leading_zeros(), 3);
633 #[unstable(feature = "saturating_int_impl", issue = "87920")]
634 pub const fn leading_zeros(self) -> u32 {
635 self.0.leading_zeros()
638 /// Saturating absolute value. Computes `self.abs()`, returning `MAX` if `self == MIN`
639 /// instead of overflowing.
646 /// #![feature(saturating_int_impl)]
647 /// use std::num::Saturating;
649 #[doc = concat!("assert_eq!(Saturating(100", stringify!($t), ").abs(), Saturating(100));")]
650 #[doc = concat!("assert_eq!(Saturating(-100", stringify!($t), ").abs(), Saturating(100));")]
651 #[doc = concat!("assert_eq!(Saturating(", stringify!($t), "::MIN).abs(), Saturating((", stringify!($t), "::MIN + 1).abs()));")]
652 #[doc = concat!("assert_eq!(Saturating(", stringify!($t), "::MIN).abs(), Saturating(", stringify!($t), "::MIN.saturating_abs()));")]
653 /// assert_eq!(Saturating(-128i8).abs().0 as u8, i8::MAX as u8);
656 #[unstable(feature = "saturating_int_impl", issue = "87920")]
657 pub fn abs(self) -> Saturating<$t> {
658 Saturating(self.0.saturating_abs())
661 /// Returns a number representing sign of `self`.
663 /// - `0` if the number is zero
664 /// - `1` if the number is positive
665 /// - `-1` if the number is negative
672 /// #![feature(saturating_int_impl)]
673 /// use std::num::Saturating;
675 #[doc = concat!("assert_eq!(Saturating(10", stringify!($t), ").signum(), Saturating(1));")]
676 #[doc = concat!("assert_eq!(Saturating(0", stringify!($t), ").signum(), Saturating(0));")]
677 #[doc = concat!("assert_eq!(Saturating(-10", stringify!($t), ").signum(), Saturating(-1));")]
680 #[unstable(feature = "saturating_int_impl", issue = "87920")]
681 pub fn signum(self) -> Saturating<$t> {
682 Saturating(self.0.signum())
685 /// Returns `true` if `self` is positive and `false` if the number is zero or
693 /// #![feature(saturating_int_impl)]
694 /// use std::num::Saturating;
696 #[doc = concat!("assert!(Saturating(10", stringify!($t), ").is_positive());")]
697 #[doc = concat!("assert!(!Saturating(-10", stringify!($t), ").is_positive());")]
700 #[unstable(feature = "saturating_int_impl", issue = "87920")]
701 pub const fn is_positive(self) -> bool {
705 /// Returns `true` if `self` is negative and `false` if the number is zero or
713 /// #![feature(saturating_int_impl)]
714 /// use std::num::Saturating;
716 #[doc = concat!("assert!(Saturating(-10", stringify!($t), ").is_negative());")]
717 #[doc = concat!("assert!(!Saturating(10", stringify!($t), ").is_negative());")]
720 #[unstable(feature = "saturating_int_impl", issue = "87920")]
721 pub const fn is_negative(self) -> bool {
728 saturating_int_impl_signed! { isize i8 i16 i32 i64 i128 }
730 macro_rules! saturating_int_impl_unsigned {
732 impl Saturating<$t> {
733 /// Returns the number of leading zeros in the binary representation of `self`.
740 /// #![feature(saturating_int_impl)]
741 /// use std::num::Saturating;
743 #[doc = concat!("let n = Saturating(", stringify!($t), "::MAX >> 2);")]
745 /// assert_eq!(n.leading_zeros(), 2);
748 #[unstable(feature = "saturating_int_impl", issue = "87920")]
749 pub const fn leading_zeros(self) -> u32 {
750 self.0.leading_zeros()
753 /// Returns `true` if and only if `self == 2^k` for some `k`.
760 /// #![feature(saturating_int_impl)]
761 /// use std::num::Saturating;
763 #[doc = concat!("assert!(Saturating(16", stringify!($t), ").is_power_of_two());")]
764 #[doc = concat!("assert!(!Saturating(10", stringify!($t), ").is_power_of_two());")]
767 #[unstable(feature = "saturating_int_impl", issue = "87920")]
768 pub fn is_power_of_two(self) -> bool {
769 self.0.is_power_of_two()
776 saturating_int_impl_unsigned! { usize u8 u16 u32 u64 u128 }