Function af_set_intersect

src/api/c/set.cpp:122–172

}

af_err af_set_intersect(af_array* out, const af_array first,
                        const af_array second, const bool is_unique) {
    try {
        const ArrayInfo& first_info  = getInfo(first);
        const ArrayInfo& second_info = getInfo(second);

        // TODO(umar): fix for set intersect from union
        if (first_info.isEmpty()) { return af_retain_array(out, first); }

        if (second_info.isEmpty()) { return af_retain_array(out, second); }

        ARG_ASSERT(1, (first_info.isVector() || first_info.isScalar()));
        ARG_ASSERT(1, (second_info.isVector() || second_info.isScalar()));

        af_dtype first_type  = first_info.getType();
        af_dtype second_type = second_info.getType();

        ARG_ASSERT(1, first_type == second_type);

        af_array res;
        switch (first_type) {
            case f32:
                res = setIntersect<float>(first, second, is_unique);
                break;
            case f64:
                res = setIntersect<double>(first, second, is_unique);
                break;
            case s32: res = setIntersect<int>(first, second, is_unique); break;
            case u32: res = setIntersect<uint>(first, second, is_unique); break;
            case s16:
                res = setIntersect<short>(first, second, is_unique);
                break;
            case u16:
                res = setIntersect<ushort>(first, second, is_unique);
                break;
            case s64: res = setIntersect<intl>(first, second, is_unique); break;
            case u64:
                res = setIntersect<uintl>(first, second, is_unique);
                break;
            case b8: res = setIntersect<char>(first, second, is_unique); break;
            case s8: res = setIntersect<schar>(first, second, is_unique); break;
            case u8: res = setIntersect<uchar>(first, second, is_unique); break;
            default: TYPE_ERROR(1, first_type);
        }

        std::swap(*out, res);
    }
    CATCHALL;

    return AF_SUCCESS;
}