simulate_vector.hpp

Go to the documentation of this file.

#ifndef TATAMI_TEST_SIMULATE_VECTOR_HPP
#define TATAMI_TEST_SIMULATE_VECTOR_HPP
 
#include <random>
#include <vector>
#include <cstdint>
 
namespace tatami_test {
 
struct SimulateVectorOptions {
    double lower = 0;
 
    double upper = 100;
 
    double density = 1;
 
    uint64_t seed = 1234567890;
};
 
template<typename Type_>
std::vector<Type_> simulate_vector(size_t length, const SimulateVectorOptions& options) {
    std::vector<Type_> output(length);
    std::mt19937_64 rng(options.seed);
    std::uniform_real_distribution<> unif(options.lower, options.upper);
 
    if (options.density == 1) {
        for (auto& v : output) {
            v = unif(rng);
        }
    } else {
        std::uniform_real_distribution<> nonzero(0.0, 1.0);
        for (auto& v : output) {
            if (nonzero(rng) <= options.density) {
                v = unif(rng);
            }
        }
    }
 
    return output;
}
 
}
 
#endif