dp2d_gs.cc

来自「Flens库-一个在C++的矩阵运算库」· CC 代码 · 共 31 行

#include <poisson_solver/flens_impl.h>

#include <dirichlet_poisson.h>

using namespace flens;
using namespace std;

typedef DirichletPoisson2D MatType;
typedef GridVector2D       VecType;

typedef GaussSeidelRedBlack<MatType, VecType>  Smoother;

int
main()
{
    int l = 11;
    int rh = (1<<l);
    MatType A(rh);
    VecType f(rh), u(rh), solution(rh);

    problem2(f, u, solution);

    Smoother S(A, f);

    errorStat(0, A, f, u, solution);
    for (int it=1; it<=10; ++it) {
        u = S*u;
        errorStat(it, A, f, u, solution);
    }
}