Function print

src/api/c/print.cpp:73–116

template<typename T>
static void print(const char *exp, af_array arr, const int precision,
                  std::ostream &os = std::cout, bool transpose = true) {
    if (exp == NULL) {
        os << "No Name Array" << std::endl;
    } else {
        os << exp << std::endl;
    }

    const ArrayInfo &info = getInfo(arr);

    std::ios_base::fmtflags backup = os.flags();

    os << "[" << info.dims() << "]\n";
#ifndef NDEBUG
    os << "   Offset: " << info.getOffset() << std::endl;
    os << "   Strides: [" << info.strides() << "]" << std::endl;
#endif

    // Handle empty array
    if (info.elements() == 0) {
        os << "<empty>" << std::endl;
        os.flags(backup);
        return;
    }

    vector<T> data(info.elements());

    af_array arrT;
    if (transpose) {
        AF_CHECK(af_reorder(&arrT, arr, 1, 0, 2, 3));
    } else {
        arrT = arr;
    }

    // FIXME: Use alternative function to avoid copies if possible
    AF_CHECK(af_get_data_ptr(&data.front(), arrT));
    const ArrayInfo &infoT = getInfo(arrT);

    printer(os, &data.front(), infoT, infoT.ndims() - 1, precision);

    if (transpose) { AF_CHECK(af_release_array(arrT)); }

    os.flags(backup);
}

template<typename T>
static void printSparse(const char *exp, af_array arr, const int precision,