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_* x, const PrecomputedWindows<Data_>& windows, const Data_* y, Data_* 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_* x, const Data_* y, Data_* fitted, Data_* robust_weights, const Options<Data_>& opt) {
    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<decltype(internal::I(fitted.size()))>(n)),
        robust_weights(sanisizer::cast<decltype(internal::I(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_* x, const Data_* 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