block_weights.hpp

Go to the documentation of this file.

#ifndef SCRAN_BLOCKS_BLOCK_WEIGHTS_HPP
#define SCRAN_BLOCKS_BLOCK_WEIGHTS_HPP
 
#include <vector>
#include <cstddef>
#include <algorithm>
 
#include "sanisizer/sanisizer.hpp"
 
#include "utils.hpp"
 
namespace scran_blocks {
 
enum class WeightPolicy : char { NONE, SIZE, VARIABLE, EQUAL };
 
struct VariableWeightParameters {
    double lower_bound = 0;
 
    double upper_bound = 1000;
};
 
inline double compute_variable_weight(const double s, const VariableWeightParameters& params) {
    if (s < params.lower_bound || s == 0) {
        return 0;
    }
 
    if (s > params.upper_bound) {
        return 1;
    }
 
    return (s - params.lower_bound) / (params.upper_bound - params.lower_bound);
}
 
template<typename Size_, typename Weight_>
void compute_weights(const std::size_t num_blocks, const Size_* const sizes, const WeightPolicy policy, const VariableWeightParameters& variable, Weight_* const weights) {
    if (policy == WeightPolicy::NONE || policy == WeightPolicy::SIZE) {
        std::copy_n(sizes, num_blocks, weights);
    } else if (policy == WeightPolicy::EQUAL) {
        for (decltype(I(num_blocks)) s = 0; s < num_blocks; ++s) {
            weights[s] = sizes[s] > 0;
        }
    } else {
        for (decltype(I(num_blocks)) s = 0; s < num_blocks; ++s) {
            weights[s] = compute_variable_weight(sizes[s], variable);
        }
    }
}
 
template<typename Weight_ = double, typename Size_>
std::vector<Weight_> compute_weights(const std::vector<Size_>& sizes, const WeightPolicy policy, const VariableWeightParameters& variable) {
    auto output = sanisizer::create<std::vector<Weight_> >(sizes.size());
    compute_weights(sizes.size(), sizes.data(), policy, variable, output.data());
    return output;
}
 
}
 
#endif