ashv2 9.0.1

Implementation of the Asynchronous Serial Host (ASH) protocol.
Documentation
//! Data frame header.

use bitflags::bitflags;

use crate::seq::Seq;

/// Data frame header.
#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash)]
pub struct Header(u8);

bitflags! {
    impl Header: u8 {
        /// The frame number mask.
        const FRAME_NUM = 0b0111_0000;

        /// The retransmit flag.
        const RETRANSMIT = 0b0000_1000;

        /// The acknowledgement number mask.
        const ACK_NUM = 0b0000_0111;
    }
}

impl Header {
    /// Creates a new data header.
    #[must_use]
    pub const fn new(frame_num: Seq, retransmit: bool, ack_num: Seq) -> Self {
        let mut raw = Self::FRAME_NUM.bits()
            & (frame_num.as_u8() << Self::FRAME_NUM.bits().trailing_zeros())
            | Self::ACK_NUM.bits() & ack_num.as_u8();

        if retransmit {
            raw |= Self::RETRANSMIT.bits();
        }

        Self(raw)
    }

    /// Returns the frame number.
    #[must_use]
    pub fn frame_num(self) -> Seq {
        Seq::try_from(
            (self.bits() & Self::FRAME_NUM.bits()) >> Self::FRAME_NUM.bits().trailing_zeros(),
        )
        .expect("Seq always fits.")
    }

    /// Returns the ACK number.
    #[must_use]
    pub fn ack_num(self) -> Seq {
        Seq::try_from(self.bits() & Self::ACK_NUM.bits()).expect("Seq always fits.")
    }
}

#[cfg(test)]
mod tests {
    use super::Header;

    #[test]
    fn test_offset() {
        assert_eq!(Header::FRAME_NUM.bits().trailing_zeros(), 4);
    }
}