simulate_compressed_sparse.hpp

Go to the documentation of this file.

#ifndef TATAMI_TEST_SIMULATE_COMPRESSED_SPARSE_HPP
#define TATAMI_TEST_SIMULATE_COMPRESSED_SPARSE_HPP
 
#include <random>
#include <vector>
#include <cstdint>
 
namespace tatami_test {
 
struct SimulateCompressedSparseOptions {
    double lower = 0;
 
    double upper = 100;
 
    double density = 0.1;
 
    uint64_t seed = 1234567890;
};
struct SimulateCompressedSparseOptions {…};
 
template<typename Value_, typename Index_>
struct SimulateCompressedSparseResult {
    std::vector<Value_> data;
 
    std::vector<Index_> index;
 
    std::vector<size_t> indptr;
};
struct SimulateCompressedSparseResult {…};
 
template<typename Value_, typename Index_>
SimulateCompressedSparseResult<Value_, Index_> simulate_compressed_sparse(size_t primary, size_t secondary, const SimulateCompressedSparseOptions& options) {
    std::mt19937_64 rng(options.seed);
    std::uniform_real_distribution<> nonzero(0.0, 1.0);
    std::uniform_real_distribution<> unif(options.lower, options.upper);
 
    SimulateCompressedSparseResult<Value_, Index_> output;
    output.indptr.resize(primary + 1);
    for (size_t p = 0; p < primary; ++p) {
        size_t idx = output.indptr[p];
        for (size_t s = 0; s < secondary; ++s) {
            if (nonzero(rng) < options.density) {
                output.data.push_back(unif(rng));
                output.index.push_back(s);
                ++idx;
            }
        }
        output.indptr[p + 1] = idx;
    }
 
    return output;
}
SimulateCompressedSparseResult<Value_, Index_> simulate_compressed_sparse(size_t primary, size_t secondary, const SimulateCompressedSparseOptions& options) {…}
 
}
 
#endif