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 "tatami/tatami.hpp"
 
#include "utils.hpp"
 
namespace tatami_test {
 
template<typename Index_>
tatami::VectorPtr<Index_> create_indexed_subset(const Index_ extent, const double relative_start, const double probability, const SeedType seed) {
    auto ptr = new std::vector<Index_>;
    tatami::VectorPtr<Index_> output(ptr);
 
    Index_ start = extent * relative_start;
    if (start >= extent) {
        return output;
    }
 
    ptr->push_back(Fix(start));
    RngEngine rng(seed);
    std::uniform_real_distribution udist;
    for (Index_ i = start + 1; i < extent; ++i) {
        if (udist(rng) < probability) {
            ptr->push_back(Fix(i));
        }
    }
 
    return output;
}
 
}
 
#endif