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 <cstddef>
 
#include "sanisizer/sanisizer.hpp"
 
#include "utils.hpp"
 
namespace tatami_test {
 
struct SimulateVectorOptions {
    double lower = 0;
 
    double upper = 100;
 
    double density = 1;
 
    SeedType seed = 1234567890;
};
 
template<typename Type_, typename Length_>
std::vector<Type_> simulate_vector(const Length_ length, const SimulateVectorOptions& options) {
    auto output = sanisizer::create<std::vector<Type_> >(length);
    RngEngine 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;
}
 
template<typename Type_, typename Index_>
std::vector<Type_> simulate_vector(const Index_ nrow, const Index_ ncol, const SimulateVectorOptions& options) {
    return simulate_vector<Type_>(sanisizer::product<typename std::vector<Type_>::size_type>(nrow, ncol), options);
}
 
}
 
#endif