ChooseHvgs.hpp

Go to the documentation of this file.

#ifndef SCRAN_CHOOSE_HVGS_HPP
#define SCRAN_CHOOSE_HVGS_HPP
 
#include "../utils/macros.hpp"
 
#include <vector>
#include <algorithm>
#include <numeric>
#include <cstdint>
 
namespace scran {
 
class ChooseHvgs {
public:
    struct Defaults {
        static constexpr size_t top = 4000;
    };
 
private:
    size_t top = Defaults::top;
 
public:
    ChooseHvgs& set_top(size_t t = Defaults::top) {
        top = t;
        return *this;
    }
 
public:
    template<typename V, typename T>
    void run(size_t n, const V* statistic, T* output) const {
        std::vector<size_t> collected(n);
        std::iota(collected.begin(), collected.end(), 0);
        std::sort(collected.begin(), collected.end(), [&](size_t l, size_t r) -> bool { return statistic[l] > statistic[r]; });
        
        auto limit = std::min(n, top);
        std::fill(output, output + n, false);
        for (size_t i = 0; i < limit; ++i) {
            output[collected[i]] = true; 
        }
    }
 
    template<typename T = uint8_t, typename V>
    std::vector<T> run(size_t n, const V* statistic) const {
        std::vector<T> output(n);
        run(n, statistic, output.data());
        return output;
    }
};
 
}
#endif