90 lines
2.1 KiB
Rust
90 lines
2.1 KiB
Rust
use std::fmt::Display;
|
|
|
|
#[cfg(debug_assertions)]
|
|
use checked_int_cast::CheckedIntCast;
|
|
|
|
pub trait Truncate {
|
|
fn truncate_as_u8(self) -> u8;
|
|
}
|
|
|
|
impl Truncate for u16 {
|
|
#[allow(clippy::cast_possible_truncation)]
|
|
fn truncate_as_u8(self) -> u8 {
|
|
(self & 0xff) as u8
|
|
}
|
|
}
|
|
|
|
pub trait As {
|
|
fn as_u16(self) -> u16;
|
|
fn as_i16(self) -> i16;
|
|
fn as_u32(self) -> u32;
|
|
fn as_usize(self) -> usize;
|
|
fn as_isize(self) -> isize;
|
|
}
|
|
|
|
trait ExpectOrOverflow {
|
|
type Output;
|
|
fn expect_or_overflow<D: Display>(self, value: D, ty: &str) -> Self::Output;
|
|
}
|
|
|
|
impl<T> ExpectOrOverflow for Option<T> {
|
|
type Output = T;
|
|
fn expect_or_overflow<D: Display>(self, value: D, ty: &str) -> Self::Output {
|
|
match self {
|
|
Some(v) => v,
|
|
None => panic!("{} overflows {}", value, ty),
|
|
}
|
|
}
|
|
}
|
|
|
|
macro_rules! impl_as {
|
|
($ty:ty) => {
|
|
#[cfg(debug_assertions)]
|
|
impl As for $ty {
|
|
fn as_u16(self) -> u16 {
|
|
self.as_u16_checked().expect_or_overflow(self, "u16")
|
|
}
|
|
|
|
fn as_i16(self) -> i16 {
|
|
self.as_i16_checked().expect_or_overflow(self, "i16")
|
|
}
|
|
|
|
fn as_u32(self) -> u32 {
|
|
self.as_u32_checked().expect_or_overflow(self, "u32")
|
|
}
|
|
|
|
fn as_usize(self) -> usize {
|
|
self.as_usize_checked().expect_or_overflow(self, "usize")
|
|
}
|
|
|
|
fn as_isize(self) -> isize {
|
|
self.as_isize_checked().expect_or_overflow(self, "usize")
|
|
}
|
|
}
|
|
|
|
#[cfg(not(debug_assertions))]
|
|
impl As for $ty {
|
|
fn as_u16(self) -> u16 {
|
|
self as u16
|
|
}
|
|
fn as_i16(self) -> i16 {
|
|
self as i16
|
|
}
|
|
fn as_u32(self) -> u32 {
|
|
self as u32
|
|
}
|
|
fn as_usize(self) -> usize {
|
|
self as usize
|
|
}
|
|
fn as_isize(self) -> isize {
|
|
self as isize
|
|
}
|
|
}
|
|
};
|
|
}
|
|
|
|
impl_as!(i16);
|
|
impl_as!(u32);
|
|
impl_as!(usize);
|
|
impl_as!(isize);
|