takumi 1.0.11

use color::{AlphaColor, ColorSpaceTag, DynamicColor, HueDirection, Rgba8, Srgb};
#[cfg(test)]
use image::RgbaImage;
use smallvec::SmallVec;
use taffy::Point;
use tiny_skia::{ColorU8, PremultipliedColorU8};

use super::{Color, GradientStop, ResolvedGradientStop};
use crate::rendering::{RenderContext, fast_div_255};

const MIN_GRADIENT_LUT_SIZE: usize = 2;
const MAX_GRADIENT_LUT_SIZE: usize = 8193;

/// Interpolates between two colors in RGBA space, if t is 0.0 or 1.0, returns the first or second color.
pub(crate) fn interpolate_rgba(c1: Color, c2: Color, t: f32) -> Color {
  if t <= f32::EPSILON {
    return c1;
  }

  if t >= 1.0 - f32::EPSILON {
    return c2;
  }

  let [r1, g1, b1, a1] = c1.0;
  let [r2, g2, b2, a2] = c2.0;
  let premul_1 = [
    fast_div_255(r1 as u32 * a1 as u32),
    fast_div_255(g1 as u32 * a1 as u32),
    fast_div_255(b1 as u32 * a1 as u32),
    a1,
  ];
  let premul_2 = [
    fast_div_255(r2 as u32 * a2 as u32),
    fast_div_255(g2 as u32 * a2 as u32),
    fast_div_255(b2 as u32 * a2 as u32),
    a2,
  ];

  let mut result = [0u8; 4];
  for i in 0..4 {
    result[i] = (premul_1[i] as f32 * (1.0 - t) + premul_2[i] as f32 * t)
      .round()
      .clamp(0.0, 255.0) as u8;
  }

  let premul = PremultipliedColorU8::from_rgba(
    result[0].min(result[3]),
    result[1].min(result[3]),
    result[2].min(result[3]),
    result[3],
  )
  .unwrap_or(PremultipliedColorU8::TRANSPARENT);
  let demul: ColorU8 = premul.demultiply();
  Color([demul.red(), demul.green(), demul.blue(), demul.alpha()])
}

pub(crate) fn interpolate_with_color_space(
  c1: Color,
  c2: Color,
  t: f32,
  color_space: ColorSpaceTag,
  hue_direction: HueDirection,
) -> Color {
  if c1 == c2 {
    return c1;
  }

  if color_space == ColorSpaceTag::Srgb && hue_direction == HueDirection::Shorter {
    return interpolate_rgba(c1, c2, t);
  }

  if t <= f32::EPSILON {
    return c1;
  }

  if t >= 1.0 - f32::EPSILON {
    return c2;
  }

  let dynamic_1 =
    DynamicColor::from_alpha_color(AlphaColor::<Srgb>::from(Rgba8::from_u8_array(c1.0)));
  let dynamic_2 =
    DynamicColor::from_alpha_color(AlphaColor::<Srgb>::from(Rgba8::from_u8_array(c2.0)));

  let mixed = dynamic_1
    .interpolate(dynamic_2, color_space, hue_direction)
    .eval(t);
  let rgba = mixed.to_alpha_color::<Srgb>().to_rgba8().to_u8_array();

  Color(rgba)
}

/// Interpolates two premultiplied colors directly in premultiplied RGBA space.
pub(crate) fn interpolate_rgba_premultiplied(
  c1: PremultipliedColorU8,
  c2: PremultipliedColorU8,
  t: f32,
) -> PremultipliedColorU8 {
  if t <= f32::EPSILON {
    return c1;
  }

  if t >= 1.0 - f32::EPSILON {
    return c2;
  }

  let mut result = [0u8; 4];
  let c1_rgba = [c1.red(), c1.green(), c1.blue(), c1.alpha()];
  let c2_rgba = [c2.red(), c2.green(), c2.blue(), c2.alpha()];

  for i in 0..4 {
    result[i] = (c1_rgba[i] as f32 * (1.0 - t) + c2_rgba[i] as f32 * t)
      .round()
      .clamp(0.0, 255.0) as u8;
  }

  PremultipliedColorU8::from_rgba(
    result[0].min(result[3]),
    result[1].min(result[3]),
    result[2].min(result[3]),
    result[3],
  )
  .unwrap_or(PremultipliedColorU8::TRANSPARENT)
}

pub(crate) trait GradientOverlayTile {
  type RowState;

  fn width(&self) -> u32;
  fn height(&self) -> u32;
  fn lut_len(&self) -> usize;
  fn sample_at(&self, lut_idx: usize) -> PremultipliedColorU8;
  fn sample_pixel(&self, x: u32, y: u32) -> PremultipliedColorU8;
  fn begin_row(&self, src_x_start: u32, src_y: u32, lut_len: usize) -> Self::RowState;
  /// Returns an index in `0..lut_len` where `lut_len` is the value passed to `begin_row`.
  fn next_lut_index(&self, row_state: &mut Self::RowState) -> usize;
}

#[cfg(test)]
pub(crate) fn overlay_gradient_tile_fast_normal_unconstrained_test<T: GradientOverlayTile>(
  bottom: &mut RgbaImage,
  tile: &T,
  offset: Point<f32>,
) {
  let bottom_width = bottom.width();
  let bottom_height = bottom.height();
  overlay_gradient_tile_fast_normal_unconstrained(
    bottom.as_mut(),
    bottom_width,
    bottom_height,
    tile,
    offset,
  );
}

pub(crate) fn overlay_gradient_tile_fast_normal_unconstrained<T: GradientOverlayTile>(
  data: &mut [u8],
  bottom_width: u32,
  bottom_height: u32,
  tile: &T,
  offset: Point<f32>,
) {
  let Some((offset_x, offset_y, dest_x_min, dest_x_max, dest_y_min, dest_y_max)) =
    compute_overlay_bounds_raw(
      bottom_width,
      bottom_height,
      offset,
      tile.width(),
      tile.height(),
    )
  else {
    return;
  };

  let lut_len = tile.lut_len();
  if lut_len == 0 {
    return;
  }
  let row_stride = bottom_width as usize * 4;

  for dest_y in dest_y_min..dest_y_max {
    let src_y = (dest_y - offset_y) as u32;
    let src_x_start = (dest_x_min - offset_x) as u32;
    let mut row_state = tile.begin_row(src_x_start, src_y, lut_len);
    for dest_x in dest_x_min..dest_x_max {
      let lut_idx = tile.next_lut_index(&mut row_state);
      debug_assert!(lut_idx < lut_len);
      let pixel = tile.sample_at(lut_idx);
      let src_a = pixel.alpha();
      if src_a == 0 {
        continue;
      }

      let pixel_offset = dest_y as usize * row_stride + dest_x as usize * 4;
      if src_a == u8::MAX {
        data[pixel_offset] = pixel.red();
        data[pixel_offset + 1] = pixel.green();
        data[pixel_offset + 2] = pixel.blue();
        data[pixel_offset + 3] = src_a;
      } else {
        let inv_src_a = u8::MAX - src_a;
        data[pixel_offset] = pixel
          .red()
          .saturating_add(fast_div_255(data[pixel_offset] as u32 * inv_src_a as u32));
        data[pixel_offset + 1] = pixel.green().saturating_add(fast_div_255(
          data[pixel_offset + 1] as u32 * inv_src_a as u32,
        ));
        data[pixel_offset + 2] = pixel.blue().saturating_add(fast_div_255(
          data[pixel_offset + 2] as u32 * inv_src_a as u32,
        ));
        data[pixel_offset + 3] = src_a.saturating_add(fast_div_255(
          data[pixel_offset + 3] as u32 * inv_src_a as u32,
        ));
      }
    }
  }
}

#[inline(always)]
fn compute_overlay_bounds_raw(
  bottom_width: u32,
  bottom_height: u32,
  offset: Point<f32>,
  width: u32,
  height: u32,
) -> Option<(i32, i32, i32, i32, i32, i32)> {
  if width == 0 || height == 0 {
    return None;
  }

  let offset_x = offset.x.trunc() as i32;
  let offset_y = offset.y.trunc() as i32;
  let bottom_width = bottom_width as i32;
  let bottom_height = bottom_height as i32;
  let dest_y_min = offset_y.max(0);
  let dest_y_max = (offset_y + height as i32).min(bottom_height);
  if dest_y_min >= dest_y_max {
    return None;
  }

  let dest_x_min = offset_x.max(0);
  let dest_x_max = (offset_x + width as i32).min(bottom_width);
  if dest_x_min >= dest_x_max {
    return None;
  }

  Some((
    offset_x, offset_y, dest_x_min, dest_x_max, dest_y_min, dest_y_max,
  ))
}

#[inline(always)]
fn position_to_sample_index(position: f32, axis_length: f32, lut_size: usize) -> usize {
  if lut_size <= 1 || axis_length.abs() <= f32::EPSILON {
    return 0;
  }

  let max_index = lut_size - 1;
  ((position.clamp(0.0, axis_length) * max_index as f32 / axis_length).round() as usize)
    .min(max_index)
}

fn assign_stop_sample_indices(
  resolved_stops: &[ResolvedGradientStop],
  axis_length: f32,
  lut_size: usize,
) -> Vec<usize> {
  if resolved_stops.is_empty() || lut_size == 0 {
    return Vec::new();
  }

  let stop_count = resolved_stops.len();
  let max_index = lut_size - 1;
  let mut indices = vec![0usize; stop_count];
  let mut i = 0usize;

  while i < stop_count {
    let position = resolved_stops[i].position;
    let preferred = position_to_sample_index(position, axis_length, lut_size);
    let mut run_end = i + 1;
    while run_end < stop_count
      && (resolved_stops[run_end].position - position).abs() <= f32::EPSILON
    {
      run_end += 1;
    }

    let run_len = run_end - i;
    let run_start_index = preferred.saturating_sub(run_len.saturating_sub(1));
    for (offset, slot) in indices[i..run_end].iter_mut().enumerate() {
      let logical_index = run_start_index.saturating_add(offset).min(max_index);
      let stop_index = i + offset;
      let lower_bound = stop_index.min(max_index);
      let upper_bound = max_index.saturating_sub(stop_count - 1 - stop_index);
      *slot = logical_index.clamp(lower_bound, upper_bound);
    }

    i = run_end;
  }

  for i in 1..stop_count {
    indices[i] = indices[i].max(indices[i - 1].saturating_add(1));
  }

  for i in (0..stop_count.saturating_sub(1)).rev() {
    indices[i] = indices[i].min(indices[i + 1].saturating_sub(1));
  }

  indices
}

fn snap_stop_samples(
  typed_lut: &mut [PremultipliedColorU8],
  resolved_stops: &[ResolvedGradientStop],
  axis_length: f32,
) {
  if typed_lut.is_empty() || resolved_stops.is_empty() {
    return;
  }

  let stop_indices = assign_stop_sample_indices(resolved_stops, axis_length, typed_lut.len());
  for (stop, &sample_index) in resolved_stops.iter().zip(&stop_indices) {
    typed_lut[sample_index] = stop.color.into();
  }
}

#[inline(always)]
fn interpolation_position(left_position: f32, right_position: f32, sample_position: f32) -> f32 {
  let denominator = right_position - left_position;
  if denominator.abs() < f32::EPSILON {
    return 0.0;
  }

  ((sample_position - left_position) / denominator).clamp(0.0, 1.0)
}

/// Builds a pre-computed high-precision color lookup table for a gradient.
/// This allows O(1) color sampling instead of O(n) search + interpolation per pixel.
pub(crate) fn build_color_lut_with_interpolation(
  resolved_stops: &[ResolvedGradientStop],
  axis_length: f32,
  lut_size: usize,
  color_space: ColorSpaceTag,
  hue_direction: HueDirection,
) -> Vec<PremultipliedColorU8> {
  if lut_size == 0 {
    return Vec::new();
  }

  // Fast path: if only one color, fill just 16 bytes
  if resolved_stops.len() <= 1 {
    let color = resolved_stops
      .first()
      .map(|s| s.color)
      .unwrap_or(crate::layout::style::Color::transparent());

    return vec![color.into()];
  }

  let mut left_index = 0usize;
  let mut right_index = 1usize;
  let sample_step = if lut_size <= 1 {
    0.0
  } else {
    axis_length / (lut_size - 1) as f32
  };

  let mut write_sample = |sample_index: usize| -> PremultipliedColorU8 {
    let position_px = sample_index as f32 * sample_step;

    while right_index < resolved_stops.len() && resolved_stops[right_index].position <= position_px
    {
      left_index = right_index;
      right_index += 1;
    }

    let color = if right_index >= resolved_stops.len() {
      resolved_stops[left_index].color
    } else {
      let left_stop = &resolved_stops[left_index];
      let right_stop = &resolved_stops[right_index];
      if left_stop.color == right_stop.color {
        return left_stop.color.into();
      }

      let t = interpolation_position(left_stop.position, right_stop.position, position_px);
      if color_space == ColorSpaceTag::Srgb && hue_direction == HueDirection::Shorter {
        return interpolate_rgba_premultiplied(left_stop.color.into(), right_stop.color.into(), t);
      }

      interpolate_with_color_space(
        left_stop.color,
        right_stop.color,
        t,
        color_space,
        hue_direction,
      )
    };

    color.into()
  };

  let mut typed_lut = vec![PremultipliedColorU8::TRANSPARENT; lut_size];
  for (sample_index, chunk) in typed_lut.iter_mut().enumerate() {
    *chunk = write_sample(sample_index);
  }
  snap_stop_samples(&mut typed_lut, resolved_stops, axis_length);
  typed_lut
}

/// Calculates an adaptive LUT size based on the gradient axis length.
pub(crate) fn adaptive_lut_size(
  axis_length: f32,
  resolved_stops: &[ResolvedGradientStop],
) -> usize {
  adaptive_lut_size_with_visible_samples(
    (axis_length.ceil() as usize)
      .saturating_add(1)
      .max(MIN_GRADIENT_LUT_SIZE),
    axis_length,
    resolved_stops,
  )
}

pub(crate) fn adaptive_lut_size_with_visible_samples(
  visible_samples: usize,
  axis_length: f32,
  resolved_stops: &[ResolvedGradientStop],
) -> usize {
  let visible_samples = visible_samples.max(MIN_GRADIENT_LUT_SIZE);

  let min_interval = resolved_stops
    .windows(2)
    .map(|stops| stops[1].position - stops[0].position)
    .filter(|interval| *interval > f32::EPSILON)
    .fold(f32::INFINITY, f32::min);

  let segment_aware_size = if min_interval.is_finite() {
    ((axis_length / min_interval).ceil() as usize)
      .saturating_add(resolved_stops.len())
      .saturating_add(1)
      .max(MIN_GRADIENT_LUT_SIZE)
  } else {
    resolved_stops
      .len()
      .saturating_add(1)
      .max(MIN_GRADIENT_LUT_SIZE)
  };

  let size = visible_samples
    .max(segment_aware_size)
    .max(resolved_stops.len().saturating_mul(2))
    .max(MIN_GRADIENT_LUT_SIZE);
  size.min(MAX_GRADIENT_LUT_SIZE)
}

const UNDEFINED_POSITION: f32 = -1.0;

pub(crate) fn resolve_stops_along_axis(
  stops: &[GradientStop],
  axis_size_px: f32,
  context: &RenderContext,
) -> SmallVec<[ResolvedGradientStop; 4]> {
  let mut resolved: SmallVec<[ResolvedGradientStop; 4]> = SmallVec::new();
  let mut last_position = 0.0;

  for (i, step) in stops.iter().enumerate() {
    match step {
      GradientStop::ColorHint {
        color,
        hint: Some(hint),
      } => {
        let position = hint
          .0
          .to_px(&context.sizing, axis_size_px)
          .max(last_position);

        last_position = position;

        resolved.push(ResolvedGradientStop {
          color: color.resolve(context.current_color),
          position,
        });
      }
      GradientStop::ColorHint { color, hint: None } => {
        resolved.push(ResolvedGradientStop {
          color: color.resolve(context.current_color),
          position: UNDEFINED_POSITION,
        });
      }
      GradientStop::Hint(hint) => {
        let Some(before) = resolved.last() else {
          continue;
        };

        let Some(after_color) = stops.get(i + 1).and_then(|stop| match stop {
          GradientStop::ColorHint { color, hint: _ } => Some(color.resolve(context.current_color)),
          GradientStop::Hint(_) => None,
        }) else {
          continue;
        };

        let interpolated_color = interpolate_rgba(before.color, after_color, 0.5);

        let position = hint
          .0
          .to_px(&context.sizing, axis_size_px)
          .max(last_position);

        resolved.push(ResolvedGradientStop {
          color: interpolated_color,
          position,
        });

        last_position = position;
      }
    }
  }

  // If there are no color stops, return an empty vector
  if resolved.is_empty() {
    return resolved;
  }

  // if there is only one stop, treat it as pure color image
  if resolved.len() == 1 {
    if let Some(first_stop) = resolved.first_mut() {
      first_stop.position = axis_size_px;
    }

    return resolved;
  }

  if let Some(first_stop) = resolved.first_mut()
    && first_stop.position == UNDEFINED_POSITION
  {
    first_stop.position = 0.0;
  }

  if let Some(last_stop) = resolved.last_mut()
    && last_stop.position == UNDEFINED_POSITION
  {
    last_stop.position = axis_size_px;
  }

  // Distribute unspecified or non-increasing positions in pixel domain
  let mut i = 1usize;
  while i < resolved.len() - 1 {
    // if the position is defined and valid, skip it
    if resolved[i].position != UNDEFINED_POSITION {
      i += 1;
      continue;
    }

    let last_defined_position = resolved.get(i - 1).map(|s| s.position).unwrap_or(0.0);

    // try to find next defined position
    let next_index = resolved
      .iter()
      .skip(i + 1)
      .position(|s| s.position != UNDEFINED_POSITION)
      .map(|idx| i + 1 + idx)
      .unwrap_or(resolved.len() - 1);

    let next_position = resolved[next_index].position;

    // number of segments between last defined and next position
    let segments_count = (next_index - i + 1) as f32;
    let step_for_each_segment = (next_position - last_defined_position) / segments_count;

    // distribute the step evenly between the stops
    for j in i..next_index {
      let offset = (j - i + 1) as f32;
      resolved[j].position = last_defined_position + step_for_each_segment * offset;
    }

    i = next_index + 1;
  }

  resolved
}

#[cfg(test)]
mod tests {
  use image::{Rgba, RgbaImage};
  use taffy::Point;

  use crate::layout::Viewport;
  use crate::rendering::blend_pixel;
  use crate::{
    GlobalContext,
    layout::style::{BlendMode, Color, Length, StopPosition},
  };

  use super::*;

  #[derive(Debug, Clone, Copy)]
  struct MockTile {
    width: u32,
    height: u32,
  }

  #[derive(Debug, Clone, Copy)]
  struct MockRowState {
    value: usize,
    lut_len: usize,
  }

  impl GradientOverlayTile for MockTile {
    type RowState = MockRowState;

    fn width(&self) -> u32 {
      self.width
    }

    fn height(&self) -> u32 {
      self.height
    }

    fn lut_len(&self) -> usize {
      2
    }

    fn sample_at(&self, lut_idx: usize) -> PremultipliedColorU8 {
      match lut_idx {
        0 => Color([255, 0, 0, 255]).into(),
        _ => Color([0, 0, 255, 255]).into(),
      }
    }

    fn sample_pixel(&self, x: u32, y: u32) -> PremultipliedColorU8 {
      let lut_idx = ((x + y) % self.lut_len() as u32) as usize;
      self.sample_at(lut_idx)
    }

    fn begin_row(&self, src_x_start: u32, src_y: u32, lut_len: usize) -> Self::RowState {
      MockRowState {
        value: ((src_x_start + src_y) as usize) % lut_len.max(1),
        lut_len,
      }
    }

    fn next_lut_index(&self, row_state: &mut Self::RowState) -> usize {
      let value = row_state.value;
      row_state.value = (row_state.value + 1) % row_state.lut_len.max(1);
      value
    }
  }

  fn overlay_reference(bottom: &mut RgbaImage, tile: &MockTile, offset: Point<f32>) {
    let offset_x = offset.x as i32;
    let offset_y = offset.y as i32;
    let dest_x_min = offset_x.max(0);
    let dest_x_max = (offset_x + tile.width as i32).min(bottom.width() as i32);
    let dest_y_min = offset_y.max(0);
    let dest_y_max = (offset_y + tile.height as i32).min(bottom.height() as i32);
    for dest_y in dest_y_min..dest_y_max {
      let src_y = (dest_y - offset_y) as u32;
      let src_x_start = (dest_x_min - offset_x) as u32;
      let mut row_state = tile.begin_row(src_x_start, src_y, tile.lut_len());

      for dest_x in dest_x_min..dest_x_max {
        let lut_idx = tile.next_lut_index(&mut row_state);
        let pixel = tile.sample_at(lut_idx);
        let pixel = pixel.demultiply();
        let current = bottom.get_pixel_mut(dest_x as u32, dest_y as u32);
        blend_pixel(
          current,
          Rgba([pixel.red(), pixel.green(), pixel.blue(), pixel.alpha()]),
          BlendMode::Normal,
        );
      }
    }
  }

  #[test]
  fn test_overlay_gradient_tile_fast_matches_reference() {
    let tile = MockTile {
      width: 4,
      height: 3,
    };
    let offset = Point { x: 2.0, y: 1.0 };
    let mut actual = RgbaImage::from_pixel(10, 7, Rgba([20, 30, 40, 255]));
    let mut expected = actual.clone();

    overlay_gradient_tile_fast_normal_unconstrained_test(&mut actual, &tile, offset);
    overlay_reference(&mut expected, &tile, offset);

    assert_eq!(actual, expected);
  }

  #[test]
  fn test_resolve_stops_along_axis() {
    let stops = vec![
      GradientStop::ColorHint {
        color: Color([255, 0, 0, 255]).into(),
        hint: Some(StopPosition(Length::Px(10.0))),
      },
      GradientStop::ColorHint {
        color: Color([0, 255, 0, 255]).into(),
        hint: Some(StopPosition(Length::Px(20.0))),
      },
      GradientStop::ColorHint {
        color: Color([0, 0, 255, 255]).into(),
        hint: Some(StopPosition(Length::Percentage(30.0))),
      },
    ];

    let context = GlobalContext::default();
    let render_context = RenderContext::new_test(&context, Viewport::new((40, 40)));

    let width = render_context.sizing.viewport.size.width;

    assert!(width.is_some());

    let resolved =
      resolve_stops_along_axis(&stops, width.unwrap_or_default() as f32, &render_context);

    assert_eq!(
      resolved[0],
      ResolvedGradientStop {
        color: Color([255, 0, 0, 255]),
        position: 10.0,
      },
    );

    assert_eq!(
      resolved[1],
      ResolvedGradientStop {
        color: Color([0, 255, 0, 255]),
        position: 20.0,
      },
    );

    assert_eq!(
      resolved[2],
      ResolvedGradientStop {
        color: Color([0, 0, 255, 255]),
        position: 20.0, // since 30% (12px) is smaller than the last
      },
    );
  }

  #[test]
  fn test_distribute_evenly_between_positions() {
    let stops = vec![
      GradientStop::ColorHint {
        color: Color([255, 0, 0, 255]).into(),
        hint: None,
      },
      GradientStop::ColorHint {
        color: Color([0, 255, 0, 255]).into(),
        hint: None,
      },
      GradientStop::ColorHint {
        color: Color([0, 0, 255, 255]).into(),
        hint: None,
      },
    ];

    let context = GlobalContext::default();
    let render_context = RenderContext::new_test(&context, Viewport::new((40, 40)));

    let resolved = resolve_stops_along_axis(
      &stops,
      render_context
        .sizing
        .viewport
        .size
        .width
        .unwrap_or_default() as f32,
      &render_context,
    );

    assert_eq!(
      resolved.as_slice(),
      &[
        ResolvedGradientStop {
          color: Color([255, 0, 0, 255]),
          position: 0.0,
        },
        ResolvedGradientStop {
          color: Color([0, 255, 0, 255]),
          position: render_context
            .sizing
            .viewport
            .size
            .width
            .unwrap_or_default() as f32
            / 2.0,
        },
        ResolvedGradientStop {
          color: Color([0, 0, 255, 255]),
          position: render_context
            .sizing
            .viewport
            .size
            .width
            .unwrap_or_default() as f32,
        },
      ]
    );
  }

  #[test]
  fn test_hint_only() {
    let stops = vec![
      GradientStop::ColorHint {
        color: Color([255, 0, 0, 255]).into(),
        hint: None,
      },
      GradientStop::Hint(StopPosition(Length::Percentage(10.0))),
      GradientStop::ColorHint {
        color: Color([0, 0, 255, 255]).into(),
        hint: None,
      },
    ];

    let context = GlobalContext::default();
    let render_context = RenderContext::new_test(&context, Viewport::new((40, 40)));

    let resolved = resolve_stops_along_axis(
      &stops,
      render_context
        .sizing
        .viewport
        .size
        .width
        .unwrap_or_default() as f32,
      &render_context,
    );

    assert_eq!(
      resolved[0],
      ResolvedGradientStop {
        color: Color([255, 0, 0, 255]),
        position: 0.0,
      },
    );

    // the mid color between red and blue should be at 10%
    assert_eq!(
      resolved[1],
      ResolvedGradientStop {
        color: interpolate_rgba(Color([255, 0, 0, 255]), Color([0, 0, 255, 255]), 0.5),
        position: render_context
          .sizing
          .viewport
          .size
          .width
          .unwrap_or_default() as f32
          * 0.1,
      },
    );

    assert_eq!(
      resolved[2],
      ResolvedGradientStop {
        color: Color([0, 0, 255, 255]),
        position: render_context
          .sizing
          .viewport
          .size
          .width
          .unwrap_or_default() as f32,
      },
    );
  }

  #[test]
  fn test_adaptive_lut_size_grows_for_tight_stop_clusters() {
    let resolved = [
      ResolvedGradientStop {
        color: Color([255, 0, 0, 255]),
        position: 0.0,
      },
      ResolvedGradientStop {
        color: Color([0, 255, 0, 255]),
        position: 0.25,
      },
      ResolvedGradientStop {
        color: Color([0, 0, 255, 255]),
        position: 256.0,
      },
    ];

    let size = adaptive_lut_size(256.0, &resolved);

    assert!(size > 1025);
    assert!(size <= MAX_GRADIENT_LUT_SIZE);
  }

  #[test]
  fn test_build_color_lut_preserves_hard_stop_transition() {
    let resolved = [
      ResolvedGradientStop {
        color: Color([255, 0, 0, 255]),
        position: 0.0,
      },
      ResolvedGradientStop {
        color: Color([255, 0, 0, 255]),
        position: 8.0,
      },
      ResolvedGradientStop {
        color: Color([0, 0, 255, 255]),
        position: 8.0,
      },
      ResolvedGradientStop {
        color: Color([0, 0, 255, 255]),
        position: 16.0,
      },
    ];

    let lut = build_color_lut_with_interpolation(
      &resolved,
      16.0,
      17,
      ColorSpaceTag::Srgb,
      HueDirection::Shorter,
    );

    assert_eq!(lut[7], Color([255, 0, 0, 255]).into());
    assert_eq!(lut[8], Color([0, 0, 255, 255]).into());
  }

  #[test]
  fn test_build_color_lut_gives_distinct_samples_to_narrow_interval() {
    let resolved = [
      ResolvedGradientStop {
        color: Color([255, 0, 0, 255]),
        position: 0.0,
      },
      ResolvedGradientStop {
        color: Color([0, 255, 0, 255]),
        position: 0.05,
      },
      ResolvedGradientStop {
        color: Color([0, 0, 255, 255]),
        position: 32.0,
      },
    ];

    let lut_size = adaptive_lut_size(32.0, &resolved);
    let lut = build_color_lut_with_interpolation(
      &resolved,
      32.0,
      lut_size,
      ColorSpaceTag::Srgb,
      HueDirection::Shorter,
    );
    let stop_indices = assign_stop_sample_indices(&resolved, 32.0, lut.len());

    assert!(stop_indices[0] < stop_indices[1]);
    assert_eq!(lut[stop_indices[0]], resolved[0].color.into());
    assert_eq!(lut[stop_indices[1]], resolved[1].color.into());
  }

  #[test]
  fn test_build_color_lut_remains_monotonic_for_even_spacing() {
    let resolved = [
      ResolvedGradientStop {
        color: Color([0, 0, 0, 255]),
        position: 0.0,
      },
      ResolvedGradientStop {
        color: Color([255, 255, 255, 255]),
        position: 10.0,
      },
    ];

    let lut = build_color_lut_with_interpolation(
      &resolved,
      10.0,
      33,
      ColorSpaceTag::Srgb,
      HueDirection::Shorter,
    );

    for pair in lut.windows(2) {
      assert!(pair[0].red() <= pair[1].red());
      assert!(pair[0].green() <= pair[1].green());
      assert!(pair[0].blue() <= pair[1].blue());
      assert_eq!(pair[0].alpha(), 255);
      assert_eq!(pair[1].alpha(), 255);
    }
  }

  #[test]
  fn test_interpolate_rgba_uses_premultiplied_alpha() {
    let mixed = interpolate_rgba(Color([255, 255, 255, 255]), Color([0, 0, 0, 0]), 0.5);
    assert_eq!(mixed, Color([255, 255, 255, 128]));
  }
}