simulate_vector.hpp

Go to the documentation of this file.

#ifndef SCRAN_TESTS_SIMULATE_VECTOR_HPP
#define SCRAN_TESTS_SIMULATE_VECTOR_HPP
 
#include <random>
#include <vector>
#include <cstdint>
#include <type_traits>
 
namespace scran_tests {
 
template<typename Type_ = double>
struct SimulationParameters {
    SimulationParameters() {
        if constexpr(std::is_unsigned<Type_>::value) {
            lower = 0;
        }
    }
    double density = 1;
 
    Type_ lower = -10;
 
    Type_ upper = 10;
 
    uint64_t seed = 1234567890;
};
 
template<typename Type_ = double>
std::vector<Type_> simulate_vector(size_t length, const SimulationParameters<Type_>& params) {
    std::mt19937_64 rng(params.seed);
 
    typename std::conditional<
        std::is_floating_point<Type_>::value, 
        std::uniform_real_distribution<Type_>,
        std::uniform_int_distribution<Type_>
    >::type unif(params.lower, params.upper);
 
    std::vector<Type_> values(length);
 
    if (params.density == 1) {
        for (auto& v : values) {
            v = unif(rng);
        }
    } else {
        std::uniform_real_distribution<double> nonzero(0.0, 1.0);
        for (auto& v : values) {
            if (nonzero(rng) < params.density) {
                v = unif(rng);
            }
        }
    }
 
    return values;
}
 
}
 
#endif