//! Memory allocation for TrueType scaling.

use std::mem::{align_of, size_of};

use read_fonts::{
    tables::glyf::PointFlags,
    types::{F26Dot6, Fixed, Point},
};

use super::{super::Hinting, Outline};

/// Buffers used during HarfBuzz-style glyph scaling.
pub(crate) struct HarfBuzzOutlineMemory<'a> {
    pub points: &'a mut [Point<f32>],
    pub contours: &'a mut [u16],
    pub flags: &'a mut [PointFlags],
    pub deltas: &'a mut [Point<f32>],
    pub iup_buffer: &'a mut [Point<f32>],
    pub composite_deltas: &'a mut [Point<f32>],
}

impl<'a> HarfBuzzOutlineMemory<'a> {
    pub(super) fn new(outline: &Outline, buf: &'a mut [u8]) -> Option<Self> {
        let (points, buf) = alloc_slice(buf, outline.points)?;
        let (contours, buf) = alloc_slice(buf, outline.contours)?;
        let (flags, buf) = alloc_slice(buf, outline.points)?;
        // Don't allocate any delta buffers if we don't have variations
        let (deltas, iup_buffer, composite_deltas, _buf) = if outline.has_variations {
            let (deltas, buf) = alloc_slice(buf, outline.max_simple_points)?;
            let (iup_buffer, buf) = alloc_slice(buf, outline.max_simple_points)?;
            let (composite_deltas, buf) = alloc_slice(buf, outline.max_component_delta_stack)?;
            (deltas, iup_buffer, composite_deltas, buf)
        } else {
            (
                Default::default(),
                Default::default(),
                Default::default(),
                buf,
            )
        };
        Some(Self {
            points,
            contours,
            flags,
            deltas,
            iup_buffer,
            composite_deltas,
        })
    }
}

/// Buffers used during glyph scaling.
pub(crate) struct FreeTypeOutlineMemory<'a> {
    pub unscaled: &'a mut [Point<i32>],
    pub scaled: &'a mut [Point<F26Dot6>],
    pub original_scaled: &'a mut [Point<F26Dot6>],
    pub contours: &'a mut [u16],
    pub flags: &'a mut [PointFlags],
    pub deltas: &'a mut [Point<Fixed>],
    pub iup_buffer: &'a mut [Point<Fixed>],
    pub composite_deltas: &'a mut [Point<Fixed>],
    pub stack: &'a mut [i32],
    pub cvt: &'a mut [i32],
    pub storage: &'a mut [i32],
    pub twilight_scaled: &'a mut [Point<F26Dot6>],
    pub twilight_original_scaled: &'a mut [Point<F26Dot6>],
    pub twilight_flags: &'a mut [PointFlags],
}

impl<'a> FreeTypeOutlineMemory<'a> {
    pub(super) fn new(outline: &Outline, buf: &'a mut [u8], hinting: Hinting) -> Option<Self> {
        let hinted = outline.has_hinting && hinting == Hinting::Embedded;
        let (scaled, buf) = alloc_slice(buf, outline.points)?;
        let (unscaled, buf) = alloc_slice(buf, outline.max_other_points)?;
        // We only need original scaled points when hinting
        let (original_scaled, buf) = if hinted {
            alloc_slice(buf, outline.max_other_points)?
        } else {
            (Default::default(), buf)
        };
        // Don't allocate any delta buffers if we don't have variations
        let (deltas, iup_buffer, composite_deltas, buf) = if outline.has_variations {
            let (deltas, buf) = alloc_slice(buf, outline.max_simple_points)?;
            let (iup_buffer, buf) = alloc_slice(buf, outline.max_simple_points)?;
            let (composite_deltas, buf) = alloc_slice(buf, outline.max_component_delta_stack)?;
            (deltas, iup_buffer, composite_deltas, buf)
        } else {
            (
                Default::default(),
                Default::default(),
                Default::default(),
                buf,
            )
        };
        // Hinting value stack
        let (stack, buf) = if hinted {
            alloc_slice(buf, outline.max_stack)?
        } else {
            (Default::default(), buf)
        };
        // Copy-on-write buffers for CVT and storage area
        let (cvt, storage, buf) = if hinted {
            let (cvt, buf) = alloc_slice(buf, outline.cvt_count)?;
            let (storage, buf) = alloc_slice(buf, outline.storage_count)?;
            (cvt, storage, buf)
        } else {
            (Default::default(), Default::default(), buf)
        };
        // Twilight zone point buffers
        let (twilight_scaled, twilight_original_scaled, buf) = if hinted {
            let (scaled, buf) = alloc_slice(buf, outline.max_twilight_points)?;
            let (original_scaled, buf) = alloc_slice(buf, outline.max_twilight_points)?;
            (scaled, original_scaled, buf)
        } else {
            (Default::default(), Default::default(), buf)
        };
        let (contours, buf) = alloc_slice(buf, outline.contours)?;
        let (flags, buf) = alloc_slice(buf, outline.points)?;
        // Twilight zone point flags
        let twilight_flags = if hinted {
            alloc_slice(buf, outline.max_twilight_points)?.0
        } else {
            Default::default()
        };
        Some(Self {
            unscaled,
            scaled,
            original_scaled,
            contours,
            flags,
            deltas,
            iup_buffer,
            composite_deltas,
            stack,
            cvt,
            storage,
            twilight_scaled,
            twilight_original_scaled,
            twilight_flags,
        })
    }
}

/// Allocates a mutable slice of `T` of the given length from the specified
/// buffer.
///
/// Returns the allocated slice and the remainder of the buffer.
fn alloc_slice<T>(buf: &mut [u8], len: usize) -> Option<(&mut [T], &mut [u8])>
where
    T: bytemuck::AnyBitPattern + bytemuck::NoUninit,
{
    if len == 0 {
        return Some((Default::default(), buf));
    }
    // 1) Ensure we slice the buffer at a position that is properly aligned
    // for T.
    let base_ptr = buf.as_ptr() as usize;
    let aligned_ptr = align_up(base_ptr, align_of::<T>());
    let aligned_offset = aligned_ptr - base_ptr;
    let buf = buf.get_mut(aligned_offset..)?;
    // 2) Ensure we have enough space in the buffer to allocate our slice.
    let len_in_bytes = len * size_of::<T>();
    if len_in_bytes > buf.len() {
        return None;
    }
    let (slice_buf, rest) = buf.split_at_mut(len_in_bytes);
    // Bytemuck handles all safety guarantees here.
    let slice = bytemuck::try_cast_slice_mut(slice_buf).ok()?;
    Some((slice, rest))
}

fn align_up(len: usize, alignment: usize) -> usize {
    len + (len.wrapping_neg() & (alignment - 1))
}

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

    #[test]
    fn unaligned_buffer() {
        let mut buf = [0u8; 40];
        let alignment = align_of::<i32>();
        let addr = buf.as_ptr() as usize;
        let mut unaligned_addr = addr;
        // Force an unaligned offset
        if unaligned_addr % alignment == 0 {
            unaligned_addr += 1;
        }
        let unaligned_offset = unaligned_addr - addr;
        let unaligned = &mut buf[unaligned_offset..];
        assert!(unaligned.as_ptr() as usize % alignment != 0);
        let (slice, _) = alloc_slice::<i32>(unaligned, 8).unwrap();
        assert_eq!(slice.as_ptr() as usize % alignment, 0);
    }

    #[test]
    fn fail_unaligned_buffer() {
        let mut buf = [0u8; 40];
        let alignment = align_of::<i32>();
        let addr = buf.as_ptr() as usize;
        let mut unaligned_addr = addr;
        // Force an unaligned offset
        if unaligned_addr % alignment == 0 {
            unaligned_addr += 1;
        }
        let unaligned_offset = unaligned_addr - addr;
        let unaligned = &mut buf[unaligned_offset..];
        assert_eq!(alloc_slice::<i32>(unaligned, 16), None);
    }

    #[test]
    fn outline_memory() {
        let outline_info = Outline {
            glyph: None,
            glyph_id: Default::default(),
            points: 10,
            contours: 4,
            max_simple_points: 4,
            max_other_points: 4,
            max_component_delta_stack: 4,
            max_stack: 0,
            cvt_count: 0,
            storage_count: 0,
            max_twilight_points: 0,
            has_hinting: false,
            has_variations: true,
            has_overlaps: false,
        };
        let required_size = outline_info.required_buffer_size(Hinting::None);
        let mut buf = vec![0u8; required_size];
        let memory = FreeTypeOutlineMemory::new(&outline_info, &mut buf, Hinting::None).unwrap();
        assert_eq!(memory.scaled.len(), outline_info.points);
        assert_eq!(memory.unscaled.len(), outline_info.max_other_points);
        // We don't allocate this buffer when hinting is disabled
        assert_eq!(memory.original_scaled.len(), 0);
        assert_eq!(memory.flags.len(), outline_info.points);
        assert_eq!(memory.contours.len(), outline_info.contours);
        assert_eq!(memory.deltas.len(), outline_info.max_simple_points);
        assert_eq!(memory.iup_buffer.len(), outline_info.max_simple_points);
        assert_eq!(
            memory.composite_deltas.len(),
            outline_info.max_component_delta_stack
        );
    }

    #[test]
    fn fail_outline_memory() {
        let outline_info = Outline {
            glyph: None,
            glyph_id: Default::default(),
            points: 10,
            contours: 4,
            max_simple_points: 4,
            max_other_points: 4,
            max_component_delta_stack: 4,
            max_stack: 0,
            cvt_count: 0,
            storage_count: 0,
            max_twilight_points: 0,
            has_hinting: false,
            has_variations: true,
            has_overlaps: false,
        };
        // Required size adds 4 bytes slop to account for internal alignment
        // requirements. So subtract 5 to force a failure.
        let not_enough = outline_info.required_buffer_size(Hinting::None) - 5;
        let mut buf = vec![0u8; not_enough];
        assert!(FreeTypeOutlineMemory::new(&outline_info, &mut buf, Hinting::None).is_none());
    }
}