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
|
#include "equal.hpp"
#include "fcn.hpp"
#include "pybind11.hpp"
#include <Minuit2/FunctionMinimum.h>
#include <Minuit2/MinimumSeed.h>
#include <Minuit2/MinimumState.h>
#include <Minuit2/MnStrategy.h>
#include <Minuit2/MnUserFcn.h>
#include <Minuit2/MnUserParameterState.h>
#include <type_traits>
namespace py = pybind11;
using namespace ROOT::Minuit2;
FunctionMinimum init(const FCN& fcn, const MnUserParameterState& st,
const MnStrategy& str, double edm_goal);
FunctionMinimum init2(const MnUserTransformation& tr, py::sequence par,
py::sequence cov, py::sequence grad, double fval, double up,
double edm_goal, int nfcn, int max_nfcn, bool exact_hess_inv);
py::tuple fmin_getstate(const FunctionMinimum&);
FunctionMinimum fmin_setstate(py::tuple);
bool hesse_failed_workaround(const FunctionMinimum& self) {
// Calling FunctionMinimum::HesseFailed is not reliable.
// It calls MinimumError::HesseFailed, but this is false if
// there were the failures MinimumError::MnReachedCallLimit
// or MinimumError::MnInvertFailed, which cannot be queried
// from FunctionMinimum.
//
// As a workaround, we return true if covariance exists, but
// is not accurate or made positive definite.
return self.HasCovariance() && !(self.HasPosDefCovar() || self.HasMadePosDefCovar());
}
void bind_functionminimum(py::module m) {
py::class_<FunctionMinimum>(m, "FunctionMinimum")
.def(py::init(&init))
.def(py::init(&init2))
.def_property_readonly("state", &FunctionMinimum::UserState)
.def_property_readonly("edm", &FunctionMinimum::Edm)
.def_property_readonly("fval", &FunctionMinimum::Fval)
.def_property_readonly("is_valid", &FunctionMinimum::IsValid)
// No need to wrap HasValidParameters
// .def_property_readonly("has_valid_parameters",
// &FunctionMinimum::HasValidParameters)
.def_property_readonly("has_accurate_covar", &FunctionMinimum::HasAccurateCovar)
.def_property_readonly("has_posdef_covar", &FunctionMinimum::HasPosDefCovar)
.def_property_readonly("has_made_posdef_covar",
&FunctionMinimum::HasMadePosDefCovar)
.def_property_readonly("hesse_failed", hesse_failed_workaround)
.def_property_readonly("has_covariance", &FunctionMinimum::HasCovariance)
.def_property_readonly("is_above_max_edm", &FunctionMinimum::IsAboveMaxEdm)
.def_property_readonly("has_reached_call_limit",
&FunctionMinimum::HasReachedCallLimit)
.def_property("errordef", &FunctionMinimum::Up, &FunctionMinimum::SetErrorDef)
.def(py::self == py::self)
.def(py::pickle(&fmin_getstate, &fmin_setstate))
;
}
|