context-allocator 0.2.3

context-allocator
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

// This file is part of context-allocator. It is subject to the license terms in the COPYRIGHT file found in the top-level directory of this distribution and at https://raw.githubusercontent.com/lemonrock/context-allocator/master/COPYRIGHT. No part of context-allocator, including this file, may be copied, modified, propagated, or distributed except according to the terms contained in the COPYRIGHT file.
// Copyright © 2019 The developers of context-allocator. See the COPYRIGHT file in the top-level directory of this distribution and at https://raw.githubusercontent.com/lemonrock/context-allocator/master/COPYRIGHT.


/// Useful extensions.
pub(crate) trait NonZeroU32Ext: Sized + Copy
{
	/// Add.
	#[inline(always)]
	fn checked_add(self, increment: Self) -> Option<Self>
	{
		self.to_u32().checked_add(increment.to_u32()).map(Self::non_zero_unchecked)
	}

	/// Add.
	#[inline(always)]
	fn add_assign(&mut self, increment: Self)
	{
		*self = Self::non_zero_unchecked(self.to_u32() + increment.to_u32())
	}

	/// Next power of two.
	#[inline(always)]
	fn next_power_of_two(self) -> Self
	{
		Self::non_zero_unchecked(self.to_u32().next_power_of_two())
	}

	/// Non zero.
	#[inline(always)]
	fn non_zero(value: u32) -> Self
	{
		debug_assert_ne!(value, 0, "value is zero");

		Self::non_zero_unchecked(value)
	}

	/// Non zero.
	fn non_zero_unchecked(value: u32) -> Self;

	#[doc(hidden)]
	fn to_u32(self) -> u32;
}

impl NonZeroU32Ext for NonZeroU32
{
	#[inline(always)]
	fn to_u32(self) -> u32
	{
		self.get()
	}

	#[inline(always)]
	fn non_zero_unchecked(value: u32) -> Self
	{
		unsafe { NonZeroU32::new_unchecked(value) }
	}
}