compute.hpp

Go to the documentation of this file.

#ifndef WEIGHTEDLOWESS_COMPUTE_HPP
#define WEIGHTEDLOWESS_COMPUTE_HPP
 
#include <vector>
 
#include "fit.hpp"
#include "Options.hpp"
 
namespace WeightedLowess {
 
template<typename Data_>
void compute(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) {
        rbuffer.resize(num_points);
        robust_weights = rbuffer.data();
    }
    internal::fit_trend(num_points, x, windows, y, fitted, robust_weights, opt);
}
 
template<typename Data_>
void compute(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(size_t n) : fitted(n), robust_weights(n) {}
 
    std::vector<Data_> fitted;
 
    std::vector<Data_> robust_weights;
};
 
template<typename Data_>
Results<Data_> compute(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