/
UserFunction.h
80 lines (70 loc) · 2.76 KB
/
UserFunction.h
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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
// Mantid Repository : https://github.com/mantidproject/mantid
//
// Copyright © 2007 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 +
#pragma once
//----------------------------------------------------------------------
// Includes
//----------------------------------------------------------------------
#include "MantidAPI/IFunction1D.h"
#include "MantidAPI/ParamFunction.h"
#include "MantidCurveFitting/DllConfig.h"
#include <memory>
namespace mu {
class Parser;
}
namespace Mantid {
namespace CurveFitting {
namespace Functions {
/**
A user defined function.
@author Roman Tolchenov, Tessella plc
@date 15/01/2010
*/
class MANTID_CURVEFITTING_DLL UserFunction : public API::ParamFunction, public API::IFunction1D {
public:
/// Constructor
UserFunction();
/// Destructor
~UserFunction() override;
/// Returns the function's name
std::string name() const override { return "UserFunction"; }
// Returns Category
const std::string category() const override { return "General"; }
/// Function you want to fit to.
void function1D(double *out, const double *xValues, const size_t nData) const override;
/// Derivatives of function with respect to active parameters
void functionDeriv(const API::FunctionDomain &domain, API::Jacobian &jacobian) override;
/// Returns the number of attributes associated with the function
size_t nAttributes() const override { return 1; }
/// Returns a list of attribute names
std::vector<std::string> getAttributeNames() const override { return std::vector<std::string>(1, "Formula"); }
/// Return a value of attribute attName
Attribute getAttribute(const std::string &attName) const override {
return attName == "Formula" ? Attribute(m_formula) : getAttribute(attName);
}
/// Set a value to attribute attName
void setAttribute(const std::string &attName, const Attribute &value) override;
/// Check if attribute attName exists
bool hasAttribute(const std::string &attName) const override { return attName == "Formula"; }
private:
/// The formula
std::string m_formula;
/// extended muParser instance
mu::Parser *m_parser;
/// Used as 'x' variable in m_parser.
mutable double m_x;
/// True indicates that input formula contains 'x' variable
bool m_x_set;
/// Temporary data storage used in functionDeriv
mutable std::vector<double> m_tmp;
/// Temporary data storage used in functionDeriv
mutable std::vector<double> m_tmp1;
/// mu::Parser callback function for setting variables.
static double *AddVariable(const char *varName, void *pufun);
};
} // namespace Functions
} // namespace CurveFitting
} // namespace Mantid