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 modular arithmetic (e.g.,
17 /// Saturating arithmetic can be achieved either through methods like
18 /// `saturating_add`, or through the `Saturating<T>` type, which says that
19 /// all standard arithmetic operations on the underlying value are
20 /// intended to have saturating semantics.
22 /// The underlying value can be retrieved through the `.0` index of the
23 /// `Saturating` tuple.
28 /// #![feature(saturating_int_impl)]
29 /// use std::num::Saturating;
31 /// let max = Saturating(u32::MAX);
32 /// let one = Saturating(1u32);
34 /// assert_eq!(u32::MAX, (max + one).0);
36 #[unstable(feature = "saturating_int_impl", issue = "87920")]
37 #[derive(PartialEq, Eq, PartialOrd, Ord, Clone, Copy, Default, Hash)]
39 pub struct Saturating<T>(#[stable(feature = "rust1", since = "1.0.0")] pub T);
41 #[stable(feature = "rust1", since = "1.0.0")]
42 impl<T: fmt::Debug> fmt::Debug for Saturating<T> {
43 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
48 #[stable(feature = "saturating_display", since = "1.10.0")]
49 impl<T: fmt::Display> fmt::Display for Saturating<T> {
50 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
55 #[stable(feature = "saturating_fmt", since = "1.11.0")]
56 impl<T: fmt::Binary> fmt::Binary for Saturating<T> {
57 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
62 #[stable(feature = "saturating_fmt", since = "1.11.0")]
63 impl<T: fmt::Octal> fmt::Octal for Saturating<T> {
64 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
69 #[stable(feature = "saturating_fmt", since = "1.11.0")]
70 impl<T: fmt::LowerHex> fmt::LowerHex for Saturating<T> {
71 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
76 #[stable(feature = "saturating_fmt", since = "1.11.0")]
77 impl<T: fmt::UpperHex> fmt::UpperHex for Saturating<T> {
78 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
83 #[allow(unused_macros)]
84 macro_rules! sh_impl_signed {
85 ($t:ident, $f:ident) => {
90 macro_rules! sh_impl_unsigned {
91 ($t:ident, $f:ident) => {
96 // FIXME (#23545): uncomment the remaining impls
97 macro_rules! sh_impl_all {
99 //sh_impl_unsigned! { $t, u8 }
100 //sh_impl_unsigned! { $t, u16 }
101 //sh_impl_unsigned! { $t, u32 }
102 //sh_impl_unsigned! { $t, u64 }
103 //sh_impl_unsigned! { $t, u128 }
104 sh_impl_unsigned! { $t, usize }
106 //sh_impl_signed! { $t, i8 }
107 //sh_impl_signed! { $t, i16 }
108 //sh_impl_signed! { $t, i32 }
109 //sh_impl_signed! { $t, i64 }
110 //sh_impl_signed! { $t, i128 }
111 //sh_impl_signed! { $t, isize }
115 sh_impl_all! { u8 u16 u32 u64 u128 usize i8 i16 i32 i64 i128 isize }
117 // FIXME(30524): impl Op<T> for Saturating<T>, impl OpAssign<T> for Saturating<T>
118 macro_rules! saturating_impl {
120 #[stable(feature = "rust1", since = "1.0.0")]
121 impl Add for Saturating<$t> {
122 type Output = Saturating<$t>;
125 fn add(self, other: Saturating<$t>) -> Saturating<$t> {
126 Saturating(self.0.saturating_add(other.0))
129 forward_ref_binop! { impl Add, add for Saturating<$t>, Saturating<$t>,
130 #[stable(feature = "saturating_ref", since = "1.14.0")] }
132 #[stable(feature = "op_assign_traits", since = "1.8.0")]
133 impl AddAssign for Saturating<$t> {
135 fn add_assign(&mut self, other: Saturating<$t>) {
136 *self = *self + other;
139 forward_ref_op_assign! { impl AddAssign, add_assign for Saturating<$t>, Saturating<$t> }
141 #[stable(feature = "rust1", since = "1.0.0")]
142 impl Sub for Saturating<$t> {
143 type Output = Saturating<$t>;
146 fn sub(self, other: Saturating<$t>) -> Saturating<$t> {
147 Saturating(self.0.saturating_sub(other.0))
150 forward_ref_binop! { impl Sub, sub for Saturating<$t>, Saturating<$t>,
151 #[stable(feature = "saturating_ref", since = "1.14.0")] }
153 #[stable(feature = "op_assign_traits", since = "1.8.0")]
154 impl SubAssign for Saturating<$t> {
156 fn sub_assign(&mut self, other: Saturating<$t>) {
157 *self = *self - other;
160 forward_ref_op_assign! { impl SubAssign, sub_assign for Saturating<$t>, Saturating<$t> }
162 #[stable(feature = "rust1", since = "1.0.0")]
163 impl Mul for Saturating<$t> {
164 type Output = Saturating<$t>;
167 fn mul(self, other: Saturating<$t>) -> Saturating<$t> {
168 Saturating(self.0.saturating_mul(other.0))
171 forward_ref_binop! { impl Mul, mul for Saturating<$t>, Saturating<$t>,
172 #[stable(feature = "saturating_ref", since = "1.14.0")] }
174 #[stable(feature = "op_assign_traits", since = "1.8.0")]
175 impl MulAssign for Saturating<$t> {
177 fn mul_assign(&mut self, other: Saturating<$t>) {
178 *self = *self * other;
181 forward_ref_op_assign! { impl MulAssign, mul_assign for Saturating<$t>, Saturating<$t> }
183 #[stable(feature = "saturating_div", since = "1.3.0")]
184 impl Div for Saturating<$t> {
185 type Output = Saturating<$t>;
188 fn div(self, other: Saturating<$t>) -> Saturating<$t> {
189 // saturating div is the default behavior?
190 Saturating(self.0.div(other.0))
193 forward_ref_binop! { impl Div, div for Saturating<$t>, Saturating<$t>,
194 #[stable(feature = "saturating_ref", since = "1.14.0")] }
196 #[stable(feature = "op_assign_traits", since = "1.8.0")]
197 impl DivAssign for Saturating<$t> {
199 fn div_assign(&mut self, other: Saturating<$t>) {
200 *self = *self / other;
203 forward_ref_op_assign! { impl DivAssign, div_assign for Saturating<$t>, Saturating<$t> }
205 #[stable(feature = "rust1", since = "1.0.0")]
206 impl Not for Saturating<$t> {
207 type Output = Saturating<$t>;
210 fn not(self) -> Saturating<$t> {
214 forward_ref_unop! { impl Not, not for Saturating<$t>,
215 #[stable(feature = "saturating_ref", since = "1.14.0")] }
217 #[stable(feature = "rust1", since = "1.0.0")]
218 impl BitXor for Saturating<$t> {
219 type Output = Saturating<$t>;
222 fn bitxor(self, other: Saturating<$t>) -> Saturating<$t> {
223 Saturating(self.0 ^ other.0)
226 forward_ref_binop! { impl BitXor, bitxor for Saturating<$t>, Saturating<$t>,
227 #[stable(feature = "saturating_ref", since = "1.14.0")] }
229 #[stable(feature = "op_assign_traits", since = "1.8.0")]
230 impl BitXorAssign for Saturating<$t> {
232 fn bitxor_assign(&mut self, other: Saturating<$t>) {
233 *self = *self ^ other;
236 forward_ref_op_assign! { impl BitXorAssign, bitxor_assign for Saturating<$t>, Saturating<$t> }
238 #[stable(feature = "rust1", since = "1.0.0")]
239 impl BitOr for Saturating<$t> {
240 type Output = Saturating<$t>;
243 fn bitor(self, other: Saturating<$t>) -> Saturating<$t> {
244 Saturating(self.0 | other.0)
247 forward_ref_binop! { impl BitOr, bitor for Saturating<$t>, Saturating<$t>,
248 #[stable(feature = "saturating_ref", since = "1.14.0")] }
250 #[stable(feature = "op_assign_traits", since = "1.8.0")]
251 impl BitOrAssign for Saturating<$t> {
253 fn bitor_assign(&mut self, other: Saturating<$t>) {
254 *self = *self | other;
257 forward_ref_op_assign! { impl BitOrAssign, bitor_assign for Saturating<$t>, Saturating<$t> }
259 #[stable(feature = "rust1", since = "1.0.0")]
260 impl BitAnd for Saturating<$t> {
261 type Output = Saturating<$t>;
264 fn bitand(self, other: Saturating<$t>) -> Saturating<$t> {
265 Saturating(self.0 & other.0)
268 forward_ref_binop! { impl BitAnd, bitand for Saturating<$t>, Saturating<$t>,
269 #[stable(feature = "saturating_ref", since = "1.14.0")] }
271 #[stable(feature = "op_assign_traits", since = "1.8.0")]
272 impl BitAndAssign for Saturating<$t> {
274 fn bitand_assign(&mut self, other: Saturating<$t>) {
275 *self = *self & other;
278 forward_ref_op_assign! { impl BitAndAssign, bitand_assign for Saturating<$t>, Saturating<$t> }
280 #[stable(feature = "saturating_neg", since = "1.45.0")]
281 impl Neg for Saturating<$t> {
284 fn neg(self) -> Self {
288 forward_ref_unop! { impl Neg, neg for Saturating<$t>,
289 #[stable(feature = "saturating_ref", since = "1.14.0")] }
294 saturating_impl! { usize u8 u16 u32 u64 u128 isize i8 i16 i32 i64 i128 }
296 macro_rules! saturating_int_impl {
298 impl Saturating<$t> {
299 /// Returns the smallest value that can be represented by this integer type.
306 /// #![feature(saturating_int_impl)]
307 /// use std::num::Saturating;
309 #[doc = concat!("assert_eq!(<Saturating<", stringify!($t), ">>::MIN, Saturating(", stringify!($t), "::MIN));")]
311 #[unstable(feature = "saturating_int_impl", issue = "87920")]
312 pub const MIN: Self = Self(<$t>::MIN);
314 /// Returns the largest value that can be represented by this integer type.
321 /// #![feature(saturating_int_impl)]
322 /// use std::num::Saturating;
324 #[doc = concat!("assert_eq!(<Saturating<", stringify!($t), ">>::MAX, Saturating(", stringify!($t), "::MAX));")]
326 #[unstable(feature = "saturating_int_impl", issue = "87920")]
327 pub const MAX: Self = Self(<$t>::MAX);
329 /// Returns the size of this integer type in bits.
336 /// #![feature(saturating_int_impl)]
337 /// use std::num::Saturating;
339 #[doc = concat!("assert_eq!(<Saturating<", stringify!($t), ">>::BITS, ", stringify!($t), "::BITS);")]
341 #[unstable(feature = "saturating_int_impl", issue = "87920")]
342 pub const BITS: u32 = <$t>::BITS;
344 /// Returns the number of ones in the binary representation of `self`.
351 /// #![feature(saturating_int_impl)]
352 /// use std::num::Saturating;
354 #[doc = concat!("let n = Saturating(0b01001100", stringify!($t), ");")]
356 /// assert_eq!(n.count_ones(), 3);
359 #[doc(alias = "popcount")]
360 #[doc(alias = "popcnt")]
361 #[unstable(feature = "saturating_int_impl", issue = "87920")]
362 pub const fn count_ones(self) -> u32 {
366 /// Returns the number of zeros in the binary representation of `self`.
373 /// #![feature(saturating_int_impl)]
374 /// use std::num::Saturating;
376 #[doc = concat!("assert_eq!(Saturating(!0", stringify!($t), ").count_zeros(), 0);")]
379 #[unstable(feature = "saturating_int_impl", issue = "87920")]
380 pub const fn count_zeros(self) -> u32 {
384 /// Returns the number of trailing zeros in the binary representation of `self`.
391 /// #![feature(saturating_int_impl)]
392 /// use std::num::Saturating;
394 #[doc = concat!("let n = Saturating(0b0101000", stringify!($t), ");")]
396 /// assert_eq!(n.trailing_zeros(), 3);
399 #[unstable(feature = "saturating_int_impl", issue = "87920")]
400 pub const fn trailing_zeros(self) -> u32 {
401 self.0.trailing_zeros()
404 /// Shifts the bits to the left by a specified amount, `n`,
405 /// saturating the truncated bits to the end of the resulting
408 /// Please note this isn't the same operation as the `<<` shifting
416 /// #![feature(saturating_int_impl)]
417 /// use std::num::Saturating;
419 /// let n: Saturating<i64> = Saturating(0x0123456789ABCDEF);
420 /// let m: Saturating<i64> = Saturating(-0x76543210FEDCBA99);
422 /// assert_eq!(n.rotate_left(32), m);
425 #[unstable(feature = "saturating_int_impl", issue = "87920")]
426 pub const fn rotate_left(self, n: u32) -> Self {
427 Saturating(self.0.rotate_left(n))
430 /// Shifts the bits to the right by a specified amount, `n`,
431 /// saturating the truncated bits to the beginning of the resulting
434 /// Please note this isn't the same operation as the `>>` shifting
442 /// #![feature(saturating_int_impl)]
443 /// use std::num::Saturating;
445 /// let n: Saturating<i64> = Saturating(0x0123456789ABCDEF);
446 /// let m: Saturating<i64> = Saturating(-0xFEDCBA987654322);
448 /// assert_eq!(n.rotate_right(4), m);
451 #[unstable(feature = "saturating_int_impl", issue = "87920")]
452 pub const fn rotate_right(self, n: u32) -> Self {
453 Saturating(self.0.rotate_right(n))
456 /// Reverses the byte order of the integer.
463 /// #![feature(saturating_int_impl)]
464 /// use std::num::Saturating;
466 /// let n: Saturating<i16> = Saturating(0b0000000_01010101);
467 /// assert_eq!(n, Saturating(85));
469 /// let m = n.swap_bytes();
471 /// assert_eq!(m, Saturating(0b01010101_00000000));
472 /// assert_eq!(m, Saturating(21760));
475 #[unstable(feature = "saturating_int_impl", issue = "87920")]
476 pub const fn swap_bytes(self) -> Self {
477 Saturating(self.0.swap_bytes())
480 /// Reverses the bit pattern of the integer.
484 /// Please note that this example is shared between integer types.
485 /// Which explains why `i16` is used here.
490 /// #![feature(saturating_int_impl)]
491 /// use std::num::Saturating;
493 /// let n = Saturating(0b0000000_01010101i16);
494 /// assert_eq!(n, Saturating(85));
496 /// let m = n.reverse_bits();
498 /// assert_eq!(m.0 as u16, 0b10101010_00000000);
499 /// assert_eq!(m, Saturating(-22016));
501 #[stable(feature = "reverse_bits", since = "1.37.0")]
502 #[rustc_const_stable(feature = "const_reverse_bits", since = "1.37.0")]
505 pub const fn reverse_bits(self) -> Self {
506 Saturating(self.0.reverse_bits())
509 /// Converts an integer from big endian to the target's endianness.
511 /// On big endian this is a no-op. On little endian the bytes are
519 /// #![feature(saturating_int_impl)]
520 /// use std::num::Saturating;
522 #[doc = concat!("let n = Saturating(0x1A", stringify!($t), ");")]
524 /// if cfg!(target_endian = "big") {
525 #[doc = concat!(" assert_eq!(<Saturating<", stringify!($t), ">>::from_be(n), n)")]
527 #[doc = concat!(" assert_eq!(<Saturating<", stringify!($t), ">>::from_be(n), n.swap_bytes())")]
531 #[unstable(feature = "saturating_int_impl", issue = "87920")]
532 pub const fn from_be(x: Self) -> Self {
533 Saturating(<$t>::from_be(x.0))
536 /// Converts an integer from little endian to the target's endianness.
538 /// On little endian this is a no-op. On big endian the bytes are
546 /// #![feature(saturating_int_impl)]
547 /// use std::num::Saturating;
549 #[doc = concat!("let n = Saturating(0x1A", stringify!($t), ");")]
551 /// if cfg!(target_endian = "little") {
552 #[doc = concat!(" assert_eq!(<Saturating<", stringify!($t), ">>::from_le(n), n)")]
554 #[doc = concat!(" assert_eq!(<Saturating<", stringify!($t), ">>::from_le(n), n.swap_bytes())")]
558 #[unstable(feature = "saturating_int_impl", issue = "87920")]
559 pub const fn from_le(x: Self) -> Self {
560 Saturating(<$t>::from_le(x.0))
563 /// Converts `self` to big endian from the target's endianness.
565 /// On big endian this is a no-op. On little endian the bytes are
573 /// #![feature(saturating_int_impl)]
574 /// use std::num::Saturating;
576 #[doc = concat!("let n = Saturating(0x1A", stringify!($t), ");")]
578 /// if cfg!(target_endian = "big") {
579 /// assert_eq!(n.to_be(), n)
581 /// assert_eq!(n.to_be(), n.swap_bytes())
585 #[unstable(feature = "saturating_int_impl", issue = "87920")]
586 pub const fn to_be(self) -> Self {
587 Saturating(self.0.to_be())
590 /// Converts `self` to little endian from the target's endianness.
592 /// On little endian this is a no-op. On big endian the bytes are
600 /// #![feature(saturating_int_impl)]
601 /// use std::num::Saturating;
603 #[doc = concat!("let n = Saturating(0x1A", stringify!($t), ");")]
605 /// if cfg!(target_endian = "little") {
606 /// assert_eq!(n.to_le(), n)
608 /// assert_eq!(n.to_le(), n.swap_bytes())
612 #[unstable(feature = "saturating_int_impl", issue = "87920")]
613 pub const fn to_le(self) -> Self {
614 Saturating(self.0.to_le())
617 /// Raises self to the power of `exp`, using exponentiation by squaring.
624 /// #![feature(saturating_int_impl)]
625 /// use std::num::Saturating;
627 #[doc = concat!("assert_eq!(Saturating(3", stringify!($t), ").pow(4), Saturating(81));")]
630 /// Results that are too large are wrapped:
633 /// #![feature(saturating_int_impl)]
634 /// use std::num::Saturating;
636 /// assert_eq!(Saturating(3i8).pow(5), Saturating(127));
637 /// assert_eq!(Saturating(3i8).pow(6), Saturating(127));
640 #[unstable(feature = "saturating_int_impl", issue = "87920")]
641 pub fn pow(self, exp: u32) -> Self {
642 Saturating(self.0.saturating_pow(exp))
648 saturating_int_impl! { usize u8 u16 u32 u64 u128 isize i8 i16 i32 i64 i128 }
650 macro_rules! saturating_int_impl_signed {
652 impl Saturating<$t> {
653 /// Returns the number of leading zeros in the binary representation of `self`.
660 /// #![feature(saturating_int_impl)]
661 /// use std::num::Saturating;
663 #[doc = concat!("let n = Saturating(", stringify!($t), "::MAX) / 4;")]
665 /// assert_eq!(n.leading_zeros(), 3);
668 #[unstable(feature = "saturating_int_impl", issue = "87920")]
669 pub const fn leading_zeros(self) -> u32 {
670 self.0.leading_zeros()
673 /// Computes the absolute value of `self`, saturating around at
674 /// the boundary of the type.
676 /// The only case where such saturating can occur is when one takes the absolute value of the negative
677 /// minimal value for the type this is a positive value that is too large to represent in the type. In
678 /// such a case, this function returns `MIN` itself.
685 /// #![feature(saturating_int_impl)]
686 /// use std::num::Saturating;
688 #[doc = concat!("assert_eq!(Saturating(100", stringify!($t), ").abs(), Saturating(100));")]
689 #[doc = concat!("assert_eq!(Saturating(-100", stringify!($t), ").abs(), Saturating(100));")]
690 #[doc = concat!("assert_eq!(Saturating(", stringify!($t), "::MIN).abs(), Saturating(", stringify!($t), "::MIN));")]
691 /// assert_eq!(Saturating(-128i8).abs().0 as u8, 128u8);
694 #[unstable(feature = "saturating_int_impl", issue = "87920")]
695 pub fn abs(self) -> Saturating<$t> {
696 Saturating(self.0.saturating_abs())
699 /// Returns a number representing sign of `self`.
701 /// - `0` if the number is zero
702 /// - `1` if the number is positive
703 /// - `-1` if the number is negative
710 /// #![feature(saturating_int_impl)]
711 /// use std::num::Saturating;
713 #[doc = concat!("assert_eq!(Saturating(10", stringify!($t), ").signum(), Saturating(1));")]
714 #[doc = concat!("assert_eq!(Saturating(0", stringify!($t), ").signum(), Saturating(0));")]
715 #[doc = concat!("assert_eq!(Saturating(-10", stringify!($t), ").signum(), Saturating(-1));")]
718 #[unstable(feature = "saturating_int_impl", issue = "87920")]
719 pub fn signum(self) -> Saturating<$t> {
720 Saturating(self.0.signum())
723 /// Returns `true` if `self` is positive and `false` if the number is zero or
731 /// #![feature(saturating_int_impl)]
732 /// use std::num::Saturating;
734 #[doc = concat!("assert!(Saturating(10", stringify!($t), ").is_positive());")]
735 #[doc = concat!("assert!(!Saturating(-10", stringify!($t), ").is_positive());")]
738 #[unstable(feature = "saturating_int_impl", issue = "87920")]
739 pub const fn is_positive(self) -> bool {
743 /// Returns `true` if `self` is negative and `false` if the number is zero or
751 /// #![feature(saturating_int_impl)]
752 /// use std::num::Saturating;
754 #[doc = concat!("assert!(Saturating(-10", stringify!($t), ").is_negative());")]
755 #[doc = concat!("assert!(!Saturating(10", stringify!($t), ").is_negative());")]
758 #[unstable(feature = "saturating_int_impl", issue = "87920")]
759 pub const fn is_negative(self) -> bool {
766 saturating_int_impl_signed! { isize i8 i16 i32 i64 i128 }
768 macro_rules! saturating_int_impl_unsigned {
770 impl Saturating<$t> {
771 /// Returns the number of leading zeros in the binary representation of `self`.
778 /// #![feature(saturating_int_impl)]
779 /// use std::num::Saturating;
781 #[doc = concat!("let n = Saturating(", stringify!($t), "::MAX) / 4;")]
783 /// assert_eq!(n.leading_zeros(), 2);
786 #[unstable(feature = "saturating_int_impl", issue = "87920")]
787 pub const fn leading_zeros(self) -> u32 {
788 self.0.leading_zeros()
791 /// Returns `true` if and only if `self == 2^k` for some `k`.
798 /// #![feature(saturating_int_impl)]
799 /// use std::num::Saturating;
801 #[doc = concat!("assert!(Saturating(16", stringify!($t), ").is_power_of_two());")]
802 #[doc = concat!("assert!(!Saturating(10", stringify!($t), ").is_power_of_two());")]
805 #[unstable(feature = "saturating_int_impl", issue = "87920")]
806 pub fn is_power_of_two(self) -> bool {
807 self.0.is_power_of_two()
814 saturating_int_impl_unsigned! { usize u8 u16 u32 u64 u128 }