#[cfg(any(
target_arch = "aarch64",
target_arch = "x86_64",
target_arch = "wasm32"
))]
use crate::row::arch;
#[cfg(target_arch = "aarch64")]
use crate::row::neon_available;
#[cfg(target_arch = "wasm32")]
use crate::row::simd128_available;
#[cfg(target_arch = "x86_64")]
use crate::row::{avx2_available, avx512_available, sse41_available};
use crate::{
ColorMatrix,
row::{rgb_row_bytes, rgb_row_elems, rgba_row_bytes, rgba_row_elems, scalar},
};
#[cfg_attr(not(tarpaulin), inline(always))]
pub fn v410_to_rgb_row(
packed: &[u32],
rgb_out: &mut [u8],
width: usize,
matrix: ColorMatrix,
full_range: bool,
use_simd: bool,
be_input: bool,
) {
assert!(packed.len() >= width, "packed row too short");
assert!(
rgb_out.len() >= rgb_row_bytes(width),
"rgb_out row too short"
);
if use_simd {
cfg_select! {
target_arch = "aarch64" => {
if neon_available() {
if be_input {
unsafe { arch::neon::v410_to_rgb_or_rgba_row::<false, true>(packed, rgb_out, width, matrix, full_range); }
} else {
unsafe { arch::neon::v410_to_rgb_or_rgba_row::<false, false>(packed, rgb_out, width, matrix, full_range); }
}
return;
}
},
target_arch = "x86_64" => {
if avx512_available() {
if be_input {
unsafe { arch::x86_avx512::v410_to_rgb_or_rgba_row::<false, true>(packed, rgb_out, width, matrix, full_range); }
} else {
unsafe { arch::x86_avx512::v410_to_rgb_or_rgba_row::<false, false>(packed, rgb_out, width, matrix, full_range); }
}
return;
}
if avx2_available() {
if be_input {
unsafe { arch::x86_avx2::v410_to_rgb_or_rgba_row::<false, true>(packed, rgb_out, width, matrix, full_range); }
} else {
unsafe { arch::x86_avx2::v410_to_rgb_or_rgba_row::<false, false>(packed, rgb_out, width, matrix, full_range); }
}
return;
}
if sse41_available() {
if be_input {
unsafe { arch::x86_sse41::v410_to_rgb_or_rgba_row::<false, true>(packed, rgb_out, width, matrix, full_range); }
} else {
unsafe { arch::x86_sse41::v410_to_rgb_or_rgba_row::<false, false>(packed, rgb_out, width, matrix, full_range); }
}
return;
}
},
target_arch = "wasm32" => {
if simd128_available() {
if be_input {
unsafe { arch::wasm_simd128::v410_to_rgb_or_rgba_row::<false, true>(packed, rgb_out, width, matrix, full_range); }
} else {
unsafe { arch::wasm_simd128::v410_to_rgb_or_rgba_row::<false, false>(packed, rgb_out, width, matrix, full_range); }
}
return;
}
},
_ => {}
}
}
if be_input {
scalar::v410_to_rgb_or_rgba_row::<false, true>(packed, rgb_out, width, matrix, full_range);
} else {
scalar::v410_to_rgb_or_rgba_row::<false, false>(packed, rgb_out, width, matrix, full_range);
}
}
#[cfg_attr(not(tarpaulin), inline(always))]
pub fn v410_to_rgba_row(
packed: &[u32],
rgba_out: &mut [u8],
width: usize,
matrix: ColorMatrix,
full_range: bool,
use_simd: bool,
be_input: bool,
) {
assert!(packed.len() >= width, "packed row too short");
assert!(
rgba_out.len() >= rgba_row_bytes(width),
"rgba_out row too short"
);
if use_simd {
cfg_select! {
target_arch = "aarch64" => {
if neon_available() {
if be_input {
unsafe { arch::neon::v410_to_rgb_or_rgba_row::<true, true>(packed, rgba_out, width, matrix, full_range); }
} else {
unsafe { arch::neon::v410_to_rgb_or_rgba_row::<true, false>(packed, rgba_out, width, matrix, full_range); }
}
return;
}
},
target_arch = "x86_64" => {
if avx512_available() {
if be_input {
unsafe { arch::x86_avx512::v410_to_rgb_or_rgba_row::<true, true>(packed, rgba_out, width, matrix, full_range); }
} else {
unsafe { arch::x86_avx512::v410_to_rgb_or_rgba_row::<true, false>(packed, rgba_out, width, matrix, full_range); }
}
return;
}
if avx2_available() {
if be_input {
unsafe { arch::x86_avx2::v410_to_rgb_or_rgba_row::<true, true>(packed, rgba_out, width, matrix, full_range); }
} else {
unsafe { arch::x86_avx2::v410_to_rgb_or_rgba_row::<true, false>(packed, rgba_out, width, matrix, full_range); }
}
return;
}
if sse41_available() {
if be_input {
unsafe { arch::x86_sse41::v410_to_rgb_or_rgba_row::<true, true>(packed, rgba_out, width, matrix, full_range); }
} else {
unsafe { arch::x86_sse41::v410_to_rgb_or_rgba_row::<true, false>(packed, rgba_out, width, matrix, full_range); }
}
return;
}
},
target_arch = "wasm32" => {
if simd128_available() {
if be_input {
unsafe { arch::wasm_simd128::v410_to_rgb_or_rgba_row::<true, true>(packed, rgba_out, width, matrix, full_range); }
} else {
unsafe { arch::wasm_simd128::v410_to_rgb_or_rgba_row::<true, false>(packed, rgba_out, width, matrix, full_range); }
}
return;
}
},
_ => {}
}
}
if be_input {
scalar::v410_to_rgb_or_rgba_row::<true, true>(packed, rgba_out, width, matrix, full_range);
} else {
scalar::v410_to_rgb_or_rgba_row::<true, false>(packed, rgba_out, width, matrix, full_range);
}
}
#[cfg_attr(not(tarpaulin), inline(always))]
pub fn v410_to_rgb_u16_row(
packed: &[u32],
rgb_out: &mut [u16],
width: usize,
matrix: ColorMatrix,
full_range: bool,
use_simd: bool,
be_input: bool,
) {
assert!(packed.len() >= width, "packed row too short");
assert!(
rgb_out.len() >= rgb_row_elems(width),
"rgb_out row too short"
);
if use_simd {
cfg_select! {
target_arch = "aarch64" => {
if neon_available() {
if be_input {
unsafe { arch::neon::v410_to_rgb_u16_or_rgba_u16_row::<false, true>(packed, rgb_out, width, matrix, full_range); }
} else {
unsafe { arch::neon::v410_to_rgb_u16_or_rgba_u16_row::<false, false>(packed, rgb_out, width, matrix, full_range); }
}
return;
}
},
target_arch = "x86_64" => {
if avx512_available() {
if be_input {
unsafe { arch::x86_avx512::v410_to_rgb_u16_or_rgba_u16_row::<false, true>(packed, rgb_out, width, matrix, full_range); }
} else {
unsafe { arch::x86_avx512::v410_to_rgb_u16_or_rgba_u16_row::<false, false>(packed, rgb_out, width, matrix, full_range); }
}
return;
}
if avx2_available() {
if be_input {
unsafe { arch::x86_avx2::v410_to_rgb_u16_or_rgba_u16_row::<false, true>(packed, rgb_out, width, matrix, full_range); }
} else {
unsafe { arch::x86_avx2::v410_to_rgb_u16_or_rgba_u16_row::<false, false>(packed, rgb_out, width, matrix, full_range); }
}
return;
}
if sse41_available() {
if be_input {
unsafe { arch::x86_sse41::v410_to_rgb_u16_or_rgba_u16_row::<false, true>(packed, rgb_out, width, matrix, full_range); }
} else {
unsafe { arch::x86_sse41::v410_to_rgb_u16_or_rgba_u16_row::<false, false>(packed, rgb_out, width, matrix, full_range); }
}
return;
}
},
target_arch = "wasm32" => {
if simd128_available() {
if be_input {
unsafe { arch::wasm_simd128::v410_to_rgb_u16_or_rgba_u16_row::<false, true>(packed, rgb_out, width, matrix, full_range); }
} else {
unsafe { arch::wasm_simd128::v410_to_rgb_u16_or_rgba_u16_row::<false, false>(packed, rgb_out, width, matrix, full_range); }
}
return;
}
},
_ => {}
}
}
if be_input {
scalar::v410_to_rgb_u16_or_rgba_u16_row::<false, true>(
packed, rgb_out, width, matrix, full_range,
);
} else {
scalar::v410_to_rgb_u16_or_rgba_u16_row::<false, false>(
packed, rgb_out, width, matrix, full_range,
);
}
}
#[cfg_attr(not(tarpaulin), inline(always))]
pub fn v410_to_rgba_u16_row(
packed: &[u32],
rgba_out: &mut [u16],
width: usize,
matrix: ColorMatrix,
full_range: bool,
use_simd: bool,
be_input: bool,
) {
assert!(packed.len() >= width, "packed row too short");
assert!(
rgba_out.len() >= rgba_row_elems(width),
"rgba_out row too short"
);
if use_simd {
cfg_select! {
target_arch = "aarch64" => {
if neon_available() {
if be_input {
unsafe { arch::neon::v410_to_rgb_u16_or_rgba_u16_row::<true, true>(packed, rgba_out, width, matrix, full_range); }
} else {
unsafe { arch::neon::v410_to_rgb_u16_or_rgba_u16_row::<true, false>(packed, rgba_out, width, matrix, full_range); }
}
return;
}
},
target_arch = "x86_64" => {
if avx512_available() {
if be_input {
unsafe { arch::x86_avx512::v410_to_rgb_u16_or_rgba_u16_row::<true, true>(packed, rgba_out, width, matrix, full_range); }
} else {
unsafe { arch::x86_avx512::v410_to_rgb_u16_or_rgba_u16_row::<true, false>(packed, rgba_out, width, matrix, full_range); }
}
return;
}
if avx2_available() {
if be_input {
unsafe { arch::x86_avx2::v410_to_rgb_u16_or_rgba_u16_row::<true, true>(packed, rgba_out, width, matrix, full_range); }
} else {
unsafe { arch::x86_avx2::v410_to_rgb_u16_or_rgba_u16_row::<true, false>(packed, rgba_out, width, matrix, full_range); }
}
return;
}
if sse41_available() {
if be_input {
unsafe { arch::x86_sse41::v410_to_rgb_u16_or_rgba_u16_row::<true, true>(packed, rgba_out, width, matrix, full_range); }
} else {
unsafe { arch::x86_sse41::v410_to_rgb_u16_or_rgba_u16_row::<true, false>(packed, rgba_out, width, matrix, full_range); }
}
return;
}
},
target_arch = "wasm32" => {
if simd128_available() {
if be_input {
unsafe { arch::wasm_simd128::v410_to_rgb_u16_or_rgba_u16_row::<true, true>(packed, rgba_out, width, matrix, full_range); }
} else {
unsafe { arch::wasm_simd128::v410_to_rgb_u16_or_rgba_u16_row::<true, false>(packed, rgba_out, width, matrix, full_range); }
}
return;
}
},
_ => {}
}
}
if be_input {
scalar::v410_to_rgb_u16_or_rgba_u16_row::<true, true>(
packed, rgba_out, width, matrix, full_range,
);
} else {
scalar::v410_to_rgb_u16_or_rgba_u16_row::<true, false>(
packed, rgba_out, width, matrix, full_range,
);
}
}
#[cfg_attr(not(tarpaulin), inline(always))]
pub fn v410_to_luma_row(
packed: &[u32],
luma_out: &mut [u8],
width: usize,
use_simd: bool,
be_input: bool,
) {
assert!(packed.len() >= width, "packed row too short");
assert!(luma_out.len() >= width, "luma_out row too short");
if use_simd {
cfg_select! {
target_arch = "aarch64" => {
if neon_available() {
if be_input {
unsafe { arch::neon::v410_to_luma_row::<true>(packed, luma_out, width); }
} else {
unsafe { arch::neon::v410_to_luma_row::<false>(packed, luma_out, width); }
}
return;
}
},
target_arch = "x86_64" => {
if avx512_available() {
if be_input {
unsafe { arch::x86_avx512::v410_to_luma_row::<true>(packed, luma_out, width); }
} else {
unsafe { arch::x86_avx512::v410_to_luma_row::<false>(packed, luma_out, width); }
}
return;
}
if avx2_available() {
if be_input {
unsafe { arch::x86_avx2::v410_to_luma_row::<true>(packed, luma_out, width); }
} else {
unsafe { arch::x86_avx2::v410_to_luma_row::<false>(packed, luma_out, width); }
}
return;
}
if sse41_available() {
if be_input {
unsafe { arch::x86_sse41::v410_to_luma_row::<true>(packed, luma_out, width); }
} else {
unsafe { arch::x86_sse41::v410_to_luma_row::<false>(packed, luma_out, width); }
}
return;
}
},
target_arch = "wasm32" => {
if simd128_available() {
if be_input {
unsafe { arch::wasm_simd128::v410_to_luma_row::<true>(packed, luma_out, width); }
} else {
unsafe { arch::wasm_simd128::v410_to_luma_row::<false>(packed, luma_out, width); }
}
return;
}
},
_ => {}
}
}
if be_input {
scalar::v410_to_luma_row::<true>(packed, luma_out, width);
} else {
scalar::v410_to_luma_row::<false>(packed, luma_out, width);
}
}
#[cfg_attr(not(tarpaulin), inline(always))]
pub fn v410_to_luma_u16_row(
packed: &[u32],
luma_out: &mut [u16],
width: usize,
use_simd: bool,
be_input: bool,
) {
assert!(packed.len() >= width, "packed row too short");
assert!(luma_out.len() >= width, "luma_out row too short");
if use_simd {
cfg_select! {
target_arch = "aarch64" => {
if neon_available() {
if be_input {
unsafe { arch::neon::v410_to_luma_u16_row::<true>(packed, luma_out, width); }
} else {
unsafe { arch::neon::v410_to_luma_u16_row::<false>(packed, luma_out, width); }
}
return;
}
},
target_arch = "x86_64" => {
if avx512_available() {
if be_input {
unsafe { arch::x86_avx512::v410_to_luma_u16_row::<true>(packed, luma_out, width); }
} else {
unsafe { arch::x86_avx512::v410_to_luma_u16_row::<false>(packed, luma_out, width); }
}
return;
}
if avx2_available() {
if be_input {
unsafe { arch::x86_avx2::v410_to_luma_u16_row::<true>(packed, luma_out, width); }
} else {
unsafe { arch::x86_avx2::v410_to_luma_u16_row::<false>(packed, luma_out, width); }
}
return;
}
if sse41_available() {
if be_input {
unsafe { arch::x86_sse41::v410_to_luma_u16_row::<true>(packed, luma_out, width); }
} else {
unsafe { arch::x86_sse41::v410_to_luma_u16_row::<false>(packed, luma_out, width); }
}
return;
}
},
target_arch = "wasm32" => {
if simd128_available() {
if be_input {
unsafe { arch::wasm_simd128::v410_to_luma_u16_row::<true>(packed, luma_out, width); }
} else {
unsafe { arch::wasm_simd128::v410_to_luma_u16_row::<false>(packed, luma_out, width); }
}
return;
}
},
_ => {}
}
}
if be_input {
scalar::v410_to_luma_u16_row::<true>(packed, luma_out, width);
} else {
scalar::v410_to_luma_u16_row::<false>(packed, luma_out, width);
}
}
#[cfg(all(test, feature = "std"))]
mod tests {
use super::*;
fn pack_v410(u: u32, y: u32, v: u32) -> u32 {
debug_assert!(u < 1024 && y < 1024 && v < 1024);
(v << 20) | (y << 10) | u
}
fn solid_v410(width: usize, u: u32, y: u32, v: u32) -> std::vec::Vec<u32> {
(0..width)
.map(|_| u32::from_ne_bytes(pack_v410(u, y, v).to_le_bytes()))
.collect()
}
fn solid_v410_be(width: usize, u: u32, y: u32, v: u32) -> std::vec::Vec<u32> {
(0..width)
.map(|_| u32::from_ne_bytes(pack_v410(u, y, v).to_be_bytes()))
.collect()
}
#[test]
#[should_panic(expected = "packed row too short")]
fn v410_dispatcher_rejects_short_packed() {
let packed = [0u32; 2];
let mut rgb = [0u8; 4 * 3];
v410_to_rgb_row(&packed, &mut rgb, 4, ColorMatrix::Bt709, true, false, false);
}
#[test]
#[should_panic(expected = "rgb_out row too short")]
fn v410_dispatcher_rejects_short_output() {
let packed = [0u32; 4];
let mut rgb = [0u8; 2];
v410_to_rgb_row(&packed, &mut rgb, 4, ColorMatrix::Bt709, true, false, false);
}
#[test]
fn v410_dispatchers_route_with_simd_false() {
let buf = solid_v410(8, 512, 512, 512);
let mut rgb = [0u8; 8 * 3];
v410_to_rgb_row(&buf, &mut rgb, 8, ColorMatrix::Bt709, true, false, false);
for px in rgb.chunks(3) {
assert!(px[0].abs_diff(128) <= 1);
assert_eq!(px[0], px[1]);
assert_eq!(px[1], px[2]);
}
let mut rgba = [0u8; 8 * 4];
v410_to_rgba_row(&buf, &mut rgba, 8, ColorMatrix::Bt709, true, false, false);
for px in rgba.chunks(4) {
assert!(px[0].abs_diff(128) <= 1);
assert_eq!(px[3], 0xFF);
}
let mut rgb_u16 = [0u16; 8 * 3];
v410_to_rgb_u16_row(
&buf,
&mut rgb_u16,
8,
ColorMatrix::Bt709,
true,
false,
false,
);
for px in rgb_u16.chunks(3) {
assert!(px[0].abs_diff(512) <= 2);
assert_eq!(px[0], px[1]);
assert_eq!(px[1], px[2]);
}
let mut rgba_u16 = [0u16; 8 * 4];
v410_to_rgba_u16_row(
&buf,
&mut rgba_u16,
8,
ColorMatrix::Bt709,
true,
false,
false,
);
for px in rgba_u16.chunks(4) {
assert_eq!(px[3], 1023);
}
let mut luma = [0u8; 8];
v410_to_luma_row(&buf, &mut luma, 8, false, false);
for &y in &luma {
assert_eq!(y, (512u32 >> 2) as u8);
}
let mut luma_u16 = [0u16; 8];
v410_to_luma_u16_row(&buf, &mut luma_u16, 8, false, false);
for &y in &luma_u16 {
assert_eq!(y, 512);
}
}
#[test]
fn v410_be_and_le_dispatchers_agree() {
let le_buf = solid_v410(8, 512, 512, 512);
let be_buf = solid_v410_be(8, 512, 512, 512);
let mut rgb_le = [0u8; 8 * 3];
let mut rgb_be = [0u8; 8 * 3];
v410_to_rgb_row(
&le_buf,
&mut rgb_le,
8,
ColorMatrix::Bt709,
true,
false,
false,
);
v410_to_rgb_row(
&be_buf,
&mut rgb_be,
8,
ColorMatrix::Bt709,
true,
false,
true,
);
assert_eq!(
rgb_le, rgb_be,
"LE and BE must produce identical RGB output"
);
let mut luma_le = [0u8; 8];
let mut luma_be = [0u8; 8];
v410_to_luma_row(&le_buf, &mut luma_le, 8, false, false);
v410_to_luma_row(&be_buf, &mut luma_be, 8, false, true);
assert_eq!(
luma_le, luma_be,
"LE and BE must produce identical luma output"
);
}
}