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 <cstddef>
 
#include "sanisizer/sanisizer.hpp"
 
#include "utils.hpp"
 
namespace scran_qc {
 
template<typename Keep_, typename Index_>
void filter_index(const std::size_t num, const Keep_* const filter, std::vector<Index_>& output) {
    output.clear();
    for (decltype(I(num)) i = 0; i < num; ++i) {
        if (filter[i]) {
            output.push_back(i);
        }
    }
}
 
template<typename Index_, typename Keep_>
std::vector<Index_> filter_index(const std::size_t num, const Keep_* const filter) {
    std::vector<Index_> output;
#ifdef SCRAN_QC_TEST_INIT
    output.resize(10, SCRAN_QC_TEST_INIT);
#endif
    filter_index(num, filter, output);
    return output;
}
 
template<typename Keep_, typename Output_>
void combine_filters(const std::size_t num, const std::vector<Keep_*>& filters, Output_* const output) {
    std::copy_n(filters.front(), num, output);
    const auto nfilters = filters.size();
    for (decltype(I(nfilters)) f = 1; f < nfilters; ++f) {
        const auto filt = filters[f];
        for (decltype(I(num)) i = 0; i < num; ++i) {
            output[i] = output[i] && filt[i];
        }
    }
}
 
template<typename Output_ = unsigned char, typename Keep_>
std::vector<Output_> combine_filters(const std::size_t num, const std::vector<const Keep_*>& filters) {
    auto output = sanisizer::create<std::vector<Output_> >(num
#ifdef SCRAN_QC_TEST_INIT
        , SCRAN_QC_TEST_INIT
#endif
    );
    combine_filters(num, filters, output.data());
    return output;
}
 
template<typename Index_, typename Keep_>
void combine_filters_index(const Index_ num, const std::vector<const Keep_*>& filters, std::vector<Index_>& output) {
    output.clear();
 
    const auto nfilters = filters.size();
    for (decltype(I(num)) i = 0; i < num; ++i) {
        bool keep = true;
        for (decltype(I(nfilters)) 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(const Index_ num, const std::vector<const Keep_*>& filters) {
    std::vector<Index_> output;
#ifdef SCRAN_QC_TEST_INIT
    output.resize(10, SCRAN_QC_TEST_INIT);
#endif
    combine_filters_index(num, filters, output);
    return output;
}
 
}
 
#endif