subset_vector.hpp

Go to the documentation of this file.

#ifndef SUBSET_VECTOR_HPP
#define SUBSET_VECTOR_HPP
 
#include "macros.hpp"
 
namespace scran {
 
template<bool retain, class Vector, typename Subset>
Vector subset_vector(const Vector& vec, const Subset* sub) {
    int n = 0;
    for (size_t i = 0; i < vec.size(); ++i) {
        if (sub[i]) {
            ++n;
        }
    }
 
    if constexpr(!retain) {
        n = vec.size() - n;        
    }
 
    Vector output(n);
    size_t counter = 0;
    for (size_t i = 0; i < vec.size(); ++i) {
        if constexpr(retain) {
            if (sub[i]) {
                output[counter] = vec[i];
                ++counter;
            }
        } else {
            if (!sub[i]) {
                output[counter] = vec[i];
                ++counter;
            }
        }
    }
 
    return output;
}
 
}
 
#endif