compute.hpp

Go to the documentation of this file.

#ifndef WEIGHTEDLOWESS_COMPUTE_HPP
#define WEIGHTEDLOWESS_COMPUTE_HPP
 
#include <vector>
#include <cstddef>
 
#include "sanisizer/sanisizer.hpp"
 
#include "fit.hpp"
#include "Options.hpp"
#include "utils.hpp"
 
namespace WeightedLowess {
 
template<typename Data_>
void compute(
    const std::size_t num_points,
    const Data_* const x,
    const PrecomputedWindows<Data_>& windows,
    const Data_* const y,
    Data_* const fitted,
    Data_* robust_weights,
    const Options<Data_>& opt
) {
    std::vector<Data_> rbuffer;
    if (robust_weights == NULL) {
        sanisizer::resize(rbuffer, num_points);
        robust_weights = rbuffer.data();
    }
    internal::fit_trend(num_points, x, windows, y, fitted, robust_weights, opt);
}
 
template<typename Data_>
void compute(
    const std::size_t num_points,
    const Data_* const x,
    const Data_* const y,
    Data_* const fitted,
    Data_* const robust_weights,
    const Options<Data_>& opt
) {
    const auto win = define_windows(num_points, x, opt);
    compute(num_points, x, win, y, fitted, robust_weights, opt);
}
 
template<typename Data_>
struct Results {
    Results(const std::size_t n) :
        fitted(sanisizer::cast<I<decltype(fitted.size())> >(n)),
        robust_weights(sanisizer::cast<I<decltype(robust_weights.size())> >(n))
    {}
 
    std::vector<Data_> fitted;
 
    std::vector<Data_> robust_weights;
};
 
template<typename Data_>
Results<Data_> compute(
    const std::size_t num_points,
    const Data_* const x,
    const Data_* const y,
    const Options<Data_>& opt
) {
    Results<Data_> output(num_points);
    compute(num_points, x, y, output.fitted.data(), output.robust_weights.data(), opt);
    return output;
}
 
}
 
#endif