core/num/nonzero.rs
1//! Definitions of integer that is known not to equal zero.
2
3use super::{IntErrorKind, ParseIntError};
4use crate::clone::{TrivialClone, UseCloned};
5use crate::cmp::Ordering;
6use crate::hash::{Hash, Hasher};
7use crate::marker::{Destruct, Freeze, StructuralPartialEq};
8use crate::num::imp;
9use crate::ops::{BitOr, BitOrAssign, Div, DivAssign, Neg, Rem, RemAssign};
10use crate::panic::{RefUnwindSafe, UnwindSafe};
11use crate::str::FromStr;
12use crate::{fmt, intrinsics, ptr, ub_checks};
13
14/// A marker trait for primitive types which can be zero.
15///
16/// This is an implementation detail for <code>[NonZero]\<T></code> which may disappear or be replaced at any time.
17///
18/// # Safety
19///
20/// Types implementing this trait must be primitives that are valid when zeroed.
21///
22/// The associated `Self::NonZeroInner` type must have the same size+align as `Self`,
23/// but with a niche and bit validity making it so the following `transmutes` are sound:
24///
25/// - `Self::NonZeroInner` to `Option<Self::NonZeroInner>`
26/// - `Option<Self::NonZeroInner>` to `Self`
27///
28/// (And, consequently, `Self::NonZeroInner` to `Self`.)
29#[unstable(
30 feature = "nonzero_internals",
31 reason = "implementation detail which may disappear or be replaced at any time",
32 issue = "none"
33)]
34pub impl(self) unsafe trait ZeroablePrimitive: Sized + Copy {
35 /// A type like `Self` but with a niche that includes zero.
36 type NonZeroInner: Sized + Copy;
37}
38
39macro_rules! impl_zeroable_primitive {
40 ($($NonZeroInner:ident ( $primitive:ty )),+ $(,)?) => {
41 $(
42 #[unstable(
43 feature = "nonzero_internals",
44 reason = "implementation detail which may disappear or be replaced at any time",
45 issue = "none"
46 )]
47 unsafe impl ZeroablePrimitive for $primitive {
48 type NonZeroInner = super::niche_types::$NonZeroInner;
49 }
50 )+
51 };
52}
53
54impl_zeroable_primitive!(
55 NonZeroU8Inner(u8),
56 NonZeroU16Inner(u16),
57 NonZeroU32Inner(u32),
58 NonZeroU64Inner(u64),
59 NonZeroU128Inner(u128),
60 NonZeroUsizeInner(usize),
61 NonZeroI8Inner(i8),
62 NonZeroI16Inner(i16),
63 NonZeroI32Inner(i32),
64 NonZeroI64Inner(i64),
65 NonZeroI128Inner(i128),
66 NonZeroIsizeInner(isize),
67 NonZeroCharInner(char),
68);
69
70/// A value that is known not to equal zero.
71///
72/// This enables some memory layout optimization.
73/// For example, `Option<NonZero<u32>>` is the same size as `u32`:
74///
75/// ```
76/// use core::{num::NonZero};
77///
78/// assert_eq!(size_of::<Option<NonZero<u32>>>(), size_of::<u32>());
79/// ```
80///
81/// # Layout
82///
83/// `NonZero<T>` is guaranteed to have the same layout and bit validity as `T`
84/// with the exception that the all-zero bit pattern is invalid.
85/// `Option<NonZero<T>>` is guaranteed to be compatible with `T`, including in
86/// FFI.
87///
88/// Thanks to the [null pointer optimization], `NonZero<T>` and
89/// `Option<NonZero<T>>` are guaranteed to have the same size and alignment:
90///
91/// ```
92/// use std::num::NonZero;
93///
94/// assert_eq!(size_of::<NonZero<u32>>(), size_of::<Option<NonZero<u32>>>());
95/// assert_eq!(align_of::<NonZero<u32>>(), align_of::<Option<NonZero<u32>>>());
96/// ```
97///
98/// [null pointer optimization]: crate::option#representation
99///
100/// # Note on generic usage
101///
102/// `NonZero<T>` can only be used with some standard library primitive types
103/// (such as `u8`, `i32`, and etc.). The type parameter `T` must implement the
104/// internal trait [`ZeroablePrimitive`], which is currently permanently unstable
105/// and cannot be implemented by users. Therefore, you cannot use `NonZero<T>`
106/// with your own types, nor can you implement traits for all `NonZero<T>`,
107/// only for concrete types.
108#[stable(feature = "generic_nonzero", since = "1.79.0")]
109#[repr(transparent)]
110#[rustc_nonnull_optimization_guaranteed]
111#[rustc_diagnostic_item = "NonZero"]
112pub struct NonZero<T: ZeroablePrimitive>(T::NonZeroInner);
113
114macro_rules! impl_nonzero_fmt {
115 ($(#[$Attribute:meta] $Trait:ident)*) => {
116 $(
117 #[$Attribute]
118 impl<T> fmt::$Trait for NonZero<T>
119 where
120 T: ZeroablePrimitive + fmt::$Trait,
121 {
122 #[inline]
123 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
124 self.get().fmt(f)
125 }
126 }
127 )*
128 };
129}
130
131impl_nonzero_fmt! {
132 #[stable(feature = "nonzero", since = "1.28.0")]
133 Debug
134 #[stable(feature = "nonzero", since = "1.28.0")]
135 Display
136 #[stable(feature = "nonzero", since = "1.28.0")]
137 Binary
138 #[stable(feature = "nonzero", since = "1.28.0")]
139 Octal
140 #[stable(feature = "nonzero", since = "1.28.0")]
141 LowerHex
142 #[stable(feature = "nonzero", since = "1.28.0")]
143 UpperHex
144 #[stable(feature = "nonzero_fmt_exp", since = "1.84.0")]
145 LowerExp
146 #[stable(feature = "nonzero_fmt_exp", since = "1.84.0")]
147 UpperExp
148}
149
150macro_rules! impl_nonzero_auto_trait {
151 (unsafe $Trait:ident) => {
152 #[stable(feature = "nonzero", since = "1.28.0")]
153 unsafe impl<T> $Trait for NonZero<T> where T: ZeroablePrimitive + $Trait {}
154 };
155 ($Trait:ident) => {
156 #[stable(feature = "nonzero", since = "1.28.0")]
157 impl<T> $Trait for NonZero<T> where T: ZeroablePrimitive + $Trait {}
158 };
159}
160
161// Implement auto-traits manually based on `T` to avoid docs exposing
162// the `ZeroablePrimitive::NonZeroInner` implementation detail.
163impl_nonzero_auto_trait!(unsafe Freeze);
164impl_nonzero_auto_trait!(RefUnwindSafe);
165impl_nonzero_auto_trait!(unsafe Send);
166impl_nonzero_auto_trait!(unsafe Sync);
167impl_nonzero_auto_trait!(Unpin);
168impl_nonzero_auto_trait!(UnwindSafe);
169
170#[stable(feature = "nonzero", since = "1.28.0")]
171#[rustc_const_unstable(feature = "const_clone", issue = "142757")]
172const impl<T> Clone for NonZero<T>
173where
174 T: ZeroablePrimitive,
175{
176 #[inline]
177 fn clone(&self) -> Self {
178 *self
179 }
180}
181
182#[unstable(feature = "ergonomic_clones", issue = "132290")]
183impl<T> UseCloned for NonZero<T> where T: ZeroablePrimitive {}
184
185#[stable(feature = "nonzero", since = "1.28.0")]
186impl<T> Copy for NonZero<T> where T: ZeroablePrimitive {}
187
188#[doc(hidden)]
189#[unstable(feature = "trivial_clone", issue = "none")]
190#[rustc_const_unstable(feature = "const_clone", issue = "142757")]
191const unsafe impl<T> TrivialClone for NonZero<T> where T: ZeroablePrimitive {}
192
193#[stable(feature = "nonzero", since = "1.28.0")]
194#[rustc_const_unstable(feature = "const_cmp", issue = "143800")]
195const impl<T> PartialEq for NonZero<T>
196where
197 T: ZeroablePrimitive + [const] PartialEq,
198{
199 #[inline]
200 fn eq(&self, other: &Self) -> bool {
201 self.get() == other.get()
202 }
203
204 #[inline]
205 fn ne(&self, other: &Self) -> bool {
206 self.get() != other.get()
207 }
208}
209
210#[unstable(feature = "structural_match", issue = "31434")]
211impl<T> StructuralPartialEq for NonZero<T> where T: ZeroablePrimitive + StructuralPartialEq {}
212
213#[stable(feature = "nonzero", since = "1.28.0")]
214#[rustc_const_unstable(feature = "const_cmp", issue = "143800")]
215const impl<T> Eq for NonZero<T> where T: ZeroablePrimitive + [const] Eq {}
216
217#[stable(feature = "nonzero", since = "1.28.0")]
218#[rustc_const_unstable(feature = "const_cmp", issue = "143800")]
219const impl<T> PartialOrd for NonZero<T>
220where
221 T: ZeroablePrimitive + [const] PartialOrd,
222{
223 #[inline]
224 fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
225 self.get().partial_cmp(&other.get())
226 }
227
228 #[inline]
229 fn lt(&self, other: &Self) -> bool {
230 self.get() < other.get()
231 }
232
233 #[inline]
234 fn le(&self, other: &Self) -> bool {
235 self.get() <= other.get()
236 }
237
238 #[inline]
239 fn gt(&self, other: &Self) -> bool {
240 self.get() > other.get()
241 }
242
243 #[inline]
244 fn ge(&self, other: &Self) -> bool {
245 self.get() >= other.get()
246 }
247}
248
249#[stable(feature = "nonzero", since = "1.28.0")]
250#[rustc_const_unstable(feature = "const_cmp", issue = "143800")]
251const impl<T> Ord for NonZero<T>
252where
253 // FIXME(const_hack): the T: ~const Destruct should be inferred from the Self: ~const Destruct.
254 // See https://github.com/rust-lang/rust/issues/144207
255 T: ZeroablePrimitive + [const] Ord + [const] Destruct,
256{
257 #[inline]
258 fn cmp(&self, other: &Self) -> Ordering {
259 self.get().cmp(&other.get())
260 }
261
262 #[inline]
263 fn max(self, other: Self) -> Self {
264 // SAFETY: The maximum of two non-zero values is still non-zero.
265 unsafe { Self::new_unchecked(self.get().max(other.get())) }
266 }
267
268 #[inline]
269 fn min(self, other: Self) -> Self {
270 // SAFETY: The minimum of two non-zero values is still non-zero.
271 unsafe { Self::new_unchecked(self.get().min(other.get())) }
272 }
273
274 #[inline]
275 fn clamp(self, min: Self, max: Self) -> Self {
276 // SAFETY: A non-zero value clamped between two non-zero values is still non-zero.
277 unsafe { Self::new_unchecked(self.get().clamp(min.get(), max.get())) }
278 }
279}
280
281#[stable(feature = "nonzero", since = "1.28.0")]
282impl<T> Hash for NonZero<T>
283where
284 T: ZeroablePrimitive + Hash,
285{
286 #[inline]
287 fn hash<H>(&self, state: &mut H)
288 where
289 H: Hasher,
290 {
291 self.get().hash(state)
292 }
293}
294
295#[stable(feature = "from_nonzero", since = "1.31.0")]
296#[rustc_const_unstable(feature = "const_convert", issue = "143773")]
297const impl<T> From<NonZero<T>> for T
298where
299 T: ZeroablePrimitive,
300{
301 #[inline]
302 fn from(nonzero: NonZero<T>) -> Self {
303 // Call `get` method to keep range information.
304 nonzero.get()
305 }
306}
307
308#[stable(feature = "nonzero_bitor", since = "1.45.0")]
309#[rustc_const_unstable(feature = "const_ops", issue = "143802")]
310const impl<T> BitOr for NonZero<T>
311where
312 T: ZeroablePrimitive + [const] BitOr<Output = T>,
313{
314 type Output = Self;
315
316 #[inline]
317 fn bitor(self, rhs: Self) -> Self::Output {
318 // SAFETY: Bitwise OR of two non-zero values is still non-zero.
319 unsafe { Self::new_unchecked(self.get() | rhs.get()) }
320 }
321}
322
323#[stable(feature = "nonzero_bitor", since = "1.45.0")]
324#[rustc_const_unstable(feature = "const_ops", issue = "143802")]
325const impl<T> BitOr<T> for NonZero<T>
326where
327 T: ZeroablePrimitive + [const] BitOr<Output = T>,
328{
329 type Output = Self;
330
331 #[inline]
332 fn bitor(self, rhs: T) -> Self::Output {
333 // SAFETY: Bitwise OR of a non-zero value with anything is still non-zero.
334 unsafe { Self::new_unchecked(self.get() | rhs) }
335 }
336}
337
338#[stable(feature = "nonzero_bitor", since = "1.45.0")]
339#[rustc_const_unstable(feature = "const_ops", issue = "143802")]
340const impl<T> BitOr<NonZero<T>> for T
341where
342 T: ZeroablePrimitive + [const] BitOr<Output = T>,
343{
344 type Output = NonZero<T>;
345
346 #[inline]
347 fn bitor(self, rhs: NonZero<T>) -> Self::Output {
348 // SAFETY: Bitwise OR of anything with a non-zero value is still non-zero.
349 unsafe { NonZero::new_unchecked(self | rhs.get()) }
350 }
351}
352
353#[stable(feature = "nonzero_bitor", since = "1.45.0")]
354#[rustc_const_unstable(feature = "const_ops", issue = "143802")]
355const impl<T> BitOrAssign for NonZero<T>
356where
357 T: ZeroablePrimitive,
358 Self: [const] BitOr<Output = Self>,
359{
360 #[inline]
361 fn bitor_assign(&mut self, rhs: Self) {
362 *self = *self | rhs;
363 }
364}
365
366#[stable(feature = "nonzero_bitor", since = "1.45.0")]
367#[rustc_const_unstable(feature = "const_ops", issue = "143802")]
368const impl<T> BitOrAssign<T> for NonZero<T>
369where
370 T: ZeroablePrimitive,
371 Self: [const] BitOr<T, Output = Self>,
372{
373 #[inline]
374 fn bitor_assign(&mut self, rhs: T) {
375 *self = *self | rhs;
376 }
377}
378
379impl<T> NonZero<T>
380where
381 T: ZeroablePrimitive,
382{
383 /// Creates a non-zero if the given value is not zero.
384 #[stable(feature = "nonzero", since = "1.28.0")]
385 #[rustc_const_stable(feature = "const_nonzero_int_methods", since = "1.47.0")]
386 #[must_use]
387 #[inline]
388 pub const fn new(n: T) -> Option<Self> {
389 // SAFETY: Memory layout optimization guarantees that `Option<NonZero<T>>` has
390 // the same layout and size as `T`, with `0` representing `None`.
391 unsafe { intrinsics::transmute_unchecked(n) }
392 }
393
394 /// Creates a non-zero without checking whether the value is non-zero.
395 /// This results in undefined behavior if the value is zero.
396 ///
397 /// # Safety
398 ///
399 /// The value must not be zero.
400 #[stable(feature = "nonzero", since = "1.28.0")]
401 #[rustc_const_stable(feature = "nonzero", since = "1.28.0")]
402 #[must_use]
403 #[inline]
404 #[track_caller]
405 pub const unsafe fn new_unchecked(n: T) -> Self {
406 match Self::new(n) {
407 Some(n) => n,
408 None => {
409 // SAFETY: The caller guarantees that `n` is non-zero, so this is unreachable.
410 unsafe {
411 ub_checks::assert_unsafe_precondition!(
412 check_language_ub,
413 "NonZero::new_unchecked requires the argument to be non-zero",
414 () => false,
415 );
416 intrinsics::unreachable()
417 }
418 }
419 }
420 }
421
422 /// Converts a reference to a non-zero mutable reference
423 /// if the referenced value is not zero.
424 #[unstable(feature = "nonzero_from_mut", issue = "106290")]
425 #[must_use]
426 #[inline]
427 pub fn from_mut(n: &mut T) -> Option<&mut Self> {
428 // SAFETY: Memory layout optimization guarantees that `Option<NonZero<T>>` has
429 // the same layout and size as `T`, with `0` representing `None`.
430 let opt_n = unsafe { &mut *(ptr::from_mut(n).cast::<Option<Self>>()) };
431
432 opt_n.as_mut()
433 }
434
435 /// Converts a mutable reference to a non-zero mutable reference
436 /// without checking whether the referenced value is non-zero.
437 /// This results in undefined behavior if the referenced value is zero.
438 ///
439 /// # Safety
440 ///
441 /// The referenced value must not be zero.
442 #[unstable(feature = "nonzero_from_mut", issue = "106290")]
443 #[must_use]
444 #[inline]
445 #[track_caller]
446 pub unsafe fn from_mut_unchecked(n: &mut T) -> &mut Self {
447 match Self::from_mut(n) {
448 Some(n) => n,
449 None => {
450 // SAFETY: The caller guarantees that `n` references a value that is non-zero, so this is unreachable.
451 unsafe {
452 ub_checks::assert_unsafe_precondition!(
453 check_library_ub,
454 "NonZero::from_mut_unchecked requires the argument to dereference as non-zero",
455 () => false,
456 );
457 intrinsics::unreachable()
458 }
459 }
460 }
461 }
462
463 /// Returns the contained value as a primitive type.
464 #[stable(feature = "nonzero", since = "1.28.0")]
465 #[rustc_const_stable(feature = "const_nonzero_get", since = "1.34.0")]
466 #[inline]
467 pub const fn get(self) -> T {
468 // Rustc can set range metadata only if it loads `self` from
469 // memory somewhere. If the value of `self` was from by-value argument
470 // of some not-inlined function, LLVM don't have range metadata
471 // to understand that the value cannot be zero.
472 //
473 // Using the transmute `assume`s the range at runtime.
474 //
475 // Even once LLVM supports `!range` metadata for function arguments
476 // (see <https://github.com/llvm/llvm-project/issues/76628>), this can't
477 // be `.0` because MCP#807 bans field-projecting into `scalar_valid_range`
478 // types, and it arguably wouldn't want to be anyway because if this is
479 // MIR-inlined, there's no opportunity to put that argument metadata anywhere.
480 //
481 // The good answer here will eventually be pattern types, which will hopefully
482 // allow it to go back to `.0`, maybe with a cast of some sort.
483 //
484 // SAFETY: `ZeroablePrimitive` guarantees that the size and bit validity
485 // of `.0` is such that this transmute is sound.
486 unsafe { intrinsics::transmute_unchecked(self) }
487 }
488}
489
490macro_rules! nonzero_integer {
491 (
492 #[$stability:meta]
493 Self = $Ty:ident,
494 Primitive = $signedness:ident $Int:ident,
495 SignedPrimitive = $Sint:ty,
496 UnsignedPrimitive = $Uint:ty,
497
498 // Used in doc comments.
499 rot = $rot:literal,
500 rot_op = $rot_op:literal,
501 rot_result = $rot_result:literal,
502 swap_op = $swap_op:literal,
503 swapped = $swapped:literal,
504 reversed = $reversed:literal,
505 leading_zeros_test = $leading_zeros_test:expr,
506 ) => {
507 #[doc = sign_dependent_expr!{
508 $signedness ?
509 if signed {
510 concat!("An [`", stringify!($Int), "`] that is known not to equal zero.")
511 }
512 if unsigned {
513 concat!("A [`", stringify!($Int), "`] that is known not to equal zero.")
514 }
515 }]
516 ///
517 /// This enables some memory layout optimization.
518 #[doc = concat!("For example, `Option<", stringify!($Ty), ">` is the same size as `", stringify!($Int), "`:")]
519 ///
520 /// ```rust
521 #[doc = concat!("assert_eq!(size_of::<Option<core::num::", stringify!($Ty), ">>(), size_of::<", stringify!($Int), ">());")]
522 /// ```
523 ///
524 /// # Layout
525 ///
526 #[doc = concat!("`", stringify!($Ty), "` is guaranteed to have the same layout and bit validity as `", stringify!($Int), "`")]
527 /// with the exception that `0` is not a valid instance.
528 #[doc = concat!("`Option<", stringify!($Ty), ">` is guaranteed to be compatible with `", stringify!($Int), "`,")]
529 /// including in FFI.
530 ///
531 /// Thanks to the [null pointer optimization],
532 #[doc = concat!("`", stringify!($Ty), "` and `Option<", stringify!($Ty), ">`")]
533 /// are guaranteed to have the same size and alignment:
534 ///
535 /// ```
536 #[doc = concat!("use std::num::", stringify!($Ty), ";")]
537 ///
538 #[doc = concat!("assert_eq!(size_of::<", stringify!($Ty), ">(), size_of::<Option<", stringify!($Ty), ">>());")]
539 #[doc = concat!("assert_eq!(align_of::<", stringify!($Ty), ">(), align_of::<Option<", stringify!($Ty), ">>());")]
540 /// ```
541 ///
542 /// # Compile-time creation
543 ///
544 /// Since both [`Option::unwrap()`] and [`Option::expect()`] are `const`, it is possible to
545 /// define a new
546 #[doc = concat!("`", stringify!($Ty), "`")]
547 /// at compile time via:
548 /// ```
549 #[doc = concat!("use std::num::", stringify!($Ty), ";")]
550 ///
551 #[doc = concat!("const TEN: ", stringify!($Ty), " = ", stringify!($Ty) , r#"::new(10).expect("ten is non-zero");"#)]
552 /// ```
553 ///
554 /// [null pointer optimization]: crate::option#representation
555 #[$stability]
556 pub type $Ty = NonZero<$Int>;
557
558 impl NonZero<$Int> {
559 /// The size of this non-zero integer type in bits.
560 ///
561 #[doc = concat!("This value is equal to [`", stringify!($Int), "::BITS`].")]
562 ///
563 /// # Examples
564 ///
565 /// ```
566 /// # use std::num::NonZero;
567 /// #
568 #[doc = concat!("assert_eq!(NonZero::<", stringify!($Int), ">::BITS, ", stringify!($Int), "::BITS);")]
569 /// ```
570 #[stable(feature = "nonzero_bits", since = "1.67.0")]
571 pub const BITS: u32 = <$Int>::BITS;
572
573 /// Returns the number of leading zeros in the binary representation of `self`.
574 ///
575 /// On many architectures, this function can perform better than `leading_zeros()` on the underlying integer type, as special handling of zero can be avoided.
576 ///
577 /// # Examples
578 ///
579 /// ```
580 /// # use std::num::NonZero;
581 /// #
582 /// # fn main() { test().unwrap(); }
583 /// # fn test() -> Option<()> {
584 #[doc = concat!("let n = NonZero::<", stringify!($Int), ">::new(", $leading_zeros_test, ")?;")]
585 ///
586 /// assert_eq!(n.leading_zeros(), 0);
587 /// # Some(())
588 /// # }
589 /// ```
590 #[stable(feature = "nonzero_leading_trailing_zeros", since = "1.53.0")]
591 #[rustc_const_stable(feature = "nonzero_leading_trailing_zeros", since = "1.53.0")]
592 #[must_use = "this returns the result of the operation, \
593 without modifying the original"]
594 #[inline]
595 pub const fn leading_zeros(self) -> u32 {
596 // SAFETY: since `self` cannot be zero, it is safe to call `ctlz_nonzero`.
597 unsafe {
598 intrinsics::ctlz_nonzero(self.get() as $Uint)
599 }
600 }
601
602 /// Returns the number of trailing zeros in the binary representation
603 /// of `self`.
604 ///
605 /// On many architectures, this function can perform better than `trailing_zeros()` on the underlying integer type, as special handling of zero can be avoided.
606 ///
607 /// # Examples
608 ///
609 /// ```
610 /// # use std::num::NonZero;
611 /// #
612 /// # fn main() { test().unwrap(); }
613 /// # fn test() -> Option<()> {
614 #[doc = concat!("let n = NonZero::<", stringify!($Int), ">::new(0b0101000)?;")]
615 ///
616 /// assert_eq!(n.trailing_zeros(), 3);
617 /// # Some(())
618 /// # }
619 /// ```
620 #[stable(feature = "nonzero_leading_trailing_zeros", since = "1.53.0")]
621 #[rustc_const_stable(feature = "nonzero_leading_trailing_zeros", since = "1.53.0")]
622 #[must_use = "this returns the result of the operation, \
623 without modifying the original"]
624 #[inline]
625 pub const fn trailing_zeros(self) -> u32 {
626 // SAFETY: since `self` cannot be zero, it is safe to call `cttz_nonzero`.
627 unsafe {
628 intrinsics::cttz_nonzero(self.get() as $Uint)
629 }
630 }
631
632 /// Returns `self` with only the most significant bit set.
633 ///
634 /// # Example
635 ///
636 /// ```
637 /// # use core::num::NonZero;
638 /// # fn main() { test().unwrap(); }
639 /// # fn test() -> Option<()> {
640 #[doc = concat!("let a = NonZero::<", stringify!($Int), ">::new(0b_01100100)?;")]
641 #[doc = concat!("let b = NonZero::<", stringify!($Int), ">::new(0b_01000000)?;")]
642 ///
643 /// assert_eq!(a.isolate_highest_one(), b);
644 /// # Some(())
645 /// # }
646 /// ```
647 #[stable(feature = "isolate_most_least_significant_one", since = "1.97.0")]
648 #[rustc_const_stable(feature = "isolate_most_least_significant_one", since = "1.97.0")]
649 #[must_use = "this returns the result of the operation, \
650 without modifying the original"]
651 #[inline(always)]
652 pub const fn isolate_highest_one(self) -> Self {
653 // SAFETY:
654 // `self` is non-zero, so masking to preserve only the most
655 // significant set bit will result in a non-zero `n`.
656 // and self.leading_zeros() is always < $INT::BITS since
657 // at least one of the bits in the number is not zero
658 unsafe {
659 let bit = (((1 as $Uint) << (<$Uint>::BITS - 1)).unchecked_shr(self.leading_zeros()));
660 NonZero::new_unchecked(bit as $Int)
661 }
662 }
663
664 /// Returns `self` with only the least significant bit set.
665 ///
666 /// # Example
667 ///
668 /// ```
669 /// # use core::num::NonZero;
670 /// # fn main() { test().unwrap(); }
671 /// # fn test() -> Option<()> {
672 #[doc = concat!("let a = NonZero::<", stringify!($Int), ">::new(0b_01100100)?;")]
673 #[doc = concat!("let b = NonZero::<", stringify!($Int), ">::new(0b_00000100)?;")]
674 ///
675 /// assert_eq!(a.isolate_lowest_one(), b);
676 /// # Some(())
677 /// # }
678 /// ```
679 #[stable(feature = "isolate_most_least_significant_one", since = "1.97.0")]
680 #[rustc_const_stable(feature = "isolate_most_least_significant_one", since = "1.97.0")]
681 #[must_use = "this returns the result of the operation, \
682 without modifying the original"]
683 #[inline(always)]
684 pub const fn isolate_lowest_one(self) -> Self {
685 let n = self.get();
686 let n = n & n.wrapping_neg();
687
688 // SAFETY: `self` is non-zero, so `self` with only its least
689 // significant set bit will remain non-zero.
690 unsafe { NonZero::new_unchecked(n) }
691 }
692
693 /// Returns the index of the highest bit set to one in `self`.
694 ///
695 #[doc = sign_dependent_expr!{
696 $signedness ?
697 if signed {
698 ""
699 }
700 if unsigned {
701 "Note that this is equivalent to [`ilog2`](Self::ilog2)."
702 }
703 }]
704 ///
705 /// # Examples
706 ///
707 /// ```
708 /// # use core::num::NonZero;
709 /// # fn main() { test().unwrap(); }
710 /// # fn test() -> Option<()> {
711 #[doc = concat!("assert_eq!(NonZero::<", stringify!($Int), ">::new(0b1)?.highest_one(), 0);")]
712 #[doc = concat!("assert_eq!(NonZero::<", stringify!($Int), ">::new(0b1_0000)?.highest_one(), 4);")]
713 #[doc = concat!("assert_eq!(NonZero::<", stringify!($Int), ">::new(0b1_1111)?.highest_one(), 4);")]
714 /// # Some(())
715 /// # }
716 /// ```
717 #[stable(feature = "int_lowest_highest_one", since = "1.97.0")]
718 #[rustc_const_stable(feature = "int_lowest_highest_one", since = "1.97.0")]
719 #[must_use = "this returns the result of the operation, \
720 without modifying the original"]
721 #[inline(always)]
722 pub const fn highest_one(self) -> u32 {
723 Self::BITS - 1 - self.leading_zeros()
724 }
725
726 /// Returns the index of the lowest bit set to one in `self`.
727 ///
728 /// # Examples
729 ///
730 /// ```
731 /// # use core::num::NonZero;
732 /// # fn main() { test().unwrap(); }
733 /// # fn test() -> Option<()> {
734 #[doc = concat!("assert_eq!(NonZero::<", stringify!($Int), ">::new(0b1)?.lowest_one(), 0);")]
735 #[doc = concat!("assert_eq!(NonZero::<", stringify!($Int), ">::new(0b1_0000)?.lowest_one(), 4);")]
736 #[doc = concat!("assert_eq!(NonZero::<", stringify!($Int), ">::new(0b1_1111)?.lowest_one(), 0);")]
737 /// # Some(())
738 /// # }
739 /// ```
740 #[stable(feature = "int_lowest_highest_one", since = "1.97.0")]
741 #[rustc_const_stable(feature = "int_lowest_highest_one", since = "1.97.0")]
742 #[must_use = "this returns the result of the operation, \
743 without modifying the original"]
744 #[inline(always)]
745 pub const fn lowest_one(self) -> u32 {
746 self.trailing_zeros()
747 }
748
749 /// Returns the number of ones in the binary representation of `self`.
750 ///
751 /// # Examples
752 ///
753 /// ```
754 /// # use std::num::NonZero;
755 /// #
756 /// # fn main() { test().unwrap(); }
757 /// # fn test() -> Option<()> {
758 #[doc = concat!("let a = NonZero::<", stringify!($Int), ">::new(0b100_0000)?;")]
759 #[doc = concat!("let b = NonZero::<", stringify!($Int), ">::new(0b100_0011)?;")]
760 ///
761 /// assert_eq!(a.count_ones(), NonZero::new(1)?);
762 /// assert_eq!(b.count_ones(), NonZero::new(3)?);
763 /// # Some(())
764 /// # }
765 /// ```
766 ///
767 #[stable(feature = "non_zero_count_ones", since = "1.86.0")]
768 #[rustc_const_stable(feature = "non_zero_count_ones", since = "1.86.0")]
769 #[doc(alias = "popcount")]
770 #[doc(alias = "popcnt")]
771 #[must_use = "this returns the result of the operation, \
772 without modifying the original"]
773 #[inline(always)]
774 pub const fn count_ones(self) -> NonZero<u32> {
775 // SAFETY:
776 // `self` is non-zero, which means it has at least one bit set, which means
777 // that the result of `count_ones` is non-zero.
778 unsafe { NonZero::new_unchecked(self.get().count_ones()) }
779 }
780
781 /// Shifts the bits to the left by a specified amount, `n`,
782 /// wrapping the truncated bits to the end of the resulting integer.
783 ///
784 /// Please note this isn't the same operation as the `<<` shifting operator!
785 ///
786 /// # Examples
787 ///
788 /// ```
789 /// #![feature(nonzero_bitwise)]
790 /// # use std::num::NonZero;
791 /// #
792 /// # fn main() { test().unwrap(); }
793 /// # fn test() -> Option<()> {
794 #[doc = concat!("let n = NonZero::new(", $rot_op, stringify!($Int), ")?;")]
795 #[doc = concat!("let m = NonZero::new(", $rot_result, ")?;")]
796 ///
797 #[doc = concat!("assert_eq!(n.rotate_left(", $rot, "), m);")]
798 /// # Some(())
799 /// # }
800 /// ```
801 #[unstable(feature = "nonzero_bitwise", issue = "128281")]
802 #[must_use = "this returns the result of the operation, \
803 without modifying the original"]
804 #[inline(always)]
805 pub const fn rotate_left(self, n: u32) -> Self {
806 let result = self.get().rotate_left(n);
807 // SAFETY: Rotating bits preserves the property int > 0.
808 unsafe { Self::new_unchecked(result) }
809 }
810
811 /// Shifts the bits to the right by a specified amount, `n`,
812 /// wrapping the truncated bits to the beginning of the resulting
813 /// integer.
814 ///
815 /// Please note this isn't the same operation as the `>>` shifting operator!
816 ///
817 /// # Examples
818 ///
819 /// ```
820 /// #![feature(nonzero_bitwise)]
821 /// # use std::num::NonZero;
822 /// #
823 /// # fn main() { test().unwrap(); }
824 /// # fn test() -> Option<()> {
825 #[doc = concat!("let n = NonZero::new(", $rot_result, stringify!($Int), ")?;")]
826 #[doc = concat!("let m = NonZero::new(", $rot_op, ")?;")]
827 ///
828 #[doc = concat!("assert_eq!(n.rotate_right(", $rot, "), m);")]
829 /// # Some(())
830 /// # }
831 /// ```
832 #[unstable(feature = "nonzero_bitwise", issue = "128281")]
833 #[must_use = "this returns the result of the operation, \
834 without modifying the original"]
835 #[inline(always)]
836 pub const fn rotate_right(self, n: u32) -> Self {
837 let result = self.get().rotate_right(n);
838 // SAFETY: Rotating bits preserves the property int > 0.
839 unsafe { Self::new_unchecked(result) }
840 }
841
842 /// Reverses the byte order of the integer.
843 ///
844 /// # Examples
845 ///
846 /// ```
847 /// #![feature(nonzero_bitwise)]
848 /// # use std::num::NonZero;
849 /// #
850 /// # fn main() { test().unwrap(); }
851 /// # fn test() -> Option<()> {
852 #[doc = concat!("let n = NonZero::new(", $swap_op, stringify!($Int), ")?;")]
853 /// let m = n.swap_bytes();
854 ///
855 #[doc = concat!("assert_eq!(m, NonZero::new(", $swapped, ")?);")]
856 /// # Some(())
857 /// # }
858 /// ```
859 #[unstable(feature = "nonzero_bitwise", issue = "128281")]
860 #[must_use = "this returns the result of the operation, \
861 without modifying the original"]
862 #[inline(always)]
863 pub const fn swap_bytes(self) -> Self {
864 let result = self.get().swap_bytes();
865 // SAFETY: Shuffling bytes preserves the property int > 0.
866 unsafe { Self::new_unchecked(result) }
867 }
868
869 /// Reverses the order of bits in the integer. The least significant bit becomes the most significant bit,
870 /// second least-significant bit becomes second most-significant bit, etc.
871 ///
872 /// # Examples
873 ///
874 /// ```
875 /// #![feature(nonzero_bitwise)]
876 /// # use std::num::NonZero;
877 /// #
878 /// # fn main() { test().unwrap(); }
879 /// # fn test() -> Option<()> {
880 #[doc = concat!("let n = NonZero::new(", $swap_op, stringify!($Int), ")?;")]
881 /// let m = n.reverse_bits();
882 ///
883 #[doc = concat!("assert_eq!(m, NonZero::new(", $reversed, ")?);")]
884 /// # Some(())
885 /// # }
886 /// ```
887 #[unstable(feature = "nonzero_bitwise", issue = "128281")]
888 #[must_use = "this returns the result of the operation, \
889 without modifying the original"]
890 #[inline(always)]
891 pub const fn reverse_bits(self) -> Self {
892 let result = self.get().reverse_bits();
893 // SAFETY: Reversing bits preserves the property int > 0.
894 unsafe { Self::new_unchecked(result) }
895 }
896
897 /// Converts an integer from big endian to the target's endianness.
898 ///
899 /// On big endian this is a no-op. On little endian the bytes are
900 /// swapped.
901 ///
902 /// # Examples
903 ///
904 /// ```
905 /// #![feature(nonzero_bitwise)]
906 /// # use std::num::NonZero;
907 #[doc = concat!("use std::num::", stringify!($Ty), ";")]
908 /// #
909 /// # fn main() { test().unwrap(); }
910 /// # fn test() -> Option<()> {
911 #[doc = concat!("let n = NonZero::new(0x1A", stringify!($Int), ")?;")]
912 ///
913 /// if cfg!(target_endian = "big") {
914 #[doc = concat!(" assert_eq!(", stringify!($Ty), "::from_be(n), n)")]
915 /// } else {
916 #[doc = concat!(" assert_eq!(", stringify!($Ty), "::from_be(n), n.swap_bytes())")]
917 /// }
918 /// # Some(())
919 /// # }
920 /// ```
921 #[unstable(feature = "nonzero_bitwise", issue = "128281")]
922 #[must_use]
923 #[inline(always)]
924 pub const fn from_be(x: Self) -> Self {
925 let result = $Int::from_be(x.get());
926 // SAFETY: Shuffling bytes preserves the property int > 0.
927 unsafe { Self::new_unchecked(result) }
928 }
929
930 /// Converts an integer from little endian to the target's endianness.
931 ///
932 /// On little endian this is a no-op. On big endian the bytes are
933 /// swapped.
934 ///
935 /// # Examples
936 ///
937 /// ```
938 /// #![feature(nonzero_bitwise)]
939 /// # use std::num::NonZero;
940 #[doc = concat!("use std::num::", stringify!($Ty), ";")]
941 /// #
942 /// # fn main() { test().unwrap(); }
943 /// # fn test() -> Option<()> {
944 #[doc = concat!("let n = NonZero::new(0x1A", stringify!($Int), ")?;")]
945 ///
946 /// if cfg!(target_endian = "little") {
947 #[doc = concat!(" assert_eq!(", stringify!($Ty), "::from_le(n), n)")]
948 /// } else {
949 #[doc = concat!(" assert_eq!(", stringify!($Ty), "::from_le(n), n.swap_bytes())")]
950 /// }
951 /// # Some(())
952 /// # }
953 /// ```
954 #[unstable(feature = "nonzero_bitwise", issue = "128281")]
955 #[must_use]
956 #[inline(always)]
957 pub const fn from_le(x: Self) -> Self {
958 let result = $Int::from_le(x.get());
959 // SAFETY: Shuffling bytes preserves the property int > 0.
960 unsafe { Self::new_unchecked(result) }
961 }
962
963 /// Converts `self` to big endian from the target's endianness.
964 ///
965 /// On big endian this is a no-op. On little endian the bytes are
966 /// swapped.
967 ///
968 /// # Examples
969 ///
970 /// ```
971 /// #![feature(nonzero_bitwise)]
972 /// # use std::num::NonZero;
973 /// #
974 /// # fn main() { test().unwrap(); }
975 /// # fn test() -> Option<()> {
976 #[doc = concat!("let n = NonZero::new(0x1A", stringify!($Int), ")?;")]
977 ///
978 /// if cfg!(target_endian = "big") {
979 /// assert_eq!(n.to_be(), n)
980 /// } else {
981 /// assert_eq!(n.to_be(), n.swap_bytes())
982 /// }
983 /// # Some(())
984 /// # }
985 /// ```
986 #[unstable(feature = "nonzero_bitwise", issue = "128281")]
987 #[must_use = "this returns the result of the operation, \
988 without modifying the original"]
989 #[inline(always)]
990 pub const fn to_be(self) -> Self {
991 let result = self.get().to_be();
992 // SAFETY: Shuffling bytes preserves the property int > 0.
993 unsafe { Self::new_unchecked(result) }
994 }
995
996 /// Converts `self` to little endian from the target's endianness.
997 ///
998 /// On little endian this is a no-op. On big endian the bytes are
999 /// swapped.
1000 ///
1001 /// # Examples
1002 ///
1003 /// ```
1004 /// #![feature(nonzero_bitwise)]
1005 /// # use std::num::NonZero;
1006 /// #
1007 /// # fn main() { test().unwrap(); }
1008 /// # fn test() -> Option<()> {
1009 #[doc = concat!("let n = NonZero::new(0x1A", stringify!($Int), ")?;")]
1010 ///
1011 /// if cfg!(target_endian = "little") {
1012 /// assert_eq!(n.to_le(), n)
1013 /// } else {
1014 /// assert_eq!(n.to_le(), n.swap_bytes())
1015 /// }
1016 /// # Some(())
1017 /// # }
1018 /// ```
1019 #[unstable(feature = "nonzero_bitwise", issue = "128281")]
1020 #[must_use = "this returns the result of the operation, \
1021 without modifying the original"]
1022 #[inline(always)]
1023 pub const fn to_le(self) -> Self {
1024 let result = self.get().to_le();
1025 // SAFETY: Shuffling bytes preserves the property int > 0.
1026 unsafe { Self::new_unchecked(result) }
1027 }
1028
1029 nonzero_integer_signedness_dependent_methods! {
1030 Primitive = $signedness $Int,
1031 SignedPrimitive = $Sint,
1032 UnsignedPrimitive = $Uint,
1033 }
1034
1035 /// Multiplies two non-zero integers together.
1036 /// Checks for overflow and returns [`None`] on overflow.
1037 /// As a consequence, the result cannot wrap to zero.
1038 ///
1039 /// # Examples
1040 ///
1041 /// ```
1042 /// # use std::num::NonZero;
1043 /// #
1044 /// # fn main() { test().unwrap(); }
1045 /// # fn test() -> Option<()> {
1046 #[doc = concat!("let two = NonZero::new(2", stringify!($Int), ")?;")]
1047 #[doc = concat!("let four = NonZero::new(4", stringify!($Int), ")?;")]
1048 #[doc = concat!("let max = NonZero::new(", stringify!($Int), "::MAX)?;")]
1049 ///
1050 /// assert_eq!(Some(four), two.checked_mul(two));
1051 /// assert_eq!(None, max.checked_mul(two));
1052 /// # Some(())
1053 /// # }
1054 /// ```
1055 #[stable(feature = "nonzero_checked_ops", since = "1.64.0")]
1056 #[rustc_const_stable(feature = "const_nonzero_checked_ops", since = "1.64.0")]
1057 #[must_use = "this returns the result of the operation, \
1058 without modifying the original"]
1059 #[inline]
1060 pub const fn checked_mul(self, other: Self) -> Option<Self> {
1061 if let Some(result) = self.get().checked_mul(other.get()) {
1062 // SAFETY:
1063 // - `checked_mul` returns `None` on overflow
1064 // - `self` and `other` are non-zero
1065 // - the only way to get zero from a multiplication without overflow is for one
1066 // of the sides to be zero
1067 //
1068 // So the result cannot be zero.
1069 Some(unsafe { Self::new_unchecked(result) })
1070 } else {
1071 None
1072 }
1073 }
1074
1075 /// Multiplies two non-zero integers together.
1076 #[doc = concat!("Return [`NonZero::<", stringify!($Int), ">::MAX`] on overflow.")]
1077 ///
1078 /// # Examples
1079 ///
1080 /// ```
1081 /// # use std::num::NonZero;
1082 /// #
1083 /// # fn main() { test().unwrap(); }
1084 /// # fn test() -> Option<()> {
1085 #[doc = concat!("let two = NonZero::new(2", stringify!($Int), ")?;")]
1086 #[doc = concat!("let four = NonZero::new(4", stringify!($Int), ")?;")]
1087 #[doc = concat!("let max = NonZero::new(", stringify!($Int), "::MAX)?;")]
1088 ///
1089 /// assert_eq!(four, two.saturating_mul(two));
1090 /// assert_eq!(max, four.saturating_mul(max));
1091 /// # Some(())
1092 /// # }
1093 /// ```
1094 #[stable(feature = "nonzero_checked_ops", since = "1.64.0")]
1095 #[rustc_const_stable(feature = "const_nonzero_checked_ops", since = "1.64.0")]
1096 #[must_use = "this returns the result of the operation, \
1097 without modifying the original"]
1098 #[inline]
1099 pub const fn saturating_mul(self, other: Self) -> Self {
1100 // SAFETY:
1101 // - `saturating_mul` returns `u*::MAX`/`i*::MAX`/`i*::MIN` on overflow/underflow,
1102 // all of which are non-zero
1103 // - `self` and `other` are non-zero
1104 // - the only way to get zero from a multiplication without overflow is for one
1105 // of the sides to be zero
1106 //
1107 // So the result cannot be zero.
1108 unsafe { Self::new_unchecked(self.get().saturating_mul(other.get())) }
1109 }
1110
1111 /// Multiplies two non-zero integers together,
1112 /// assuming overflow cannot occur.
1113 /// Overflow is unchecked, and it is undefined behavior to overflow
1114 /// *even if the result would wrap to a non-zero value*.
1115 ///
1116 /// # Safety
1117 ///
1118 /// This results in undefined behavior when
1119 #[doc = sign_dependent_expr!{
1120 $signedness ?
1121 if signed {
1122 concat!("`self * rhs > ", stringify!($Int), "::MAX`, ",
1123 "or `self * rhs < ", stringify!($Int), "::MIN`.")
1124 }
1125 if unsigned {
1126 concat!("`self * rhs > ", stringify!($Int), "::MAX`.")
1127 }
1128 }]
1129 ///
1130 /// # Examples
1131 ///
1132 /// ```
1133 /// #![feature(nonzero_ops)]
1134 ///
1135 /// # use std::num::NonZero;
1136 /// #
1137 /// # fn main() { test().unwrap(); }
1138 /// # fn test() -> Option<()> {
1139 #[doc = concat!("let two = NonZero::new(2", stringify!($Int), ")?;")]
1140 #[doc = concat!("let four = NonZero::new(4", stringify!($Int), ")?;")]
1141 ///
1142 /// assert_eq!(four, unsafe { two.unchecked_mul(two) });
1143 /// # Some(())
1144 /// # }
1145 /// ```
1146 #[unstable(feature = "nonzero_ops", issue = "84186")]
1147 #[must_use = "this returns the result of the operation, \
1148 without modifying the original"]
1149 #[inline]
1150 pub const unsafe fn unchecked_mul(self, other: Self) -> Self {
1151 // SAFETY: The caller ensures there is no overflow.
1152 unsafe { Self::new_unchecked(self.get().unchecked_mul(other.get())) }
1153 }
1154
1155 /// Raises non-zero value to an integer power.
1156 /// Checks for overflow and returns [`None`] on overflow.
1157 /// As a consequence, the result cannot wrap to zero.
1158 ///
1159 /// # Examples
1160 ///
1161 /// ```
1162 /// # use std::num::NonZero;
1163 /// #
1164 /// # fn main() { test().unwrap(); }
1165 /// # fn test() -> Option<()> {
1166 #[doc = concat!("let three = NonZero::new(3", stringify!($Int), ")?;")]
1167 #[doc = concat!("let twenty_seven = NonZero::new(27", stringify!($Int), ")?;")]
1168 #[doc = concat!("let half_max = NonZero::new(", stringify!($Int), "::MAX / 2)?;")]
1169 ///
1170 /// assert_eq!(Some(twenty_seven), three.checked_pow(3));
1171 /// assert_eq!(None, half_max.checked_pow(3));
1172 /// # Some(())
1173 /// # }
1174 /// ```
1175 #[stable(feature = "nonzero_checked_ops", since = "1.64.0")]
1176 #[rustc_const_stable(feature = "const_nonzero_checked_ops", since = "1.64.0")]
1177 #[must_use = "this returns the result of the operation, \
1178 without modifying the original"]
1179 #[inline]
1180 pub const fn checked_pow(self, other: u32) -> Option<Self> {
1181 if let Some(result) = self.get().checked_pow(other) {
1182 // SAFETY:
1183 // - `checked_pow` returns `None` on overflow/underflow
1184 // - `self` is non-zero
1185 // - the only way to get zero from an exponentiation without overflow is
1186 // for base to be zero
1187 //
1188 // So the result cannot be zero.
1189 Some(unsafe { Self::new_unchecked(result) })
1190 } else {
1191 None
1192 }
1193 }
1194
1195 /// Raise non-zero value to an integer power.
1196 #[doc = sign_dependent_expr!{
1197 $signedness ?
1198 if signed {
1199 concat!("Return [`NonZero::<", stringify!($Int), ">::MIN`] ",
1200 "or [`NonZero::<", stringify!($Int), ">::MAX`] on overflow.")
1201 }
1202 if unsigned {
1203 concat!("Return [`NonZero::<", stringify!($Int), ">::MAX`] on overflow.")
1204 }
1205 }]
1206 ///
1207 /// # Examples
1208 ///
1209 /// ```
1210 /// # use std::num::NonZero;
1211 /// #
1212 /// # fn main() { test().unwrap(); }
1213 /// # fn test() -> Option<()> {
1214 #[doc = concat!("let three = NonZero::new(3", stringify!($Int), ")?;")]
1215 #[doc = concat!("let twenty_seven = NonZero::new(27", stringify!($Int), ")?;")]
1216 #[doc = concat!("let max = NonZero::new(", stringify!($Int), "::MAX)?;")]
1217 ///
1218 /// assert_eq!(twenty_seven, three.saturating_pow(3));
1219 /// assert_eq!(max, max.saturating_pow(3));
1220 /// # Some(())
1221 /// # }
1222 /// ```
1223 #[stable(feature = "nonzero_checked_ops", since = "1.64.0")]
1224 #[rustc_const_stable(feature = "const_nonzero_checked_ops", since = "1.64.0")]
1225 #[must_use = "this returns the result of the operation, \
1226 without modifying the original"]
1227 #[inline]
1228 pub const fn saturating_pow(self, other: u32) -> Self {
1229 // SAFETY:
1230 // - `saturating_pow` returns `u*::MAX`/`i*::MAX`/`i*::MIN` on overflow/underflow,
1231 // all of which are non-zero
1232 // - `self` is non-zero
1233 // - the only way to get zero from an exponentiation without overflow is
1234 // for base to be zero
1235 //
1236 // So the result cannot be zero.
1237 unsafe { Self::new_unchecked(self.get().saturating_pow(other)) }
1238 }
1239
1240 /// Parses a non-zero integer from an ASCII-byte slice with decimal digits.
1241 ///
1242 /// The characters are expected to be an optional
1243 #[doc = sign_dependent_expr!{
1244 $signedness ?
1245 if signed {
1246 " `+` or `-` "
1247 }
1248 if unsigned {
1249 " `+` "
1250 }
1251 }]
1252 /// sign followed by only digits. Leading and trailing non-digit characters (including
1253 /// whitespace) represent an error. Underscores (which are accepted in Rust literals)
1254 /// also represent an error.
1255 ///
1256 /// # Examples
1257 ///
1258 /// ```
1259 /// #![feature(int_from_ascii)]
1260 ///
1261 /// # use std::num::NonZero;
1262 /// #
1263 /// # fn main() { test().unwrap(); }
1264 /// # fn test() -> Option<()> {
1265 #[doc = concat!("assert_eq!(NonZero::<", stringify!($Int), ">::from_ascii(b\"+10\"), Ok(NonZero::new(10)?));")]
1266 /// # Some(())
1267 /// # }
1268 /// ```
1269 ///
1270 /// Trailing space returns error:
1271 ///
1272 /// ```
1273 /// #![feature(int_from_ascii)]
1274 ///
1275 /// # use std::num::NonZero;
1276 /// #
1277 #[doc = concat!("assert!(NonZero::<", stringify!($Int), ">::from_ascii(b\"1 \").is_err());")]
1278 /// ```
1279 #[unstable(feature = "int_from_ascii", issue = "134821")]
1280 #[inline]
1281 pub const fn from_ascii(src: &[u8]) -> Result<Self, ParseIntError> {
1282 Self::from_ascii_radix(src, 10)
1283 }
1284
1285 /// Parses a non-zero integer from an ASCII-byte slice with digits in a given base.
1286 ///
1287 /// The characters are expected to be an optional
1288 #[doc = sign_dependent_expr!{
1289 $signedness ?
1290 if signed {
1291 " `+` or `-` "
1292 }
1293 if unsigned {
1294 " `+` "
1295 }
1296 }]
1297 /// sign followed by only digits. Leading and trailing non-digit characters (including
1298 /// whitespace) represent an error. Underscores (which are accepted in Rust literals)
1299 /// also represent an error.
1300 ///
1301 /// Digits are a subset of these characters, depending on `radix`:
1302 ///
1303 /// - `0-9`
1304 /// - `a-z`
1305 /// - `A-Z`
1306 ///
1307 /// # Panics
1308 ///
1309 /// This method panics if `radix` is not in the range from 2 to 36.
1310 ///
1311 /// # Examples
1312 ///
1313 /// ```
1314 /// #![feature(int_from_ascii)]
1315 ///
1316 /// # use std::num::NonZero;
1317 /// #
1318 /// # fn main() { test().unwrap(); }
1319 /// # fn test() -> Option<()> {
1320 #[doc = concat!("assert_eq!(NonZero::<", stringify!($Int), ">::from_ascii_radix(b\"A\", 16), Ok(NonZero::new(10)?));")]
1321 /// # Some(())
1322 /// # }
1323 /// ```
1324 ///
1325 /// Trailing space returns error:
1326 ///
1327 /// ```
1328 /// #![feature(int_from_ascii)]
1329 ///
1330 /// # use std::num::NonZero;
1331 /// #
1332 #[doc = concat!("assert!(NonZero::<", stringify!($Int), ">::from_ascii_radix(b\"1 \", 10).is_err());")]
1333 /// ```
1334 #[unstable(feature = "int_from_ascii", issue = "134821")]
1335 #[inline]
1336 pub const fn from_ascii_radix(src: &[u8], radix: u32) -> Result<Self, ParseIntError> {
1337 let n = match <$Int>::from_ascii_radix(src, radix) {
1338 Ok(n) => n,
1339 Err(err) => return Err(err),
1340 };
1341 if let Some(n) = Self::new(n) {
1342 Ok(n)
1343 } else {
1344 Err(ParseIntError { kind: IntErrorKind::Zero })
1345 }
1346 }
1347
1348 /// Parses a non-zero integer from a string slice with digits in a given base.
1349 ///
1350 /// The string is expected to be an optional
1351 #[doc = sign_dependent_expr!{
1352 $signedness ?
1353 if signed {
1354 " `+` or `-` "
1355 }
1356 if unsigned {
1357 " `+` "
1358 }
1359 }]
1360 /// sign followed by only digits. Leading and trailing non-digit characters (including
1361 /// whitespace) represent an error. Underscores (which are accepted in Rust literals)
1362 /// also represent an error.
1363 ///
1364 /// Digits are a subset of these characters, depending on `radix`:
1365 ///
1366 /// - `0-9`
1367 /// - `a-z`
1368 /// - `A-Z`
1369 ///
1370 /// # Panics
1371 ///
1372 /// This method panics if `radix` is not in the range from 2 to 36.
1373 ///
1374 /// # Examples
1375 ///
1376 /// ```
1377 /// #![feature(nonzero_from_str_radix)]
1378 ///
1379 /// # use std::num::NonZero;
1380 /// #
1381 /// # fn main() { test().unwrap(); }
1382 /// # fn test() -> Option<()> {
1383 #[doc = concat!("assert_eq!(NonZero::<", stringify!($Int), ">::from_str_radix(\"A\", 16), Ok(NonZero::new(10)?));")]
1384 /// # Some(())
1385 /// # }
1386 /// ```
1387 ///
1388 /// Trailing space returns error:
1389 ///
1390 /// ```
1391 /// #![feature(nonzero_from_str_radix)]
1392 ///
1393 /// # use std::num::NonZero;
1394 /// #
1395 #[doc = concat!("assert!(NonZero::<", stringify!($Int), ">::from_str_radix(\"1 \", 10).is_err());")]
1396 /// ```
1397 #[unstable(feature = "nonzero_from_str_radix", issue = "152193")]
1398 #[inline]
1399 pub const fn from_str_radix(src: &str, radix: u32) -> Result<Self, ParseIntError> {
1400 Self::from_ascii_radix(src.as_bytes(), radix)
1401 }
1402 }
1403
1404 #[stable(feature = "nonzero_parse", since = "1.35.0")]
1405 impl FromStr for NonZero<$Int> {
1406 type Err = ParseIntError;
1407 fn from_str(src: &str) -> Result<Self, Self::Err> {
1408 Self::from_str_radix(src, 10)
1409 }
1410 }
1411
1412 nonzero_integer_signedness_dependent_impls!($signedness $Int);
1413 };
1414
1415 (
1416 Self = $Ty:ident,
1417 Primitive = unsigned $Int:ident,
1418 SignedPrimitive = $Sint:ident,
1419 rot = $rot:literal,
1420 rot_op = $rot_op:literal,
1421 rot_result = $rot_result:literal,
1422 swap_op = $swap_op:literal,
1423 swapped = $swapped:literal,
1424 reversed = $reversed:literal,
1425 $(,)?
1426 ) => {
1427 nonzero_integer! {
1428 #[stable(feature = "nonzero", since = "1.28.0")]
1429 Self = $Ty,
1430 Primitive = unsigned $Int,
1431 SignedPrimitive = $Sint,
1432 UnsignedPrimitive = $Int,
1433 rot = $rot,
1434 rot_op = $rot_op,
1435 rot_result = $rot_result,
1436 swap_op = $swap_op,
1437 swapped = $swapped,
1438 reversed = $reversed,
1439 leading_zeros_test = concat!(stringify!($Int), "::MAX"),
1440 }
1441 };
1442
1443 (
1444 Self = $Ty:ident,
1445 Primitive = signed $Int:ident,
1446 UnsignedPrimitive = $Uint:ident,
1447 rot = $rot:literal,
1448 rot_op = $rot_op:literal,
1449 rot_result = $rot_result:literal,
1450 swap_op = $swap_op:literal,
1451 swapped = $swapped:literal,
1452 reversed = $reversed:literal,
1453 ) => {
1454 nonzero_integer! {
1455 #[stable(feature = "signed_nonzero", since = "1.34.0")]
1456 Self = $Ty,
1457 Primitive = signed $Int,
1458 SignedPrimitive = $Int,
1459 UnsignedPrimitive = $Uint,
1460 rot = $rot,
1461 rot_op = $rot_op,
1462 rot_result = $rot_result,
1463 swap_op = $swap_op,
1464 swapped = $swapped,
1465 reversed = $reversed,
1466 leading_zeros_test = concat!("-1", stringify!($Int)),
1467 }
1468 };
1469}
1470
1471macro_rules! nonzero_integer_signedness_dependent_impls {
1472 // Impls for unsigned nonzero types only.
1473 (unsigned $Int:ty) => {
1474 #[stable(feature = "nonzero_div", since = "1.51.0")]
1475 #[rustc_const_unstable(feature = "const_ops", issue = "143802")]
1476 const impl Div<NonZero<$Int>> for $Int {
1477 type Output = $Int;
1478
1479 /// Same as `self / other.get()`, but because `other` is a `NonZero<_>`,
1480 /// there's never a runtime check for division-by-zero.
1481 ///
1482 /// This operation rounds towards zero, truncating any fractional
1483 /// part of the exact result, and cannot panic.
1484 #[doc(alias = "unchecked_div")]
1485 #[inline]
1486 fn div(self, other: NonZero<$Int>) -> $Int {
1487 // SAFETY: Division by zero is checked because `other` is non-zero,
1488 // and MIN/-1 is checked because `self` is an unsigned int.
1489 unsafe { intrinsics::unchecked_div(self, other.get()) }
1490 }
1491 }
1492
1493 #[stable(feature = "nonzero_div_assign", since = "1.79.0")]
1494 #[rustc_const_unstable(feature = "const_ops", issue = "143802")]
1495 const impl DivAssign<NonZero<$Int>> for $Int {
1496 /// Same as `self /= other.get()`, but because `other` is a `NonZero<_>`,
1497 /// there's never a runtime check for division-by-zero.
1498 ///
1499 /// This operation rounds towards zero, truncating any fractional
1500 /// part of the exact result, and cannot panic.
1501 #[inline]
1502 fn div_assign(&mut self, other: NonZero<$Int>) {
1503 *self = *self / other;
1504 }
1505 }
1506
1507 #[stable(feature = "nonzero_div", since = "1.51.0")]
1508 #[rustc_const_unstable(feature = "const_ops", issue = "143802")]
1509 const impl Rem<NonZero<$Int>> for $Int {
1510 type Output = $Int;
1511
1512 /// This operation satisfies `n % d == n - (n / d) * d`, and cannot panic.
1513 #[inline]
1514 fn rem(self, other: NonZero<$Int>) -> $Int {
1515 // SAFETY: Remainder by zero is checked because `other` is non-zero,
1516 // and MIN/-1 is checked because `self` is an unsigned int.
1517 unsafe { intrinsics::unchecked_rem(self, other.get()) }
1518 }
1519 }
1520
1521 #[stable(feature = "nonzero_div_assign", since = "1.79.0")]
1522 #[rustc_const_unstable(feature = "const_ops", issue = "143802")]
1523 const impl RemAssign<NonZero<$Int>> for $Int {
1524 /// This operation satisfies `n % d == n - (n / d) * d`, and cannot panic.
1525 #[inline]
1526 fn rem_assign(&mut self, other: NonZero<$Int>) {
1527 *self = *self % other;
1528 }
1529 }
1530
1531 impl NonZero<$Int> {
1532 /// Calculates the quotient of `self` and `rhs`, rounding the result towards positive infinity.
1533 ///
1534 /// The result is guaranteed to be non-zero.
1535 ///
1536 /// # Examples
1537 ///
1538 /// ```
1539 /// # use std::num::NonZero;
1540 #[doc = concat!("let one = NonZero::new(1", stringify!($Int), ").unwrap();")]
1541 #[doc = concat!("let max = NonZero::new(", stringify!($Int), "::MAX).unwrap();")]
1542 /// assert_eq!(one.div_ceil(max), one);
1543 ///
1544 #[doc = concat!("let two = NonZero::new(2", stringify!($Int), ").unwrap();")]
1545 #[doc = concat!("let three = NonZero::new(3", stringify!($Int), ").unwrap();")]
1546 /// assert_eq!(three.div_ceil(two), two);
1547 /// ```
1548 #[stable(feature = "unsigned_nonzero_div_ceil", since = "1.92.0")]
1549 #[rustc_const_stable(feature = "unsigned_nonzero_div_ceil", since = "1.92.0")]
1550 #[must_use = "this returns the result of the operation, \
1551 without modifying the original"]
1552 #[inline]
1553 pub const fn div_ceil(self, rhs: Self) -> Self {
1554 let v = self.get().div_ceil(rhs.get());
1555 // SAFETY: ceiled division of two positive integers can never be zero.
1556 unsafe { Self::new_unchecked(v) }
1557 }
1558 }
1559 };
1560 // Impls for signed nonzero types only.
1561 (signed $Int:ty) => {
1562 #[stable(feature = "signed_nonzero_neg", since = "1.71.0")]
1563 #[rustc_const_unstable(feature = "const_ops", issue = "143802")]
1564 const impl Neg for NonZero<$Int> {
1565 type Output = Self;
1566
1567 #[inline]
1568 fn neg(self) -> Self {
1569 // SAFETY: negation of nonzero cannot yield zero values.
1570 unsafe { Self::new_unchecked(self.get().neg()) }
1571 }
1572 }
1573
1574 forward_ref_unop! { impl Neg, neg for NonZero<$Int>,
1575 #[stable(feature = "signed_nonzero_neg", since = "1.71.0")]
1576 #[rustc_const_unstable(feature = "const_ops", issue = "143802")] }
1577 };
1578}
1579
1580#[rustfmt::skip] // https://github.com/rust-lang/rustfmt/issues/5974
1581macro_rules! nonzero_integer_signedness_dependent_methods {
1582 // Associated items for unsigned nonzero types only.
1583 (
1584 Primitive = unsigned $Int:ident,
1585 SignedPrimitive = $Sint:ty,
1586 UnsignedPrimitive = $Uint:ty,
1587 ) => {
1588 /// The smallest value that can be represented by this non-zero
1589 /// integer type, 1.
1590 ///
1591 /// # Examples
1592 ///
1593 /// ```
1594 /// # use std::num::NonZero;
1595 /// #
1596 #[doc = concat!("assert_eq!(NonZero::<", stringify!($Int), ">::MIN.get(), 1", stringify!($Int), ");")]
1597 /// ```
1598 #[stable(feature = "nonzero_min_max", since = "1.70.0")]
1599 pub const MIN: Self = Self::new(1).unwrap();
1600
1601 /// The largest value that can be represented by this non-zero
1602 /// integer type,
1603 #[doc = concat!("equal to [`", stringify!($Int), "::MAX`].")]
1604 ///
1605 /// # Examples
1606 ///
1607 /// ```
1608 /// # use std::num::NonZero;
1609 /// #
1610 #[doc = concat!("assert_eq!(NonZero::<", stringify!($Int), ">::MAX.get(), ", stringify!($Int), "::MAX);")]
1611 /// ```
1612 #[stable(feature = "nonzero_min_max", since = "1.70.0")]
1613 pub const MAX: Self = Self::new(<$Int>::MAX).unwrap();
1614
1615 /// Adds an unsigned integer to a non-zero value.
1616 /// Checks for overflow and returns [`None`] on overflow.
1617 /// As a consequence, the result cannot wrap to zero.
1618 ///
1619 ///
1620 /// # Examples
1621 ///
1622 /// ```
1623 /// # use std::num::NonZero;
1624 /// #
1625 /// # fn main() { test().unwrap(); }
1626 /// # fn test() -> Option<()> {
1627 #[doc = concat!("let one = NonZero::new(1", stringify!($Int), ")?;")]
1628 #[doc = concat!("let two = NonZero::new(2", stringify!($Int), ")?;")]
1629 #[doc = concat!("let max = NonZero::new(", stringify!($Int), "::MAX)?;")]
1630 ///
1631 /// assert_eq!(Some(two), one.checked_add(1));
1632 /// assert_eq!(None, max.checked_add(1));
1633 /// # Some(())
1634 /// # }
1635 /// ```
1636 #[stable(feature = "nonzero_checked_ops", since = "1.64.0")]
1637 #[rustc_const_stable(feature = "const_nonzero_checked_ops", since = "1.64.0")]
1638 #[must_use = "this returns the result of the operation, \
1639 without modifying the original"]
1640 #[inline]
1641 pub const fn checked_add(self, other: $Int) -> Option<Self> {
1642 if let Some(result) = self.get().checked_add(other) {
1643 // SAFETY:
1644 // - `checked_add` returns `None` on overflow
1645 // - `self` is non-zero
1646 // - the only way to get zero from an addition without overflow is for both
1647 // sides to be zero
1648 //
1649 // So the result cannot be zero.
1650 Some(unsafe { Self::new_unchecked(result) })
1651 } else {
1652 None
1653 }
1654 }
1655
1656 /// Adds an unsigned integer to a non-zero value.
1657 #[doc = concat!("Return [`NonZero::<", stringify!($Int), ">::MAX`] on overflow.")]
1658 ///
1659 /// # Examples
1660 ///
1661 /// ```
1662 /// # use std::num::NonZero;
1663 /// #
1664 /// # fn main() { test().unwrap(); }
1665 /// # fn test() -> Option<()> {
1666 #[doc = concat!("let one = NonZero::new(1", stringify!($Int), ")?;")]
1667 #[doc = concat!("let two = NonZero::new(2", stringify!($Int), ")?;")]
1668 #[doc = concat!("let max = NonZero::new(", stringify!($Int), "::MAX)?;")]
1669 ///
1670 /// assert_eq!(two, one.saturating_add(1));
1671 /// assert_eq!(max, max.saturating_add(1));
1672 /// # Some(())
1673 /// # }
1674 /// ```
1675 #[stable(feature = "nonzero_checked_ops", since = "1.64.0")]
1676 #[rustc_const_stable(feature = "const_nonzero_checked_ops", since = "1.64.0")]
1677 #[must_use = "this returns the result of the operation, \
1678 without modifying the original"]
1679 #[inline]
1680 pub const fn saturating_add(self, other: $Int) -> Self {
1681 // SAFETY:
1682 // - `saturating_add` returns `u*::MAX` on overflow, which is non-zero
1683 // - `self` is non-zero
1684 // - the only way to get zero from an addition without overflow is for both
1685 // sides to be zero
1686 //
1687 // So the result cannot be zero.
1688 unsafe { Self::new_unchecked(self.get().saturating_add(other)) }
1689 }
1690
1691 /// Adds an unsigned integer to a non-zero value,
1692 /// assuming overflow cannot occur.
1693 /// Overflow is unchecked, and it is undefined behavior to overflow
1694 /// *even if the result would wrap to a non-zero value*.
1695 ///
1696 /// # Safety
1697 ///
1698 /// This results in undefined behavior when
1699 #[doc = concat!("`self + rhs > ", stringify!($Int), "::MAX`.")]
1700 ///
1701 /// # Examples
1702 ///
1703 /// ```
1704 /// #![feature(nonzero_ops)]
1705 ///
1706 /// # use std::num::NonZero;
1707 /// #
1708 /// # fn main() { test().unwrap(); }
1709 /// # fn test() -> Option<()> {
1710 #[doc = concat!("let one = NonZero::new(1", stringify!($Int), ")?;")]
1711 #[doc = concat!("let two = NonZero::new(2", stringify!($Int), ")?;")]
1712 ///
1713 /// assert_eq!(two, unsafe { one.unchecked_add(1) });
1714 /// # Some(())
1715 /// # }
1716 /// ```
1717 #[unstable(feature = "nonzero_ops", issue = "84186")]
1718 #[must_use = "this returns the result of the operation, \
1719 without modifying the original"]
1720 #[inline]
1721 pub const unsafe fn unchecked_add(self, other: $Int) -> Self {
1722 // SAFETY: The caller ensures there is no overflow.
1723 unsafe { Self::new_unchecked(self.get().unchecked_add(other)) }
1724 }
1725
1726 /// Returns the smallest power of two greater than or equal to `self`.
1727 /// Checks for overflow and returns [`None`]
1728 /// if the next power of two is greater than the type’s maximum value.
1729 /// As a consequence, the result cannot wrap to zero.
1730 ///
1731 /// # Examples
1732 ///
1733 /// ```
1734 /// # use std::num::NonZero;
1735 /// #
1736 /// # fn main() { test().unwrap(); }
1737 /// # fn test() -> Option<()> {
1738 #[doc = concat!("let two = NonZero::new(2", stringify!($Int), ")?;")]
1739 #[doc = concat!("let three = NonZero::new(3", stringify!($Int), ")?;")]
1740 #[doc = concat!("let four = NonZero::new(4", stringify!($Int), ")?;")]
1741 #[doc = concat!("let max = NonZero::new(", stringify!($Int), "::MAX)?;")]
1742 ///
1743 /// assert_eq!(Some(two), two.checked_next_power_of_two() );
1744 /// assert_eq!(Some(four), three.checked_next_power_of_two() );
1745 /// assert_eq!(None, max.checked_next_power_of_two() );
1746 /// # Some(())
1747 /// # }
1748 /// ```
1749 #[stable(feature = "nonzero_checked_ops", since = "1.64.0")]
1750 #[rustc_const_stable(feature = "const_nonzero_checked_ops", since = "1.64.0")]
1751 #[must_use = "this returns the result of the operation, \
1752 without modifying the original"]
1753 #[inline]
1754 pub const fn checked_next_power_of_two(self) -> Option<Self> {
1755 if let Some(nz) = self.get().checked_next_power_of_two() {
1756 // SAFETY: The next power of two is positive
1757 // and overflow is checked.
1758 Some(unsafe { Self::new_unchecked(nz) })
1759 } else {
1760 None
1761 }
1762 }
1763
1764 /// Returns the base 2 logarithm of the number, rounded down.
1765 ///
1766 /// This is the same operation as
1767 #[doc = concat!("[`", stringify!($Int), "::ilog2`],")]
1768 /// except that it has no failure cases to worry about
1769 /// since this value can never be zero.
1770 ///
1771 /// Note that this is equivalent to [`highest_one`](Self::highest_one).
1772 ///
1773 /// # Examples
1774 ///
1775 /// ```
1776 /// # use std::num::NonZero;
1777 /// #
1778 /// # fn main() { test().unwrap(); }
1779 /// # fn test() -> Option<()> {
1780 #[doc = concat!("assert_eq!(NonZero::new(7", stringify!($Int), ")?.ilog2(), 2);")]
1781 #[doc = concat!("assert_eq!(NonZero::new(8", stringify!($Int), ")?.ilog2(), 3);")]
1782 #[doc = concat!("assert_eq!(NonZero::new(9", stringify!($Int), ")?.ilog2(), 3);")]
1783 /// # Some(())
1784 /// # }
1785 /// ```
1786 #[stable(feature = "int_log", since = "1.67.0")]
1787 #[rustc_const_stable(feature = "int_log", since = "1.67.0")]
1788 #[must_use = "this returns the result of the operation, \
1789 without modifying the original"]
1790 #[inline]
1791 pub const fn ilog2(self) -> u32 {
1792 Self::BITS - 1 - self.leading_zeros()
1793 }
1794
1795 /// Returns the base 10 logarithm of the number, rounded down.
1796 ///
1797 /// This is the same operation as
1798 #[doc = concat!("[`", stringify!($Int), "::ilog10`],")]
1799 /// except that it has no failure cases to worry about
1800 /// since this value can never be zero.
1801 ///
1802 /// # Examples
1803 ///
1804 /// ```
1805 /// # use std::num::NonZero;
1806 /// #
1807 /// # fn main() { test().unwrap(); }
1808 /// # fn test() -> Option<()> {
1809 #[doc = concat!("assert_eq!(NonZero::new(99", stringify!($Int), ")?.ilog10(), 1);")]
1810 #[doc = concat!("assert_eq!(NonZero::new(100", stringify!($Int), ")?.ilog10(), 2);")]
1811 #[doc = concat!("assert_eq!(NonZero::new(101", stringify!($Int), ")?.ilog10(), 2);")]
1812 /// # Some(())
1813 /// # }
1814 /// ```
1815 #[stable(feature = "int_log", since = "1.67.0")]
1816 #[rustc_const_stable(feature = "int_log", since = "1.67.0")]
1817 #[must_use = "this returns the result of the operation, \
1818 without modifying the original"]
1819 #[inline]
1820 pub const fn ilog10(self) -> u32 {
1821 imp::int_log10::$Int(self)
1822 }
1823
1824 /// Calculates the midpoint (average) between `self` and `rhs`.
1825 ///
1826 /// `midpoint(a, b)` is `(a + b) >> 1` as if it were performed in a
1827 /// sufficiently-large signed integral type. This implies that the result is
1828 /// always rounded towards negative infinity and that no overflow will ever occur.
1829 ///
1830 /// # Examples
1831 ///
1832 /// ```
1833 /// # use std::num::NonZero;
1834 /// #
1835 /// # fn main() { test().unwrap(); }
1836 /// # fn test() -> Option<()> {
1837 #[doc = concat!("let one = NonZero::new(1", stringify!($Int), ")?;")]
1838 #[doc = concat!("let two = NonZero::new(2", stringify!($Int), ")?;")]
1839 #[doc = concat!("let four = NonZero::new(4", stringify!($Int), ")?;")]
1840 ///
1841 /// assert_eq!(one.midpoint(four), two);
1842 /// assert_eq!(four.midpoint(one), two);
1843 /// # Some(())
1844 /// # }
1845 /// ```
1846 #[stable(feature = "num_midpoint", since = "1.85.0")]
1847 #[rustc_const_stable(feature = "num_midpoint", since = "1.85.0")]
1848 #[must_use = "this returns the result of the operation, \
1849 without modifying the original"]
1850 #[doc(alias = "average_floor")]
1851 #[doc(alias = "average")]
1852 #[inline]
1853 pub const fn midpoint(self, rhs: Self) -> Self {
1854 // SAFETY: The only way to get `0` with midpoint is to have two opposite or
1855 // near opposite numbers: (-5, 5), (0, 1), (0, 0) which is impossible because
1856 // of the unsignedness of this number and also because `Self` is guaranteed to
1857 // never being 0.
1858 unsafe { Self::new_unchecked(self.get().midpoint(rhs.get())) }
1859 }
1860
1861 /// Returns `true` if and only if `self == (1 << k)` for some `k`.
1862 ///
1863 /// On many architectures, this function can perform better than `is_power_of_two()`
1864 /// on the underlying integer type, as special handling of zero can be avoided.
1865 ///
1866 /// # Examples
1867 ///
1868 /// ```
1869 /// # use std::num::NonZero;
1870 /// #
1871 /// # fn main() { test().unwrap(); }
1872 /// # fn test() -> Option<()> {
1873 #[doc = concat!("let eight = NonZero::new(8", stringify!($Int), ")?;")]
1874 /// assert!(eight.is_power_of_two());
1875 #[doc = concat!("let ten = NonZero::new(10", stringify!($Int), ")?;")]
1876 /// assert!(!ten.is_power_of_two());
1877 /// # Some(())
1878 /// # }
1879 /// ```
1880 #[must_use]
1881 #[stable(feature = "nonzero_is_power_of_two", since = "1.59.0")]
1882 #[rustc_const_stable(feature = "nonzero_is_power_of_two", since = "1.59.0")]
1883 #[inline]
1884 pub const fn is_power_of_two(self) -> bool {
1885 // LLVM 11 normalizes `unchecked_sub(x, 1) & x == 0` to the implementation seen here.
1886 // On the basic x86-64 target, this saves 3 instructions for the zero check.
1887 // On x86_64 with BMI1, being nonzero lets it codegen to `BLSR`, which saves an instruction
1888 // compared to the `POPCNT` implementation on the underlying integer type.
1889
1890 intrinsics::ctpop(self.get()) < 2
1891 }
1892
1893 /// Returns the square root of the number, rounded down.
1894 ///
1895 /// # Examples
1896 ///
1897 /// ```
1898 /// # use std::num::NonZero;
1899 /// #
1900 /// # fn main() { test().unwrap(); }
1901 /// # fn test() -> Option<()> {
1902 #[doc = concat!("let ten = NonZero::new(10", stringify!($Int), ")?;")]
1903 #[doc = concat!("let three = NonZero::new(3", stringify!($Int), ")?;")]
1904 ///
1905 /// assert_eq!(ten.isqrt(), three);
1906 /// # Some(())
1907 /// # }
1908 /// ```
1909 #[stable(feature = "isqrt", since = "1.84.0")]
1910 #[rustc_const_stable(feature = "isqrt", since = "1.84.0")]
1911 #[must_use = "this returns the result of the operation, \
1912 without modifying the original"]
1913 #[inline]
1914 pub const fn isqrt(self) -> Self {
1915 let result = self.get().isqrt();
1916
1917 // SAFETY: Integer square root is a monotonically nondecreasing
1918 // function, which means that increasing the input will never cause
1919 // the output to decrease. Thus, since the input for nonzero
1920 // unsigned integers has a lower bound of 1, the lower bound of the
1921 // results will be sqrt(1), which is 1, so a result can't be zero.
1922 unsafe { Self::new_unchecked(result) }
1923 }
1924
1925 /// Returns the bit pattern of `self` reinterpreted as a signed integer of the same size.
1926 ///
1927 /// # Examples
1928 ///
1929 /// ```
1930 /// # use std::num::NonZero;
1931 ///
1932 #[doc = concat!("let n = NonZero::<", stringify!($Int), ">::MAX;")]
1933 ///
1934 #[doc = concat!("assert_eq!(n.cast_signed(), NonZero::new(-1", stringify!($Sint), ").unwrap());")]
1935 /// ```
1936 #[stable(feature = "integer_sign_cast", since = "1.87.0")]
1937 #[rustc_const_stable(feature = "integer_sign_cast", since = "1.87.0")]
1938 #[must_use = "this returns the result of the operation, \
1939 without modifying the original"]
1940 #[inline(always)]
1941 pub const fn cast_signed(self) -> NonZero<$Sint> {
1942 // SAFETY: `self.get()` can't be zero
1943 unsafe { NonZero::new_unchecked(self.get().cast_signed()) }
1944 }
1945
1946 /// Returns the minimum number of bits required to represent `self`.
1947 ///
1948 /// # Examples
1949 ///
1950 /// ```
1951 /// # use core::num::NonZero;
1952 /// #
1953 /// # fn main() { test().unwrap(); }
1954 /// # fn test() -> Option<()> {
1955 #[doc = concat!("assert_eq!(NonZero::<", stringify!($Int), ">::new(0b1)?.bit_width(), NonZero::new(1)?);")]
1956 #[doc = concat!("assert_eq!(NonZero::<", stringify!($Int), ">::new(0b111)?.bit_width(), NonZero::new(3)?);")]
1957 #[doc = concat!("assert_eq!(NonZero::<", stringify!($Int), ">::new(0b1110)?.bit_width(), NonZero::new(4)?);")]
1958 /// # Some(())
1959 /// # }
1960 /// ```
1961 #[stable(feature = "uint_bit_width", since = "1.97.0")]
1962 #[rustc_const_stable(feature = "uint_bit_width", since = "1.97.0")]
1963 #[must_use = "this returns the result of the operation, \
1964 without modifying the original"]
1965 #[inline(always)]
1966 pub const fn bit_width(self) -> NonZero<u32> {
1967 // SAFETY: Since `self.leading_zeros()` is always less than
1968 // `Self::BITS`, this subtraction can never be zero.
1969 unsafe { NonZero::new_unchecked(Self::BITS - self.leading_zeros()) }
1970 }
1971 };
1972
1973 // Associated items for signed nonzero types only.
1974 (
1975 Primitive = signed $Int:ident,
1976 SignedPrimitive = $Sint:ty,
1977 UnsignedPrimitive = $Uint:ty,
1978 ) => {
1979 /// The smallest value that can be represented by this non-zero
1980 /// integer type,
1981 #[doc = concat!("equal to [`", stringify!($Int), "::MIN`].")]
1982 ///
1983 /// Note: While most integer types are defined for every whole
1984 /// number between `MIN` and `MAX`, signed non-zero integers are
1985 /// a special case. They have a "gap" at 0.
1986 ///
1987 /// # Examples
1988 ///
1989 /// ```
1990 /// # use std::num::NonZero;
1991 /// #
1992 #[doc = concat!("assert_eq!(NonZero::<", stringify!($Int), ">::MIN.get(), ", stringify!($Int), "::MIN);")]
1993 /// ```
1994 #[stable(feature = "nonzero_min_max", since = "1.70.0")]
1995 pub const MIN: Self = Self::new(<$Int>::MIN).unwrap();
1996
1997 /// The largest value that can be represented by this non-zero
1998 /// integer type,
1999 #[doc = concat!("equal to [`", stringify!($Int), "::MAX`].")]
2000 ///
2001 /// Note: While most integer types are defined for every whole
2002 /// number between `MIN` and `MAX`, signed non-zero integers are
2003 /// a special case. They have a "gap" at 0.
2004 ///
2005 /// # Examples
2006 ///
2007 /// ```
2008 /// # use std::num::NonZero;
2009 /// #
2010 #[doc = concat!("assert_eq!(NonZero::<", stringify!($Int), ">::MAX.get(), ", stringify!($Int), "::MAX);")]
2011 /// ```
2012 #[stable(feature = "nonzero_min_max", since = "1.70.0")]
2013 pub const MAX: Self = Self::new(<$Int>::MAX).unwrap();
2014
2015 /// Computes the absolute value of self.
2016 #[doc = concat!("See [`", stringify!($Int), "::abs`]")]
2017 /// for documentation on overflow behavior.
2018 ///
2019 /// # Example
2020 ///
2021 /// ```
2022 /// # use std::num::NonZero;
2023 /// #
2024 /// # fn main() { test().unwrap(); }
2025 /// # fn test() -> Option<()> {
2026 #[doc = concat!("let pos = NonZero::new(1", stringify!($Int), ")?;")]
2027 #[doc = concat!("let neg = NonZero::new(-1", stringify!($Int), ")?;")]
2028 ///
2029 /// assert_eq!(pos, pos.abs());
2030 /// assert_eq!(pos, neg.abs());
2031 /// # Some(())
2032 /// # }
2033 /// ```
2034 #[stable(feature = "nonzero_checked_ops", since = "1.64.0")]
2035 #[rustc_const_stable(feature = "const_nonzero_checked_ops", since = "1.64.0")]
2036 #[must_use = "this returns the result of the operation, \
2037 without modifying the original"]
2038 #[inline]
2039 pub const fn abs(self) -> Self {
2040 // SAFETY: This cannot overflow to zero.
2041 unsafe { Self::new_unchecked(self.get().abs()) }
2042 }
2043
2044 /// Checked absolute value.
2045 /// Checks for overflow and returns [`None`] if
2046 #[doc = concat!("`self == NonZero::<", stringify!($Int), ">::MIN`.")]
2047 /// The result cannot be zero.
2048 ///
2049 /// # Example
2050 ///
2051 /// ```
2052 /// # use std::num::NonZero;
2053 /// #
2054 /// # fn main() { test().unwrap(); }
2055 /// # fn test() -> Option<()> {
2056 #[doc = concat!("let pos = NonZero::new(1", stringify!($Int), ")?;")]
2057 #[doc = concat!("let neg = NonZero::new(-1", stringify!($Int), ")?;")]
2058 #[doc = concat!("let min = NonZero::new(", stringify!($Int), "::MIN)?;")]
2059 ///
2060 /// assert_eq!(Some(pos), neg.checked_abs());
2061 /// assert_eq!(None, min.checked_abs());
2062 /// # Some(())
2063 /// # }
2064 /// ```
2065 #[stable(feature = "nonzero_checked_ops", since = "1.64.0")]
2066 #[rustc_const_stable(feature = "const_nonzero_checked_ops", since = "1.64.0")]
2067 #[must_use = "this returns the result of the operation, \
2068 without modifying the original"]
2069 #[inline]
2070 pub const fn checked_abs(self) -> Option<Self> {
2071 if let Some(nz) = self.get().checked_abs() {
2072 // SAFETY: absolute value of nonzero cannot yield zero values.
2073 Some(unsafe { Self::new_unchecked(nz) })
2074 } else {
2075 None
2076 }
2077 }
2078
2079 /// Computes the absolute value of self,
2080 /// with overflow information, see
2081 #[doc = concat!("[`", stringify!($Int), "::overflowing_abs`].")]
2082 ///
2083 /// # Example
2084 ///
2085 /// ```
2086 /// # use std::num::NonZero;
2087 /// #
2088 /// # fn main() { test().unwrap(); }
2089 /// # fn test() -> Option<()> {
2090 #[doc = concat!("let pos = NonZero::new(1", stringify!($Int), ")?;")]
2091 #[doc = concat!("let neg = NonZero::new(-1", stringify!($Int), ")?;")]
2092 #[doc = concat!("let min = NonZero::new(", stringify!($Int), "::MIN)?;")]
2093 ///
2094 /// assert_eq!((pos, false), pos.overflowing_abs());
2095 /// assert_eq!((pos, false), neg.overflowing_abs());
2096 /// assert_eq!((min, true), min.overflowing_abs());
2097 /// # Some(())
2098 /// # }
2099 /// ```
2100 #[stable(feature = "nonzero_checked_ops", since = "1.64.0")]
2101 #[rustc_const_stable(feature = "const_nonzero_checked_ops", since = "1.64.0")]
2102 #[must_use = "this returns the result of the operation, \
2103 without modifying the original"]
2104 #[inline]
2105 pub const fn overflowing_abs(self) -> (Self, bool) {
2106 let (nz, flag) = self.get().overflowing_abs();
2107 (
2108 // SAFETY: absolute value of nonzero cannot yield zero values.
2109 unsafe { Self::new_unchecked(nz) },
2110 flag,
2111 )
2112 }
2113
2114 /// Saturating absolute value, see
2115 #[doc = concat!("[`", stringify!($Int), "::saturating_abs`].")]
2116 ///
2117 /// # Example
2118 ///
2119 /// ```
2120 /// # use std::num::NonZero;
2121 /// #
2122 /// # fn main() { test().unwrap(); }
2123 /// # fn test() -> Option<()> {
2124 #[doc = concat!("let pos = NonZero::new(1", stringify!($Int), ")?;")]
2125 #[doc = concat!("let neg = NonZero::new(-1", stringify!($Int), ")?;")]
2126 #[doc = concat!("let min = NonZero::new(", stringify!($Int), "::MIN)?;")]
2127 #[doc = concat!("let min_plus = NonZero::new(", stringify!($Int), "::MIN + 1)?;")]
2128 #[doc = concat!("let max = NonZero::new(", stringify!($Int), "::MAX)?;")]
2129 ///
2130 /// assert_eq!(pos, pos.saturating_abs());
2131 /// assert_eq!(pos, neg.saturating_abs());
2132 /// assert_eq!(max, min.saturating_abs());
2133 /// assert_eq!(max, min_plus.saturating_abs());
2134 /// # Some(())
2135 /// # }
2136 /// ```
2137 #[stable(feature = "nonzero_checked_ops", since = "1.64.0")]
2138 #[rustc_const_stable(feature = "const_nonzero_checked_ops", since = "1.64.0")]
2139 #[must_use = "this returns the result of the operation, \
2140 without modifying the original"]
2141 #[inline]
2142 pub const fn saturating_abs(self) -> Self {
2143 // SAFETY: absolute value of nonzero cannot yield zero values.
2144 unsafe { Self::new_unchecked(self.get().saturating_abs()) }
2145 }
2146
2147 /// Wrapping absolute value, see
2148 #[doc = concat!("[`", stringify!($Int), "::wrapping_abs`].")]
2149 ///
2150 /// # Example
2151 ///
2152 /// ```
2153 /// # use std::num::NonZero;
2154 /// #
2155 /// # fn main() { test().unwrap(); }
2156 /// # fn test() -> Option<()> {
2157 #[doc = concat!("let pos = NonZero::new(1", stringify!($Int), ")?;")]
2158 #[doc = concat!("let neg = NonZero::new(-1", stringify!($Int), ")?;")]
2159 #[doc = concat!("let min = NonZero::new(", stringify!($Int), "::MIN)?;")]
2160 #[doc = concat!("# let max = NonZero::new(", stringify!($Int), "::MAX)?;")]
2161 ///
2162 /// assert_eq!(pos, pos.wrapping_abs());
2163 /// assert_eq!(pos, neg.wrapping_abs());
2164 /// assert_eq!(min, min.wrapping_abs());
2165 /// assert_eq!(max, (-max).wrapping_abs());
2166 /// # Some(())
2167 /// # }
2168 /// ```
2169 #[stable(feature = "nonzero_checked_ops", since = "1.64.0")]
2170 #[rustc_const_stable(feature = "const_nonzero_checked_ops", since = "1.64.0")]
2171 #[must_use = "this returns the result of the operation, \
2172 without modifying the original"]
2173 #[inline]
2174 pub const fn wrapping_abs(self) -> Self {
2175 // SAFETY: absolute value of nonzero cannot yield zero values.
2176 unsafe { Self::new_unchecked(self.get().wrapping_abs()) }
2177 }
2178
2179 /// Computes the absolute value of self
2180 /// without any wrapping or panicking.
2181 ///
2182 /// # Example
2183 ///
2184 /// ```
2185 /// # use std::num::NonZero;
2186 /// #
2187 /// # fn main() { test().unwrap(); }
2188 /// # fn test() -> Option<()> {
2189 #[doc = concat!("let u_pos = NonZero::new(1", stringify!($Uint), ")?;")]
2190 #[doc = concat!("let i_pos = NonZero::new(1", stringify!($Int), ")?;")]
2191 #[doc = concat!("let i_neg = NonZero::new(-1", stringify!($Int), ")?;")]
2192 #[doc = concat!("let i_min = NonZero::new(", stringify!($Int), "::MIN)?;")]
2193 #[doc = concat!("let u_max = NonZero::new(", stringify!($Uint), "::MAX / 2 + 1)?;")]
2194 ///
2195 /// assert_eq!(u_pos, i_pos.unsigned_abs());
2196 /// assert_eq!(u_pos, i_neg.unsigned_abs());
2197 /// assert_eq!(u_max, i_min.unsigned_abs());
2198 /// # Some(())
2199 /// # }
2200 /// ```
2201 #[stable(feature = "nonzero_checked_ops", since = "1.64.0")]
2202 #[rustc_const_stable(feature = "const_nonzero_checked_ops", since = "1.64.0")]
2203 #[must_use = "this returns the result of the operation, \
2204 without modifying the original"]
2205 #[inline]
2206 pub const fn unsigned_abs(self) -> NonZero<$Uint> {
2207 // SAFETY: absolute value of nonzero cannot yield zero values.
2208 unsafe { NonZero::new_unchecked(self.get().unsigned_abs()) }
2209 }
2210
2211 /// Returns `true` if `self` is positive and `false` if the
2212 /// number is negative.
2213 ///
2214 /// # Example
2215 ///
2216 /// ```
2217 /// # use std::num::NonZero;
2218 /// #
2219 /// # fn main() { test().unwrap(); }
2220 /// # fn test() -> Option<()> {
2221 #[doc = concat!("let pos_five = NonZero::new(5", stringify!($Int), ")?;")]
2222 #[doc = concat!("let neg_five = NonZero::new(-5", stringify!($Int), ")?;")]
2223 ///
2224 /// assert!(pos_five.is_positive());
2225 /// assert!(!neg_five.is_positive());
2226 /// # Some(())
2227 /// # }
2228 /// ```
2229 #[must_use]
2230 #[inline]
2231 #[stable(feature = "nonzero_negation_ops", since = "1.71.0")]
2232 #[rustc_const_stable(feature = "nonzero_negation_ops", since = "1.71.0")]
2233 pub const fn is_positive(self) -> bool {
2234 self.get().is_positive()
2235 }
2236
2237 /// Returns `true` if `self` is negative and `false` if the
2238 /// number is positive.
2239 ///
2240 /// # Example
2241 ///
2242 /// ```
2243 /// # use std::num::NonZero;
2244 /// #
2245 /// # fn main() { test().unwrap(); }
2246 /// # fn test() -> Option<()> {
2247 #[doc = concat!("let pos_five = NonZero::new(5", stringify!($Int), ")?;")]
2248 #[doc = concat!("let neg_five = NonZero::new(-5", stringify!($Int), ")?;")]
2249 ///
2250 /// assert!(neg_five.is_negative());
2251 /// assert!(!pos_five.is_negative());
2252 /// # Some(())
2253 /// # }
2254 /// ```
2255 #[must_use]
2256 #[inline]
2257 #[stable(feature = "nonzero_negation_ops", since = "1.71.0")]
2258 #[rustc_const_stable(feature = "nonzero_negation_ops", since = "1.71.0")]
2259 pub const fn is_negative(self) -> bool {
2260 self.get().is_negative()
2261 }
2262
2263 /// Checked negation. Computes `-self`,
2264 #[doc = concat!("returning `None` if `self == NonZero::<", stringify!($Int), ">::MIN`.")]
2265 ///
2266 /// # Example
2267 ///
2268 /// ```
2269 /// # use std::num::NonZero;
2270 /// #
2271 /// # fn main() { test().unwrap(); }
2272 /// # fn test() -> Option<()> {
2273 #[doc = concat!("let pos_five = NonZero::new(5", stringify!($Int), ")?;")]
2274 #[doc = concat!("let neg_five = NonZero::new(-5", stringify!($Int), ")?;")]
2275 #[doc = concat!("let min = NonZero::new(", stringify!($Int), "::MIN)?;")]
2276 ///
2277 /// assert_eq!(pos_five.checked_neg(), Some(neg_five));
2278 /// assert_eq!(min.checked_neg(), None);
2279 /// # Some(())
2280 /// # }
2281 /// ```
2282 #[inline]
2283 #[stable(feature = "nonzero_negation_ops", since = "1.71.0")]
2284 #[rustc_const_stable(feature = "nonzero_negation_ops", since = "1.71.0")]
2285 pub const fn checked_neg(self) -> Option<Self> {
2286 if let Some(result) = self.get().checked_neg() {
2287 // SAFETY: negation of nonzero cannot yield zero values.
2288 return Some(unsafe { Self::new_unchecked(result) });
2289 }
2290 None
2291 }
2292
2293 /// Negates self, overflowing if this is equal to the minimum value.
2294 ///
2295 #[doc = concat!("See [`", stringify!($Int), "::overflowing_neg`]")]
2296 /// for documentation on overflow behavior.
2297 ///
2298 /// # Example
2299 ///
2300 /// ```
2301 /// # use std::num::NonZero;
2302 /// #
2303 /// # fn main() { test().unwrap(); }
2304 /// # fn test() -> Option<()> {
2305 #[doc = concat!("let pos_five = NonZero::new(5", stringify!($Int), ")?;")]
2306 #[doc = concat!("let neg_five = NonZero::new(-5", stringify!($Int), ")?;")]
2307 #[doc = concat!("let min = NonZero::new(", stringify!($Int), "::MIN)?;")]
2308 ///
2309 /// assert_eq!(pos_five.overflowing_neg(), (neg_five, false));
2310 /// assert_eq!(min.overflowing_neg(), (min, true));
2311 /// # Some(())
2312 /// # }
2313 /// ```
2314 #[inline]
2315 #[stable(feature = "nonzero_negation_ops", since = "1.71.0")]
2316 #[rustc_const_stable(feature = "nonzero_negation_ops", since = "1.71.0")]
2317 pub const fn overflowing_neg(self) -> (Self, bool) {
2318 let (result, overflow) = self.get().overflowing_neg();
2319 // SAFETY: negation of nonzero cannot yield zero values.
2320 ((unsafe { Self::new_unchecked(result) }), overflow)
2321 }
2322
2323 /// Saturating negation. Computes `-self`,
2324 #[doc = concat!("returning [`NonZero::<", stringify!($Int), ">::MAX`]")]
2325 #[doc = concat!("if `self == NonZero::<", stringify!($Int), ">::MIN`")]
2326 /// instead of overflowing.
2327 ///
2328 /// # Example
2329 ///
2330 /// ```
2331 /// # use std::num::NonZero;
2332 /// #
2333 /// # fn main() { test().unwrap(); }
2334 /// # fn test() -> Option<()> {
2335 #[doc = concat!("let pos_five = NonZero::new(5", stringify!($Int), ")?;")]
2336 #[doc = concat!("let neg_five = NonZero::new(-5", stringify!($Int), ")?;")]
2337 #[doc = concat!("let min = NonZero::new(", stringify!($Int), "::MIN)?;")]
2338 #[doc = concat!("let min_plus_one = NonZero::new(", stringify!($Int), "::MIN + 1)?;")]
2339 #[doc = concat!("let max = NonZero::new(", stringify!($Int), "::MAX)?;")]
2340 ///
2341 /// assert_eq!(pos_five.saturating_neg(), neg_five);
2342 /// assert_eq!(min.saturating_neg(), max);
2343 /// assert_eq!(max.saturating_neg(), min_plus_one);
2344 /// # Some(())
2345 /// # }
2346 /// ```
2347 #[inline]
2348 #[stable(feature = "nonzero_negation_ops", since = "1.71.0")]
2349 #[rustc_const_stable(feature = "nonzero_negation_ops", since = "1.71.0")]
2350 pub const fn saturating_neg(self) -> Self {
2351 if let Some(result) = self.checked_neg() {
2352 return result;
2353 }
2354 Self::MAX
2355 }
2356
2357 /// Wrapping (modular) negation. Computes `-self`, wrapping around at the boundary
2358 /// of the type.
2359 ///
2360 #[doc = concat!("See [`", stringify!($Int), "::wrapping_neg`]")]
2361 /// for documentation on overflow behavior.
2362 ///
2363 /// # Example
2364 ///
2365 /// ```
2366 /// # use std::num::NonZero;
2367 /// #
2368 /// # fn main() { test().unwrap(); }
2369 /// # fn test() -> Option<()> {
2370 #[doc = concat!("let pos_five = NonZero::new(5", stringify!($Int), ")?;")]
2371 #[doc = concat!("let neg_five = NonZero::new(-5", stringify!($Int), ")?;")]
2372 #[doc = concat!("let min = NonZero::new(", stringify!($Int), "::MIN)?;")]
2373 ///
2374 /// assert_eq!(pos_five.wrapping_neg(), neg_five);
2375 /// assert_eq!(min.wrapping_neg(), min);
2376 /// # Some(())
2377 /// # }
2378 /// ```
2379 #[inline]
2380 #[stable(feature = "nonzero_negation_ops", since = "1.71.0")]
2381 #[rustc_const_stable(feature = "nonzero_negation_ops", since = "1.71.0")]
2382 pub const fn wrapping_neg(self) -> Self {
2383 let result = self.get().wrapping_neg();
2384 // SAFETY: negation of nonzero cannot yield zero values.
2385 unsafe { Self::new_unchecked(result) }
2386 }
2387
2388 /// Returns the bit pattern of `self` reinterpreted as an unsigned integer of the same size.
2389 ///
2390 /// # Examples
2391 ///
2392 /// ```
2393 /// # use std::num::NonZero;
2394 ///
2395 #[doc = concat!("let n = NonZero::new(-1", stringify!($Int), ").unwrap();")]
2396 ///
2397 #[doc = concat!("assert_eq!(n.cast_unsigned(), NonZero::<", stringify!($Uint), ">::MAX);")]
2398 /// ```
2399 #[stable(feature = "integer_sign_cast", since = "1.87.0")]
2400 #[rustc_const_stable(feature = "integer_sign_cast", since = "1.87.0")]
2401 #[must_use = "this returns the result of the operation, \
2402 without modifying the original"]
2403 #[inline(always)]
2404 pub const fn cast_unsigned(self) -> NonZero<$Uint> {
2405 // SAFETY: `self.get()` can't be zero
2406 unsafe { NonZero::new_unchecked(self.get().cast_unsigned()) }
2407 }
2408
2409 };
2410}
2411
2412nonzero_integer! {
2413 Self = NonZeroU8,
2414 Primitive = unsigned u8,
2415 SignedPrimitive = i8,
2416 rot = 2,
2417 rot_op = "0x82",
2418 rot_result = "0xa",
2419 swap_op = "0x12",
2420 swapped = "0x12",
2421 reversed = "0x48",
2422}
2423
2424nonzero_integer! {
2425 Self = NonZeroU16,
2426 Primitive = unsigned u16,
2427 SignedPrimitive = i16,
2428 rot = 4,
2429 rot_op = "0xa003",
2430 rot_result = "0x3a",
2431 swap_op = "0x1234",
2432 swapped = "0x3412",
2433 reversed = "0x2c48",
2434}
2435
2436nonzero_integer! {
2437 Self = NonZeroU32,
2438 Primitive = unsigned u32,
2439 SignedPrimitive = i32,
2440 rot = 8,
2441 rot_op = "0x10000b3",
2442 rot_result = "0xb301",
2443 swap_op = "0x12345678",
2444 swapped = "0x78563412",
2445 reversed = "0x1e6a2c48",
2446}
2447
2448nonzero_integer! {
2449 Self = NonZeroU64,
2450 Primitive = unsigned u64,
2451 SignedPrimitive = i64,
2452 rot = 12,
2453 rot_op = "0xaa00000000006e1",
2454 rot_result = "0x6e10aa",
2455 swap_op = "0x1234567890123456",
2456 swapped = "0x5634129078563412",
2457 reversed = "0x6a2c48091e6a2c48",
2458}
2459
2460nonzero_integer! {
2461 Self = NonZeroU128,
2462 Primitive = unsigned u128,
2463 SignedPrimitive = i128,
2464 rot = 16,
2465 rot_op = "0x13f40000000000000000000000004f76",
2466 rot_result = "0x4f7613f4",
2467 swap_op = "0x12345678901234567890123456789012",
2468 swapped = "0x12907856341290785634129078563412",
2469 reversed = "0x48091e6a2c48091e6a2c48091e6a2c48",
2470}
2471
2472#[cfg(target_pointer_width = "16")]
2473nonzero_integer! {
2474 Self = NonZeroUsize,
2475 Primitive = unsigned usize,
2476 SignedPrimitive = isize,
2477 rot = 4,
2478 rot_op = "0xa003",
2479 rot_result = "0x3a",
2480 swap_op = "0x1234",
2481 swapped = "0x3412",
2482 reversed = "0x2c48",
2483}
2484
2485#[cfg(target_pointer_width = "32")]
2486nonzero_integer! {
2487 Self = NonZeroUsize,
2488 Primitive = unsigned usize,
2489 SignedPrimitive = isize,
2490 rot = 8,
2491 rot_op = "0x10000b3",
2492 rot_result = "0xb301",
2493 swap_op = "0x12345678",
2494 swapped = "0x78563412",
2495 reversed = "0x1e6a2c48",
2496}
2497
2498#[cfg(target_pointer_width = "64")]
2499nonzero_integer! {
2500 Self = NonZeroUsize,
2501 Primitive = unsigned usize,
2502 SignedPrimitive = isize,
2503 rot = 12,
2504 rot_op = "0xaa00000000006e1",
2505 rot_result = "0x6e10aa",
2506 swap_op = "0x1234567890123456",
2507 swapped = "0x5634129078563412",
2508 reversed = "0x6a2c48091e6a2c48",
2509}
2510
2511nonzero_integer! {
2512 Self = NonZeroI8,
2513 Primitive = signed i8,
2514 UnsignedPrimitive = u8,
2515 rot = 2,
2516 rot_op = "-0x7e",
2517 rot_result = "0xa",
2518 swap_op = "0x12",
2519 swapped = "0x12",
2520 reversed = "0x48",
2521}
2522
2523nonzero_integer! {
2524 Self = NonZeroI16,
2525 Primitive = signed i16,
2526 UnsignedPrimitive = u16,
2527 rot = 4,
2528 rot_op = "-0x5ffd",
2529 rot_result = "0x3a",
2530 swap_op = "0x1234",
2531 swapped = "0x3412",
2532 reversed = "0x2c48",
2533}
2534
2535nonzero_integer! {
2536 Self = NonZeroI32,
2537 Primitive = signed i32,
2538 UnsignedPrimitive = u32,
2539 rot = 8,
2540 rot_op = "0x10000b3",
2541 rot_result = "0xb301",
2542 swap_op = "0x12345678",
2543 swapped = "0x78563412",
2544 reversed = "0x1e6a2c48",
2545}
2546
2547nonzero_integer! {
2548 Self = NonZeroI64,
2549 Primitive = signed i64,
2550 UnsignedPrimitive = u64,
2551 rot = 12,
2552 rot_op = "0xaa00000000006e1",
2553 rot_result = "0x6e10aa",
2554 swap_op = "0x1234567890123456",
2555 swapped = "0x5634129078563412",
2556 reversed = "0x6a2c48091e6a2c48",
2557}
2558
2559nonzero_integer! {
2560 Self = NonZeroI128,
2561 Primitive = signed i128,
2562 UnsignedPrimitive = u128,
2563 rot = 16,
2564 rot_op = "0x13f40000000000000000000000004f76",
2565 rot_result = "0x4f7613f4",
2566 swap_op = "0x12345678901234567890123456789012",
2567 swapped = "0x12907856341290785634129078563412",
2568 reversed = "0x48091e6a2c48091e6a2c48091e6a2c48",
2569}
2570
2571#[cfg(target_pointer_width = "16")]
2572nonzero_integer! {
2573 Self = NonZeroIsize,
2574 Primitive = signed isize,
2575 UnsignedPrimitive = usize,
2576 rot = 4,
2577 rot_op = "-0x5ffd",
2578 rot_result = "0x3a",
2579 swap_op = "0x1234",
2580 swapped = "0x3412",
2581 reversed = "0x2c48",
2582}
2583
2584#[cfg(target_pointer_width = "32")]
2585nonzero_integer! {
2586 Self = NonZeroIsize,
2587 Primitive = signed isize,
2588 UnsignedPrimitive = usize,
2589 rot = 8,
2590 rot_op = "0x10000b3",
2591 rot_result = "0xb301",
2592 swap_op = "0x12345678",
2593 swapped = "0x78563412",
2594 reversed = "0x1e6a2c48",
2595}
2596
2597#[cfg(target_pointer_width = "64")]
2598nonzero_integer! {
2599 Self = NonZeroIsize,
2600 Primitive = signed isize,
2601 UnsignedPrimitive = usize,
2602 rot = 12,
2603 rot_op = "0xaa00000000006e1",
2604 rot_result = "0x6e10aa",
2605 swap_op = "0x1234567890123456",
2606 swapped = "0x5634129078563412",
2607 reversed = "0x6a2c48091e6a2c48",
2608}