Function af_medfilt2

src/api/c/filters.cpp:86–147

}

af_err af_medfilt2(af_array *out, const af_array in, const dim_t wind_length,
                   const dim_t wind_width, const af_border_type edge_pad) {
    try {
        ARG_ASSERT(2, (wind_length == wind_width));
        ARG_ASSERT(2, (wind_length > 0));
        ARG_ASSERT(3, (wind_width > 0));
        ARG_ASSERT(4, (edge_pad >= AF_PAD_ZERO && edge_pad <= AF_PAD_SYM));

        const ArrayInfo &info = getInfo(in);
        af::dim4 dims         = info.dims();

        if (info.isColumn()) {
            return af_medfilt1(out, in, wind_width, edge_pad);
        }

        dim_t input_ndims = dims.ndims();
        DIM_ASSERT(1, (input_ndims >= 2));

        if (wind_length == 1) {
            *out = retain(in);
            return AF_SUCCESS;
        }
        af_array output = nullptr;
        af_dtype type   = info.getType();
        switch (type) {
            case f32:
                output = medfilt2<float>(in, wind_length, wind_width, edge_pad);
                break;
            case f64:
                output =
                    medfilt2<double>(in, wind_length, wind_width, edge_pad);
                break;
            case b8:
                output = medfilt2<char>(in, wind_length, wind_width, edge_pad);
                break;
            case s32:
                output = medfilt2<int>(in, wind_length, wind_width, edge_pad);
                break;
            case u32:
                output = medfilt2<uint>(in, wind_length, wind_width, edge_pad);
                break;
            case s16:
                output = medfilt2<short>(in, wind_length, wind_width, edge_pad);
                break;
            case u16:
                output =
                    medfilt2<ushort>(in, wind_length, wind_width, edge_pad);
                break;
            case s8:
                output = medfilt2<schar>(in, wind_length, wind_width, edge_pad);
                break;
            case u8:
                output = medfilt2<uchar>(in, wind_length, wind_width, edge_pad);
                break;
            default: TYPE_ERROR(1, type);
        }
        std::swap(*out, output);
    }