base64-ng 2.0.1

no_std-first Base64 encoding and decoding with strict RFC 4648 APIs and optional SIMD
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95

//! Pure encode dispatch policy and frozen automatic thresholds.

#[cfg(any(
    all(feature = "simd", any(target_arch = "x86", target_arch = "x86_64")),
    all(feature = "simd", target_arch = "aarch64", target_endian = "little"),
    all(
        feature = "std",
        feature = "simd",
        target_arch = "riscv64",
        target_os = "linux"
    )
))]
use super::EncodeBackend;

pub(super) const MIN_SIMD_INPUT: usize = 12;

#[cfg(all(feature = "simd", any(target_arch = "x86", target_arch = "x86_64")))]
pub(super) const X86_AVX2_MIN_INPUT: usize = 24;
#[cfg(all(feature = "simd", any(target_arch = "x86", target_arch = "x86_64")))]
pub(super) const X86_AVX512_MIN_INPUT: usize = 192;

#[cfg(all(feature = "simd", target_arch = "aarch64", target_endian = "little"))]
pub(super) const NEON_MIN_INPUT: usize = 192;

#[cfg(all(
    feature = "std",
    feature = "simd",
    target_arch = "riscv64",
    target_os = "linux"
))]
pub(super) const RVV_MIN_INPUT: usize = 384;

#[cfg(all(feature = "simd", any(target_arch = "x86", target_arch = "x86_64")))]
pub(super) fn select_x86(
    candidate: EncodeBackend,
    input_len: usize,
    mut admit: impl FnMut(EncodeBackend) -> bool,
) -> EncodeBackend {
    if candidate == EncodeBackend::Avx512Vbmi
        && input_len >= X86_AVX512_MIN_INPUT
        && admit(EncodeBackend::Avx512Vbmi)
    {
        return EncodeBackend::Avx512Vbmi;
    }
    if matches!(candidate, EncodeBackend::Avx512Vbmi | EncodeBackend::Avx2)
        && input_len >= X86_AVX2_MIN_INPUT
        && admit(EncodeBackend::Avx2)
    {
        return EncodeBackend::Avx2;
    }
    if matches!(
        candidate,
        EncodeBackend::Avx512Vbmi | EncodeBackend::Avx2 | EncodeBackend::Ssse3Sse41
    ) && input_len >= MIN_SIMD_INPUT
        && admit(EncodeBackend::Ssse3Sse41)
    {
        return EncodeBackend::Ssse3Sse41;
    }
    EncodeBackend::Scalar
}

#[cfg(all(feature = "simd", target_arch = "aarch64", target_endian = "little"))]
pub(super) fn select_neon(
    candidate: EncodeBackend,
    input_len: usize,
    mut admit: impl FnMut(EncodeBackend) -> bool,
) -> EncodeBackend {
    if candidate == EncodeBackend::Neon && input_len >= NEON_MIN_INPUT && admit(EncodeBackend::Neon)
    {
        EncodeBackend::Neon
    } else {
        EncodeBackend::Scalar
    }
}

#[cfg(all(
    feature = "std",
    feature = "simd",
    target_arch = "riscv64",
    target_os = "linux"
))]
pub(super) fn select_rvv(
    candidate: EncodeBackend,
    input_len: usize,
    mut admit: impl FnMut(EncodeBackend) -> bool,
) -> EncodeBackend {
    if candidate == EncodeBackend::Rvv && input_len >= RVV_MIN_INPUT && admit(EncodeBackend::Rvv) {
        EncodeBackend::Rvv
    } else {
        EncodeBackend::Scalar
    }
}

#[cfg(test)]
mod tests;