block_weights.hpp

Go to the documentation of this file.

#ifndef SCRAN_BLOCKS_BLOCK_WEIGHTS_HPP
#define SCRAN_BLOCKS_BLOCK_WEIGHTS_HPP
 
namespace scran_blocks {
 
enum class WeightPolicy : char { NONE, VARIABLE, EQUAL };
 
struct VariableWeightParameters {
    double lower_bound = 0;
 
    double upper_bound = 1000;
};
 
inline double compute_variable_weight(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(size_t num_blocks, const Size_* sizes, WeightPolicy policy, const VariableWeightParameters& variable, Weight_* weights) {
    if (policy == WeightPolicy::NONE) {
        std::copy_n(sizes, num_blocks, weights);
    } else if (policy == WeightPolicy::EQUAL) {
        for (size_t s = 0; s < num_blocks; ++s) {
            weights[s] = sizes[s] > 0;
        }
    } else {
        for (size_t 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, WeightPolicy policy, const VariableWeightParameters& variable) {
    std::vector<Weight_> output(sizes.size());
    compute_weights(sizes.size(), sizes.data(), policy, variable, output.data());
    return output;
}
 
}
 
#endif