#![no_std]
use core::fmt::{Debug, Display, Formatter};
use core::mem::size_of;
use core::ops::Deref;
use core::str;
pub trait NumToA {
fn numtoa(self, base: Self, string: &mut [u8]) -> &[u8];
fn numtoa_str(self, base: Self, buf: &mut [u8]) -> &str;
}
const LOOKUP: &[u8] = b"0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ";
const DEC_LOOKUP: &[u8; 200] = b"0001020304050607080910111213141516171819\
2021222324252627282930313233343536373839\
4041424344454647484950515253545556575859\
6061626364656667686970717273747576777879\
8081828384858687888990919293949596979899";
pub struct AsciiNumber<const N: usize> {
string: [u8; N],
start: usize,
}
impl <const N: usize> AsciiNumber<N> {
pub const fn as_slice(&self) -> &[u8] {
self.string.split_at(self.start).1
}
pub const fn as_str(&self) -> &str {
unsafe { core::str::from_utf8_unchecked(Self::as_slice(self)) }
}
pub const fn into_inner(self) -> ([u8;N], usize) {
(self.string, self.start)
}
}
impl <const N: usize> Deref for AsciiNumber<N> {
type Target = str;
fn deref(&self) -> &<Self as Deref>::Target {
self.as_str()
}
}
impl <const N: usize> Display for AsciiNumber<N> {
fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), core::fmt::Error> {
Display::fmt(self.as_str(), f)
}
}
impl <const N: usize> Debug for AsciiNumber<N> {
fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), core::fmt::Error> {
Debug::fmt(self.as_str(), f)
}
}
macro_rules! copy_2_dec_lut_bytes {
($to:ident,$to_index:expr,$lut_index:expr) => {
$to[$to_index as usize] = DEC_LOOKUP[$lut_index as usize];
$to[$to_index as usize+1] = DEC_LOOKUP[$lut_index as usize+1];
};
}
macro_rules! base_10 {
($number:ident, $index:ident, $string:ident) => {
while $number > 9999 {
let rem = ($number % 10000) as u16;
let (frst, scnd) = ((rem / 100) * 2, (rem % 100) * 2);
copy_2_dec_lut_bytes!($string, $index-3, frst);
copy_2_dec_lut_bytes!($string, $index-1, scnd);
$index = $index.wrapping_sub(4);
$number /= 10000;
}
if $number > 999 {
let (frst, scnd) = (($number / 100) * 2, ($number % 100) * 2);
copy_2_dec_lut_bytes!($string, $index-3, frst);
copy_2_dec_lut_bytes!($string, $index-1, scnd);
$index = $index.wrapping_sub(4);
} else if $number > 99 {
let section = ($number as u16 / 10) * 2;
copy_2_dec_lut_bytes!($string, $index-2, section);
$string[$index] = LOOKUP[($number % 10) as usize];
$index = $index.wrapping_sub(3);
} else if $number > 9 {
$number *= 2;
copy_2_dec_lut_bytes!($string, $index-1, $number);
$index = $index.wrapping_sub(2);
} else {
$string[$index] = LOOKUP[$number as usize];
$index = $index.wrapping_sub(1);
}
}
}
macro_rules! impl_unsigned_numtoa_for {
(
$type_name:ty,
$core_function_name:ident,
$str_function_name:ident
) => {
pub const fn $core_function_name(mut num: $type_name, base: $type_name, string: &mut [u8]) -> &[u8] {
if cfg!(debug_assertions) {
if base == 10 {
match size_of::<$type_name>() {
2 => debug_assert!(string.len() >= 5, "u16 base 10 conversions require at least 5 bytes"),
4 => debug_assert!(string.len() >= 10, "u32 base 10 conversions require at least 10 bytes"),
8 => debug_assert!(string.len() >= 20, "u64 base 10 conversions require at least 20 bytes"),
16 => debug_assert!(string.len() >= 39, "u128 base 10 conversions require at least 39 bytes"),
_ => unreachable!()
}
}
}
let mut index = string.len() - 1;
if num == 0 {
string[index] = b'0';
return string.split_at(index).1;
}
if base == 10 {
base_10!(num, index, string);
} else {
while num != 0 {
let rem = num % base;
string[index] = LOOKUP[rem as usize];
index = index.wrapping_sub(1);
num /= base;
}
}
string.split_at(index.wrapping_add(1)).1
}
pub const fn $str_function_name(num: $type_name, base: $type_name, string: &mut [u8]) -> &str {
unsafe { core::str::from_utf8_unchecked($core_function_name(num, base, string)) }
}
impl NumToA for $type_name {
fn numtoa(self, base: $type_name, string: &mut [u8]) -> &[u8] {
$core_function_name(self, base, string)
}
fn numtoa_str(self, base: $type_name, buf: &mut [u8]) -> &str {
$str_function_name(self, base, buf)
}
}
}
}
macro_rules! impl_signed_numtoa_for {
(
$type_name:ty,
$core_function_name:ident,
$str_function_name:ident
) => {
pub const fn $core_function_name(mut num: $type_name, base: $type_name, string: &mut [u8]) -> &[u8] {
if cfg!(debug_assertions) {
if base == 10 {
match size_of::<$type_name>() {
2 => debug_assert!(string.len() >= 6, "i16 base 10 conversions require at least 6 bytes"),
4 => debug_assert!(string.len() >= 11, "i32 base 10 conversions require at least 11 bytes"),
8 => debug_assert!(string.len() >= 19, "i64 base 10 conversions require at least 19 bytes"),
16 => debug_assert!(string.len() >= 39, "i128 base 10 conversions require at least 39 bytes"),
_ => unreachable!()
}
}
}
let mut index = string.len() - 1;
let mut is_negative = false;
if num < 0 {
is_negative = true;
num = match num.checked_abs() {
Some(value) => value,
None => {
let value = <$type_name>::max_value();
string[index] = LOOKUP[((value % base + 1) % base) as usize];
index -= 1;
value / base + ((value % base == base - 1) as $type_name)
}
};
} else if num == 0 {
string[index] = b'0';
return string.split_at(index).1;
}
if base == 10 {
base_10!(num, index, string);
} else {
while num != 0 {
let rem = num % base;
string[index] = LOOKUP[rem as usize];
index = index.wrapping_sub(1);
num /= base;
}
}
if is_negative {
string[index] = b'-';
index = index.wrapping_sub(1);
}
string.split_at(index.wrapping_add(1)).1
}
pub const fn $str_function_name(num: $type_name, base: $type_name, string: &mut [u8]) -> &str {
unsafe { core::str::from_utf8_unchecked($core_function_name(num, base, string)) }
}
impl NumToA for $type_name {
fn numtoa(self, base: $type_name, string: &mut [u8]) -> &[u8] {
$core_function_name(self, base, string)
}
fn numtoa_str(self, base: $type_name, buf: &mut [u8]) -> &str {
$str_function_name(self, base, buf)
}
}
}
}
impl_signed_numtoa_for!(i16,numtoa_i16,numtoa_i16_str);
impl_signed_numtoa_for!(i32,numtoa_i32,numtoa_i32_str);
impl_signed_numtoa_for!(i64,numtoa_i64,numtoa_i64_str);
impl_signed_numtoa_for!(i128,numtoa_i128,numtoa_i128_str);
impl_signed_numtoa_for!(isize,numtoa_isize,numtoa_isize_str);
impl_unsigned_numtoa_for!(u16,numtoa_u16,numtoa_u16_str);
impl_unsigned_numtoa_for!(u32,numtoa_u32,numtoa_u32_str);
impl_unsigned_numtoa_for!(u64,numtoa_u64,numtoa_u64_str);
impl_unsigned_numtoa_for!(u128,numtoa_u128,numtoa_u128_str);
impl_unsigned_numtoa_for!(usize,numtoa_usize,numtoa_usize_str);
pub const fn numtoa_i8(mut num: i8, base: i8, string: &mut [u8]) -> &[u8] {
if cfg!(debug_assertions) {
if base == 10 {
debug_assert!(string.len() >= 4, "i8 conversions need at least 4 bytes");
}
}
let mut index = string.len() - 1;
let mut is_negative = false;
if num < 0 {
is_negative = true;
num = match num.checked_abs() {
Some(value) => value,
None => {
let value = <i8>::max_value();
string[index] = LOOKUP[((value % base + 1) % base) as usize];
index -= 1;
value / base + ((value % base == base - 1) as i8)
}
};
} else if num == 0 {
string[index] = b'0';
return string.split_at(index).1;
}
if base == 10 {
if num > 99 {
let section = (num / 10) * 2;
copy_2_dec_lut_bytes!(string, index-2, section);
string[index] = LOOKUP[(num % 10) as usize];
index = index.wrapping_sub(3);
} else if num > 9 {
let idx = num as usize * 2;
copy_2_dec_lut_bytes!(string, index-1, idx);
index = index.wrapping_sub(2);
} else {
string[index] = LOOKUP[num as usize];
index = index.wrapping_sub(1);
}
} else {
while num != 0 {
let rem = num % base;
string[index] = LOOKUP[rem as usize];
index = index.wrapping_sub(1);
num /= base;
}
}
if is_negative {
string[index] = b'-';
index = index.wrapping_sub(1);
}
string.split_at(index.wrapping_add(1)).1
}
pub const fn numtoa_i8_str(num: i8, base: i8, string: &mut [u8]) -> &str {
unsafe { str::from_utf8_unchecked(numtoa_i8(num, base, string)) }
}
impl NumToA for i8 {
fn numtoa(self, base: i8, string: &mut [u8]) -> &[u8] {
numtoa_i8(self, base, string)
}
fn numtoa_str(self, base: Self, buf: &mut [u8]) -> &str {
numtoa_i8_str(self, base, buf)
}
}
pub const fn numtoa_u8(mut num: u8, base: u8, string: &mut [u8]) -> &[u8] {
if cfg!(debug_assertions) {
if base == 10 {
debug_assert!(string.len() >= 3, "u8 conversions need at least 3 bytes");
}
}
let mut index = string.len() - 1;
if num == 0 {
string[index] = b'0';
return string.split_at(index).1;
}
if base == 10 {
if num > 99 {
let section = (num / 10) * 2;
copy_2_dec_lut_bytes!(string, index-2, section);
string[index] = LOOKUP[(num % 10) as usize];
index = index.wrapping_sub(3);
} else if num > 9 {
num *= 2;
copy_2_dec_lut_bytes!(string, index-1, num);
index = index.wrapping_sub(2);
} else {
string[index] = LOOKUP[num as usize];
index = index.wrapping_sub(1);
}
} else {
while num != 0 {
let rem = num % base;
string[index] = LOOKUP[rem as usize];
index = index.wrapping_sub(1);
num /= base;
}
}
string.split_at(1).1
}
pub const fn numtoa_u8_str(num: u8, base: u8, string: &mut [u8]) -> &str {
unsafe { str::from_utf8_unchecked(numtoa_u8(num, base, string)) }
}
impl NumToA for u8 {
fn numtoa(self, base: u8, string: &mut [u8]) -> &[u8] {
numtoa_u8(self, base, string)
}
fn numtoa_str(self, base: Self, buf: &mut [u8]) -> &str {
numtoa_u8_str(self, base, buf)
}
}
macro_rules! impl_numtoa_base_n_init_for {
(
$type_name:ty,
$base:expr,
$core_function_name:ident,
$base_n_function_name:ident,
$needed_buffer_size:expr) => {
pub const fn $base_n_function_name(num: $type_name) -> AsciiNumber<$needed_buffer_size> {
let mut string = [0_u8; $needed_buffer_size];
let len = $core_function_name(num, $base, &mut string).len();
return AsciiNumber { string, start:$needed_buffer_size-len}
}
};
}
pub mod base10 {
use AsciiNumber;
use numtoa_u8;
use numtoa_u16;
use numtoa_u32;
use numtoa_u64;
use numtoa_u128;
use numtoa_i8;
use numtoa_i16;
use numtoa_i32;
use numtoa_i64;
use numtoa_i128;
impl_numtoa_base_n_init_for!(u8,10,numtoa_u8,u8,3); impl_numtoa_base_n_init_for!(u16,10,numtoa_u16,u16,5); impl_numtoa_base_n_init_for!(u32,10,numtoa_u32,u32,10); impl_numtoa_base_n_init_for!(u64,10,numtoa_u64,u64,20); impl_numtoa_base_n_init_for!(u128,10,numtoa_u128,u128,39); impl_numtoa_base_n_init_for!(i8,10,numtoa_i8,i8,4); impl_numtoa_base_n_init_for!(i16,10,numtoa_i16,i16,6); impl_numtoa_base_n_init_for!(i32,10,numtoa_i32,i32,11); impl_numtoa_base_n_init_for!(i64,10,numtoa_i64,i64,20); impl_numtoa_base_n_init_for!(i128,10,numtoa_i128,i128,40);
}
pub mod base16 {
use AsciiNumber;
use numtoa_u8;
use numtoa_u16;
use numtoa_u32;
use numtoa_u64;
use numtoa_u128;
use numtoa_i8;
use numtoa_i16;
use numtoa_i32;
use numtoa_i64;
use numtoa_i128;
impl_numtoa_base_n_init_for!(u8,16,numtoa_u8,u8,2);
impl_numtoa_base_n_init_for!(u16,16,numtoa_u16,u16,4);
impl_numtoa_base_n_init_for!(u32,16,numtoa_u32,u32,8);
impl_numtoa_base_n_init_for!(u64,16,numtoa_u64,u64,16);
impl_numtoa_base_n_init_for!(u128,16,numtoa_u128,u128,32);
impl_numtoa_base_n_init_for!(i8,16,numtoa_i8,i8,3);
impl_numtoa_base_n_init_for!(i16,16,numtoa_i16,i16,5);
impl_numtoa_base_n_init_for!(i32,16,numtoa_i32,i32,9);
impl_numtoa_base_n_init_for!(i64,16,numtoa_i64,i64,17);
impl_numtoa_base_n_init_for!(i128,16,numtoa_i128,i128,33);
}
#[test]
fn str_convenience_core() {
assert_eq!("256123", numtoa_i32_str(256123_i32, 10, &mut [0u8; 20]));
}
#[test]
fn str_convenience_trait() {
assert_eq!("256123", 256123.numtoa_str(10, &mut [0u8; 20]));
}
#[test]
fn str_convenience_base10() {
assert_eq!("256123", base10::i32(256123).as_str());
}
#[test]
fn str_convenience_base16() {
assert_eq!("3E87B", base16::i32(256123).as_str());
}
#[test]
#[should_panic]
fn base10_u8_array_too_small_core() {
let _ = numtoa_u8(0_u8, 10, &mut [0u8; 2]);
}
#[test]
#[should_panic]
fn base10_u8_array_too_small_trait() {
let _ = 0u8.numtoa(10, &mut [0u8; 2]);
}
#[test]
fn base10_u8_array_just_right_core() {
let _ = numtoa_u8(0, 10, &mut [0u8; 3]);
}
#[test]
fn base10_u8_array_just_right_trait() {
let _ = 0u8.numtoa(10, &mut [0u8; 3]);
}
#[test]
#[should_panic]
fn base10_i8_array_too_small() {
let mut buffer = [0u8; 3];
let _ = 0i8.numtoa(10, &mut buffer);
}
#[test]
fn base10_i8_array_just_right() {
let mut buffer = [0u8; 4];
assert_eq!((-127i8).numtoa(10, &mut buffer), b"-127");
}
#[test]
#[should_panic]
fn base10_i16_array_too_small() {
let mut buffer = [0u8; 5];
let _ = 0i16.numtoa(10, &mut buffer);
}
#[test]
fn base10_i16_array_just_right() {
let mut buffer = [0u8; 6];
assert_eq!((-12768i16).numtoa(10, &mut buffer), b"-12768");
}
#[test]
#[should_panic]
fn base10_u16_array_too_small() {
let mut buffer = [0u8; 4];
let _ = 0u16.numtoa(10, &mut buffer);
}
#[test]
fn base10_u16_array_just_right() {
let mut buffer = [0u8; 5];
let _ = 0u16.numtoa(10, &mut buffer);
}
#[test]
#[should_panic]
fn base10_i32_array_too_small() {
let mut buffer = [0u8; 10];
let _ = 0i32.numtoa(10, &mut buffer);
}
#[test]
fn base10_i32_array_just_right() {
let mut buffer = [0u8; 11];
let _ = 0i32.numtoa(10, &mut buffer);
}
#[test]
#[should_panic]
fn base10_u32_array_too_small() {
let mut buffer = [0u8; 9];
let _ = 0u32.numtoa(10, &mut buffer);
}
#[test]
fn base10_u32_array_just_right() {
let mut buffer = [0u8; 10];
let _ = 0u32.numtoa(10, &mut buffer);
}
#[test]
#[should_panic]
fn base10_i64_array_too_small() {
let mut buffer = [0u8; 18];
let _ = 0i64.numtoa(10, &mut buffer);
}
#[test]
fn base10_i64_array_just_right() {
let mut buffer = [0u8; 19];
let _ = 0i64.numtoa(10, &mut buffer);
}
#[test]
#[should_panic]
fn base10_u64_array_too_small() {
let mut buffer = [0u8; 19];
let _ = 0u64.numtoa(10, &mut buffer);
}
#[test]
fn base10_u64_array_just_right() {
let mut buffer = [0u8; 20];
let _ = 0u64.numtoa(10, &mut buffer);
}
#[test]
fn base10_i8_all_core() {
for i in i8::MIN..i8::MAX {
let _ = numtoa_i8(i, 10, &mut [0u8; 4]);
}
}
#[test]
fn base10_i8_all_trait() {
for i in i8::MIN..i8::MAX {
let _ = i.numtoa(10, &mut [0u8; 4]);
}
}
#[test]
fn base10_i8_all_base10() {
for i in i8::MIN..i8::MAX {
let _ = base10::i8(i);
}
}
#[test]
fn base16_i8_all_core() {
for i in i8::MIN..i8::MAX {
let _ = numtoa_i8(i, 16, &mut [0u8; 3]);
}
}
#[test]
fn base16_i8_all_trait() {
for i in i8::MIN..i8::MAX {
let _ = i.numtoa(16, &mut [0u8; 3]);
}
}
#[test]
fn base16_i8_all_base16() {
for i in i8::MIN..i8::MAX {
let _ = base16::i8(i);
}
}
#[test]
fn base10_u8_all_core() {
for i in u8::MIN..u8::MAX {
let _ = numtoa_u8(i, 10, &mut [0u8; 3]);
}
}
#[test]
fn base10_u8_all_trait() {
for i in u8::MIN..u8::MAX {
let _ = i.numtoa(10, &mut [0u8; 3]);
}
}
#[test]
fn base10_u8_all_base10() {
for i in u8::MIN..u8::MAX {
let _ = base10::u8(i);
}
}
#[test]
fn base16_u8_all_core() {
for i in u8::MIN..u8::MAX {
let _ = numtoa_u8(i, 16, &mut [0u8; 3]);
}
}
#[test]
fn base16_u8_all_trait() {
for i in u8::MIN..u8::MAX {
let _ = i.numtoa(16, &mut [0u8; 3]);
}
}
#[test]
fn base16_u8_all_base16() {
for i in u8::MIN..u8::MAX {
let _ = base16::u8(i);
}
}
#[test]
#[should_panic]
fn base10_i128_array_too_small() {
let mut buffer = [0u8; 38];
let _ = 0i128.numtoa(10, &mut buffer);
}
#[test]
fn base10_i128_array_just_right() {
let mut buffer = [0u8; 39];
let _ = 0i128.numtoa(10, &mut buffer);
}
#[test]
#[should_panic]
fn base10_u128_array_too_small() {
let mut buffer = [0u8; 38];
let _ = 0u128.numtoa(10, &mut buffer);
}
#[test]
fn base10_u128_array_just_right() {
let mut buffer = [0u8; 39];
let _ = 0u128.numtoa(10, &mut buffer);
}
#[test]
fn base8_min_signed_number() {
let mut buffer = [0u8; 50];
assert_eq!((-128i8).numtoa(8, &mut buffer), b"-200");
assert_eq!((-32768i16).numtoa(8, &mut buffer), b"-100000");
assert_eq!((-2147483648i32).numtoa(8, &mut buffer), b"-20000000000");
assert_eq!((-9223372036854775808i64).numtoa(8, &mut buffer), b"-1000000000000000000000");
assert_eq!((i128::MIN).numtoa(8, &mut buffer), b"-2000000000000000000000000000000000000000000");
}
#[test]
fn base16_min_signed_number() {
let mut buffer = [0u8; 40];
assert_eq!((-128i8).numtoa(16, &mut buffer), b"-80");
assert_eq!((-32768i16).numtoa(16, &mut buffer), b"-8000");
assert_eq!((-2147483648i32).numtoa(16, &mut buffer), b"-80000000");
assert_eq!((-9223372036854775808i64).numtoa(16, &mut buffer), b"-8000000000000000");
assert_eq!((i128::MIN).numtoa(16, &mut buffer), b"-80000000000000000000000000000000");
}