Function replaceScalar

src/api/c/replace.cpp:96–129

template<typename ScalarType>
af_err replaceScalar(af_array a, const af_array cond, const ScalarType b) {
    try {
        const ArrayInfo& ainfo = getInfo(a);
        const ArrayInfo& cinfo = getInfo(cond);

        ARG_ASSERT(1, cinfo.getType() == b8);
        DIM_ASSERT(1, cinfo.ndims() == ainfo.ndims());

        dim4 adims = ainfo.dims();
        dim4 cdims = cinfo.dims();

        for (int i = 0; i < 4; i++) { DIM_ASSERT(1, cdims[i] == adims[i]); }

        switch (ainfo.getType()) {
            case f16: replace_scalar<half>(a, cond, b); break;
            case f32: replace_scalar<float>(a, cond, b); break;
            case f64: replace_scalar<double>(a, cond, b); break;
            case c32: replace_scalar<cfloat>(a, cond, b); break;
            case c64: replace_scalar<cdouble>(a, cond, b); break;
            case s32: replace_scalar<int>(a, cond, b); break;
            case u32: replace_scalar<uint>(a, cond, b); break;
            case s64: replace_scalar<intl>(a, cond, b); break;
            case u64: replace_scalar<uintl>(a, cond, b); break;
            case s16: replace_scalar<short>(a, cond, b); break;
            case u16: replace_scalar<ushort>(a, cond, b); break;
            case s8: replace_scalar<schar>(a, cond, b); break;
            case u8: replace_scalar<uchar>(a, cond, b); break;
            case b8: replace_scalar<char>(a, cond, b); break;
            default: TYPE_ERROR(2, ainfo.getType());
        }
    }
    CATCHALL;
    return AF_SUCCESS;
}

af_err af_replace_scalar(af_array a, const af_array cond, const double b) {
    return replaceScalar(a, cond, b);