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

use crate::simd::intrinsics;
use crate::simd::{LaneCount, Mask, MaskElement, Simd, SimdElement, SupportedLaneCount};

impl<T, const LANES: usize> Mask<T, LANES>
where
    T: MaskElement,
    LaneCount<LANES>: SupportedLaneCount,
{
    /// 从两个 vectors 中选择 lanes。
    ///
    /// 对于掩码中的每个 lane，如果该 lane 的掩码为 true，则从 `true_values` 中选择相应的 lane，如果该 lane 的掩码为 false，则从 `false_values` 中选择相应的 lane。
    ///
    /// # Examples
    ///
    /// ```
    /// # #![feature(portable_simd)]
    /// # use core::simd::{Simd, Mask};
    /// let a = Simd::from_array([0, 1, 2, 3]);
    /// let b = Simd::from_array([4, 5, 6, 7]);
    /// let mask = Mask::from_array([true, false, false, true]);
    /// let c = mask.select(a, b);
    /// assert_eq!(c.to_array(), [0, 5, 6, 3]);
    /// ```
    #[inline]
    #[must_use = "method returns a new vector and does not mutate the original inputs"]
    pub fn select<U>(
        self,
        true_values: Simd<U, LANES>,
        false_values: Simd<U, LANES>,
    ) -> Simd<U, LANES>
    where
        U: SimdElement<Mask = T>,
    {
        // 安全性: 掩码已转换为整数的 vector，要选择的操作数是相同类型和长度的 vectors。
        //
        unsafe { intrinsics::simd_select(self.to_int(), true_values, false_values) }
    }

    /// 从两个掩码中选择 lanes。
    ///
    /// 对于掩码中的每个 lane，如果该 lane 的掩码为 true，则从 `true_values` 中选择相应的 lane，如果该 lane 的掩码为 false，则从 `false_values` 中选择相应的 lane。
    ///
    /// # Examples
    ///
    /// ```
    /// # #![feature(portable_simd)]
    /// # use core::simd::Mask;
    /// let a = Mask::<i32, 4>::from_array([true, true, false, false]);
    /// let b = Mask::<i32, 4>::from_array([false, false, true, true]);
    /// let mask = Mask::<i32, 4>::from_array([true, false, false, true]);
    /// let c = mask.select_mask(a, b);
    /// assert_eq!(c.to_array(), [true, false, true, false]);
    /// ```
    #[inline]
    #[must_use = "method returns a new mask and does not mutate the original inputs"]
    pub fn select_mask(self, true_values: Self, false_values: Self) -> Self {
        self & true_values | !self & false_values
    }
}