use core::arch::wasm32::*;
use super::{endian, *};
use crate::{ColorMatrix, row::scalar};
#[inline]
#[target_feature(enable = "simd128")]
unsafe fn deinterleave_xv36_8px<const BE: bool>(ptr: *const u16) -> (v128, v128, v128) {
unsafe {
let raw0 = endian::load_endian_u16x8::<BE>(ptr as *const u8); let raw1 = endian::load_endian_u16x8::<BE>(ptr.add(8) as *const u8); let raw2 = endian::load_endian_u16x8::<BE>(ptr.add(16) as *const u8); let raw3 = endian::load_endian_u16x8::<BE>(ptr.add(24) as *const u8);
let u_idx = i8x16(0, 1, 8, 9, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1);
let y_idx = i8x16(2, 3, 10, 11, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1);
let v_idx = i8x16(4, 5, 12, 13, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1);
let u0 = u8x16_swizzle(raw0, u_idx); let u1 = u8x16_swizzle(raw1, u_idx); let u2 = u8x16_swizzle(raw2, u_idx); let u3 = u8x16_swizzle(raw3, u_idx);
let y0 = u8x16_swizzle(raw0, y_idx);
let y1 = u8x16_swizzle(raw1, y_idx);
let y2 = u8x16_swizzle(raw2, y_idx);
let y3 = u8x16_swizzle(raw3, y_idx);
let v0 = u8x16_swizzle(raw0, v_idx);
let v1 = u8x16_swizzle(raw1, v_idx);
let v2 = u8x16_swizzle(raw2, v_idx);
let v3 = u8x16_swizzle(raw3, v_idx);
let u01 = i8x16_shuffle::<0, 1, 2, 3, 16, 17, 18, 19, 0, 0, 0, 0, 0, 0, 0, 0>(u0, u1);
let u23 = i8x16_shuffle::<0, 1, 2, 3, 16, 17, 18, 19, 0, 0, 0, 0, 0, 0, 0, 0>(u2, u3);
let u_raw = i8x16_shuffle::<0, 1, 2, 3, 4, 5, 6, 7, 16, 17, 18, 19, 20, 21, 22, 23>(u01, u23);
let y01 = i8x16_shuffle::<0, 1, 2, 3, 16, 17, 18, 19, 0, 0, 0, 0, 0, 0, 0, 0>(y0, y1);
let y23 = i8x16_shuffle::<0, 1, 2, 3, 16, 17, 18, 19, 0, 0, 0, 0, 0, 0, 0, 0>(y2, y3);
let y_raw = i8x16_shuffle::<0, 1, 2, 3, 4, 5, 6, 7, 16, 17, 18, 19, 20, 21, 22, 23>(y01, y23);
let v01 = i8x16_shuffle::<0, 1, 2, 3, 16, 17, 18, 19, 0, 0, 0, 0, 0, 0, 0, 0>(v0, v1);
let v23 = i8x16_shuffle::<0, 1, 2, 3, 16, 17, 18, 19, 0, 0, 0, 0, 0, 0, 0, 0>(v2, v3);
let v_raw = i8x16_shuffle::<0, 1, 2, 3, 4, 5, 6, 7, 16, 17, 18, 19, 20, 21, 22, 23>(v01, v23);
(u_raw, y_raw, v_raw)
}
}
#[inline]
#[target_feature(enable = "simd128")]
pub(crate) unsafe fn xv36_to_rgb_or_rgba_row<const ALPHA: bool, const BE: bool>(
packed: &[u16],
out: &mut [u8],
width: usize,
matrix: ColorMatrix,
full_range: bool,
) {
debug_assert!(packed.len() >= width * 4, "packed row too short");
let bpp: usize = if ALPHA { 4 } else { 3 };
debug_assert!(out.len() >= width * bpp, "out row too short");
let coeffs = scalar::Coefficients::for_matrix(matrix);
let (y_off, y_scale, c_scale) = scalar::range_params_n::<12, 8>(full_range);
let bias = scalar::chroma_bias::<12>();
const RND: i32 = 1 << 14;
unsafe {
let rnd_v = i32x4_splat(RND);
let y_off_v = i16x8_splat(y_off as i16);
let y_scale_v = i32x4_splat(y_scale);
let c_scale_v = i32x4_splat(c_scale);
let bias_v = i16x8_splat(bias as i16);
let cru = i32x4_splat(coeffs.r_u());
let crv = i32x4_splat(coeffs.r_v());
let cgu = i32x4_splat(coeffs.g_u());
let cgv = i32x4_splat(coeffs.g_v());
let cbu = i32x4_splat(coeffs.b_u());
let cbv = i32x4_splat(coeffs.b_v());
let mut x = 0usize;
while x + 8 <= width {
let (u_raw, y_raw, v_raw) = deinterleave_xv36_8px::<BE>(packed.as_ptr().add(x * 4));
let u_i16 = u16x8_shr(u_raw, 4);
let y_i16 = u16x8_shr(y_raw, 4);
let v_i16 = u16x8_shr(v_raw, 4);
let u_sub = i16x8_sub(u_i16, bias_v);
let v_sub = i16x8_sub(v_i16, bias_v);
let u_lo_i32 = i32x4_extend_low_i16x8(u_sub);
let u_hi_i32 = i32x4_extend_high_i16x8(u_sub);
let v_lo_i32 = i32x4_extend_low_i16x8(v_sub);
let v_hi_i32 = i32x4_extend_high_i16x8(v_sub);
let u_d_lo = q15_shift(i32x4_add(i32x4_mul(u_lo_i32, c_scale_v), rnd_v));
let u_d_hi = q15_shift(i32x4_add(i32x4_mul(u_hi_i32, c_scale_v), rnd_v));
let v_d_lo = q15_shift(i32x4_add(i32x4_mul(v_lo_i32, c_scale_v), rnd_v));
let v_d_hi = q15_shift(i32x4_add(i32x4_mul(v_hi_i32, c_scale_v), rnd_v));
let r_chroma = chroma_i16x8(cru, crv, u_d_lo, v_d_lo, u_d_hi, v_d_hi, rnd_v);
let g_chroma = chroma_i16x8(cgu, cgv, u_d_lo, v_d_lo, u_d_hi, v_d_hi, rnd_v);
let b_chroma = chroma_i16x8(cbu, cbv, u_d_lo, v_d_lo, u_d_hi, v_d_hi, rnd_v);
let y_scaled = scale_y(y_i16, y_off_v, y_scale_v, rnd_v);
let r_sum = i16x8_add_sat(y_scaled, r_chroma);
let g_sum = i16x8_add_sat(y_scaled, g_chroma);
let b_sum = i16x8_add_sat(y_scaled, b_chroma);
let r_u8 = u8x16_narrow_i16x8(r_sum, r_sum);
let g_u8 = u8x16_narrow_i16x8(g_sum, g_sum);
let b_u8 = u8x16_narrow_i16x8(b_sum, b_sum);
let mut r_tmp = [0u8; 16];
let mut g_tmp = [0u8; 16];
let mut b_tmp = [0u8; 16];
v128_store(r_tmp.as_mut_ptr().cast(), r_u8);
v128_store(g_tmp.as_mut_ptr().cast(), g_u8);
v128_store(b_tmp.as_mut_ptr().cast(), b_u8);
if ALPHA {
let dst = &mut out[x * 4..x * 4 + 8 * 4];
for i in 0..8 {
dst[i * 4] = r_tmp[i];
dst[i * 4 + 1] = g_tmp[i];
dst[i * 4 + 2] = b_tmp[i];
dst[i * 4 + 3] = 0xFF;
}
} else {
let dst = &mut out[x * 3..x * 3 + 8 * 3];
for i in 0..8 {
dst[i * 3] = r_tmp[i];
dst[i * 3 + 1] = g_tmp[i];
dst[i * 3 + 2] = b_tmp[i];
}
}
x += 8;
}
if x < width {
let tail_packed = &packed[x * 4..width * 4];
let tail_out = &mut out[x * bpp..width * bpp];
let tail_w = width - x;
scalar::xv36_to_rgb_or_rgba_row::<ALPHA, BE>(
tail_packed,
tail_out,
tail_w,
matrix,
full_range,
);
}
}
}
#[inline]
#[target_feature(enable = "simd128")]
pub(crate) unsafe fn xv36_to_rgb_u16_or_rgba_u16_row<const ALPHA: bool, const BE: bool>(
packed: &[u16],
out: &mut [u16],
width: usize,
matrix: ColorMatrix,
full_range: bool,
) {
debug_assert!(packed.len() >= width * 4, "packed row too short");
let bpp: usize = if ALPHA { 4 } else { 3 };
debug_assert!(out.len() >= width * bpp, "out row too short");
let coeffs = scalar::Coefficients::for_matrix(matrix);
let (y_off, y_scale, c_scale) = scalar::range_params_n::<12, 12>(full_range);
let bias = scalar::chroma_bias::<12>();
const RND: i32 = 1 << 14;
let out_max: i16 = 0x0FFF;
let alpha_u16: u16 = 0x0FFF;
unsafe {
let rnd_v = i32x4_splat(RND);
let y_off_v = i16x8_splat(y_off as i16);
let y_scale_v = i32x4_splat(y_scale);
let c_scale_v = i32x4_splat(c_scale);
let bias_v = i16x8_splat(bias as i16);
let max_v = i16x8_splat(out_max);
let zero_v = i16x8_splat(0);
let cru = i32x4_splat(coeffs.r_u());
let crv = i32x4_splat(coeffs.r_v());
let cgu = i32x4_splat(coeffs.g_u());
let cgv = i32x4_splat(coeffs.g_v());
let cbu = i32x4_splat(coeffs.b_u());
let cbv = i32x4_splat(coeffs.b_v());
let mut x = 0usize;
while x + 8 <= width {
let (u_raw, y_raw, v_raw) = deinterleave_xv36_8px::<BE>(packed.as_ptr().add(x * 4));
let u_i16 = u16x8_shr(u_raw, 4);
let y_i16 = u16x8_shr(y_raw, 4);
let v_i16 = u16x8_shr(v_raw, 4);
let u_sub = i16x8_sub(u_i16, bias_v);
let v_sub = i16x8_sub(v_i16, bias_v);
let u_lo_i32 = i32x4_extend_low_i16x8(u_sub);
let u_hi_i32 = i32x4_extend_high_i16x8(u_sub);
let v_lo_i32 = i32x4_extend_low_i16x8(v_sub);
let v_hi_i32 = i32x4_extend_high_i16x8(v_sub);
let u_d_lo = q15_shift(i32x4_add(i32x4_mul(u_lo_i32, c_scale_v), rnd_v));
let u_d_hi = q15_shift(i32x4_add(i32x4_mul(u_hi_i32, c_scale_v), rnd_v));
let v_d_lo = q15_shift(i32x4_add(i32x4_mul(v_lo_i32, c_scale_v), rnd_v));
let v_d_hi = q15_shift(i32x4_add(i32x4_mul(v_hi_i32, c_scale_v), rnd_v));
let r_chroma = chroma_i16x8(cru, crv, u_d_lo, v_d_lo, u_d_hi, v_d_hi, rnd_v);
let g_chroma = chroma_i16x8(cgu, cgv, u_d_lo, v_d_lo, u_d_hi, v_d_hi, rnd_v);
let b_chroma = chroma_i16x8(cbu, cbv, u_d_lo, v_d_lo, u_d_hi, v_d_hi, rnd_v);
let y_scaled = scale_y(y_i16, y_off_v, y_scale_v, rnd_v);
let r = clamp_u16_max_wasm(i16x8_add_sat(y_scaled, r_chroma), zero_v, max_v);
let g = clamp_u16_max_wasm(i16x8_add_sat(y_scaled, g_chroma), zero_v, max_v);
let b = clamp_u16_max_wasm(i16x8_add_sat(y_scaled, b_chroma), zero_v, max_v);
let mut r_tmp = [0u16; 8];
let mut g_tmp = [0u16; 8];
let mut b_tmp = [0u16; 8];
v128_store(r_tmp.as_mut_ptr().cast(), r);
v128_store(g_tmp.as_mut_ptr().cast(), g);
v128_store(b_tmp.as_mut_ptr().cast(), b);
if ALPHA {
let dst = &mut out[x * 4..x * 4 + 8 * 4];
for i in 0..8 {
dst[i * 4] = r_tmp[i];
dst[i * 4 + 1] = g_tmp[i];
dst[i * 4 + 2] = b_tmp[i];
dst[i * 4 + 3] = alpha_u16;
}
} else {
let dst = &mut out[x * 3..x * 3 + 8 * 3];
for i in 0..8 {
dst[i * 3] = r_tmp[i];
dst[i * 3 + 1] = g_tmp[i];
dst[i * 3 + 2] = b_tmp[i];
}
}
x += 8;
}
if x < width {
let tail_packed = &packed[x * 4..width * 4];
let tail_out = &mut out[x * bpp..width * bpp];
let tail_w = width - x;
scalar::xv36_to_rgb_u16_or_rgba_u16_row::<ALPHA, BE>(
tail_packed,
tail_out,
tail_w,
matrix,
full_range,
);
}
}
}
#[inline]
#[target_feature(enable = "simd128")]
pub(crate) unsafe fn xv36_to_luma_row<const BE: bool>(
packed: &[u16],
out: &mut [u8],
width: usize,
) {
debug_assert!(packed.len() >= width * 4);
debug_assert!(out.len() >= width);
unsafe {
let y_idx = i8x16(2, 3, 10, 11, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1);
let mut x = 0usize;
while x + 8 <= width {
let ptr = packed.as_ptr().add(x * 4);
let raw0 = endian::load_endian_u16x8::<BE>(ptr as *const u8); let raw1 = endian::load_endian_u16x8::<BE>(ptr.add(8) as *const u8); let raw2 = endian::load_endian_u16x8::<BE>(ptr.add(16) as *const u8); let raw3 = endian::load_endian_u16x8::<BE>(ptr.add(24) as *const u8);
let y0 = u8x16_swizzle(raw0, y_idx); let y1 = u8x16_swizzle(raw1, y_idx); let y2 = u8x16_swizzle(raw2, y_idx); let y3 = u8x16_swizzle(raw3, y_idx);
let y01 = i8x16_shuffle::<0, 1, 2, 3, 16, 17, 18, 19, 0, 0, 0, 0, 0, 0, 0, 0>(y0, y1);
let y23 = i8x16_shuffle::<0, 1, 2, 3, 16, 17, 18, 19, 0, 0, 0, 0, 0, 0, 0, 0>(y2, y3);
let y_vec = i8x16_shuffle::<0, 1, 2, 3, 4, 5, 6, 7, 16, 17, 18, 19, 20, 21, 22, 23>(y01, y23);
let y_shr = u16x8_shr(y_vec, 8);
let y_u8 = u8x16_narrow_i16x8(y_shr, y_shr);
let mut tmp = [0u8; 16];
v128_store(tmp.as_mut_ptr().cast(), y_u8);
out[x..x + 8].copy_from_slice(&tmp[..8]);
x += 8;
}
if x < width {
scalar::xv36_to_luma_row::<BE>(&packed[x * 4..width * 4], &mut out[x..width], width - x);
}
}
}
#[inline]
#[target_feature(enable = "simd128")]
pub(crate) unsafe fn xv36_to_luma_u16_row<const BE: bool>(
packed: &[u16],
out: &mut [u16],
width: usize,
) {
debug_assert!(packed.len() >= width * 4);
debug_assert!(out.len() >= width);
unsafe {
let y_idx = i8x16(2, 3, 10, 11, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1);
let mut x = 0usize;
while x + 8 <= width {
let ptr = packed.as_ptr().add(x * 4);
let raw0 = endian::load_endian_u16x8::<BE>(ptr as *const u8);
let raw1 = endian::load_endian_u16x8::<BE>(ptr.add(8) as *const u8);
let raw2 = endian::load_endian_u16x8::<BE>(ptr.add(16) as *const u8);
let raw3 = endian::load_endian_u16x8::<BE>(ptr.add(24) as *const u8);
let y0 = u8x16_swizzle(raw0, y_idx);
let y1 = u8x16_swizzle(raw1, y_idx);
let y2 = u8x16_swizzle(raw2, y_idx);
let y3 = u8x16_swizzle(raw3, y_idx);
let y01 = i8x16_shuffle::<0, 1, 2, 3, 16, 17, 18, 19, 0, 0, 0, 0, 0, 0, 0, 0>(y0, y1);
let y23 = i8x16_shuffle::<0, 1, 2, 3, 16, 17, 18, 19, 0, 0, 0, 0, 0, 0, 0, 0>(y2, y3);
let y_vec = i8x16_shuffle::<0, 1, 2, 3, 4, 5, 6, 7, 16, 17, 18, 19, 20, 21, 22, 23>(y01, y23);
let y_low = u16x8_shr(y_vec, 4);
v128_store(out.as_mut_ptr().add(x).cast(), y_low);
x += 8;
}
if x < width {
scalar::xv36_to_luma_u16_row::<BE>(&packed[x * 4..width * 4], &mut out[x..width], width - x);
}
}
}