use core::ptr::write;
use crate::shared::U8_MASK_BOT_6;
use super::alphabet::{
BITS, WIDTH_1, WIDTH_10, WIDTH_2, WIDTH_3, WIDTH_4, WIDTH_5, WIDTH_6, WIDTH_7, WIDTH_8, WIDTH_9,
};
#[inline]
pub const fn capacity_encode_u64(n: u64) -> usize {
if let Some(log) = n.checked_ilog2() {
1 + (log / 6) as usize
} else {
1
}
}
pub fn encode_u64(enc: &'static [u8; BITS], n: u64) -> String {
let cap = capacity_encode_u64(n);
let mut b = Vec::<u8>::with_capacity(cap);
encode_u64_inner(enc, n, &mut b, cap, 0, cap);
unsafe { String::from_utf8_unchecked(b) }
}
pub fn encode_u64_into(enc: &'static [u8; BITS], n: u64, b: &mut Vec<u8>) {
let cap = capacity_encode_u64(n);
let len = b.len();
let new_len = len + cap;
assert!(b.capacity() >= new_len, "Missing capacity for encoding");
encode_u64_inner(enc, n, b, cap, len, new_len);
}
#[rustfmt::skip]
#[inline(always)]
fn encode_u64_inner(enc: &'static [u8; BITS], n: u64, b: &mut Vec<u8>, cap: usize, len: usize, new_len: usize) {
match cap {
11 => unsafe {
let end = b.as_mut_ptr().add(len);
write(end , enc[((n >> WIDTH_10) as u8 ) as usize]);
write(end.add( 1), enc[((n >> WIDTH_9) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 2), enc[((n >> WIDTH_8) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 3), enc[((n >> WIDTH_7) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 4), enc[((n >> WIDTH_6) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 5), enc[((n >> WIDTH_5) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 6), enc[((n >> WIDTH_4) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 7), enc[((n >> WIDTH_3) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 8), enc[((n >> WIDTH_2) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 9), enc[((n >> WIDTH_1) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add(10), enc[( n as u8 & U8_MASK_BOT_6) as usize]);
b.set_len(new_len);
}
10 => unsafe {
let end = b.as_mut_ptr().add(len);
write(end , enc[((n >> WIDTH_9) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 1), enc[((n >> WIDTH_8) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 2), enc[((n >> WIDTH_7) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 3), enc[((n >> WIDTH_6) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 4), enc[((n >> WIDTH_5) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 5), enc[((n >> WIDTH_4) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 6), enc[((n >> WIDTH_3) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 7), enc[((n >> WIDTH_2) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 8), enc[((n >> WIDTH_1) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 9), enc[( n as u8 & U8_MASK_BOT_6) as usize]);
b.set_len(new_len);
}
9 => unsafe {
let end = b.as_mut_ptr().add(len);
write(end , enc[((n >> WIDTH_8) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 1), enc[((n >> WIDTH_7) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 2), enc[((n >> WIDTH_6) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 3), enc[((n >> WIDTH_5) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 4), enc[((n >> WIDTH_4) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 5), enc[((n >> WIDTH_3) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 6), enc[((n >> WIDTH_2) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 7), enc[((n >> WIDTH_1) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 8), enc[( n as u8 & U8_MASK_BOT_6) as usize]);
b.set_len(new_len);
}
8 => unsafe {
let end = b.as_mut_ptr().add(len);
write(end , enc[((n >> WIDTH_7) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 1), enc[((n >> WIDTH_6) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 2), enc[((n >> WIDTH_5) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 3), enc[((n >> WIDTH_4) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 4), enc[((n >> WIDTH_3) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 5), enc[((n >> WIDTH_2) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 6), enc[((n >> WIDTH_1) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 7), enc[( n as u8 & U8_MASK_BOT_6) as usize]);
b.set_len(new_len);
}
7 => unsafe {
let end = b.as_mut_ptr().add(len);
write(end , enc[((n >> WIDTH_6) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 1), enc[((n >> WIDTH_5) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 2), enc[((n >> WIDTH_4) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 3), enc[((n >> WIDTH_3) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 4), enc[((n >> WIDTH_2) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 5), enc[((n >> WIDTH_1) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 6), enc[( n as u8 & U8_MASK_BOT_6) as usize]);
b.set_len(new_len);
}
6 => unsafe {
let end = b.as_mut_ptr().add(len);
write(end , enc[((n >> WIDTH_5) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 1), enc[((n >> WIDTH_4) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 2), enc[((n >> WIDTH_3) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 3), enc[((n >> WIDTH_2) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 4), enc[((n >> WIDTH_1) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 5), enc[( n as u8 & U8_MASK_BOT_6) as usize]);
b.set_len(new_len);
}
5 => unsafe {
let end = b.as_mut_ptr().add(len);
write(end , enc[((n >> WIDTH_4) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 1), enc[((n >> WIDTH_3) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 2), enc[((n >> WIDTH_2) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 3), enc[((n >> WIDTH_1) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 4), enc[( n as u8 & U8_MASK_BOT_6) as usize]);
b.set_len(new_len);
}
4 => unsafe {
let end = b.as_mut_ptr().add(len);
write(end , enc[((n >> WIDTH_3) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 1), enc[((n >> WIDTH_2) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 2), enc[((n >> WIDTH_1) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 3), enc[( n as u8 & U8_MASK_BOT_6) as usize]);
b.set_len(new_len);
}
3 => unsafe {
let end = b.as_mut_ptr().add(len);
write(end , enc[((n >> WIDTH_2) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 1), enc[((n >> WIDTH_1) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 2), enc[( n as u8 & U8_MASK_BOT_6) as usize]);
b.set_len(new_len);
}
2 => unsafe {
let end = b.as_mut_ptr().add(len);
write(end , enc[((n >> WIDTH_1) as u8 & U8_MASK_BOT_6) as usize]);
write(end.add( 1), enc[( n as u8 & U8_MASK_BOT_6) as usize]);
b.set_len(new_len);
}
1 => unsafe {
let end = b.as_mut_ptr().add(len);
write(end , enc[( n as u8 & U8_MASK_BOT_6) as usize]);
b.set_len(new_len);
}
_ => panic!("Inconceivable! Impossible length for u64")
}
}