|
| 1 | +// For some bitfield members, working with them as enums will make more sense to |
| 2 | +// the user than working with them as integers. We will require enums that have |
| 3 | +// a power-of-two number of variants so that they exhaustively cover a fixed |
| 4 | +// range of bits. |
| 5 | +// |
| 6 | +// // Works like B3, but getter and setter signatures will use |
| 7 | +// // the enum instead of u8. |
| 8 | +// #[derive(BitfieldSpecifier)] |
| 9 | +// enum DeliveryMode { |
| 10 | +// Fixed = 0b000, |
| 11 | +// Lowest = 0b001, |
| 12 | +// SMI = 0b010, |
| 13 | +// RemoteRead = 0b011, |
| 14 | +// NMI = 0b100, |
| 15 | +// Init = 0b101, |
| 16 | +// Startup = 0b110, |
| 17 | +// External = 0b111, |
| 18 | +// } |
| 19 | +// |
| 20 | +// For this test case it is okay to require that every enum variant has an |
| 21 | +// explicit discriminant that is an integer literal. We will relax this |
| 22 | +// requirement in a later test case. |
| 23 | +// |
| 24 | +// Optionally if you are interested, come up with a way to support enums with a |
| 25 | +// number of variants that is not a power of two, but this is not necessary for |
| 26 | +// the test suite. Maybe there could be a #[bits = N] attribute that determines |
| 27 | +// the bit width of the specifier, and the getter (only for such enums) would |
| 28 | +// return Result<T, Unrecognized> with the raw value accessible through the |
| 29 | +// error type as u64: |
| 30 | +// |
| 31 | +// #[derive(BitfieldSpecifier)] |
| 32 | +// #[bits = 4] |
| 33 | +// enum SmallPrime { |
| 34 | +// Two = 0b0010, |
| 35 | +// Three = 0b0011, |
| 36 | +// Five = 0b0101, |
| 37 | +// Seven = 0b0111, |
| 38 | +// Eleven = 0b1011, |
| 39 | +// Thirteen = 0b1101, |
| 40 | +// } |
| 41 | +// |
| 42 | +// ... |
| 43 | +// let mut bitfield = MyBitfield::new(); |
| 44 | +// assert_eq!(0, bitfield.small_prime().unwrap_err().raw_value()); |
| 45 | +// |
| 46 | +// bitfield.set_small_prime(SmallPrime::Seven); |
| 47 | +// let p = bitfield.small_prime().unwrap_or(SmallPrime::Two); |
| 48 | + |
| 49 | +use bitfield::*; |
| 50 | + |
| 51 | +#[bitfield] |
| 52 | +pub struct RedirectionTableEntry { |
| 53 | + acknowledged: bool, |
| 54 | + trigger_mode: TriggerMode, |
| 55 | + delivery_mode: DeliveryMode, |
| 56 | + reserved: B3, |
| 57 | +} |
| 58 | + |
| 59 | +#[derive(BitfieldSpecifier, Debug, PartialEq)] |
| 60 | +pub enum TriggerMode { |
| 61 | + Edge = 0, |
| 62 | + Level = 1, |
| 63 | +} |
| 64 | + |
| 65 | +#[derive(BitfieldSpecifier, Debug, PartialEq)] |
| 66 | +pub enum DeliveryMode { |
| 67 | + Fixed = 0b000, |
| 68 | + Lowest = 0b001, |
| 69 | + SMI = 0b010, |
| 70 | + RemoteRead = 0b011, |
| 71 | + NMI = 0b100, |
| 72 | + Init = 0b101, |
| 73 | + Startup = 0b110, |
| 74 | + External = 0b111, |
| 75 | +} |
| 76 | + |
| 77 | +fn main() { |
| 78 | + assert_eq!(std::mem::size_of::<RedirectionTableEntry>(), 1); |
| 79 | + |
| 80 | + // Initialized to all 0 bits. |
| 81 | + let mut entry = RedirectionTableEntry::new(); |
| 82 | + assert_eq!(entry.get_acknowledged(), false); |
| 83 | + assert_eq!(entry.get_trigger_mode(), TriggerMode::Edge); |
| 84 | + assert_eq!(entry.get_delivery_mode(), DeliveryMode::Fixed); |
| 85 | + |
| 86 | + entry.set_acknowledged(true); |
| 87 | + entry.set_delivery_mode(DeliveryMode::SMI); |
| 88 | + assert_eq!(entry.get_acknowledged(), true); |
| 89 | + assert_eq!(entry.get_trigger_mode(), TriggerMode::Edge); |
| 90 | + assert_eq!(entry.get_delivery_mode(), DeliveryMode::SMI); |
| 91 | +} |
0 commit comments