lilliput-core 0.1.0

use num_traits::{Signed, Unsigned};

use crate::{
    binary::bits_if,
    error::Result,
    header::{CompactIntHeader, ExtendedIntHeader, IntHeader},
    io::Write,
    num::WithPackedBeBytes,
    value::{IntValue, SignedIntValue, UnsignedIntValue},
};

use super::Encoder;

impl<W> Encoder<W>
where
    W: Write,
{
    // MARK: - Value

    /// Encodes a 8-bit signed integer value.
    pub fn encode_i8(&mut self, value: i8) -> Result<()> {
        self.encode_signed_int(value)
    }

    /// Encodes a 16-bit signed integer value.
    pub fn encode_i16(&mut self, value: i16) -> Result<()> {
        self.encode_signed_int(value)
    }

    /// Encodes a 32-bit signed integer value.
    pub fn encode_i32(&mut self, value: i32) -> Result<()> {
        self.encode_signed_int(value)
    }

    /// Encodes a 64-bit signed integer value.
    pub fn encode_i64(&mut self, value: i64) -> Result<()> {
        self.encode_signed_int(value)
    }

    /// Encodes a 8-bit unsigned integer value.
    pub fn encode_u8(&mut self, value: u8) -> Result<()> {
        self.encode_unsigned_int(value)
    }

    /// Encodes a 16-bit unsigned integer value.
    pub fn encode_u16(&mut self, value: u16) -> Result<()> {
        self.encode_unsigned_int(value)
    }

    /// Encodes a 32-bit unsigned integer value.
    pub fn encode_u32(&mut self, value: u32) -> Result<()> {
        self.encode_unsigned_int(value)
    }

    /// Encodes a 64-bit unsigned integer value.
    pub fn encode_u64(&mut self, value: u64) -> Result<()> {
        self.encode_unsigned_int(value)
    }

    /// Encodes a signed integer value, from a `SignedIntValue`.
    pub fn encode_signed_int_value(&mut self, value: &SignedIntValue) -> Result<()> {
        match value {
            SignedIntValue::I8(value) => self.encode_signed_int(*value),
            SignedIntValue::I16(value) => self.encode_signed_int(*value),
            SignedIntValue::I32(value) => self.encode_signed_int(*value),
            SignedIntValue::I64(value) => self.encode_signed_int(*value),
        }
    }

    /// Encodes an unsigned integer value, from a `UnsignedIntValue`.
    pub fn encode_unsigned_int_value(&mut self, value: &UnsignedIntValue) -> Result<()> {
        match value {
            UnsignedIntValue::U8(value) => self.encode_unsigned_int(*value),
            UnsignedIntValue::U16(value) => self.encode_unsigned_int(*value),
            UnsignedIntValue::U32(value) => self.encode_unsigned_int(*value),
            UnsignedIntValue::U64(value) => self.encode_unsigned_int(*value),
        }
    }

    /// Encodes an integer value, from a `IntValue`.
    pub fn encode_int_value(&mut self, value: &IntValue) -> Result<()> {
        match value {
            IntValue::Signed(value) => self.encode_signed_int_value(value),
            IntValue::Unsigned(value) => self.encode_unsigned_int_value(value),
        }
    }

    // MARK: - Header

    /// Encodes a integer value's header.
    pub fn encode_int_header(&mut self, header: &IntHeader) -> Result<()> {
        let mut byte = IntHeader::TYPE_BITS;

        match header {
            IntHeader::Compact(CompactIntHeader { is_signed, bits }) => {
                byte |= IntHeader::COMPACT_VARIANT_BIT;
                byte |= bits_if(IntHeader::SIGNEDNESS_BIT, *is_signed);
                byte |= bits & IntHeader::COMPACT_VALUE_BITS;

                #[cfg(feature = "tracing")]
                tracing::debug!(
                    byte = crate::binary::fmt_byte(byte),
                    is_compact = true,
                    is_signed = is_signed,
                    bits = bits
                );
            }
            IntHeader::Extended(ExtendedIntHeader { is_signed, width }) => {
                byte |= bits_if(IntHeader::SIGNEDNESS_BIT, *is_signed);
                byte |= (width - 1) & IntHeader::EXTENDED_WIDTH_BITS;

                #[cfg(feature = "tracing")]
                tracing::debug!(
                    byte = crate::binary::fmt_byte(byte),
                    is_compact = false,
                    is_signed = is_signed,
                    width = width
                );
            }
        }

        // Push the header byte:
        self.push_byte(byte)
    }

    /// Creates a header for a signed integer value.
    pub fn header_for_signed_int<T>(&self, value: T) -> IntHeader
    where
        T: Signed + WithPackedBeBytes,
    {
        IntHeader::for_signed(value, self.config.ints.packing)
    }

    /// Creates a header for an unsigned integer value.
    pub fn header_for_unsigned_int<T>(&self, value: T) -> IntHeader
    where
        T: Unsigned + WithPackedBeBytes,
    {
        IntHeader::for_unsigned(value, self.config.ints.packing)
    }

    #[inline]
    fn encode_signed_int<S>(&mut self, value: S) -> Result<()>
    where
        S: Signed + WithPackedBeBytes,
    {
        let packing_mode = self.config.ints.packing;
        value.with_packed_be_bytes(packing_mode, |bytes| {
            let header = IntHeader::for_int_be_bytes(true, bytes, packing_mode);

            self.encode_int_header(&header)?;

            #[cfg(feature = "tracing")]
            tracing::debug!(bytes = bytes);

            if matches!(header, IntHeader::Extended(_)) {
                self.push_bytes(bytes)?;
            }

            Ok(())
        })
    }

    #[inline]
    fn encode_unsigned_int<U>(&mut self, value: U) -> Result<()>
    where
        U: Unsigned + WithPackedBeBytes,
    {
        let packing_mode = self.config.ints.packing;
        value.with_packed_be_bytes(packing_mode, |bytes| {
            let header = IntHeader::for_int_be_bytes(false, bytes, packing_mode);

            self.encode_int_header(&header)?;

            #[cfg(feature = "tracing")]
            tracing::debug!(bytes = bytes);

            if matches!(header, IntHeader::Extended(_)) {
                self.push_bytes(bytes)?;
            }

            Ok(())
        })
    }
}