mirror of
https://gitlab.com/libeigen/eigen.git
synced 2026-04-10 11:34:33 +08:00
* use eigen object for callbacks for hybrd and lmdif
* use Functor instead of argument for ei_fdjac*()
This commit is contained in:
@@ -36,10 +36,6 @@ namespace Eigen {
|
||||
*/
|
||||
//@{
|
||||
|
||||
/* Declarations for minpack */
|
||||
typedef int (*minpack_func_nn)(int n, const double *x, double *fvec, int iflag );
|
||||
typedef int (*minpack_func_mn)(int m, int n, const double *x, double *fvec, int iflag );
|
||||
|
||||
#include "src/NonLinear/qrsolv.h"
|
||||
#include "src/NonLinear/r1updt.h"
|
||||
#include "src/NonLinear/r1mpyq.h"
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
|
||||
template <typename Scalar>
|
||||
int ei_fdjac1(minpack_func_nn fcn,
|
||||
template<typename Functor, typename Scalar>
|
||||
int ei_fdjac1(
|
||||
Matrix< Scalar, Dynamic, 1 > &x,
|
||||
Matrix< Scalar, Dynamic, 1 > &fvec,
|
||||
Matrix< Scalar, Dynamic, Dynamic > &fjac,
|
||||
@@ -34,7 +34,7 @@ int ei_fdjac1(minpack_func_nn fcn,
|
||||
if (h == 0.)
|
||||
h = eps;
|
||||
x[j] = temp + h;
|
||||
iflag = (*fcn)(n, x.data(), wa1.data(), 1);
|
||||
iflag = Functor::f(x, wa1, 1);
|
||||
if (iflag < 0)
|
||||
goto L30;
|
||||
x[j] = temp;
|
||||
@@ -59,7 +59,7 @@ L40:
|
||||
x[j] = wa2[j] + h;
|
||||
/* L60: */
|
||||
}
|
||||
iflag = (*fcn)(n, x.data(), wa1.data(), 1);
|
||||
iflag = Functor::f(x, wa1, 1);
|
||||
if (iflag < 0) {
|
||||
/* goto L100; */
|
||||
return iflag;
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
|
||||
template <typename Scalar>
|
||||
int ei_fdjac2(minpack_func_mn fcn,
|
||||
template<typename Functor, typename Scalar>
|
||||
int ei_fdjac2(
|
||||
Matrix< Scalar, Dynamic, 1 > &x,
|
||||
Matrix< Scalar, Dynamic, 1 > &fvec,
|
||||
Matrix< Scalar, Dynamic, Dynamic > &fjac,
|
||||
@@ -26,7 +26,7 @@ int ei_fdjac2(minpack_func_mn fcn,
|
||||
h = eps;
|
||||
}
|
||||
x[j] = temp + h;
|
||||
iflag = (*fcn)(m, n, x.data(), wa.data(), 1);
|
||||
iflag = Functor::f(x, wa, 1);
|
||||
if (iflag < 0) {
|
||||
/* goto L30; */
|
||||
return iflag;
|
||||
|
||||
@@ -75,7 +75,7 @@ L20:
|
||||
/* evaluate the function at the starting point */
|
||||
/* and calculate its norm. */
|
||||
|
||||
iflag = Functor::f(n, x.data(), fvec.data(), 1);
|
||||
iflag = Functor::f(x, fvec, 1);
|
||||
nfev = 1;
|
||||
if (iflag < 0) {
|
||||
goto L300;
|
||||
@@ -103,7 +103,7 @@ L30:
|
||||
|
||||
/* calculate the jacobian matrix. */
|
||||
|
||||
iflag = ei_fdjac1<Scalar>(Functor::f, x, fvec, fjac,
|
||||
iflag = ei_fdjac1<Functor,Scalar>(x, fvec, fjac,
|
||||
nb_of_subdiagonals, nb_of_superdiagonals, epsfcn, wa1, wa2);
|
||||
nfev += msum;
|
||||
if (iflag < 0) {
|
||||
@@ -215,7 +215,7 @@ L180:
|
||||
}
|
||||
iflag = 0;
|
||||
if ((iter - 1) % nprint == 0) {
|
||||
iflag = Functor::f(n, x.data(), fvec.data(), 0);
|
||||
iflag = Functor::f(x, fvec, 0);
|
||||
}
|
||||
if (iflag < 0) {
|
||||
goto L300;
|
||||
@@ -244,7 +244,7 @@ L190:
|
||||
|
||||
/* evaluate the function at x + p and calculate its norm. */
|
||||
|
||||
iflag = Functor::f(n, wa2.data(), wa4.data(), 1);
|
||||
iflag = Functor::f(wa2, wa4, 0);
|
||||
++nfev;
|
||||
if (iflag < 0) {
|
||||
goto L300;
|
||||
@@ -404,7 +404,7 @@ L300:
|
||||
info = iflag;
|
||||
}
|
||||
if (nprint > 0) {
|
||||
Functor::f(n, x.data(), fvec.data(), 0);
|
||||
iflag = Functor::f(x, fvec, 0);
|
||||
}
|
||||
return info;
|
||||
|
||||
|
||||
@@ -66,7 +66,7 @@ L20:
|
||||
/* evaluate the function at the starting point */
|
||||
/* and calculate its norm. */
|
||||
|
||||
iflag = Functor::f(m, n, x.data(), fvec.data(), 1);
|
||||
iflag = Functor::f(x, fvec, 1);
|
||||
nfev = 1;
|
||||
if (iflag < 0) {
|
||||
goto L300;
|
||||
@@ -84,7 +84,7 @@ L30:
|
||||
|
||||
/* calculate the jacobian matrix. */
|
||||
|
||||
iflag = ei_fdjac2<Scalar>(Functor::f, x, fvec, fjac, epsfcn, wa4);
|
||||
iflag = ei_fdjac2<Functor,Scalar>(x, fvec, fjac, epsfcn, wa4);
|
||||
nfev += n;
|
||||
if (iflag < 0) {
|
||||
goto L300;
|
||||
@@ -97,7 +97,7 @@ L30:
|
||||
}
|
||||
iflag = 0;
|
||||
if ((iter - 1) % nprint == 0) {
|
||||
iflag = Functor::f(m, n, x.data(), fvec.data(), 0);
|
||||
iflag = Functor::f(x, fvec, 0);
|
||||
}
|
||||
if (iflag < 0) {
|
||||
goto L300;
|
||||
@@ -239,7 +239,7 @@ L200:
|
||||
|
||||
/* evaluate the function at x + p and calculate its norm. */
|
||||
|
||||
iflag = Functor::f(m, n, wa2.data(), wa4.data(), 1);
|
||||
iflag = Functor::f(wa2, wa4, 1);
|
||||
++nfev;
|
||||
if (iflag < 0) {
|
||||
goto L300;
|
||||
@@ -380,7 +380,7 @@ L300:
|
||||
}
|
||||
iflag = 0;
|
||||
if (nprint > 0) {
|
||||
iflag = Functor::f(m, n, x.data(), fvec.data(), 0);
|
||||
iflag = Functor::f(x, fvec, 0);
|
||||
}
|
||||
return info;
|
||||
|
||||
|
||||
Reference in New Issue
Block a user