2024-08-26 21:58:19 +00:00
|
|
|
use bnum::BInt;
|
2024-08-27 00:28:33 +00:00
|
|
|
use typenum::Unsigned;
|
2024-08-26 21:58:19 +00:00
|
|
|
use std::marker::PhantomData;
|
2024-08-23 20:42:44 +00:00
|
|
|
|
2024-08-26 23:03:02 +00:00
|
|
|
#[derive(Clone,Copy,Debug,Hash)]
|
2024-08-26 21:58:19 +00:00
|
|
|
pub struct Fixed<const CHUNKS:usize,Frac>{
|
2024-08-27 00:28:33 +00:00
|
|
|
pub(crate)bits:BInt<{CHUNKS}>,
|
|
|
|
pub(crate)frac:PhantomData<Frac>,
|
2024-08-23 20:00:22 +00:00
|
|
|
}
|
2024-08-23 20:42:44 +00:00
|
|
|
|
2024-08-26 22:40:27 +00:00
|
|
|
impl<const CHUNKS:usize,FracDst:Unsigned,T> From<T> for Fixed<CHUNKS,FracDst>
|
|
|
|
where
|
|
|
|
BInt<CHUNKS>:From<T>
|
|
|
|
{
|
|
|
|
fn from(value:T)->Self{
|
2024-08-23 21:17:48 +00:00
|
|
|
Self{
|
2024-08-26 21:58:19 +00:00
|
|
|
bits:BInt::<{CHUNKS}>::from(value)<<FracDst::U32,
|
|
|
|
frac:PhantomData,
|
2024-08-23 21:17:48 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2024-08-26 21:58:19 +00:00
|
|
|
impl<const CHUNKS:usize,Frac> PartialEq for Fixed<CHUNKS,Frac>{
|
2024-08-23 21:17:48 +00:00
|
|
|
fn eq(&self,other:&Self)->bool{
|
2024-08-23 23:53:54 +00:00
|
|
|
self.bits.eq(&other.bits)
|
2024-08-23 21:17:48 +00:00
|
|
|
}
|
|
|
|
}
|
2024-08-26 21:58:19 +00:00
|
|
|
impl<const CHUNKS:usize,Frac> Eq for Fixed<CHUNKS,Frac>{}
|
2024-08-23 21:17:48 +00:00
|
|
|
|
2024-08-26 23:03:02 +00:00
|
|
|
impl<const CHUNKS:usize,Frac> std::ops::Neg for Fixed<CHUNKS,Frac>{
|
|
|
|
type Output=Self;
|
|
|
|
fn neg(self)->Self{
|
|
|
|
Self{
|
|
|
|
bits:self.bits.neg(),
|
|
|
|
frac:PhantomData,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2024-08-26 22:40:27 +00:00
|
|
|
macro_rules! impl_operator {
|
2024-08-26 22:57:02 +00:00
|
|
|
( $struct: ident, $trait: ident, $method: ident, $output: ty ) => {
|
|
|
|
impl<const CHUNKS:usize,Frac> core::ops::$trait<$struct<CHUNKS,Frac>> for $struct<CHUNKS,Frac>{
|
2024-08-26 22:40:27 +00:00
|
|
|
type Output = $output;
|
|
|
|
|
2024-08-26 22:57:02 +00:00
|
|
|
fn $method(self, other: Self) -> Self::Output {
|
2024-08-26 22:40:27 +00:00
|
|
|
Self {
|
2024-08-26 22:57:02 +00:00
|
|
|
bits:self.bits.$method(other.bits),
|
2024-08-26 22:40:27 +00:00
|
|
|
frac:PhantomData,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
};
|
2024-08-23 23:48:24 +00:00
|
|
|
}
|
2024-08-26 22:40:27 +00:00
|
|
|
macro_rules! impl_assign_operator {
|
|
|
|
( $struct: ident, $trait: ident, $method: ident ) => {
|
2024-08-26 22:57:02 +00:00
|
|
|
impl<const CHUNKS:usize,Frac> core::ops::$trait<$struct<CHUNKS,Frac>> for $struct<CHUNKS,Frac>{
|
|
|
|
fn $method(&mut self, other: Self) {
|
|
|
|
self.bits.$method(other.bits);
|
2024-08-26 22:40:27 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
// Impl arithmetic pperators
|
|
|
|
impl_assign_operator!( Fixed, AddAssign, add_assign );
|
|
|
|
impl_operator!( Fixed, Add, add, Self );
|
|
|
|
impl_assign_operator!( Fixed, SubAssign, sub_assign );
|
|
|
|
impl_operator!( Fixed, Sub, sub, Self );
|
|
|
|
impl_assign_operator!( Fixed, MulAssign, mul_assign );
|
|
|
|
impl_operator!( Fixed, Mul, mul, Self );
|
|
|
|
impl_assign_operator!( Fixed, DivAssign, div_assign );
|
|
|
|
impl_operator!( Fixed, Div, div, Self );
|
|
|
|
impl_assign_operator!( Fixed, RemAssign, rem_assign );
|
|
|
|
impl_operator!( Fixed, Rem, rem, Self );
|
|
|
|
|
|
|
|
// Impl bitwise operators
|
|
|
|
impl_assign_operator!( Fixed, BitAndAssign, bitand_assign );
|
|
|
|
impl_operator!( Fixed, BitAnd, bitand, Self );
|
|
|
|
impl_assign_operator!( Fixed, BitOrAssign, bitor_assign );
|
|
|
|
impl_operator!( Fixed, BitOr, bitor, Self );
|
|
|
|
impl_assign_operator!( Fixed, BitXorAssign, bitxor_assign );
|
|
|
|
impl_operator!( Fixed, BitXor, bitxor, Self );
|