//! A typesafe representation of alignment states and operations that
//! preserves LLVM's alignment invariants.

use std::cmp::Ordering;
use std::convert::{TryFrom, TryInto};
use std::fmt::{Debug, Display, Error as FmtError, Formatter, Result as FmtResult};
use std::num::ParseIntError;
use std::str::FromStr;

use paste::paste;
use thiserror::Error;

/// Errors that can occur when constructing an [`Align`](Align)
#[derive(Debug, Error)]
pub enum AlignError {
    /// The shift would exceed our maximum shift value.
    #[error("supplied shift is too large ({0} > {})", Align::MAX_SHIFT)]
    ShiftTooBig(u8),
    /// The input used to compute the shift is not a power of two.
    #[error("supplied value is not a power of two: {0}")]
    NotPowerOfTwo(u64),
    /// The input used to compute the shift is not a byte multiple.
    #[error("supplied value is not a multiple of 8: {0}")]
    NotByteMultiple(u64),
}

/// A size efficient, opaque representation of bytewise alignment.
///
/// See `Alignment.h` for LLVM's corresponding structures.
#[derive(Copy, Clone, Eq, Ord, PartialEq, PartialOrd)]
pub struct Align(u8);

impl Debug for Align {
    fn fmt(&self, f: &mut Formatter) -> Result<(), FmtError> {
        f.debug_struct("Align")
            .field("byte_align", &self.byte_align())
            .finish()
    }
}

impl Display for Align {
    fn fmt(&self, f: &mut Formatter) -> FmtResult {
        write!(f, "{}", self.byte_align())
    }
}

macro_rules! make_const_align {
    ($align:literal, $shift:literal) => {
        paste! {
            /// A convenience handle for types of exactly $width bits.
            pub const [< ALIGN $align >]: Align = Align($shift);
        }
    };
}

impl Align {
    /// The maximum alignment shift representable with this type.
    pub const MAX_SHIFT: u8 = 63;

    make_const_align!(8, 0);
    make_const_align!(16, 1);
    make_const_align!(32, 2);
    make_const_align!(64, 3);
    make_const_align!(128, 4);

    /// Returns whether `value` is a power of two, for `value > 0`.
    #[inline(always)]
    fn is_pow2(value: u64) -> bool {
        (value > 0) && ((value & (value - 1)) == 0)
    }

    /// Returns the log2 of `value`, for `value > 0`. The result is floored.
    #[inline(always)]
    fn log2(value: u64) -> u8 {
        // NOTE(ww): u8 truncation is sound here, since log2(u64::MAX) == 64.
        ((u64::from(u64::BITS) - 1) - u64::from(value.leading_zeros())) as u8
    }

    /// Create an `Align` from the given shift value, returning an error if the requested
    /// shift is invalid.
    pub fn from_shift(shift: u8) -> Result<Align, AlignError> {
        match shift > Align::MAX_SHIFT {
            false => Ok(Align(shift)),
            true => Err(AlignError::ShiftTooBig(shift)),
        }
    }

    /// Create an `Align` from the given byte alignment value, returning an error if the requested
    /// shift is invalid.
    pub fn from_byte_align(byte_align: u64) -> Result<Align, AlignError> {
        match Align::is_pow2(byte_align) {
            true => Align::from_shift(Align::log2(byte_align)),
            false => Err(AlignError::NotPowerOfTwo(byte_align)),
        }
    }

    /// Create an `Align` from the given bit alignment value, returning an error if the requested
    /// shift is invalid.
    pub fn from_bit_align(bit_align: u64) -> Result<Align, AlignError> {
        match bit_align % 8 == 0 {
            true => Align::from_byte_align(bit_align / 8),
            false => Err(AlignError::NotByteMultiple(bit_align)),
        }
    }

    /// Return this alignment's shift value, i.e. the power of two for the alignment.
    pub fn shift(&self) -> u8 {
        self.0
    }

    /// Return this alignment as a byte-granular alignment.
    pub fn byte_align(&self) -> u64 {
        1 << self.0
    }

    /// Return this alignment as a bit-granular alignment.
    pub fn bit_align(&self) -> u64 {
        self.byte_align() * 8
    }
}

/// Errors that can occur when constructing an [`AlignedTypeWidth`](AlignedTypeWidth)
#[derive(Debug, Error)]
pub enum AlignedTypeWidthError {
    /// The requested bit width is zero, which is nonsense (every non-aggregate type
    /// carries a nonzero width).
    #[error("bit width for type cannot be zero")]
    Zero,
    /// The requested bit width exceeds our support.
    #[error(
        "bit width for type is too large ({0} > {} bits)",
        AlignedTypeWidth::MAX
    )]
    TooBig(u32),
}

/// An invariant-preserving newtype for representing the bitwidth of an
/// alignable type.
#[derive(Debug, Copy, Clone, Eq, Ord, PartialEq, PartialOrd)]
pub struct AlignedTypeWidth(u32);

macro_rules! make_const_width {
    ($width:literal) => {
        paste! {
            /// A convenience handle for types of exactly $width bits.
            pub const [< WIDTH $width >]: AlignedTypeWidth = AlignedTypeWidth($width);
        }
    };
}

impl AlignedTypeWidth {
    /// The maximum type width, in bits, representable in this structure.
    pub const MAX: u32 = (1 << 23) - 1;

    // Common infallible widths, for convenience.
    make_const_width!(1);
    make_const_width!(8);
    make_const_width!(16);
    make_const_width!(32);
    make_const_width!(64);
    make_const_width!(128);
}

impl TryFrom<u32> for AlignedTypeWidth {
    type Error = AlignedTypeWidthError;
    fn try_from(value: u32) -> Result<Self, Self::Error> {
        match value > 0 && value <= AlignedTypeWidth::MAX {
            true => Ok(AlignedTypeWidth(value)),
            false => Err(AlignedTypeWidthError::TooBig(value)),
        }
    }
}

/// An enumeration of alignable non-pointer types.
#[derive(Copy, Clone, Debug, Eq, PartialEq)]
pub enum AlignedType {
    /// Aggregate types.
    Aggregate,
    /// Floating point types.
    Float(AlignedTypeWidth),
    /// Integer types.
    Integer(AlignedTypeWidth),
    /// Vector types.
    Vector(AlignedTypeWidth),
}

impl PartialOrd for AlignedType {
    fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
        Some(self.cmp(other))
    }
}

impl Ord for AlignedType {
    fn cmp(&self, other: &Self) -> Ordering {
        match (self, other) {
            // AlignedTypes are ordered first by their type, and then by their bitwidth.
            // Per LLVM, the type ordering is: aggregate < float < integer < vector.

            // Aggregate types are ordered lowest. They don't have a width, so two
            // aggregate types are always equal.
            (AlignedType::Aggregate, AlignedType::Aggregate) => Ordering::Equal,
            (AlignedType::Aggregate, _) => Ordering::Less,

            // Float types are ordered second lowest.
            (AlignedType::Float(_), AlignedType::Aggregate) => Ordering::Greater,
            (AlignedType::Float(lhs), AlignedType::Float(rhs)) => lhs.cmp(rhs),
            (AlignedType::Float(_), AlignedType::Integer(_) | AlignedType::Vector(_)) => {
                Ordering::Less
            }

            // Integer types are ordered third lowest.
            (AlignedType::Integer(_), AlignedType::Aggregate | AlignedType::Float(_)) => {
                Ordering::Greater
            }
            (AlignedType::Integer(lhs), AlignedType::Integer(rhs)) => lhs.cmp(rhs),
            (AlignedType::Integer(_), AlignedType::Vector(_)) => Ordering::Less,

            // Vector types are ordered highest.
            (
                AlignedType::Vector(_),
                AlignedType::Aggregate | AlignedType::Float(_) | AlignedType::Integer(_),
            ) => Ordering::Greater,
            (AlignedType::Vector(lhs), AlignedType::Vector(rhs)) => lhs.cmp(rhs),
        }
    }
}

macro_rules! make_const_aligned {
    ($name:ident, $width:literal) => {
        paste! {
            /// A $width bit $name:lower, subject to some alignment rules.
            pub const [< $name:upper $width >]: AlignedType = AlignedType::$name(AlignedTypeWidth::[< WIDTH $width >]);
        }
    }
}

impl AlignedType {
    // Common infallible aligned types, for convenience.
    make_const_aligned!(Float, 16);
    make_const_aligned!(Float, 32);
    make_const_aligned!(Float, 64);
    make_const_aligned!(Float, 128);
    make_const_aligned!(Integer, 1);
    make_const_aligned!(Integer, 8);
    make_const_aligned!(Integer, 16);
    make_const_aligned!(Integer, 32);
    make_const_aligned!(Integer, 64);
}

/// Errors that can occur when constructing a [`TypeAlignSpec`](TypeAlignSpec)
/// or [`PointerAlignSpec`](PointerAlignSpec).
#[derive(Debug, Error)]
pub enum AlignSpecError {
    /// The underlying type being specified has a bad width.
    #[error("impossible bit width for underlying aligned type")]
    BadTypeWidth(#[from] AlignedTypeWidthError),
    /// The supplied preferred alignment isn't greater than or equal to the ABI minimum
    #[error("impossible preferred alignment: {0} must be >= {1}")]
    AlignPref(Align, Align),
    /// The supplied ABI alignment is too large.
    #[error(
        "impossible ABI alignment for type: {0} > {}",
        TypeAlignSpec::MAX_ALIGN
    )]
    AbiAlignTooLarge(Align),
    /// We're parsing this alignment spec from a string, and it's malformed in some way.
    #[error("error while parsing alignment spec: {0}")]
    Parse(String),
    /// We're parsing this alignment spec from a string, and one of its inner alignments
    /// is malformed in some way.
    #[error("error while parsing inner alignment in spec")]
    ParseAlign(#[from] AlignError),
    /// We're parsing this alignment spec from a string, and one of its fields can't be converted
    /// into an integer.
    #[error("error while parsing integer in alignment spec")]
    BadInt(#[from] ParseIntError),
    /// The supplied address space is invalid.
    #[error("invalid address space in spec")]
    BadAddressSpace(#[from] AddressSpaceError),
}

/// Represents an alignable type, along with its ABI-mandated and
/// preferred alignments (which may differ).
#[non_exhaustive]
#[derive(Debug, Copy, Clone, Eq)]
pub struct TypeAlignSpec {
    /// The type being aligned.
    pub aligned_type: AlignedType,
    /// The ABI-enforced alignment for the type.
    pub abi_alignment: Align,
    /// The preferred alignment for the type.
    ///
    /// NOTE: This **must** be greater than or equal to the ABI alignment.
    /// This invariant is preserved during construction.
    pub preferred_alignment: Align,
}

impl PartialEq for TypeAlignSpec {
    fn eq(&self, other: &Self) -> bool {
        self.aligned_type == other.aligned_type
    }
}

impl PartialOrd for TypeAlignSpec {
    fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
        Some(self.aligned_type.cmp(&other.aligned_type))
    }
}

impl Ord for TypeAlignSpec {
    fn cmp(&self, other: &Self) -> Ordering {
        self.aligned_type.cmp(&other.aligned_type)
    }
}

impl FromStr for TypeAlignSpec {
    type Err = AlignSpecError;

    fn from_str(value: &str) -> Result<Self, Self::Err> {
        if value.is_empty() {
            return Err(AlignSpecError::Parse(
                "cannot parse type alignment from an empty string".into(),
            ));
        }

        // Unwrap safety: we check for a nonempty string above.
        #[allow(clippy::unwrap_used)]
        let id = value.chars().next().unwrap();
        let body = &value[1..];
        let parts: Vec<&str> = body.split(':').collect();

        match id {
            // `a` marks an aggregate, and is a special parsing case since it
            // doesn't include a bit size.
            'a' => {
                let parts = match body.chars().next() {
                    Some(':') => body[1..].split(':').collect::<Vec<&str>>(),
                    Some(o) => {
                        return Err(AlignSpecError::Parse(format!(
                            "unexpected character before aggregate spec: {}",
                            o
                        )))
                    }
                    None => return Err(AlignSpecError::Parse("empty aggregate specifier".into())),
                };

                if parts.is_empty() {
                    return Err(AlignSpecError::Parse(format!(
                        "wrong number of aggregate alignment parameters: expected at least 1, got {}",
                        parts.len()
                    )));
                }
                let abi = parts[0]
                    .parse::<u64>()
                    .map_err(|e| AlignSpecError::Parse(e.to_string()))
                    .and_then(|a| Align::from_bit_align(a).map_err(Into::into))?;

                let pref = parts
                    .get(1)
                    .map(|p| {
                        p.parse::<u64>()
                            .map_err(AlignSpecError::from)
                            .and_then(|a| Align::from_bit_align(a).map_err(Into::into))
                    })
                    .unwrap_or(Ok(abi))?;

                TypeAlignSpec::new(AlignedType::Aggregate, abi, pref)
            }
            // All other cases take at least two parameters, and an optional
            // third for a preferred alignment.
            id => {
                if parts.len() < 2 {
                    return Err(AlignSpecError::Parse(format!(
                        "wrong number of alignment parameters for spec '{}': expected at least 2, got {}",
                        id,
                        parts.len()
                    )));
                }

                // TODO(ww): Ugly.
                let bitsize = parts[0]
                    .parse::<u32>()
                    .map_err(|e| AlignSpecError::Parse(e.to_string()))
                    .and_then(|bs| AlignedTypeWidth::try_from(bs).map_err(Into::into))?;
                let abi = parts[1]
                    .parse::<u64>()
                    .map_err(|e| AlignSpecError::Parse(e.to_string()))
                    .and_then(|a| Align::from_bit_align(a).map_err(Into::into))?;
                let pref = parts
                    .get(2)
                    .map(|p| {
                        p.parse::<u64>()
                            .map_err(AlignSpecError::from)
                            .and_then(|a| Align::from_bit_align(a).map_err(Into::into))
                    })
                    .unwrap_or(Ok(abi))?;

                match id {
                    'i' => TypeAlignSpec::new(AlignedType::Integer(bitsize), abi, pref),
                    'v' => TypeAlignSpec::new(AlignedType::Vector(bitsize), abi, pref),
                    'f' => TypeAlignSpec::new(AlignedType::Float(bitsize), abi, pref),
                    o => Err(AlignSpecError::Parse(format!(
                        "unknown type for align spec: {}",
                        o
                    ))),
                }
            }
        }
    }
}

impl TypeAlignSpec {
    /// The maximum type width, in bits, representable in this structure.
    pub const MAX_TYPE_BIT_WIDTH: u32 = (1 << 23) - 1;

    /// The maximum alignment supported by instances of `TypeAlignSpec`.
    // NOTE(ww): On top of the normal alignment invariants, `TypeAlignSpec`
    // requires its alignments to be less than 2^16 bits. This is
    // to prevent unforeseen compatibility issues.
    // See: https://reviews.llvm.org/D67400
    pub const MAX_ALIGN: Align = Align(15);

    /// Create a new `TypeAlignSpec` for the given `AlignedType` and alignment
    /// constraints.
    pub fn new(aligned_type: AlignedType, abi: Align, pref: Align) -> Result<Self, AlignSpecError> {
        if pref < abi {
            return Err(AlignSpecError::AlignPref(pref, abi));
        }

        match ((abi <= Self::MAX_ALIGN), (pref <= Self::MAX_ALIGN)) {
            (true, true) => Ok(Self {
                aligned_type: aligned_type,
                abi_alignment: abi,
                preferred_alignment: pref,
            }),
            // NOTE(ww): We don't need a special case for the preferred alignment
            // being too large here, since it's precluded by our `pref > abi` check
            // above: `pref > MAX_ALIGN && pref >= abi` implies `abi >= MAX_ALIGN`,
            // so our ABI value is always erroneous.
            (_, _) => Err(AlignSpecError::AbiAlignTooLarge(abi)),
        }
    }
}

/// Represents a sorted collection of [`TypeAlignSpec`](TypeAlignSpec)s.
#[derive(Debug, PartialEq)]
pub struct TypeAlignSpecs(Vec<TypeAlignSpec>);

impl Default for TypeAlignSpecs {
    fn default() -> Self {
        // NOTE: The default sequence here is sorted.
        // Unwrap safety: each of these constructions is infallible.
        // TODO(ww): Use macro_rules! here to make each of these `TypeAlignSpec`s
        // into an infallible constant.
        #[allow(clippy::unwrap_used)]
        Self(vec![
            TypeAlignSpec::new(AlignedType::Aggregate, Align::ALIGN64, Align::ALIGN64).unwrap(),
            TypeAlignSpec::new(AlignedType::FLOAT16, Align::ALIGN16, Align::ALIGN16).unwrap(),
            TypeAlignSpec::new(AlignedType::FLOAT32, Align::ALIGN32, Align::ALIGN32).unwrap(),
            TypeAlignSpec::new(AlignedType::FLOAT64, Align::ALIGN64, Align::ALIGN64).unwrap(),
            TypeAlignSpec::new(AlignedType::FLOAT128, Align::ALIGN128, Align::ALIGN128).unwrap(),
            TypeAlignSpec::new(AlignedType::INTEGER1, Align::ALIGN8, Align::ALIGN8).unwrap(),
            TypeAlignSpec::new(AlignedType::INTEGER8, Align::ALIGN8, Align::ALIGN8).unwrap(),
            TypeAlignSpec::new(AlignedType::INTEGER16, Align::ALIGN16, Align::ALIGN16).unwrap(),
            TypeAlignSpec::new(AlignedType::INTEGER32, Align::ALIGN32, Align::ALIGN32).unwrap(),
            TypeAlignSpec::new(AlignedType::INTEGER64, Align::ALIGN64, Align::ALIGN64).unwrap(),
        ])
    }
}

impl TypeAlignSpecs {
    /// Update this list of type alignment specifications by inserting the given specification
    /// at the correct location, **or** rewriting an already present specification.
    pub fn update(&mut self, spec: TypeAlignSpec) {
        // Find the position of the rightmost spec that's less than or equal to
        // the spec that we're inserting.
        let pos = self.0.iter().rposition(|&other| other <= spec);
        match pos {
            // If we have a match, then we need to either insert or update
            // depending on whether our match is the same spec as us.
            // Panic safety: `pos` is a valid index returned above.
            Some(pos) => match self.0[pos] == spec {
                true => {
                    // Unwrap safety: `pos` is a valid index returned above.
                    #[allow(clippy::unwrap_used)]
                    let mut other = self.0.get_mut(pos).unwrap();

                    other.abi_alignment = spec.abi_alignment;
                    other.preferred_alignment = spec.preferred_alignment;
                }
                false => {
                    self.0.insert(pos + 1, spec);
                }
            },
            // If we don't have a match, then we're the smallest. Insert at the front.
            None => self.0.insert(0, spec),
        }
    }
}

/// Errors that can occur when constructing an [`AddressSpace`](AddressSpace)
#[derive(Debug, Error)]
pub enum AddressSpaceError {
    /// The requested address space identifier exceeds our support.
    #[error("address space identifier is too large ({0} > {})", AddressSpace::MAX)]
    TooBig(u64),
}

/// An invariant-preserving newtype for representing the address space of a pointer type.
#[derive(Copy, Clone, Debug, Eq, PartialEq, Ord, PartialOrd)]
pub struct AddressSpace(u32);

impl Default for AddressSpace {
    fn default() -> Self {
        Self(0)
    }
}

impl TryFrom<u32> for AddressSpace {
    type Error = AddressSpaceError;
    fn try_from(value: u32) -> Result<Self, Self::Error> {
        match value <= AddressSpace::MAX {
            true => Ok(AddressSpace(value)),
            false => Err(AddressSpaceError::TooBig(value.into())),
        }
    }
}

impl TryFrom<u64> for AddressSpace {
    type Error = AddressSpaceError;
    fn try_from(value: u64) -> Result<Self, Self::Error> {
        match value <= AddressSpace::MAX.into() {
            true => Ok(AddressSpace(value as u32)),
            false => Err(AddressSpaceError::TooBig(value)),
        }
    }
}

impl AddressSpace {
    /// The maximum address space identifier.
    pub const MAX: u32 = (1 << 23) - 1;
}

/// Represents a pointer width (in bits), along with its ABI-mandated and
/// preferred alignments (which may differ).
#[non_exhaustive]
#[derive(Copy, Clone, Debug, Eq, PartialEq)]
pub struct PointerAlignSpec {
    /// The address space that this pointer specification is valid in.
    pub address_space: AddressSpace,
    /// The ABI-enforced alignment for this pointer.
    pub abi_alignment: Align,
    /// The preferred alignment for this pointer.
    ///
    /// Like [`TypeAlignSpec`](TypeAlignSpec), this is enforced by construction
    /// to be no less than the ABI-enforced alignment.
    pub preferred_alignment: Align,
    /// The size of this pointer type, in bits.
    pub pointer_size: u64,
    /// The size of indexing operations with this pointer type, in bits.
    pub index_size: u64,
}

impl PartialOrd for PointerAlignSpec {
    fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
        Some(self.address_space.cmp(&other.address_space))
    }
}

impl Ord for PointerAlignSpec {
    fn cmp(&self, other: &Self) -> Ordering {
        self.address_space.cmp(&other.address_space)
    }
}

// There's only one default pointer type in LLVM datalayout specifications, so
// this is fine.
impl Default for PointerAlignSpec {
    fn default() -> Self {
        Self {
            address_space: AddressSpace::default(),
            abi_alignment: Align::ALIGN64,
            preferred_alignment: Align::ALIGN64,
            pointer_size: 64,
            index_size: 64,
        }
    }
}

impl FromStr for PointerAlignSpec {
    type Err = AlignSpecError;

    fn from_str(value: &str) -> Result<Self, Self::Err> {
        // Every pointer alignment specification looks like this:
        //     p[n]:<size>:<abi>:<pref>[:idx]
        // ...where [n] and [:idx] are optional. The absence of [n] implies
        // `0`, i.e. the default address space.

        if value.is_empty() {
            return Err(AlignSpecError::Parse(
                "cannot parse from an empty string".into(),
            ));
        }

        let parts: Vec<&str> = value[1..].split(':').collect();

        // We expect no less than 3 parts: `p[n]`, `size`, and `abi`, with
        // `pref` and `idx` being optional.
        if parts.len() < 3 {
            return Err(AlignSpecError::Parse(format!(
                "pointer align spec has too few parts ({}, expected at least 4)",
                parts.len()
            )));
        }

        let address_space = match parts[0].is_empty() {
            true => AddressSpace::default(),
            false => parts[0].parse::<u32>()?.try_into()?,
        };
        let pointer_size = parts[1].parse::<u64>()?;
        let abi = parts[2]
            .parse::<u64>()
            .map_err(|e| AlignSpecError::Parse(e.to_string()))
            .and_then(|a| Align::from_bit_align(a).map_err(Into::into))?;
        let pref = parts
            .get(3)
            .map(|idx| {
                idx.parse::<u64>()
                    .map_err(AlignSpecError::from)
                    .and_then(|a| Align::from_bit_align(a).map_err(Into::into))
            })
            .unwrap_or(Ok(abi))?;
        let index_size = parts
            .get(4)
            .map(|idx| idx.parse::<u64>())
            .unwrap_or(Ok(pointer_size))?;

        Ok(Self {
            address_space: address_space,
            abi_alignment: abi,
            preferred_alignment: pref,
            pointer_size: pointer_size,
            index_size: index_size,
        })
    }
}

impl PointerAlignSpec {
    /// Create a new `PointerAlignSpec`.
    pub fn new(
        address_space: AddressSpace,
        abi_alignment: Align,
        preferred_alignment: Align,
        pointer_size: u64,
        index_size: u64,
    ) -> Result<Self, AlignSpecError> {
        if preferred_alignment < abi_alignment {
            return Err(AlignSpecError::AlignPref(
                preferred_alignment,
                abi_alignment,
            ));
        }

        // LLVM doesn't put any constraints on the maximum alignment for pointers
        // the way it does for other types.

        Ok(Self {
            address_space,
            abi_alignment,
            preferred_alignment,
            pointer_size,
            index_size,
        })
    }
}

/// A model for function pointer alignment behavior.
#[derive(Debug, PartialEq)]
pub enum FunctionPointerAlign {
    /// The alignment of function pointers is independent of the alignment
    /// of functions, and is a multiple of the associated ABI alignment.
    Independent {
        /// The ABI-mandated alignment for function pointers.
        abi_alignment: Align,
    },
    /// The alignment of function pointers is a multiple of the explicit
    /// alignment specified on the function, **and** is a multiple of the
    /// associated ABI alignment.
    MultipleOfFunctionAlign {
        /// The ABI-mandated alignment for function pointers.
        abi_alignment: Align,
    },
}

/// Represents a sorted collection of [`PointerAlignSpec`](PointerAlignSpec)s.
#[derive(Debug, Eq, PartialEq)]
pub struct PointerAlignSpecs(Vec<PointerAlignSpec>);

impl Default for PointerAlignSpecs {
    fn default() -> Self {
        Self(vec![PointerAlignSpec::default()])
    }
}

impl PointerAlignSpecs {
    /// Update this list of pointer alignment specifications by inserting the given specification
    /// at the correct location, **or** rewriting an already present specification.
    pub fn update(&mut self, spec: PointerAlignSpec) {
        // Find the position of the rightmost spec that's less than or equal to
        // the spec that we're inserting.
        let pos = self.0.iter().rposition(|&other| other <= spec);
        match pos {
            // If we have a match, then we need to either insert or update
            // depending on whether our match is the same spec as us.
            // Panic safety: `pos` is a valid index returned above.
            Some(pos) => match self.0[pos] == spec {
                true => {
                    // Unwrap safety: `pos` is a valid index returned above.
                    #[allow(clippy::unwrap_used)]
                    let mut other = self.0.get_mut(pos).unwrap();

                    other.abi_alignment = spec.abi_alignment;
                    other.preferred_alignment = spec.preferred_alignment;
                }
                false => {
                    self.0.insert(pos + 1, spec);
                }
            },
            // If we don't have a match, then we're the smallest. Insert at the front.
            None => self.0.insert(0, spec),
        }
    }
}

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

    #[test]
    fn test_align_constants() {
        assert_eq!(Align::ALIGN8.byte_align(), 1);
        assert_eq!(Align::ALIGN16.byte_align(), 2);
        assert_eq!(Align::ALIGN32.byte_align(), 4);
        assert_eq!(Align::ALIGN64.byte_align(), 8);
        assert_eq!(Align::ALIGN128.byte_align(), 16);
    }

    #[test]
    fn test_align_is_pow2() {
        assert!(Align::is_pow2(1));
        assert!(Align::is_pow2(2));
        assert!(Align::is_pow2(4));
        assert!(Align::is_pow2(8));
        assert!(Align::is_pow2(16));
        assert!(Align::is_pow2(32));
        assert!(Align::is_pow2(64));

        assert!(!Align::is_pow2(3));
        assert!(!Align::is_pow2(6));
        assert!(!Align::is_pow2(12));
        assert!(!Align::is_pow2(65));
    }

    #[test]
    fn test_align_compares() {
        assert_eq!(Align::from_shift(1).unwrap(), Align::from_shift(1).unwrap());
        assert!(Align::from_shift(1).unwrap() < Align::from_shift(2).unwrap());
        assert!(Align::from_shift(1).unwrap() <= Align::from_shift(2).unwrap());
        assert!(Align::from_shift(2).unwrap() > Align::from_shift(1).unwrap());
        assert!(Align::from_shift(2).unwrap() >= Align::from_shift(1).unwrap());
    }

    #[test]
    fn test_align_log2() {
        assert_eq!(Align::log2(1), 0);
        assert_eq!(Align::log2(2), 1);
        assert_eq!(Align::log2(4), 2);
        assert_eq!(Align::log2(8), 3);
        assert_eq!(Align::log2(16), 4);
        assert_eq!(Align::log2(32), 5);
        assert_eq!(Align::log2(64), 6);

        // Our internal log2 is a flooring log2.
        assert_eq!(Align::log2(65), 6);
    }

    #[test]
    fn test_align_basic() {
        for (s, ba) in &[(0, 1), (1, 2), (2, 4), (3, 8), (4, 16), (5, 32), (6, 64)] {
            let align = Align(*s);

            assert_eq!(align.shift(), *s);
            assert_eq!(align.byte_align(), *ba);
            assert_eq!(align.bit_align(), (*ba) * 8);
        }
    }

    #[test]
    fn test_align_from_byte_align() {
        assert_eq!(Align::from_byte_align(1).unwrap().shift(), 0);
        assert_eq!(Align::from_byte_align(2).unwrap().shift(), 1);
        assert_eq!(Align::from_byte_align(4).unwrap().shift(), 2);
        assert_eq!(Align::from_byte_align(8).unwrap().shift(), 3);
        assert_eq!(Align::from_byte_align(16).unwrap().shift(), 4);
        assert_eq!(Align::from_byte_align(32).unwrap().shift(), 5);
        assert_eq!(Align::from_byte_align(64).unwrap().shift(), 6);
        assert_eq!(Align::from_byte_align(128).unwrap().shift(), 7);

        assert!(Align::from_byte_align(0).is_err());
        assert!(Align::from_byte_align(3).is_err());
        assert!(Align::from_byte_align(7).is_err());
        assert!(Align::from_byte_align(22).is_err());
        assert!(Align::from_byte_align(24).is_err());
    }

    #[test]
    fn test_align_from_bit_align() {
        assert_eq!(Align::from_bit_align(8).unwrap().shift(), 0);
        assert_eq!(Align::from_bit_align(16).unwrap().shift(), 1);
        assert_eq!(Align::from_bit_align(32).unwrap().shift(), 2);
        assert_eq!(Align::from_bit_align(64).unwrap().shift(), 3);
        assert_eq!(Align::from_bit_align(128).unwrap().shift(), 4);
        assert_eq!(Align::from_bit_align(256).unwrap().shift(), 5);
        assert_eq!(Align::from_bit_align(512).unwrap().shift(), 6);
        assert_eq!(Align::from_bit_align(1024).unwrap().shift(), 7);

        assert!(Align::from_bit_align(0).is_err());
        assert!(Align::from_bit_align(1).is_err());
        assert!(Align::from_bit_align(7).is_err());
        assert!(Align::from_bit_align(9).is_err());
        assert!(Align::from_bit_align(24).is_err());
        assert!(Align::from_bit_align(33).is_err());
    }

    #[test]
    fn test_aligned_type_width() {
        for i in 1..(1 << 15) {
            assert!(AlignedTypeWidth::try_from(i).is_ok());
        }

        assert!(AlignedTypeWidth::try_from(0).is_err());
        assert!(AlignedTypeWidth::try_from(u32::MAX).is_err());
    }

    #[test]
    fn test_aligned_type_ordering() {
        assert!(
            AlignedType::Integer(AlignedTypeWidth(1)) == AlignedType::Integer(AlignedTypeWidth(1))
        );
        assert!(
            AlignedType::Vector(AlignedTypeWidth(1)) == AlignedType::Vector(AlignedTypeWidth(1))
        );
        assert!(AlignedType::Float(AlignedTypeWidth(1)) == AlignedType::Float(AlignedTypeWidth(1)));

        for i in 2..(1 << 15) {
            assert!(
                AlignedType::Integer(AlignedTypeWidth(i))
                    < AlignedType::Vector(AlignedTypeWidth(i))
            );
            assert!(
                AlignedType::Integer(AlignedTypeWidth(i))
                    <= AlignedType::Vector(AlignedTypeWidth(i))
            );

            assert!(
                AlignedType::Integer(AlignedTypeWidth(i + 1))
                    < AlignedType::Vector(AlignedTypeWidth(i - 1))
            );
            assert!(
                AlignedType::Integer(AlignedTypeWidth(i + 1))
                    <= AlignedType::Vector(AlignedTypeWidth(i - 1))
            );

            assert!(
                AlignedType::Integer(AlignedTypeWidth(i)) > AlignedType::Float(AlignedTypeWidth(i))
            );
            assert!(
                AlignedType::Integer(AlignedTypeWidth(i))
                    >= AlignedType::Float(AlignedTypeWidth(i))
            );

            assert!(
                AlignedType::Integer(AlignedTypeWidth(i + 1))
                    > AlignedType::Float(AlignedTypeWidth(i - 1))
            );
            assert!(
                AlignedType::Integer(AlignedTypeWidth(i + 1))
                    >= AlignedType::Float(AlignedTypeWidth(i - 1))
            );

            assert!(AlignedType::Integer(AlignedTypeWidth(i)) > AlignedType::Aggregate);
            assert!(AlignedType::Integer(AlignedTypeWidth(i)) >= AlignedType::Aggregate);
        }

        assert!(AlignedType::Aggregate == AlignedType::Aggregate);
    }

    #[test]
    fn test_type_align_spec() {
        // Normal cases.
        assert!(TypeAlignSpec::new(
            AlignedType::Integer(AlignedTypeWidth(64)),
            Align::ALIGN64,
            Align::ALIGN64
        )
        .is_ok());
        assert!(TypeAlignSpec::new(
            AlignedType::Integer(AlignedTypeWidth(64)),
            Align::ALIGN64,
            Align::ALIGN128
        )
        .is_ok());
        assert!(TypeAlignSpec::new(
            AlignedType::Float(AlignedTypeWidth(32)),
            Align::ALIGN32,
            Align::ALIGN32
        )
        .is_ok());
        assert!(TypeAlignSpec::new(
            AlignedType::Float(AlignedTypeWidth(32)),
            Align::ALIGN32,
            Align::ALIGN64
        )
        .is_ok());

        // Can't create with an undersized preferred alignment.
        assert_eq!(
            TypeAlignSpec::new(
                AlignedType::Integer(AlignedTypeWidth(8)),
                Align(2),
                Align(1)
            )
            .unwrap_err()
            .to_string(),
            "impossible preferred alignment: 2 must be >= 4"
        );

        // Can't create with an oversized ABI alignment.
        assert_eq!(
            TypeAlignSpec::new(
                AlignedType::Integer(AlignedTypeWidth(8)),
                Align(16),
                Align(16)
            )
            .unwrap_err()
            .to_string(),
            "impossible ABI alignment for type: 65536 > 32768"
        );
    }

    #[test]
    fn test_type_align_spec_equality() {
        // Two "different" specs with the same type (+ width) compare as equal.
        let spec1 =
            TypeAlignSpec::new(AlignedType::INTEGER8, Align::ALIGN16, Align::ALIGN16).unwrap();

        let spec2 =
            TypeAlignSpec::new(AlignedType::INTEGER8, Align::ALIGN16, Align::ALIGN32).unwrap();

        assert_eq!(spec1, spec2);
    }

    #[test]
    fn test_type_align_specs_default_sorted() {
        let specs1 = TypeAlignSpecs::default();
        let mut specs2 = TypeAlignSpecs::default();
        specs2.0.sort();

        assert_eq!(specs1, specs2);
    }

    #[test]
    fn test_type_align_specs_update() {
        {
            // Trivial insertion works.
            let mut specs = TypeAlignSpecs(vec![]);
            specs.update(
                TypeAlignSpec::new(AlignedType::INTEGER8, Align::ALIGN16, Align::ALIGN16).unwrap(),
            );

            assert_eq!(specs.0.len(), 1);
        }

        {
            // Trivial updating works.
            let mut specs = TypeAlignSpecs(vec![]);
            specs.update(
                TypeAlignSpec::new(AlignedType::INTEGER8, Align::ALIGN16, Align::ALIGN16).unwrap(),
            );
            specs.update(
                TypeAlignSpec::new(AlignedType::INTEGER8, Align::ALIGN16, Align::ALIGN32).unwrap(),
            );

            assert_eq!(specs.0.len(), 1);
        }

        {
            // Ordered insertion (append) works.
            let mut specs = TypeAlignSpecs(vec![]);
            specs.update(
                TypeAlignSpec::new(AlignedType::Aggregate, Align::ALIGN64, Align::ALIGN64).unwrap(),
            );
            specs.update(
                TypeAlignSpec::new(AlignedType::INTEGER8, Align::ALIGN16, Align::ALIGN16).unwrap(),
            );

            let copy = {
                let mut copy = specs.0.clone();
                copy.sort();
                TypeAlignSpecs(copy)
            };

            assert_eq!(specs.0.len(), 2);
            assert_eq!(specs, copy);
        }

        {
            // Ordered insertion (prepend) works.
            let mut specs = TypeAlignSpecs(vec![]);
            specs.update(
                TypeAlignSpec::new(AlignedType::INTEGER8, Align::ALIGN16, Align::ALIGN16).unwrap(),
            );
            specs.update(
                TypeAlignSpec::new(AlignedType::Aggregate, Align::ALIGN64, Align::ALIGN64).unwrap(),
            );

            let copy = {
                let mut copy = specs.0.clone();
                copy.sort();
                TypeAlignSpecs(copy)
            };

            assert_eq!(specs.0.len(), 2);
            assert_eq!(specs, copy);
        }

        {
            // Ordered insertion (in the middle) works.
            let mut specs = TypeAlignSpecs(vec![]);
            specs.update(
                TypeAlignSpec::new(AlignedType::INTEGER8, Align::ALIGN16, Align::ALIGN16).unwrap(),
            );
            specs.update(
                TypeAlignSpec::new(AlignedType::Aggregate, Align::ALIGN64, Align::ALIGN64).unwrap(),
            );
            specs.update(
                TypeAlignSpec::new(AlignedType::FLOAT16, Align::ALIGN16, Align::ALIGN16).unwrap(),
            );

            let copy = {
                let mut copy = specs.0.clone();
                copy.sort();
                TypeAlignSpecs(copy)
            };

            assert_eq!(specs.0.len(), 3);
            assert_eq!(specs, copy);
        }
    }

    #[test]
    fn test_type_align_spec_parse() {
        {
            let spec = "i64:64:64".parse::<TypeAlignSpec>().unwrap();

            assert_eq!(
                spec.aligned_type,
                AlignedType::Integer(AlignedTypeWidth::WIDTH64)
            );
            assert_eq!(spec.abi_alignment, Align::ALIGN64);
            assert_eq!(spec.preferred_alignment, Align::ALIGN64);
        }

        {
            let spec = "i32:32:64".parse::<TypeAlignSpec>().unwrap();

            assert_eq!(
                spec.aligned_type,
                AlignedType::Integer(AlignedTypeWidth::WIDTH32)
            );
            assert_eq!(spec.abi_alignment, Align::ALIGN32);
            assert_eq!(spec.preferred_alignment, Align::ALIGN64);
        }

        {
            let spec = "a:32:64".parse::<TypeAlignSpec>().unwrap();

            assert_eq!(spec.aligned_type, AlignedType::Aggregate);
            assert_eq!(spec.abi_alignment, Align::ALIGN32);
            assert_eq!(spec.preferred_alignment, Align::ALIGN64);
        }

        {
            let spec = "a:32".parse::<TypeAlignSpec>().unwrap();

            assert_eq!(spec.aligned_type, AlignedType::Aggregate);
            assert_eq!(spec.abi_alignment, Align::ALIGN32);
            assert_eq!(spec.preferred_alignment, Align::ALIGN32);
        }
    }

    #[test]
    fn test_pointer_align_spec_parse() {
        {
            let spec = "p:64:64:64".parse::<PointerAlignSpec>().unwrap();

            assert_eq!(spec.address_space, AddressSpace::default());
            assert_eq!(spec.pointer_size, 64);
            assert_eq!(spec.index_size, 64);
            assert_eq!(spec.abi_alignment, Align::ALIGN64);
            assert_eq!(spec.preferred_alignment, Align::ALIGN64);
        }

        {
            let spec = "p0:64:64:64".parse::<PointerAlignSpec>().unwrap();

            assert_eq!(spec.address_space, AddressSpace::default());
            assert_eq!(spec.pointer_size, 64);
            assert_eq!(spec.index_size, 64);
            assert_eq!(spec.abi_alignment, Align::ALIGN64);
            assert_eq!(spec.preferred_alignment, Align::ALIGN64);
        }

        {
            let spec = "p:64:64:64:64".parse::<PointerAlignSpec>().unwrap();

            assert_eq!(spec.address_space, AddressSpace::default());
            assert_eq!(spec.pointer_size, 64);
            assert_eq!(spec.index_size, 64);
            assert_eq!(spec.abi_alignment, Align::ALIGN64);
            assert_eq!(spec.preferred_alignment, Align::ALIGN64);
        }

        {
            let spec = "p:64:64:64:32".parse::<PointerAlignSpec>().unwrap();

            assert_eq!(spec.address_space, AddressSpace::default());
            assert_eq!(spec.pointer_size, 64);
            assert_eq!(spec.index_size, 32);
            assert_eq!(spec.abi_alignment, Align::ALIGN64);
            assert_eq!(spec.preferred_alignment, Align::ALIGN64);
        }

        {
            let spec = "p1:64:64:64".parse::<PointerAlignSpec>().unwrap();

            assert_eq!(spec.address_space, AddressSpace(1));
            assert_eq!(spec.pointer_size, 64);
            assert_eq!(spec.index_size, 64);
            assert_eq!(spec.abi_alignment, Align::ALIGN64);
            assert_eq!(spec.preferred_alignment, Align::ALIGN64);
        }

        {
            let spec = "p1:64:64".parse::<PointerAlignSpec>().unwrap();

            assert_eq!(spec.address_space, AddressSpace(1));
            assert_eq!(spec.pointer_size, 64);
            assert_eq!(spec.index_size, 64);
            assert_eq!(spec.abi_alignment, Align::ALIGN64);
            assert_eq!(spec.preferred_alignment, Align::ALIGN64);
        }
    }

    #[test]
    fn test_address_space() {
        assert!(AddressSpace::try_from(0_u32).is_ok());
        assert!(AddressSpace::try_from(1_u32).is_ok());
        assert!(AddressSpace::try_from(AddressSpace::MAX).is_ok());

        assert!(AddressSpace::try_from(AddressSpace::MAX + 1).is_err());
    }

    #[test]
    fn test_address_space_ordering() {
        assert!(AddressSpace(0) < AddressSpace(1));
        assert!(AddressSpace(0) <= AddressSpace(1));
        assert!(AddressSpace(0) == AddressSpace(0));
    }
}