format_filters.hpp

Go to the documentation of this file.

#ifndef SCRAN_QC_FORMAT_FILTERS_HPP
#define SCRAN_QC_FORMAT_FILTERS_HPP
 
#include <vector>
#include <algorithm>
#include <cstdint>
 
namespace scran_qc {
 
template<typename Index_, typename Keep_>
void filter_index(Index_ num, const Keep_* filter, std::vector<Index_>& output) {
    output.clear();
    for (Index_ i = 0; i < num; ++i) {
        if (filter[i]) {
            output.push_back(i);
        }
    }
}
 
template<typename Index_, typename Keep_>
std::vector<Index_> filter_index(Index_ num, const Keep_* filter) {
    std::vector<Index_> output;
    filter_index(num, filter, output);
    return output;
}
 
template<typename Keep_, typename Output_>
void combine_filters(size_t num, const std::vector<Keep_*>& filters, Output_* output) {
    std::copy_n(filters.front(), num, output);
    for (size_t f = 1, nfilters = filters.size(); f < nfilters; ++f) {
        auto filt = filters[f];
        for (size_t i = 0; i < num; ++i) {
            output[i] = output[i] && filt[i];
        }
    }
}
 
template<typename Output_ = uint8_t, typename Keep_ = uint8_t>
std::vector<Output_> combine_filters(size_t num, const std::vector<const Keep_*>& filters) {
    std::vector<Output_> output(num);
    combine_filters(num, filters, output.data());
    return output;
}
 
template<typename Index_, typename Keep_>
void combine_filters_index(Index_ num, const std::vector<const Keep_*>& filters, std::vector<Index_>& output) {
    output.clear();
 
    size_t nfilters = filters.size();
    for (Index_ i = 0; i < num; ++i) {
        bool keep = true;
        for (size_t f = 0; f < nfilters; ++f) {
            if (!filters[f][i]) {
                keep = false;
                break;
            }
        }
        if (keep) {
            output.push_back(i);
        }
    }
}
 
template<typename Index_, typename Keep_>
std::vector<Index_> combine_filters_index(Index_ num, const std::vector<const Keep_*>& filters) {
    std::vector<Index_> output;
    combine_filters_index(num, filters, output);
    return output;
}
 
}
 
#endif