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
|
#include "fcn.hpp"
#include "pybind11.hpp"
#include <Minuit2/FCNBase.h>
#include <Minuit2/MnMigrad.h>
namespace py = pybind11;
using namespace ROOT::Minuit2;
MnMigrad init(const FCN& fcn, const MnUserParameterState& state,
const MnStrategy& str) {
if (fcn.grad_.is_none()) {
return MnMigrad(static_cast<const FCNBase&>(fcn), state, str);
}
return MnMigrad(static_cast<const FCNGradientBase&>(fcn), state, str);
}
void bind_migrad(py::module m) {
py::class_<MnMigrad, MnApplication>(m, "MnMigrad")
.def(py::init(&init), py::keep_alive<1, 2>())
.def("set_print_level",
[](MnMigrad& self, int lvl) {
return self.Minimizer().Builder().SetPrintLevel(lvl);
})
;
}
|