use super::{MMB16Tpl, MMB32Tpl};
use crate::mem as basic;
use crate::utils::*;
#[cfg(target_arch = "x86")]
use core::arch::x86 as mmx;
#[cfg(target_arch = "x86_64")]
use core::arch::x86_64 as mmx;
use mmx::__m128i;
use mmx::_mm_cmpeq_epi8;
use mmx::_mm_load_si128;
use mmx::_mm_loadu_si128;
use mmx::_mm_movemask_epi8;
use mmx::__m256i;
use mmx::_mm256_cmpeq_epi8;
use mmx::_mm256_load_si256;
use mmx::_mm256_loadu_si256;
use mmx::_mm256_movemask_epi8;
#[cfg(any(target_arch = "x86_64", target_arch = "x86"))]
use super::cpuid;
use core::sync::atomic::AtomicPtr;
use core::sync::atomic::Ordering;
type FuncType = fn(&[u8], B1Tpl) -> Option<usize>;
const FUNC: FuncType = fnptr_setup_func;
static FUNC_PTR_ATOM: AtomicPtr<FuncType> = AtomicPtr::new(FUNC as *mut FuncType);
#[inline(never)]
fn fnptr_setup_func(buf: &[u8], needle: B1Tpl) -> Option<usize> {
#[cfg(target_arch = "x86_64")]
let func = if cpuid::has_avx2() {
_memchr_tpl_avx2
} else {
_memchr_tpl_sse2
};
#[cfg(target_arch = "x86")]
let func = if cpuid::has_avx2() {
_memchr_tpl_avx2
} else if cpuid::has_sse2() {
_memchr_tpl_sse2
} else {
_memchr_tpl_basic
};
FUNC_PTR_ATOM.store(func as *mut FuncType, Ordering::Relaxed);
unsafe { func(buf, needle) }
}
#[inline(always)]
pub(crate) fn _memchr_tpl_impl(buf: &[u8], needle: B1Tpl) -> Option<usize> {
let func_u = FUNC_PTR_ATOM.load(Ordering::Relaxed);
#[allow(clippy::crosspointer_transmute)]
let func: FuncType = unsafe { core::mem::transmute(func_u) };
func(buf, needle)
}
unsafe fn _memchr_tpl_basic(buf: &[u8], needle: B1Tpl) -> Option<usize> {
basic::_memchr_tpl_impl(buf, needle)
}
#[cfg(any(target_arch = "x86_64", target_arch = "x86"))]
#[target_feature(enable = "sse2")]
#[allow(clippy::missing_safety_doc)]
pub unsafe fn _memchr_tpl_sse2(buf: &[u8], needle: B1Tpl) -> Option<usize> {
_memchr_tpl_sse2_impl(buf, needle)
}
#[cfg(any(target_arch = "x86_64", target_arch = "x86"))]
#[target_feature(enable = "avx2")]
#[allow(clippy::missing_safety_doc)]
pub unsafe fn _memchr_tpl_avx2(buf: &[u8], needle: B1Tpl) -> Option<usize> {
_memchr_tpl_avx2_impl(buf, needle)
}
#[inline(always)]
fn _memchr_tpl_sse2_impl(buf: &[u8], needle: B1Tpl) -> Option<usize> {
let buf_len = buf.len();
let mut buf_ptr = buf.as_ptr();
let start_ptr = buf_ptr;
let end_ptr = unsafe { buf_ptr.add(buf_len) };
buf_ptr.prefetch_read_data();
if buf_len >= 16 {
let cc: MMB16Tpl = needle.into();
{
if !buf_ptr.is_aligned_u128() {
#[cfg(not(feature = "test_alignment_check"))]
{
let r = unsafe { _chr_tpl_c16_uu_x1(buf_ptr, cc, start_ptr) };
if r.is_some() {
return r;
}
let remaining_align = 0x10_usize - ((buf_ptr as usize) & 0x0F_usize);
buf_ptr = unsafe { buf_ptr.add(remaining_align) };
}
#[cfg(feature = "test_alignment_check")]
{
let r = basic::_chr_tpl_to_aligned_u128(buf_ptr, needle, start_ptr);
if let Some(p) = r.0 {
buf_ptr = p;
} else if let Some(v) = r.1 {
return Some(v);
}
}
}
}
{
let (r, p) = _unroll_loop_with_prefetch::<8, 16, _>(buf_ptr, end_ptr, |p| unsafe {
_chr_tpl_c16_aa_x1(p, cc, start_ptr)
});
if r.is_some() {
return r;
}
buf_ptr = p;
}
{
let (r, p) = _unroll_loop::<1, 16, _>(buf_ptr, end_ptr, |p| unsafe {
_chr_tpl_c16_aa_x1(p, cc, start_ptr)
});
if r.is_some() {
return r;
}
buf_ptr = p;
}
}
let cc: B8Tpl = needle.into();
basic::_memchr_tpl_remaining_15_bytes_impl(buf_ptr, cc, start_ptr, end_ptr)
}
#[inline(always)]
fn _memchr_tpl_avx2_impl(buf: &[u8], needle: B1Tpl) -> Option<usize> {
let buf_len = buf.len();
let mut buf_ptr = buf.as_ptr();
let start_ptr = buf_ptr;
let end_ptr = unsafe { buf_ptr.add(buf_len) };
buf_ptr.prefetch_read_data();
if buf_len >= 32 {
let cc: MMB32Tpl = needle.into();
{
if !buf_ptr.is_aligned_u256() {
#[cfg(not(feature = "test_alignment_check"))]
{
let r = unsafe { _chr_tpl_c32_uu_x1(buf_ptr, cc, start_ptr) };
if r.is_some() {
return r;
}
let remaining_align = 0x20_usize - ((buf_ptr as usize) & 0x1F_usize);
buf_ptr = unsafe { buf_ptr.add(remaining_align) };
}
#[cfg(feature = "test_alignment_check")]
{
let r = basic::_chr_tpl_to_aligned_u256(buf_ptr, needle, start_ptr);
if let Some(p) = r.0 {
buf_ptr = p;
} else if let Some(v) = r.1 {
return Some(v);
}
}
}
}
{
let (r, p) = _unroll_loop::<1, 32, _>(buf_ptr, end_ptr, |p| unsafe {
_chr_tpl_c32_aa_x1(p, cc, start_ptr)
});
if r.is_some() {
return r;
}
buf_ptr = p;
}
{
let cc: MMB16Tpl = needle.into();
let (r, p) = _unroll_loop::<1, 16, _>(buf_ptr, end_ptr, |p| unsafe {
_chr_tpl_c16_aa_x1(p, cc, start_ptr)
});
if r.is_some() {
return r;
}
buf_ptr = p;
}
} else if buf_len >= 16 {
{
let cc: MMB16Tpl = needle.into();
if buf_ptr.is_aligned_u128() {
let (r, p) = _unroll_loop::<1, 16, _>(buf_ptr, end_ptr, |p| unsafe {
_chr_tpl_c16_aa_x1(p, cc, start_ptr)
});
if r.is_some() {
return r;
}
buf_ptr = p;
} else {
#[cfg(not(feature = "test_alignment_check"))]
{
let (r, p) = _unroll_loop::<1, 16, _>(buf_ptr, end_ptr, |p| unsafe {
_chr_tpl_c16_uu_x1(p, cc, start_ptr)
});
if r.is_some() {
return r;
}
buf_ptr = p;
}
#[cfg(feature = "test_alignment_check")]
{
let r = basic::_chr_tpl_to_aligned_u128(buf_ptr, needle, start_ptr);
if let Some(p) = r.0 {
buf_ptr = p;
} else if let Some(v) = r.1 {
return Some(v);
}
let (r, p) = _unroll_loop::<1, 16, _>(buf_ptr, end_ptr, |p| unsafe {
_chr_tpl_c16_aa_x1(p, cc, start_ptr)
});
if r.is_some() {
return r;
}
buf_ptr = p;
}
}
}
}
let cc: B8Tpl = needle.into();
basic::_memchr_tpl_remaining_15_bytes_impl(buf_ptr, cc, start_ptr, end_ptr)
}
#[inline(always)]
fn _return_chr_tpl<T, PU>(base: T, mask_a: PU, mask_b: PU, mask_c: PU) -> Option<usize>
where
T: core::ops::Add<usize, Output = usize>,
PU: BitOrt,
{
if !mask_a.is_zeros() {
let idx1 = mask_a.trailing_zeros() as usize;
if !mask_b.is_zeros() {
let idx2 = mask_b.trailing_zeros() as usize;
if !mask_c.is_zeros() {
let idx3 = mask_c.trailing_zeros() as usize;
Some(base + idx1.min(idx2.min(idx3)))
} else {
Some(base + idx1.min(idx2))
}
} else if !mask_c.is_zeros() {
let idx3 = mask_c.trailing_zeros() as usize;
Some(base + idx1.min(idx3))
} else {
Some(base + idx1)
}
} else if !mask_b.is_zeros() {
let idx2 = mask_b.trailing_zeros() as usize;
if !mask_c.is_zeros() {
let idx3 = mask_c.trailing_zeros() as usize;
Some(base + idx2.min(idx3))
} else {
Some(base + idx2)
}
} else if !mask_c.is_zeros() {
let idx3 = mask_c.trailing_zeros() as usize;
Some(base + idx3)
} else {
None
}
}
#[inline(always)]
unsafe fn _chr_tpl_c16_uu_x1(
buf_ptr: *const u8,
mm_c16: MMB16Tpl,
st_ptr: *const u8,
) -> Option<usize> {
unsafe {
let mm_0 = _mm_loadu_si128(buf_ptr as *const __m128i);
let mm_0_eq_a = _mm_cmpeq_epi8(mm_0, mm_c16.v1);
let mm_0_eq_b = _mm_cmpeq_epi8(mm_0, mm_c16.v2);
let mm_0_eq_c = _mm_cmpeq_epi8(mm_0, mm_c16.v3);
let mask_0_a = _mm_movemask_epi8(mm_0_eq_a) as u32;
let mask_0_b = _mm_movemask_epi8(mm_0_eq_b) as u32;
let mask_0_c = _mm_movemask_epi8(mm_0_eq_c) as u32;
let base = buf_ptr.usz_offset_from(st_ptr);
_return_chr_tpl(base, mask_0_a, mask_0_b, mask_0_c)
}
}
#[inline(always)]
unsafe fn _chr_tpl_c16_aa_x1(
buf_ptr: *const u8,
mm_c16: MMB16Tpl,
st_ptr: *const u8,
) -> Option<usize> {
unsafe {
let mm_0 = _mm_load_si128(buf_ptr as *const __m128i);
let mm_0_eq_a = _mm_cmpeq_epi8(mm_0, mm_c16.v1);
let mm_0_eq_b = _mm_cmpeq_epi8(mm_0, mm_c16.v2);
let mm_0_eq_c = _mm_cmpeq_epi8(mm_0, mm_c16.v3);
let mask_0_a = _mm_movemask_epi8(mm_0_eq_a) as u32;
let mask_0_b = _mm_movemask_epi8(mm_0_eq_b) as u32;
let mask_0_c = _mm_movemask_epi8(mm_0_eq_c) as u32;
let base = buf_ptr.usz_offset_from(st_ptr);
_return_chr_tpl(base, mask_0_a, mask_0_b, mask_0_c)
}
}
#[inline(always)]
unsafe fn _chr_tpl_c32_uu_x1(
buf_ptr: *const u8,
mm_c32: MMB32Tpl,
st_ptr: *const u8,
) -> Option<usize> {
unsafe {
let mm_0 = _mm256_loadu_si256(buf_ptr as *const __m256i);
let mm_0_eq_a = _mm256_cmpeq_epi8(mm_0, mm_c32.v1);
let mm_0_eq_b = _mm256_cmpeq_epi8(mm_0, mm_c32.v2);
let mm_0_eq_c = _mm256_cmpeq_epi8(mm_0, mm_c32.v3);
let mask_0_a = _mm256_movemask_epi8(mm_0_eq_a) as u32;
let mask_0_b = _mm256_movemask_epi8(mm_0_eq_b) as u32;
let mask_0_c = _mm256_movemask_epi8(mm_0_eq_c) as u32;
let base = buf_ptr.usz_offset_from(st_ptr);
_return_chr_tpl(base, mask_0_a, mask_0_b, mask_0_c)
}
}
#[inline(always)]
unsafe fn _chr_tpl_c32_aa_x1(
buf_ptr: *const u8,
mm_c32: MMB32Tpl,
st_ptr: *const u8,
) -> Option<usize> {
unsafe {
let mm_0 = _mm256_load_si256(buf_ptr as *const __m256i);
let mm_0_eq_a = _mm256_cmpeq_epi8(mm_0, mm_c32.v1);
let mm_0_eq_b = _mm256_cmpeq_epi8(mm_0, mm_c32.v2);
let mm_0_eq_c = _mm256_cmpeq_epi8(mm_0, mm_c32.v3);
let mask_0_a = _mm256_movemask_epi8(mm_0_eq_a) as u32;
let mask_0_b = _mm256_movemask_epi8(mm_0_eq_b) as u32;
let mask_0_c = _mm256_movemask_epi8(mm_0_eq_c) as u32;
let base = buf_ptr.usz_offset_from(st_ptr);
_return_chr_tpl(base, mask_0_a, mask_0_b, mask_0_c)
}
}