tiny.simd.copyIf

1pub fn copyIf(2    comptime D: type,3    input: []const D.Lane,4    output: []D.Lane,5    predicate: anytype,6) usize {7    if (@sizeOf(D.Lane) == 1) @compileError("copyIf requires 16/32/64-bit lanes");8    std.debug.assert(output.len >= input.len);9    var input_index: usize = 0;10    var output_index: usize = 0;11    while (input_index + D.lane_count <= input.len) : (input_index += D.lane_count) {12        const value = memory.load(D, input[input_index..]);13        output_index += compact.compressBlendedStore(14            D,15            value,16            predicate.call(D, value),17            output[output_index..],18        );19    }20    if (input_index != input.len) {21        const remaining = input.len - input_index;22        const value = memory.loadN(D, input[input_index..], remaining);23        const mask = predicate.call(D, value) & construct.firstN(D, remaining);24        output_index += compact.compressBlendedStore(D, value, mask, output[output_index..]);25    }26    std.debug.assert(output_index <= input.len);27    return output_index;28}

1pub const copyIf = algo.copyIf;