devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

some more work on Functors

Browse Source
This commit is contained in:
Thomas Capricelli
2009-08-24 21:48:22 +02:00
parent 15d2c3af90
commit 45442b8d41
11 changed files with 63 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
unsupported/Eigen/src/NonLinear/fdjac1.h
View File

@@ -1,6 +1,7 @@
template<typename Functor, typename Scalar>
template<typename FunctorType, typename Scalar>
int ei_fdjac1(
const FunctorType &Functor,
Matrix< Scalar, Dynamic, 1 > &x,
Matrix< Scalar, Dynamic, 1 > &fvec,
Matrix< Scalar, Dynamic, Dynamic > &fjac,
@@ -30,7 +31,7 @@ int ei_fdjac1(
if (h == 0.)
h = eps;
x[j] = temp + h;
iflag = Functor::f(x, wa1);
iflag = Functor.f(x, wa1);
if (iflag < 0)
return iflag;
x[j] = temp;
@@ -46,7 +47,7 @@ int ei_fdjac1(
if (h == 0.) h = eps;
x[j] = wa2[j] + h;
}
iflag = Functor::f(x, wa1);
iflag = Functor.f(x, wa1);
if (iflag < 0) {
return iflag;
}

5
unsupported/Eigen/src/NonLinear/fdjac2.h
View File

@@ -1,6 +1,7 @@
template<typename Functor, typename Scalar>
template<typename FunctorType, typename Scalar>
int ei_fdjac2(
const FunctorType &Functor,
Matrix< Scalar, Dynamic, 1 > &x,
Matrix< Scalar, Dynamic, 1 > &fvec,
Matrix< Scalar, Dynamic, Dynamic > &fjac,
@@ -23,7 +24,7 @@ int ei_fdjac2(
h = eps;
}
x[j] = temp + h;
iflag = Functor::f(x, wa);
iflag = Functor.f(x, wa);
if (iflag < 0)
return iflag;
x[j] = temp;

15
unsupported/Eigen/src/NonLinear/hybrd.h
View File

@@ -1,6 +1,7 @@
template<typename Functor, typename Scalar>
template<typename FunctorType, typename Scalar>
int ei_hybrd(
const FunctorType &Functor,
Matrix< Scalar, Dynamic, 1 > &x,
Matrix< Scalar, Dynamic, 1 > &fvec,
int &nfev,
@@ -64,7 +65,7 @@ int ei_hybrd(
/* evaluate the function at the starting point */
/* and calculate its norm. */
iflag = Functor::f(x, fvec);
iflag = Functor.f(x, fvec);
nfev = 1;
if (iflag < 0)
goto algo_end;
@@ -91,7 +92,7 @@ int ei_hybrd(
/* calculate the jacobian matrix. */
iflag = ei_fdjac1<Functor,Scalar>(x, fvec, fjac,
iflag = ei_fdjac1(Functor, x, fvec, fjac,
nb_of_subdiagonals, nb_of_superdiagonals, epsfcn, wa1, wa2);
nfev += msum;
if (iflag < 0)
@@ -163,12 +164,12 @@ int ei_hybrd(
/* beginning of the inner loop. */
while (true) {
/* if requested, call Functor::f to enable printing of iterates. */
/* if requested, call Functor.f to enable printing of iterates. */
if (nprint > 0) {
iflag = 0;
if ((iter - 1) % nprint == 0)
iflag = Functor::debug(x, fvec);
iflag = Functor.debug(x, fvec);
if (iflag < 0)
goto algo_end;
}
@@ -191,7 +192,7 @@ int ei_hybrd(
/* evaluate the function at x + p and calculate its norm. */
iflag = Functor::f(wa2, wa4);
iflag = Functor.f(wa2, wa4);
++nfev;
if (iflag < 0)
goto algo_end;
@@ -319,7 +320,7 @@ algo_end:
if (iflag < 0)
info = iflag;
if (nprint > 0)
iflag = Functor::debug(x, fvec);
iflag = Functor.debug(x, fvec);
return info;
}

6
unsupported/Eigen/src/NonLinear/hybrd1.h
View File

@@ -1,6 +1,7 @@
template<typename Functor, typename Scalar>
template<typename FunctorType, typename Scalar>
int ei_hybrd1(
const FunctorType &Functor,
Matrix< Scalar, Dynamic, 1 > &x,
Matrix< Scalar, Dynamic, 1 > &fvec,
Scalar tol = ei_sqrt(epsilon<Scalar>())
@@ -18,7 +19,8 @@ int ei_hybrd1(
}
diag.setConstant(n, 1.);
info = ei_hybrd<Functor,Scalar>(
info = ei_hybrd(
Functor,
x, fvec,
nfev,
fjac,

15
unsupported/Eigen/src/NonLinear/hybrj.h
View File

@@ -1,6 +1,7 @@
template<typename Functor, typename Scalar>
template<typename FunctorType, typename Scalar>
int ei_hybrj(
const FunctorType &Functor,
Matrix< Scalar, Dynamic, 1 > &x,
Matrix< Scalar, Dynamic, 1 > &fvec,
int &nfev,
@@ -59,7 +60,7 @@ int ei_hybrj(
/* evaluate the function at the starting point */
/* and calculate its norm. */
iflag = Functor::f(x, fvec);
iflag = Functor.f(x, fvec);
nfev = 1;
if (iflag < 0)
goto algo_end;
@@ -80,7 +81,7 @@ int ei_hybrj(
/* calculate the jacobian matrix. */
iflag = Functor::df(x, fjac);
iflag = Functor.df(x, fjac);
++njev;
if (iflag < 0)
break;
@@ -151,12 +152,12 @@ int ei_hybrj(
/* beginning of the inner loop. */
while (true) {
/* if requested, call Functor::f to enable printing of iterates. */
/* if requested, call Functor.f to enable printing of iterates. */
if (nprint > 0) {
iflag = 0;
if ((iter - 1) % nprint == 0)
iflag = Functor::debug(x, fvec, fjac);
iflag = Functor.debug(x, fvec, fjac);
if (iflag < 0)
goto algo_end;
}
@@ -179,7 +180,7 @@ int ei_hybrj(
/* evaluate the function at x + p and calculate its norm. */
iflag = Functor::f(wa2, wa4);
iflag = Functor.f(wa2, wa4);
++nfev;
if (iflag < 0)
goto algo_end;
@@ -306,7 +307,7 @@ algo_end:
if (iflag < 0)
info = iflag;
if (nprint > 0)
iflag = Functor::debug(x, fvec, fjac);
iflag = Functor.debug(x, fvec, fjac);
return info;
}

6
unsupported/Eigen/src/NonLinear/hybrj1.h
View File

@@ -1,6 +1,7 @@
template<typename Functor, typename Scalar>
template<typename FunctorType, typename Scalar>
int ei_hybrj1(
const FunctorType &Functor,
Matrix< Scalar, Dynamic, 1 > &x,
Matrix< Scalar, Dynamic, 1 > &fvec,
Matrix< Scalar, Dynamic, Dynamic > &fjac,
@@ -18,7 +19,8 @@ int ei_hybrj1(
}
diag.setConstant(n, 1.);
info = ei_hybrj<Functor,Scalar>(
info = ei_hybrj(
Functor,
x, fvec,
nfev, njev,
fjac,

15
unsupported/Eigen/src/NonLinear/lmdif.h
View File

@@ -1,6 +1,7 @@
template<typename Functor, typename Scalar>
template<typename FunctorType, typename Scalar>
int ei_lmdif(
const FunctorType &Functor,
Matrix< Scalar, Dynamic, 1 > &x,
Matrix< Scalar, Dynamic, 1 > &fvec,
int &nfev,
@@ -54,7 +55,7 @@ int ei_lmdif(
/* evaluate the function at the starting point */
/* and calculate its norm. */
iflag = Functor::f(x, fvec);
iflag = Functor.f(x, fvec);
nfev = 1;
if (iflag < 0)
goto algo_end;
@@ -71,17 +72,17 @@ int ei_lmdif(
/* calculate the jacobian matrix. */
iflag = ei_fdjac2<Functor,Scalar>(x, fvec, fjac, epsfcn, wa4);
iflag = ei_fdjac2(Functor, x, fvec, fjac, epsfcn, wa4);
nfev += n;
if (iflag < 0)
break;
/* if requested, call Functor::f to enable printing of iterates. */
/* if requested, call Functor.f to enable printing of iterates. */
if (nprint > 0) {
iflag = 0;
if ((iter - 1) % nprint == 0)
iflag = Functor::debug(x, fvec);
iflag = Functor.debug(x, fvec);
if (iflag < 0)
break;
}
@@ -177,7 +178,7 @@ int ei_lmdif(
/* evaluate the function at x + p and calculate its norm. */
iflag = Functor::f(wa2, wa4);
iflag = Functor.f(wa2, wa4);
++nfev;
if (iflag < 0)
goto algo_end;
@@ -274,7 +275,7 @@ algo_end:
if (iflag < 0)
info = iflag;
if (nprint > 0)
iflag = Functor::debug(x, fvec);
iflag = Functor.debug(x, fvec);
return info;
}

6
unsupported/Eigen/src/NonLinear/lmdif1.h
View File

@@ -1,6 +1,7 @@
template<typename Functor, typename Scalar>
template<typename FunctorType, typename Scalar>
int ei_lmdif1(
const FunctorType &Functor,
Matrix< Scalar, Dynamic, 1 > &x,
Matrix< Scalar, Dynamic, 1 > &fvec,
Scalar tol = ei_sqrt(epsilon<Scalar>())
@@ -18,7 +19,8 @@ int ei_lmdif1(
return 0;
}
info = ei_lmdif<Functor,Scalar>(
info = ei_lmdif(
Functor,
x, fvec,
nfev,
fjac, ipvt, qtf, diag,

15
unsupported/Eigen/src/NonLinear/lmstr.h
View File

@@ -1,6 +1,7 @@
template<typename Functor, typename Scalar>
template<typename FunctorType, typename Scalar>
int ei_lmstr(
const FunctorType &Functor,
Matrix< Scalar, Dynamic, 1 > &x,
Matrix< Scalar, Dynamic, 1 > &fvec,
int &nfev,
@@ -55,7 +56,7 @@ int ei_lmstr(
/* evaluate the function at the starting point */
/* and calculate its norm. */
iflag = Functor::f(x, fvec);
iflag = Functor.f(x, fvec);
nfev = 1;
if (iflag < 0)
goto algo_end;
@@ -70,12 +71,12 @@ int ei_lmstr(
while (true) {
/* if requested, call Functor::f to enable printing of iterates. */
/* if requested, call Functor.f to enable printing of iterates. */
if (nprint > 0) {
iflag = 0;
if ((iter - 1) % nprint == 0)
iflag = Functor::debug(x, fvec, wa3);
iflag = Functor.debug(x, fvec, wa3);
if (iflag < 0)
break;
}
@@ -89,7 +90,7 @@ int ei_lmstr(
fjac.fill(0.);
iflag = 2;
for (i = 0; i < m; ++i) {
if (Functor::df(x, wa3, iflag) < 0)
if (Functor.df(x, wa3, iflag) < 0)
break;
temp = fvec[i];
ei_rwupdt<Scalar>(n, fjac.data(), fjac.rows(), wa3.data(), qtf.data(), &temp, wa1.data(), wa2.data());
@@ -194,7 +195,7 @@ int ei_lmstr(
/* evaluate the function at x + p and calculate its norm. */
iflag = Functor::f(wa2, wa4);
iflag = Functor.f(wa2, wa4);
++nfev;
if (iflag < 0)
goto algo_end;
@@ -291,7 +292,7 @@ algo_end:
if (iflag < 0)
info = iflag;
if (nprint > 0)
iflag = Functor::debug(x, fvec, wa3);
iflag = Functor.debug(x, fvec, wa3);
return info;
}

6
unsupported/Eigen/src/NonLinear/lmstr1.h
View File

@@ -1,6 +1,7 @@
template<typename Functor, typename Scalar>
template<typename FunctorType, typename Scalar>
int ei_lmstr1(
const FunctorType &Functor,
Matrix< Scalar, Dynamic, 1 > &x,
Matrix< Scalar, Dynamic, 1 > &fvec,
VectorXi &ipvt,
@@ -19,7 +20,8 @@ int ei_lmstr1(
}
ipvt.resize(n);
info = ei_lmstr<Functor,Scalar>(
info = ei_lmstr(
Functor,
x, fvec,
nfev, njev,
fjac, ipvt, qtf, diag,