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

use core::fmt::Debug;
use core::mem::size_of;
use core::nonzero::Zeroable;
use core::ops::{BitAnd, BitOr, BitXor, Not, Shl, Shr, Sub};

/// Raw register value type.
pub trait RegRaw
  : Sized
  + Zeroable
  + Debug
  + Copy
  + PartialOrd
  + Not<Output = Self>
  + Sub<Output = Self>
  + BitOr<Output = Self>
  + BitXor<Output = Self>
  + BitAnd<Output = Self>
  + Shl<Self, Output = Self>
  + Shr<Self, Output = Self> {
  /// Creates a `RegRaw` from `usize`
  fn from_usize(raw: usize) -> Self;

  /// Size of the type in bits.
  fn size() -> Self;

  /// Returns one.
  fn one() -> Self;
}

macro_rules! reg_raw {
  ($type:ty) => {
    impl RegRaw for $type {
      #[inline(always)]
      fn from_usize(raw: usize) -> Self {
        raw as $type
      }

      #[inline(always)]
      fn size() -> $type {
        size_of::<$type>() as $type * 8
      }

      #[inline(always)]
      fn one() -> $type {
        1
      }
    }
  }
}

reg_raw!(u64);
reg_raw!(u32);
reg_raw!(u16);
reg_raw!(u8);