create_indexed_subset.hpp

Go to the documentation of this file.

#ifndef TATAMI_TEST_CREATE_INDEXED_SUBSET_HPP
#define TATAMI_TEST_CREATE_INDEXED_SUBSET_HPP
 
#include <vector>
#include <random>
#include <cstdint>
 
#include "tatami/tatami.hpp"
 
namespace tatami_test {
 
template<typename Index_>
tatami::VectorPtr<Index_> create_indexed_subset(Index_ extent, double relative_start, double probability, uint64_t seed) {
    auto ptr = new std::vector<Index_>;
    tatami::VectorPtr<Index_> output(ptr);
 
    Index_ start = extent * relative_start;
    if (start < extent) {
        auto& indices = *ptr;
        indices.push_back(start);
        std::mt19937_64 rng(seed);
        std::uniform_real_distribution udist;
        for (Index_ i = start + 1; i < extent; ++i) {
            if (udist(rng) < probability) {
                indices.push_back(i);
            }
        }
    }
 
    return output;
}
 
}
 
#endif