Franck HOUSSEN
43d3e5606b
* arpackmm: improve command line.
* [BUG FIX] arpackmm: run arpack with real or imag shift.
* [BUG FIX] arpackmm: make sure the restart file is written.
* [BUG FIX] arpackmm: rewrite the test.
- Do NOT use eval in bash script which does NOT return command exit code.
- Use a dedicated maxResNorm to test the residual (no point to deduce
it from tol).
* Update .gitignore.
* Update CHANGELOG.
81 lines
2.7 KiB
C++
81 lines
2.7 KiB
C++
#ifndef __PYARPACKITRSOLVER_HPP__
|
|
#define __PYARPACKITRSOLVER_HPP__
|
|
|
|
#include <string>
|
|
|
|
#include <arpackSolver.hpp>
|
|
#include <pyarpackServices.hpp>
|
|
#include "debug_c.hpp"
|
|
#include "stat_c.hpp"
|
|
|
|
#include <boost/python.hpp>
|
|
#include <boost/python/numpy.hpp>
|
|
|
|
namespace bp = boost::python;
|
|
|
|
template<typename RC, typename FD, typename EM, typename SLV>
|
|
class pyarpackSparseItrSolver: public arpackItrSolver<RC, FD, EM, SLV> {
|
|
// Public methods.
|
|
|
|
public:
|
|
|
|
pyarpackSparseItrSolver(): arpackItrSolver<RC, FD, EM, SLV>() {
|
|
debug = 0;
|
|
|
|
nopx = 0, nbx = 0, nrorth = 0, nitref = 0, nrstrt = 0;
|
|
tsaupd = 0., tsaup2 = 0., tsaitr = 0., tseigt = 0., tsgets = 0., tsapps = 0., tsconv = 0.;
|
|
tnaupd = 0., tnaup2 = 0., tnaitr = 0., tneigt = 0., tngets = 0., tnapps = 0., tnconv = 0.;
|
|
tcaupd = 0., tcaup2 = 0., tcaitr = 0., tceigt = 0., tcgets = 0., tcapps = 0., tcconv = 0.;
|
|
tmvopx = 0., tmvbx = 0., tgetv0 = 0., titref = 0., trvec = 0.;
|
|
};
|
|
|
|
int solve(bp::tuple & A, bp::tuple B = bp::tuple()) {
|
|
ARPACKSOLVERDEBUGSTAT();
|
|
EM M;
|
|
int rc = pyarpackServices<RC, EM>::buildSparseMatrice(A, M, debug, "A");
|
|
if (rc != 0) {pyarpackThrowError("build matrice from A KO"); return rc;}
|
|
bool stdPb = (bp::len(B) > 0) ? false : true;
|
|
EM N;
|
|
if (!stdPb) {
|
|
rc = pyarpackServices<RC, EM>::buildSparseMatrice(B, N, debug, "B");
|
|
if (rc != 0) {pyarpackThrowError("build matrice from B KO"); return rc;}
|
|
}
|
|
return arpackItrSolver<RC, FD, EM, SLV>::solve(M, (stdPb ? NULL : &N));
|
|
};
|
|
|
|
int checkEigVec(bp::tuple const & A, bp::tuple const B = bp::tuple(), double const maxResNorm = 1.e-3) {
|
|
ARPACKSOLVERDEBUGSTAT();
|
|
EM M;
|
|
int rc = pyarpackServices<RC, EM>::buildSparseMatrice(A, M, debug, "A");
|
|
if (rc != 0) {pyarpackThrowError("build matrice from A KO"); return rc;}
|
|
bool stdPb = (bp::len(B) > 0) ? false : true;
|
|
EM N;
|
|
if (!stdPb) {
|
|
rc = pyarpackServices<RC, EM>::buildSparseMatrice(B, N, debug, "B");
|
|
if (rc != 0) {pyarpackThrowError("build matrice from B KO"); return rc;}
|
|
}
|
|
return arpackItrSolver<RC, FD, EM, SLV>::checkEigVec(M, (stdPb ? NULL : &N), maxResNorm);
|
|
};
|
|
|
|
// Public members.
|
|
|
|
public:
|
|
|
|
a_int debug;
|
|
|
|
a_int nopx, nbx, nrorth, nitref, nrstrt;
|
|
float tsaupd, tsaup2, tsaitr, tseigt, tsgets, tsapps, tsconv;
|
|
float tnaupd, tnaup2, tnaitr, tneigt, tngets, tnapps, tnconv;
|
|
float tcaupd, tcaup2, tcaitr, tceigt, tcgets, tcapps, tcconv;
|
|
float tmvopx, tmvbx, tgetv0, titref, trvec;
|
|
};
|
|
|
|
#endif
|
|
|
|
// Local Variables:
|
|
// mode: c++
|
|
// c-file-style:"stroustrup"
|
|
// show-trailing-whitespace: t
|
|
// End:
|
|
/* vim: set sw=2 ts=2 et smartindent :*/
|