use crate::lib_bitfield;
use crate::result::{Error, Result};
mod cd;
pub use cd::*;
lib_bitfield! {
pub Argument(u32): u8 {
raw_set_cd: 0;
}
}
impl Argument {
pub const LEN: usize = 4;
pub const DEFAULT: u32 = 0;
pub const fn new() -> Self {
Self(0)
}
pub const fn set_cd(&self) -> CardDetect {
CardDetect::from_bool(self.raw_set_cd())
}
pub fn set_set_cd(&mut self, val: CardDetect) {
self.set_raw_set_cd(val.into_bool());
}
pub const fn try_from_bits(val: u32) -> Result<Self> {
Ok(Self(val))
}
pub const fn bytes(&self) -> [u8; Self::LEN] {
self.0.to_be_bytes()
}
pub const fn try_from_bytes(val: &[u8]) -> Result<Self> {
match val.len() {
len if len < Self::LEN => Err(Error::invalid_length(len, Self::LEN)),
_ => Self::try_from_bits(u32::from_be_bytes([val[0], val[1], val[2], val[3]])),
}
}
}
impl Default for Argument {
fn default() -> Self {
Self::new()
}
}
impl TryFrom<u32> for Argument {
type Error = Error;
fn try_from(val: u32) -> Result<Self> {
Self::try_from_bits(val)
}
}
impl From<Argument> for u32 {
fn from(val: Argument) -> Self {
val.bits()
}
}
impl From<Argument> for [u8; Argument::LEN] {
fn from(val: Argument) -> Self {
val.bytes()
}
}
impl TryFrom<&[u8]> for Argument {
type Error = Error;
fn try_from(val: &[u8]) -> Result<Self> {
Self::try_from_bytes(val)
}
}
impl<const N: usize> TryFrom<&[u8; N]> for Argument {
type Error = Error;
fn try_from(val: &[u8; N]) -> Result<Self> {
Self::try_from_bytes(val.as_ref())
}
}
impl<const N: usize> TryFrom<[u8; N]> for Argument {
type Error = Error;
fn try_from(val: [u8; N]) -> Result<Self> {
Self::try_from_bytes(val.as_ref())
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_fields() {
let new_set_cd = CardDetect::new();
(1..=u32::BITS)
.map(|r| ((1u64 << r) - 1) as u32)
.for_each(|raw_arg| {
let raw = raw_arg.to_be_bytes();
match Argument::try_from_bits(raw_arg) {
Ok(mut exp_arg) => {
let exp_set_cd = CardDetect::from_bool((raw_arg & 0b1) != 0);
assert_eq!(Argument::try_from_bits(raw_arg), Ok(exp_arg));
assert_eq!(Argument::try_from_bytes(&raw), Ok(exp_arg));
assert_eq!(Argument::try_from(raw_arg), Ok(exp_arg));
assert_eq!(Argument::try_from(raw), Ok(exp_arg));
assert_eq!(Argument::try_from(&raw), Ok(exp_arg));
assert_eq!(exp_arg.set_cd(), exp_set_cd);
exp_arg.set_set_cd(new_set_cd);
assert_eq!(exp_arg.set_cd(), new_set_cd);
exp_arg.set_set_cd(exp_set_cd);
assert_eq!(exp_arg.set_cd(), exp_set_cd);
}
Err(err) => {
let exp_err = Error::invalid_field_variant(
"command::argument::set_cd",
(raw_arg & 0b11) as usize,
);
assert_eq!(err, exp_err);
assert_eq!(Argument::try_from_bytes(&raw), Err(exp_err));
assert_eq!(Argument::try_from(raw_arg), Err(exp_err));
assert_eq!(Argument::try_from(raw), Err(exp_err));
assert_eq!(Argument::try_from(&raw), Err(exp_err));
}
}
});
}
}