#[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, y2xx_row_elems},
};
#[cfg_attr(not(tarpaulin), inline(always))]
pub fn y210_to_rgb_row_endian(
packed: &[u16],
rgb_out: &mut [u8],
width: usize,
matrix: ColorMatrix,
full_range: bool,
use_simd: bool,
big_endian: bool,
) {
assert!(
width.is_multiple_of(2),
"Y210 requires even width (4:2:2 chroma pair)"
);
assert!(
packed.len() >= y2xx_row_elems(width),
"packed row too short"
);
assert!(
rgb_out.len() >= rgb_row_bytes(width),
"rgb_out row too short"
);
macro_rules! dispatch_be {
($call_le:expr, $call_be:expr) => {
if big_endian { $call_be } else { $call_le }
};
}
if use_simd {
cfg_select! {
target_arch = "aarch64" => {
if neon_available() {
dispatch_be!(
unsafe { arch::neon::y2xx_n_to_rgb_or_rgba_row::<10, false, false>(packed, rgb_out, width, matrix, full_range); },
unsafe { arch::neon::y2xx_n_to_rgb_or_rgba_row::<10, false, true>(packed, rgb_out, width, matrix, full_range); }
);
return;
}
},
target_arch = "x86_64" => {
if avx512_available() {
dispatch_be!(
unsafe { arch::x86_avx512::y2xx_n_to_rgb_or_rgba_row::<10, false, false>(packed, rgb_out, width, matrix, full_range); },
unsafe { arch::x86_avx512::y2xx_n_to_rgb_or_rgba_row::<10, false, true>(packed, rgb_out, width, matrix, full_range); }
);
return;
}
if avx2_available() {
dispatch_be!(
unsafe { arch::x86_avx2::y2xx_n_to_rgb_or_rgba_row::<10, false, false>(packed, rgb_out, width, matrix, full_range); },
unsafe { arch::x86_avx2::y2xx_n_to_rgb_or_rgba_row::<10, false, true>(packed, rgb_out, width, matrix, full_range); }
);
return;
}
if sse41_available() {
dispatch_be!(
unsafe { arch::x86_sse41::y2xx_n_to_rgb_or_rgba_row::<10, false, false>(packed, rgb_out, width, matrix, full_range); },
unsafe { arch::x86_sse41::y2xx_n_to_rgb_or_rgba_row::<10, false, true>(packed, rgb_out, width, matrix, full_range); }
);
return;
}
},
target_arch = "wasm32" => {
if simd128_available() {
dispatch_be!(
unsafe { arch::wasm_simd128::y2xx_n_to_rgb_or_rgba_row::<10, false, false>(packed, rgb_out, width, matrix, full_range); },
unsafe { arch::wasm_simd128::y2xx_n_to_rgb_or_rgba_row::<10, false, true>(packed, rgb_out, width, matrix, full_range); }
);
return;
}
},
_ => {}
}
}
dispatch_be!(
scalar::y210_to_rgb_or_rgba_row::<false, false>(packed, rgb_out, width, matrix, full_range),
scalar::y210_to_rgb_or_rgba_row::<false, true>(packed, rgb_out, width, matrix, full_range)
);
}
#[cfg_attr(not(tarpaulin), inline(always))]
pub fn y210_to_rgb_row(
packed: &[u16],
rgb_out: &mut [u8],
width: usize,
matrix: ColorMatrix,
full_range: bool,
use_simd: bool,
) {
y210_to_rgb_row_endian(packed, rgb_out, width, matrix, full_range, use_simd, false)
}
#[cfg_attr(not(tarpaulin), inline(always))]
pub fn y210_to_rgba_row_endian(
packed: &[u16],
rgba_out: &mut [u8],
width: usize,
matrix: ColorMatrix,
full_range: bool,
use_simd: bool,
big_endian: bool,
) {
assert!(
width.is_multiple_of(2),
"Y210 requires even width (4:2:2 chroma pair)"
);
assert!(
packed.len() >= y2xx_row_elems(width),
"packed row too short"
);
assert!(
rgba_out.len() >= rgba_row_bytes(width),
"rgba_out row too short"
);
macro_rules! dispatch_be {
($call_le:expr, $call_be:expr) => {
if big_endian { $call_be } else { $call_le }
};
}
if use_simd {
cfg_select! {
target_arch = "aarch64" => {
if neon_available() {
dispatch_be!(
unsafe { arch::neon::y2xx_n_to_rgb_or_rgba_row::<10, true, false>(packed, rgba_out, width, matrix, full_range); },
unsafe { arch::neon::y2xx_n_to_rgb_or_rgba_row::<10, true, true>(packed, rgba_out, width, matrix, full_range); }
);
return;
}
},
target_arch = "x86_64" => {
if avx512_available() {
dispatch_be!(
unsafe { arch::x86_avx512::y2xx_n_to_rgb_or_rgba_row::<10, true, false>(packed, rgba_out, width, matrix, full_range); },
unsafe { arch::x86_avx512::y2xx_n_to_rgb_or_rgba_row::<10, true, true>(packed, rgba_out, width, matrix, full_range); }
);
return;
}
if avx2_available() {
dispatch_be!(
unsafe { arch::x86_avx2::y2xx_n_to_rgb_or_rgba_row::<10, true, false>(packed, rgba_out, width, matrix, full_range); },
unsafe { arch::x86_avx2::y2xx_n_to_rgb_or_rgba_row::<10, true, true>(packed, rgba_out, width, matrix, full_range); }
);
return;
}
if sse41_available() {
dispatch_be!(
unsafe { arch::x86_sse41::y2xx_n_to_rgb_or_rgba_row::<10, true, false>(packed, rgba_out, width, matrix, full_range); },
unsafe { arch::x86_sse41::y2xx_n_to_rgb_or_rgba_row::<10, true, true>(packed, rgba_out, width, matrix, full_range); }
);
return;
}
},
target_arch = "wasm32" => {
if simd128_available() {
dispatch_be!(
unsafe { arch::wasm_simd128::y2xx_n_to_rgb_or_rgba_row::<10, true, false>(packed, rgba_out, width, matrix, full_range); },
unsafe { arch::wasm_simd128::y2xx_n_to_rgb_or_rgba_row::<10, true, true>(packed, rgba_out, width, matrix, full_range); }
);
return;
}
},
_ => {}
}
}
dispatch_be!(
scalar::y210_to_rgb_or_rgba_row::<true, false>(packed, rgba_out, width, matrix, full_range),
scalar::y210_to_rgb_or_rgba_row::<true, true>(packed, rgba_out, width, matrix, full_range)
);
}
#[cfg_attr(not(tarpaulin), inline(always))]
pub fn y210_to_rgba_row(
packed: &[u16],
rgba_out: &mut [u8],
width: usize,
matrix: ColorMatrix,
full_range: bool,
use_simd: bool,
) {
y210_to_rgba_row_endian(packed, rgba_out, width, matrix, full_range, use_simd, false)
}
#[cfg_attr(not(tarpaulin), inline(always))]
pub fn y210_to_rgb_u16_row_endian(
packed: &[u16],
rgb_out: &mut [u16],
width: usize,
matrix: ColorMatrix,
full_range: bool,
use_simd: bool,
big_endian: bool,
) {
assert!(
width.is_multiple_of(2),
"Y210 requires even width (4:2:2 chroma pair)"
);
assert!(
packed.len() >= y2xx_row_elems(width),
"packed row too short"
);
assert!(
rgb_out.len() >= rgb_row_elems(width),
"rgb_out row too short"
);
macro_rules! dispatch_be {
($call_le:expr, $call_be:expr) => {
if big_endian { $call_be } else { $call_le }
};
}
if use_simd {
cfg_select! {
target_arch = "aarch64" => {
if neon_available() {
dispatch_be!(
unsafe { arch::neon::y2xx_n_to_rgb_u16_or_rgba_u16_row::<10, false, false>(packed, rgb_out, width, matrix, full_range); },
unsafe { arch::neon::y2xx_n_to_rgb_u16_or_rgba_u16_row::<10, false, true>(packed, rgb_out, width, matrix, full_range); }
);
return;
}
},
target_arch = "x86_64" => {
if avx512_available() {
dispatch_be!(
unsafe { arch::x86_avx512::y2xx_n_to_rgb_u16_or_rgba_u16_row::<10, false, false>(packed, rgb_out, width, matrix, full_range); },
unsafe { arch::x86_avx512::y2xx_n_to_rgb_u16_or_rgba_u16_row::<10, false, true>(packed, rgb_out, width, matrix, full_range); }
);
return;
}
if avx2_available() {
dispatch_be!(
unsafe { arch::x86_avx2::y2xx_n_to_rgb_u16_or_rgba_u16_row::<10, false, false>(packed, rgb_out, width, matrix, full_range); },
unsafe { arch::x86_avx2::y2xx_n_to_rgb_u16_or_rgba_u16_row::<10, false, true>(packed, rgb_out, width, matrix, full_range); }
);
return;
}
if sse41_available() {
dispatch_be!(
unsafe { arch::x86_sse41::y2xx_n_to_rgb_u16_or_rgba_u16_row::<10, false, false>(packed, rgb_out, width, matrix, full_range); },
unsafe { arch::x86_sse41::y2xx_n_to_rgb_u16_or_rgba_u16_row::<10, false, true>(packed, rgb_out, width, matrix, full_range); }
);
return;
}
},
target_arch = "wasm32" => {
if simd128_available() {
dispatch_be!(
unsafe { arch::wasm_simd128::y2xx_n_to_rgb_u16_or_rgba_u16_row::<10, false, false>(packed, rgb_out, width, matrix, full_range); },
unsafe { arch::wasm_simd128::y2xx_n_to_rgb_u16_or_rgba_u16_row::<10, false, true>(packed, rgb_out, width, matrix, full_range); }
);
return;
}
},
_ => {}
}
}
dispatch_be!(
scalar::y210_to_rgb_u16_or_rgba_u16_row::<false, false>(
packed, rgb_out, width, matrix, full_range
),
scalar::y210_to_rgb_u16_or_rgba_u16_row::<false, true>(
packed, rgb_out, width, matrix, full_range
)
);
}
#[cfg_attr(not(tarpaulin), inline(always))]
pub fn y210_to_rgb_u16_row(
packed: &[u16],
rgb_out: &mut [u16],
width: usize,
matrix: ColorMatrix,
full_range: bool,
use_simd: bool,
) {
y210_to_rgb_u16_row_endian(packed, rgb_out, width, matrix, full_range, use_simd, false)
}
#[cfg_attr(not(tarpaulin), inline(always))]
pub fn y210_to_rgba_u16_row_endian(
packed: &[u16],
rgba_out: &mut [u16],
width: usize,
matrix: ColorMatrix,
full_range: bool,
use_simd: bool,
big_endian: bool,
) {
assert!(
width.is_multiple_of(2),
"Y210 requires even width (4:2:2 chroma pair)"
);
assert!(
packed.len() >= y2xx_row_elems(width),
"packed row too short"
);
assert!(
rgba_out.len() >= rgba_row_elems(width),
"rgba_out row too short"
);
macro_rules! dispatch_be {
($call_le:expr, $call_be:expr) => {
if big_endian { $call_be } else { $call_le }
};
}
if use_simd {
cfg_select! {
target_arch = "aarch64" => {
if neon_available() {
dispatch_be!(
unsafe { arch::neon::y2xx_n_to_rgb_u16_or_rgba_u16_row::<10, true, false>(packed, rgba_out, width, matrix, full_range); },
unsafe { arch::neon::y2xx_n_to_rgb_u16_or_rgba_u16_row::<10, true, true>(packed, rgba_out, width, matrix, full_range); }
);
return;
}
},
target_arch = "x86_64" => {
if avx512_available() {
dispatch_be!(
unsafe { arch::x86_avx512::y2xx_n_to_rgb_u16_or_rgba_u16_row::<10, true, false>(packed, rgba_out, width, matrix, full_range); },
unsafe { arch::x86_avx512::y2xx_n_to_rgb_u16_or_rgba_u16_row::<10, true, true>(packed, rgba_out, width, matrix, full_range); }
);
return;
}
if avx2_available() {
dispatch_be!(
unsafe { arch::x86_avx2::y2xx_n_to_rgb_u16_or_rgba_u16_row::<10, true, false>(packed, rgba_out, width, matrix, full_range); },
unsafe { arch::x86_avx2::y2xx_n_to_rgb_u16_or_rgba_u16_row::<10, true, true>(packed, rgba_out, width, matrix, full_range); }
);
return;
}
if sse41_available() {
dispatch_be!(
unsafe { arch::x86_sse41::y2xx_n_to_rgb_u16_or_rgba_u16_row::<10, true, false>(packed, rgba_out, width, matrix, full_range); },
unsafe { arch::x86_sse41::y2xx_n_to_rgb_u16_or_rgba_u16_row::<10, true, true>(packed, rgba_out, width, matrix, full_range); }
);
return;
}
},
target_arch = "wasm32" => {
if simd128_available() {
dispatch_be!(
unsafe { arch::wasm_simd128::y2xx_n_to_rgb_u16_or_rgba_u16_row::<10, true, false>(packed, rgba_out, width, matrix, full_range); },
unsafe { arch::wasm_simd128::y2xx_n_to_rgb_u16_or_rgba_u16_row::<10, true, true>(packed, rgba_out, width, matrix, full_range); }
);
return;
}
},
_ => {}
}
}
dispatch_be!(
scalar::y210_to_rgb_u16_or_rgba_u16_row::<true, false>(
packed, rgba_out, width, matrix, full_range
),
scalar::y210_to_rgb_u16_or_rgba_u16_row::<true, true>(
packed, rgba_out, width, matrix, full_range
)
);
}
#[cfg_attr(not(tarpaulin), inline(always))]
pub fn y210_to_rgba_u16_row(
packed: &[u16],
rgba_out: &mut [u16],
width: usize,
matrix: ColorMatrix,
full_range: bool,
use_simd: bool,
) {
y210_to_rgba_u16_row_endian(packed, rgba_out, width, matrix, full_range, use_simd, false)
}
#[cfg_attr(not(tarpaulin), inline(always))]
pub fn y210_to_luma_row_endian(
packed: &[u16],
luma_out: &mut [u8],
width: usize,
use_simd: bool,
big_endian: bool,
) {
assert!(
width.is_multiple_of(2),
"Y210 requires even width (4:2:2 chroma pair)"
);
assert!(
packed.len() >= y2xx_row_elems(width),
"packed row too short"
);
assert!(luma_out.len() >= width, "luma_out row too short");
macro_rules! dispatch_be {
($call_le:expr, $call_be:expr) => {
if big_endian { $call_be } else { $call_le }
};
}
if use_simd {
cfg_select! {
target_arch = "aarch64" => {
if neon_available() {
dispatch_be!(
unsafe { arch::neon::y2xx_n_to_luma_row::<10, false>(packed, luma_out, width); },
unsafe { arch::neon::y2xx_n_to_luma_row::<10, true>(packed, luma_out, width); }
);
return;
}
},
target_arch = "x86_64" => {
if avx512_available() {
dispatch_be!(
unsafe { arch::x86_avx512::y2xx_n_to_luma_row::<10, false>(packed, luma_out, width); },
unsafe { arch::x86_avx512::y2xx_n_to_luma_row::<10, true>(packed, luma_out, width); }
);
return;
}
if avx2_available() {
dispatch_be!(
unsafe { arch::x86_avx2::y2xx_n_to_luma_row::<10, false>(packed, luma_out, width); },
unsafe { arch::x86_avx2::y2xx_n_to_luma_row::<10, true>(packed, luma_out, width); }
);
return;
}
if sse41_available() {
dispatch_be!(
unsafe { arch::x86_sse41::y2xx_n_to_luma_row::<10, false>(packed, luma_out, width); },
unsafe { arch::x86_sse41::y2xx_n_to_luma_row::<10, true>(packed, luma_out, width); }
);
return;
}
},
target_arch = "wasm32" => {
if simd128_available() {
dispatch_be!(
unsafe { arch::wasm_simd128::y2xx_n_to_luma_row::<10, false>(packed, luma_out, width); },
unsafe { arch::wasm_simd128::y2xx_n_to_luma_row::<10, true>(packed, luma_out, width); }
);
return;
}
},
_ => {}
}
}
dispatch_be!(
scalar::y210_to_luma_row::<false>(packed, luma_out, width),
scalar::y210_to_luma_row::<true>(packed, luma_out, width)
);
}
#[cfg_attr(not(tarpaulin), inline(always))]
pub fn y210_to_luma_row(packed: &[u16], luma_out: &mut [u8], width: usize, use_simd: bool) {
y210_to_luma_row_endian(packed, luma_out, width, use_simd, false)
}
#[cfg_attr(not(tarpaulin), inline(always))]
pub fn y210_to_luma_u16_row_endian(
packed: &[u16],
luma_out: &mut [u16],
width: usize,
use_simd: bool,
big_endian: bool,
) {
assert!(
width.is_multiple_of(2),
"Y210 requires even width (4:2:2 chroma pair)"
);
assert!(
packed.len() >= y2xx_row_elems(width),
"packed row too short"
);
assert!(luma_out.len() >= width, "luma_out row too short");
macro_rules! dispatch_be {
($call_le:expr, $call_be:expr) => {
if big_endian { $call_be } else { $call_le }
};
}
if use_simd {
cfg_select! {
target_arch = "aarch64" => {
if neon_available() {
dispatch_be!(
unsafe { arch::neon::y2xx_n_to_luma_u16_row::<10, false>(packed, luma_out, width); },
unsafe { arch::neon::y2xx_n_to_luma_u16_row::<10, true>(packed, luma_out, width); }
);
return;
}
},
target_arch = "x86_64" => {
if avx512_available() {
dispatch_be!(
unsafe { arch::x86_avx512::y2xx_n_to_luma_u16_row::<10, false>(packed, luma_out, width); },
unsafe { arch::x86_avx512::y2xx_n_to_luma_u16_row::<10, true>(packed, luma_out, width); }
);
return;
}
if avx2_available() {
dispatch_be!(
unsafe { arch::x86_avx2::y2xx_n_to_luma_u16_row::<10, false>(packed, luma_out, width); },
unsafe { arch::x86_avx2::y2xx_n_to_luma_u16_row::<10, true>(packed, luma_out, width); }
);
return;
}
if sse41_available() {
dispatch_be!(
unsafe { arch::x86_sse41::y2xx_n_to_luma_u16_row::<10, false>(packed, luma_out, width); },
unsafe { arch::x86_sse41::y2xx_n_to_luma_u16_row::<10, true>(packed, luma_out, width); }
);
return;
}
},
target_arch = "wasm32" => {
if simd128_available() {
dispatch_be!(
unsafe { arch::wasm_simd128::y2xx_n_to_luma_u16_row::<10, false>(packed, luma_out, width); },
unsafe { arch::wasm_simd128::y2xx_n_to_luma_u16_row::<10, true>(packed, luma_out, width); }
);
return;
}
},
_ => {}
}
}
dispatch_be!(
scalar::y210_to_luma_u16_row::<false>(packed, luma_out, width),
scalar::y210_to_luma_u16_row::<true>(packed, luma_out, width)
);
}
#[cfg_attr(not(tarpaulin), inline(always))]
pub fn y210_to_luma_u16_row(packed: &[u16], luma_out: &mut [u16], width: usize, use_simd: bool) {
y210_to_luma_u16_row_endian(packed, luma_out, width, use_simd, false)
}
#[cfg(all(test, feature = "std"))]
mod tests {
use super::*;
fn y210_quad(y0: u16, u: u16, y1: u16, v: u16) -> [u16; 4] {
[
(y0 & 0x3FF) << 6,
(u & 0x3FF) << 6,
(y1 & 0x3FF) << 6,
(v & 0x3FF) << 6,
]
}
fn solid_y210(width: usize, y: u16, u: u16, v: u16) -> std::vec::Vec<u16> {
let mut buf = std::vec::Vec::with_capacity(width * 2);
for _ in 0..(width / 2) {
buf.extend_from_slice(&y210_quad(y, u, y, v));
}
buf
.iter()
.map(|v| u16::from_ne_bytes(v.to_le_bytes()))
.collect()
}
#[test]
fn y210_dispatchers_route_with_simd_false() {
let buf = solid_y210(8, 512, 512, 512);
let mut rgb = [0u8; 8 * 3];
y210_to_rgb_row(&buf, &mut rgb, 8, ColorMatrix::Bt709, true, 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];
y210_to_rgba_row(&buf, &mut rgba, 8, ColorMatrix::Bt709, true, 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];
y210_to_rgb_u16_row(&buf, &mut rgb_u16, 8, ColorMatrix::Bt709, true, 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];
y210_to_rgba_u16_row(&buf, &mut rgba_u16, 8, ColorMatrix::Bt709, true, false);
for px in rgba_u16.chunks(4) {
assert_eq!(px[3], 1023);
}
let mut luma = [0u8; 8];
y210_to_luma_row(&buf, &mut luma, 8, false);
for &y in &luma {
assert_eq!(y, (512u16 >> 2) as u8);
}
let mut luma_u16 = [0u16; 8];
y210_to_luma_u16_row(&buf, &mut luma_u16, 8, false);
for &y in &luma_u16 {
assert_eq!(y, 512);
}
}
}