1 | use num_conv::prelude::*; |
2 | |
3 | /// A trait that indicates the formatted width of the value can be determined. |
4 | /// |
5 | /// Note that this should not be implemented for any signed integers. This forces the caller to |
6 | /// write the sign if desired. |
7 | pub(crate) trait DigitCount { |
8 | /// The number of digits in the stringified value. |
9 | fn num_digits(self) -> u8; |
10 | } |
11 | |
12 | /// A macro to generate implementations of `DigitCount` for unsigned integers. |
13 | macro_rules! impl_digit_count { |
14 | ($($t:ty),* $(,)?) => { |
15 | $(impl DigitCount for $t { |
16 | fn num_digits(self) -> u8 { |
17 | match self.checked_ilog10() { |
18 | Some(n) => n.truncate::<u8>() + 1, |
19 | None => 1, |
20 | } |
21 | } |
22 | })* |
23 | }; |
24 | } |
25 | |
26 | impl_digit_count!(u8, u16, u32); |
27 | |