proc_bitfield/traits/
int_impls.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
use super::{Bit, Bits, SetBit, SetBits, WithBit, WithBits};

macro_rules! impl_bits_for_int_type {
    ($storage: ty, $value: ty) => {
        impl Bits<$value> for $storage {
            #[inline]
            fn bits<const START: usize, const END: usize>(&self) -> $value {
                const VALUE_BITS: usize = <$value>::BITS as usize;
                let read_bits = END - START;
                ((*self >> START) as $value) << (VALUE_BITS - read_bits) >> (VALUE_BITS - read_bits)
            }
        }

        impl WithBits<$value> for $storage {
            #[inline]
            fn with_bits<const START: usize, const END: usize>(self, value: $value) -> Self {
                let written_bits = END - START;
                let mask = ((1 as $storage) << (written_bits - 1) << 1).wrapping_sub(1) << START;
                (self & !mask) | ((value as $storage) << START & mask)
            }
        }

        impl SetBits<$value> for $storage {
            #[inline]
            fn set_bits<const START: usize, const END: usize>(&mut self, value: $value) {
                *self = self.with_bits::<START, END>(value);
            }
        }
    };
}

macro_rules! impl_bits_for_int_types {
    (=> $($dst_ty: ty),*) => {};
    ($src_ty: ty $(, $other_src_ty: ty)* => $($dst_ty: ty),*) => {
        $(
            impl_bits_for_int_type!($src_ty, $dst_ty);
        )*
        impl_bits_for_int_types!($($other_src_ty),* => $($dst_ty),*);
    };
}

impl_bits_for_int_types!(
    u8, u16, u32, u64, u128, usize, i8, i16, i32, i64, i128, isize
        => u8, u16, u32, u64, u128, usize, i8, i16, i32, i64, i128, isize
);

macro_rules! impl_bit_for_int_type {
    ($t: ty) => {
        impl Bit for $t {
            #[inline]
            fn bit<const BIT: usize>(&self) -> bool {
                *self & 1 << BIT != 0
            }
        }

        impl WithBit for $t {
            #[inline]
            fn with_bit<const BIT: usize>(self, value: bool) -> Self {
                (self & !(1 << BIT)) | (value as $t) << BIT
            }
        }

        impl SetBit for $t {
            #[inline]
            fn set_bit<const BIT: usize>(&mut self, value: bool) {
                *self = self.with_bit::<BIT>(value);
            }
        }
    };
}

macro_rules! impl_bit_for_int_types {
    ($($t: ty),*) => {
        $(impl_bit_for_int_type!($t);)*
    };
}

impl_bit_for_int_types!(u8, u16, u32, u64, u128, usize, i8, i16, i32, i64, i128, isize);