-
Notifications
You must be signed in to change notification settings - Fork 121
/
Quadratic.cpp
66 lines (56 loc) · 2.04 KB
/
Quadratic.cpp
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
// Mantid Repository : https://github.com/mantidproject/mantid
//
// Copyright © 2018 ISIS Rutherford Appleton Laboratory UKRI,
// NScD Oak Ridge National Laboratory, European Spallation Source,
// Institut Laue - Langevin & CSNS, Institute of High Energy Physics, CAS
// SPDX - License - Identifier: GPL - 3.0 +
//----------------------------------------------------------------------
// Includes
//----------------------------------------------------------------------
#include "MantidCurveFitting/Functions/Quadratic.h"
#include "MantidAPI/FunctionFactory.h"
namespace Mantid {
namespace CurveFitting {
namespace Functions {
using namespace CurveFitting;
using namespace Kernel;
using namespace API;
DECLARE_FUNCTION(Quadratic)
/** \relates Quadratic
*Initialise function to define parameters
*/
void Quadratic::init() {
declareParameter("A0", 0.0, "coefficient for constant term");
declareParameter("A1", 0.0, "coefficient for linear term");
declareParameter("A2", 0.0, "coefficient for quadratic term");
}
/** \relates Quadratic
* Implements the quadratic function
* @param out :: The result of evaluating the function
* @param xValues :: function domain values
* @param nData :: size of the function domain
*/
void Quadratic::function1D(double *out, const double *xValues, const size_t nData) const {
const double a0 = getParameter("A0");
const double a1 = getParameter("A1");
const double a2 = getParameter("A2");
for (size_t i = 0; i < nData; i++) {
out[i] = a0 + a1 * xValues[i] + a2 * xValues[i] * xValues[i];
}
}
/** \relates Quadratic
* Calculates the derivatives of the quadratic
* @param out :: The resulting jacobian
* @param xValues :: function domain values
* @param nData :: size of the function domain
*/
void Quadratic::functionDeriv1D(API::Jacobian *out, const double *xValues, const size_t nData) {
for (size_t i = 0; i < nData; i++) {
out->set(i, 0, 1);
out->set(i, 1, xValues[i]);
out->set(i, 2, xValues[i] * xValues[i]);
}
}
} // namespace Functions
} // namespace CurveFitting
} // namespace Mantid