Function toCooVector

test/arrayfire_test.cpp:1487–1551

template<typename T>
vector<sparseCooValue<T>> toCooVector(const af::array &arr) {
    vector<sparseCooValue<T>> out;
    if (arr.issparse()) {
        switch (sparseGetStorage(arr)) {
            case AF_STORAGE_COO: {
                dim_t nnz = sparseGetNNZ(arr);
                vector<int> row(nnz), col(nnz);
                vector<T> values(nnz);
                sparseGetValues(arr).host(values.data());
                sparseGetRowIdx(arr).host(row.data());
                sparseGetColIdx(arr).host(col.data());
                out.reserve(nnz);
                for (int i = 0; i < nnz; i++) {
                    out.emplace_back(row[i], col[i], values[i]);
                }
            } break;
            case AF_STORAGE_CSR: {
                dim_t nnz = sparseGetNNZ(arr);
                vector<int> row(arr.dims(0) + 1), col(nnz);
                vector<T> values(nnz);
                sparseGetValues(arr).host(values.data());
                sparseGetRowIdx(arr).host(row.data());
                sparseGetColIdx(arr).host(col.data());
                out.reserve(nnz);
                for (int i = 0; i < row.size() - 1; i++) {
                    for (int r = row[i]; r < row[i + 1]; r++) {
                        out.emplace_back(i, col[r], values[r]);
                    }
                }
            } break;
            case AF_STORAGE_CSC: {
                dim_t nnz = sparseGetNNZ(arr);
                vector<int> row(nnz), col(arr.dims(1) + 1);
                vector<T> values(nnz);
                sparseGetValues(arr).host(values.data());
                sparseGetRowIdx(arr).host(row.data());
                sparseGetColIdx(arr).host(col.data());
                out.reserve(nnz);
                for (int i = 0; i < col.size() - 1; i++) {
                    for (int c = col[i]; c < col[i + 1]; c++) {
                        out.emplace_back(row[c], i, values[c]);
                    }
                }
            } break;
            default: throw std::logic_error("NOT SUPPORTED");
        }
    } else {
        vector<T> values(arr.elements());
        arr.host(values.data());
        int M = arr.dims(0), N = arr.dims(1);
        for (int j = 0; j < N; j++) {
            for (int i = 0; i < M; i++) {
                if (std::fpclassify(real(values[j * M + i])) == FP_ZERO) {
                    out.emplace_back(i, j, values[j * M + i]);
                }
            }
        }
    }