use core::arch::wasm32::*;
use super::*;
use crate::{ColorMatrix, row::scalar};
const HOST_NATIVE_BE: bool = cfg!(target_endian = "big");
#[inline]
#[target_feature(enable = "simd128")]
unsafe fn unpack_y216_8px_wasm(ptr: *const u16) -> (v128, v128, v128) {
unsafe {
let lo = v128_load(ptr.cast());
let hi = v128_load(ptr.add(8).cast());
let y_idx = i8x16(0, 1, 4, 5, 8, 9, 12, 13, -1, -1, -1, -1, -1, -1, -1, -1);
let y_lo = u8x16_swizzle(lo, y_idx); let y_hi = u8x16_swizzle(hi, y_idx); let y_vec = i8x16_shuffle::<0, 1, 2, 3, 4, 5, 6, 7, 16, 17, 18, 19, 20, 21, 22, 23>(y_lo, y_hi);
let c_idx = i8x16(2, 3, 6, 7, 10, 11, 14, 15, -1, -1, -1, -1, -1, -1, -1, -1);
let c_lo = u8x16_swizzle(lo, c_idx);
let c_hi = u8x16_swizzle(hi, c_idx);
let chroma =
i8x16_shuffle::<0, 1, 2, 3, 4, 5, 6, 7, 16, 17, 18, 19, 20, 21, 22, 23>(c_lo, c_hi);
let u_idx = i8x16(0, 1, 4, 5, 8, 9, 12, 13, -1, -1, -1, -1, -1, -1, -1, -1);
let v_idx = i8x16(2, 3, 6, 7, 10, 11, 14, 15, -1, -1, -1, -1, -1, -1, -1, -1);
let u_vec = u8x16_swizzle(chroma, u_idx);
let v_vec = u8x16_swizzle(chroma, v_idx);
(y_vec, u_vec, v_vec)
}
}
#[inline]
#[target_feature(enable = "simd128")]
pub(crate) unsafe fn y216_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!(width.is_multiple_of(2), "Y216 requires even width");
debug_assert!(packed.len() >= width * 2);
let bpp: usize = if ALPHA { 4 } else { 3 };
debug_assert!(out.len() >= width * bpp);
let coeffs = scalar::Coefficients::for_matrix(matrix);
let (y_off, y_scale, c_scale) = scalar::range_params_n::<16, 8>(full_range);
const RND: i32 = 1 << 14;
unsafe {
let mut x = 0usize;
if BE == HOST_NATIVE_BE {
let rnd_v = i32x4_splat(RND);
let y_off32_v = i32x4_splat(y_off);
let y_scale_v = i32x4_splat(y_scale);
let c_scale_v = i32x4_splat(c_scale);
let bias16_v = i16x8_splat(-32768i16);
let alpha_u8 = u8x16_splat(0xFF);
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());
while x + 16 <= width {
let (y_lo_vec, u_lo_vec, v_lo_vec) = unpack_y216_8px_wasm(packed.as_ptr().add(x * 2));
let (y_hi_vec, u_hi_vec, v_hi_vec) = unpack_y216_8px_wasm(packed.as_ptr().add(x * 2 + 16));
let u_lo_i16 = i16x8_sub(u_lo_vec, bias16_v);
let v_lo_i16 = i16x8_sub(v_lo_vec, bias16_v);
let u_hi_i16 = i16x8_sub(u_hi_vec, bias16_v);
let v_hi_i16 = i16x8_sub(v_hi_vec, bias16_v);
let u_lo_lo = i32x4_extend_low_i16x8(u_lo_i16);
let u_lo_hi = i32x4_extend_high_i16x8(u_lo_i16);
let v_lo_lo = i32x4_extend_low_i16x8(v_lo_i16);
let v_lo_hi = i32x4_extend_high_i16x8(v_lo_i16);
let u_hi_lo = i32x4_extend_low_i16x8(u_hi_i16);
let u_hi_hi = i32x4_extend_high_i16x8(u_hi_i16);
let v_hi_lo = i32x4_extend_low_i16x8(v_hi_i16);
let v_hi_hi = i32x4_extend_high_i16x8(v_hi_i16);
let u_d_lo_lo = q15_shift(i32x4_add(i32x4_mul(u_lo_lo, c_scale_v), rnd_v));
let u_d_lo_hi = q15_shift(i32x4_add(i32x4_mul(u_lo_hi, c_scale_v), rnd_v));
let v_d_lo_lo = q15_shift(i32x4_add(i32x4_mul(v_lo_lo, c_scale_v), rnd_v));
let v_d_lo_hi = q15_shift(i32x4_add(i32x4_mul(v_lo_hi, c_scale_v), rnd_v));
let u_d_hi_lo = q15_shift(i32x4_add(i32x4_mul(u_hi_lo, c_scale_v), rnd_v));
let u_d_hi_hi = q15_shift(i32x4_add(i32x4_mul(u_hi_hi, c_scale_v), rnd_v));
let v_d_hi_lo = q15_shift(i32x4_add(i32x4_mul(v_hi_lo, c_scale_v), rnd_v));
let v_d_hi_hi = q15_shift(i32x4_add(i32x4_mul(v_hi_hi, c_scale_v), rnd_v));
let r_chroma_lo = chroma_i16x8(cru, crv, u_d_lo_lo, v_d_lo_lo, u_d_lo_hi, v_d_lo_hi, rnd_v);
let g_chroma_lo = chroma_i16x8(cgu, cgv, u_d_lo_lo, v_d_lo_lo, u_d_lo_hi, v_d_lo_hi, rnd_v);
let b_chroma_lo = chroma_i16x8(cbu, cbv, u_d_lo_lo, v_d_lo_lo, u_d_lo_hi, v_d_lo_hi, rnd_v);
let r_chroma_hi = chroma_i16x8(cru, crv, u_d_hi_lo, v_d_hi_lo, u_d_hi_hi, v_d_hi_hi, rnd_v);
let g_chroma_hi = chroma_i16x8(cgu, cgv, u_d_hi_lo, v_d_hi_lo, u_d_hi_hi, v_d_hi_hi, rnd_v);
let b_chroma_hi = chroma_i16x8(cbu, cbv, u_d_hi_lo, v_d_hi_lo, u_d_hi_hi, v_d_hi_hi, rnd_v);
let r_dup_lo = dup_lo(r_chroma_lo);
let g_dup_lo = dup_lo(g_chroma_lo);
let b_dup_lo = dup_lo(b_chroma_lo);
let r_dup_hi = dup_lo(r_chroma_hi);
let g_dup_hi = dup_lo(g_chroma_hi);
let b_dup_hi = dup_lo(b_chroma_hi);
let y_lo_scaled = scale_y_u16_wasm(y_lo_vec, y_off32_v, y_scale_v, rnd_v);
let y_hi_scaled = scale_y_u16_wasm(y_hi_vec, y_off32_v, y_scale_v, rnd_v);
let r_lo = i16x8_add_sat(y_lo_scaled, r_dup_lo);
let r_hi = i16x8_add_sat(y_hi_scaled, r_dup_hi);
let g_lo = i16x8_add_sat(y_lo_scaled, g_dup_lo);
let g_hi = i16x8_add_sat(y_hi_scaled, g_dup_hi);
let b_lo = i16x8_add_sat(y_lo_scaled, b_dup_lo);
let b_hi = i16x8_add_sat(y_hi_scaled, b_dup_hi);
let r_u8 = u8x16_narrow_i16x8(r_lo, r_hi);
let g_u8 = u8x16_narrow_i16x8(g_lo, g_hi);
let b_u8 = u8x16_narrow_i16x8(b_lo, b_hi);
if ALPHA {
write_rgba_16(r_u8, g_u8, b_u8, alpha_u8, out.as_mut_ptr().add(x * 4));
} else {
write_rgb_16(r_u8, g_u8, b_u8, out.as_mut_ptr().add(x * 3));
}
x += 16;
}
}
if x < width {
let tail_packed = &packed[x * 2..width * 2];
let tail_out = &mut out[x * bpp..width * bpp];
let tail_w = width - x;
scalar::y216_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 y216_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!(width.is_multiple_of(2), "Y216 requires even width");
debug_assert!(packed.len() >= width * 2);
let bpp: usize = if ALPHA { 4 } else { 3 };
debug_assert!(out.len() >= width * bpp);
let coeffs = scalar::Coefficients::for_matrix(matrix);
let (y_off, y_scale, c_scale) = scalar::range_params_n::<16, 16>(full_range);
const RND_I64: i64 = 1 << 14;
const RND_I32: i32 = 1 << 14;
unsafe {
let mut x = 0usize;
if BE == HOST_NATIVE_BE {
let alpha_u16 = u16x8_splat(0xFFFF);
let rnd_i64 = i64x2_splat(RND_I64);
let rnd_i32 = i32x4_splat(RND_I32);
let y_off32 = i32x4_splat(y_off);
let y_scale_i64 = i64x2_splat(y_scale as i64);
let c_scale_i32 = i32x4_splat(c_scale);
let bias16 = i16x8_splat(-32768i16);
let cru = i64x2_extend_low_i32x4(i32x4_splat(coeffs.r_u()));
let crv = i64x2_extend_low_i32x4(i32x4_splat(coeffs.r_v()));
let cgu = i64x2_extend_low_i32x4(i32x4_splat(coeffs.g_u()));
let cgv = i64x2_extend_low_i32x4(i32x4_splat(coeffs.g_v()));
let cbu = i64x2_extend_low_i32x4(i32x4_splat(coeffs.b_u()));
let cbv = i64x2_extend_low_i32x4(i32x4_splat(coeffs.b_v()));
while x + 8 <= width {
let (y_vec, u_vec, v_vec) = unpack_y216_8px_wasm(packed.as_ptr().add(x * 2));
let u_i16 = i16x8_sub(u_vec, bias16);
let v_i16 = i16x8_sub(v_vec, bias16);
let u_i32 = i32x4_extend_low_i16x8(u_i16);
let v_i32 = i32x4_extend_low_i16x8(v_i16);
let u_d = i32x4_shr(i32x4_add(i32x4_mul(u_i32, c_scale_i32), rnd_i32), 15);
let v_d = i32x4_shr(i32x4_add(i32x4_mul(v_i32, c_scale_i32), rnd_i32), 15);
let u_d_lo = i64x2_extend_low_i32x4(u_d);
let u_d_hi = i64x2_extend_high_i32x4(u_d);
let v_d_lo = i64x2_extend_low_i32x4(v_d);
let v_d_hi = i64x2_extend_high_i32x4(v_d);
let r_ch_lo = chroma_i64x2_wasm(cru, crv, u_d_lo, v_d_lo, rnd_i64);
let r_ch_hi = chroma_i64x2_wasm(cru, crv, u_d_hi, v_d_hi, rnd_i64);
let g_ch_lo = chroma_i64x2_wasm(cgu, cgv, u_d_lo, v_d_lo, rnd_i64);
let g_ch_hi = chroma_i64x2_wasm(cgu, cgv, u_d_hi, v_d_hi, rnd_i64);
let b_ch_lo = chroma_i64x2_wasm(cbu, cbv, u_d_lo, v_d_lo, rnd_i64);
let b_ch_hi = chroma_i64x2_wasm(cbu, cbv, u_d_hi, v_d_hi, rnd_i64);
let r_ch_i32 = combine_i64x2_pair_to_i32x4(r_ch_lo, r_ch_hi);
let g_ch_i32 = combine_i64x2_pair_to_i32x4(g_ch_lo, g_ch_hi);
let b_ch_i32 = combine_i64x2_pair_to_i32x4(b_ch_lo, b_ch_hi);
let (r_dup_lo, r_dup_hi) = chroma_dup_i32x4_u16(r_ch_i32);
let (g_dup_lo, g_dup_hi) = chroma_dup_i32x4_u16(g_ch_i32);
let (b_dup_lo, b_dup_hi) = chroma_dup_i32x4_u16(b_ch_i32);
let y_lo_u32 = u32x4_extend_low_u16x8(y_vec);
let y_hi_u32 = u32x4_extend_high_u16x8(y_vec);
let y_lo_i32 = i32x4_sub(y_lo_u32, y_off32);
let y_hi_i32 = i32x4_sub(y_hi_u32, y_off32);
let y_lo_scaled = scale_y_i32x4_i64_wasm(y_lo_i32, y_scale_i64, rnd_i64);
let y_hi_scaled = scale_y_i32x4_i64_wasm(y_hi_i32, y_scale_i64, rnd_i64);
let r_u16 = u16x8_narrow_i32x4(
i32x4_add(y_lo_scaled, r_dup_lo),
i32x4_add(y_hi_scaled, r_dup_hi),
);
let g_u16 = u16x8_narrow_i32x4(
i32x4_add(y_lo_scaled, g_dup_lo),
i32x4_add(y_hi_scaled, g_dup_hi),
);
let b_u16 = u16x8_narrow_i32x4(
i32x4_add(y_lo_scaled, b_dup_lo),
i32x4_add(y_hi_scaled, b_dup_hi),
);
if ALPHA {
write_rgba_u16_8(r_u16, g_u16, b_u16, alpha_u16, out.as_mut_ptr().add(x * 4));
} else {
write_rgb_u16_8(r_u16, g_u16, b_u16, out.as_mut_ptr().add(x * 3));
}
x += 8;
}
}
if x < width {
let tail_packed = &packed[x * 2..width * 2];
let tail_out = &mut out[x * bpp..width * bpp];
let tail_w = width - x;
scalar::y216_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 y216_to_luma_row<const BE: bool>(
packed: &[u16],
luma_out: &mut [u8],
width: usize,
) {
debug_assert!(width.is_multiple_of(2));
debug_assert!(packed.len() >= width * 2);
debug_assert!(luma_out.len() >= width);
unsafe {
let mut x = 0usize;
if BE == HOST_NATIVE_BE {
let y_idx = i8x16(0, 1, 4, 5, 8, 9, 12, 13, -1, -1, -1, -1, -1, -1, -1, -1);
while x + 16 <= width {
let lo0 = v128_load(packed.as_ptr().add(x * 2).cast());
let lo1 = v128_load(packed.as_ptr().add(x * 2 + 8).cast());
let y_lo0 = u8x16_swizzle(lo0, y_idx);
let y_lo1 = u8x16_swizzle(lo1, y_idx);
let y_lo =
i8x16_shuffle::<0, 1, 2, 3, 4, 5, 6, 7, 16, 17, 18, 19, 20, 21, 22, 23>(y_lo0, y_lo1);
let hi0 = v128_load(packed.as_ptr().add(x * 2 + 16).cast());
let hi1 = v128_load(packed.as_ptr().add(x * 2 + 24).cast());
let y_hi0 = u8x16_swizzle(hi0, y_idx);
let y_hi1 = u8x16_swizzle(hi1, y_idx);
let y_hi =
i8x16_shuffle::<0, 1, 2, 3, 4, 5, 6, 7, 16, 17, 18, 19, 20, 21, 22, 23>(y_hi0, y_hi1);
let y_shr_lo = u16x8_shr(y_lo, 8);
let y_shr_hi = u16x8_shr(y_hi, 8);
let y_u8 = u8x16_narrow_i16x8(y_shr_lo, y_shr_hi);
v128_store(luma_out.as_mut_ptr().add(x).cast(), y_u8);
x += 16;
}
}
if x < width {
let tail_packed = &packed[x * 2..width * 2];
let tail_out = &mut luma_out[x..width];
let tail_w = width - x;
scalar::y216_to_luma_row::<BE>(tail_packed, tail_out, tail_w);
}
}
}
#[inline]
#[target_feature(enable = "simd128")]
pub(crate) unsafe fn y216_to_luma_u16_row<const BE: bool>(
packed: &[u16],
luma_out: &mut [u16],
width: usize,
) {
debug_assert!(width.is_multiple_of(2));
debug_assert!(packed.len() >= width * 2);
debug_assert!(luma_out.len() >= width);
unsafe {
let mut x = 0usize;
if BE == HOST_NATIVE_BE {
let y_idx = i8x16(0, 1, 4, 5, 8, 9, 12, 13, -1, -1, -1, -1, -1, -1, -1, -1);
while x + 16 <= width {
let lo0 = v128_load(packed.as_ptr().add(x * 2).cast());
let lo1 = v128_load(packed.as_ptr().add(x * 2 + 8).cast());
let y_lo0 = u8x16_swizzle(lo0, y_idx);
let y_lo1 = u8x16_swizzle(lo1, y_idx);
let y_lo =
i8x16_shuffle::<0, 1, 2, 3, 4, 5, 6, 7, 16, 17, 18, 19, 20, 21, 22, 23>(y_lo0, y_lo1);
let hi0 = v128_load(packed.as_ptr().add(x * 2 + 16).cast());
let hi1 = v128_load(packed.as_ptr().add(x * 2 + 24).cast());
let y_hi0 = u8x16_swizzle(hi0, y_idx);
let y_hi1 = u8x16_swizzle(hi1, y_idx);
let y_hi =
i8x16_shuffle::<0, 1, 2, 3, 4, 5, 6, 7, 16, 17, 18, 19, 20, 21, 22, 23>(y_hi0, y_hi1);
v128_store(luma_out.as_mut_ptr().add(x).cast(), y_lo);
v128_store(luma_out.as_mut_ptr().add(x + 8).cast(), y_hi);
x += 16;
}
}
if x < width {
let tail_packed = &packed[x * 2..width * 2];
let tail_out = &mut luma_out[x..width];
let tail_w = width - x;
scalar::y216_to_luma_u16_row::<BE>(tail_packed, tail_out, tail_w);
}
}
}