1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
// SPDX-License-Identifier: LGPL-3.0-or-later
// Author: Kristian Lytje
#include <fitter/LinearFitter.h>
#include <fitter/FitResult.h>
#include <hist/intensity_calculator/ICompositeDistanceHistogram.h>
#include <math/CubicSpline.h>
using namespace ausaxs;
using namespace ausaxs::fitter;
LinearFitter::LinearFitter(const SimpleDataset& data) : data(data) {
detail::LinearLeastSquares::data = data.y();
detail::LinearLeastSquares::inv_sigma = data.yerr();
for (unsigned i = 0; i < detail::LinearLeastSquares::inv_sigma.size(); ++i) {
detail::LinearLeastSquares::inv_sigma[i] = 1./detail::LinearLeastSquares::inv_sigma[i];
}
}
LinearFitter::LinearFitter(LinearFitter&&) noexcept = default;
LinearFitter& LinearFitter::operator=(LinearFitter&&) noexcept = default;
LinearFitter::LinearFitter(const SimpleDataset& data, std::unique_ptr<hist::DistanceHistogram> model) : data(data), model(std::move(model)) {
detail::LinearLeastSquares::data = data.y();
detail::LinearLeastSquares::model = splice(this->model->debye_transform().get_counts());
detail::LinearLeastSquares::inv_sigma = data.yerr();
for (unsigned i = 0; i < detail::LinearLeastSquares::inv_sigma.size(); ++i) {
detail::LinearLeastSquares::inv_sigma[i] = 1./detail::LinearLeastSquares::inv_sigma[i];
}
}
void LinearFitter::refresh_model() {
detail::LinearLeastSquares::model = splice(model->debye_transform().get_counts());
}
void LinearFitter::set_model(std::unique_ptr<hist::DistanceHistogram> model) {
this->model = std::move(model);
detail::LinearLeastSquares::model = splice(this->model->debye_transform().get_counts());
}
SimpleDataset LinearFitter::get_data() const {
return data;
}
std::vector<double> LinearFitter::splice(const std::vector<double>& ym) const {
std::vector<double> Im(data.size()); // spliced model values
math::CubicSpline s(model->get_q_axis(), ym);
for (unsigned int i = 0; i < data.size(); ++i) {
Im[i] = s.spline(data.x(i));
}
return Im;
}
|
