SolWrapper.c

Go to the documentation of this file.

 
#include "fasp.h"
#include "fasp_block.h"
#include "fasp_functs.h"
 
/*---------------------------------*/
/*--      Public Functions       --*/
/*---------------------------------*/
 
#if WITH_PARDISO
void fasp_fwrapper_dcsr_pardiso_(
    INT* n, INT* nnz, INT* ia, INT* ja, REAL* a, REAL* b, REAL* u, INT* ptrlvl)
{
    dCSRmat mat;      // coefficient matrix
    dvector rhs, sol; // right-hand-side, solution
 
    // set up coefficient matrix
    mat.row = *n;
    mat.col = *n;
    mat.nnz = *nnz;
    mat.IA  = ia;
    mat.JA  = ja;
    mat.val = a;
 
    rhs.row = *n;
    rhs.val = b;
    sol.row = *n;
    sol.val = u;
 
    fasp_dcsr_sort(&mat);
 
    fasp_solver_pardiso(&mat, &rhs, &sol, *ptrlvl);
}
#endif
 
#if WITH_UMFPACK
void fasp_fwrapper_dcsr_strumpack_(
    INT* n, INT* nnz, INT* ia, INT* ja, REAL* a, REAL* b, REAL* u, INT* ptrlvl)
{
    dCSRmat mat;      // coefficient matrix
    dvector rhs, sol; // right-hand-side, solution
 
    // set up coefficient matrix
    mat.row = *n;
    mat.col = *n;
    mat.nnz = *nnz;
    mat.IA  = ia;
    mat.JA  = ja;
    mat.val = a;
 
    rhs.row = *n;
    rhs.val = b;
    sol.row = *n;
    sol.val = u;
 
    fasp_dcsr_sort(&mat);
 
    fasp_solver_strumpack(&mat, &rhs, &sol, *ptrlvl);
}
#endif
 
void fasp_fwrapper_dcsr_amg_(INT*  n,
                             INT*  nnz,
                             INT*  ia,
                             INT*  ja,
                             REAL* a,
                             REAL* b,
                             REAL* u,
                             REAL* tol,
                             INT*  maxit,
                             INT*  ptrlvl)
{
    dCSRmat   mat;      // coefficient matrix
    dvector   rhs, sol; // right-hand-side, solution
    AMG_param amgparam; // parameters for AMG
 
    // setup AMG parameters
    fasp_param_amg_init(&amgparam);
 
    amgparam.tol         = *tol;
    amgparam.print_level = *ptrlvl;
    amgparam.maxit       = *maxit;
 
    // set up coefficient matrix
    mat.row = *n;
    mat.col = *n;
    mat.nnz = *nnz;
    mat.IA  = ia;
    mat.JA  = ja;
    mat.val = a;
 
    rhs.row = *n;
    rhs.val = b;
    sol.row = *n;
    sol.val = u;
 
    fasp_solver_amg(&mat, &rhs, &sol, &amgparam);
}
 
void fasp_fwrapper_dcsr_krylov_ilu_(INT*  n,
                                    INT*  nnz,
                                    INT*  ia,
                                    INT*  ja,
                                    REAL* a,
                                    REAL* b,
                                    REAL* u,
                                    REAL* tol,
                                    INT*  maxit,
                                    INT*  ptrlvl)
{
    dCSRmat   mat;      // coefficient matrix
    dvector   rhs, sol; // right-hand-side, solution
    ILU_param iluparam; // parameters for ILU
    ITS_param itsparam; // parameters for itsolver
 
    // setup ILU parameters
    fasp_param_ilu_init(&iluparam);
 
    iluparam.print_level = *ptrlvl;
 
    // setup Krylov method parameters
    fasp_param_solver_init(&itsparam);
 
    itsparam.itsolver_type = SOLVER_VFGMRES;
    itsparam.tol           = *tol;
    itsparam.maxit         = *maxit;
    itsparam.print_level   = *ptrlvl;
 
    // set up coefficient matrix
    mat.row = *n;
    mat.col = *n;
    mat.nnz = *nnz;
    mat.IA  = ia;
    mat.JA  = ja;
    mat.val = a;
 
    rhs.row = *n;
    rhs.val = b;
    sol.row = *n;
    sol.val = u;
 
    fasp_solver_dcsr_krylov_ilu(&mat, &rhs, &sol, &itsparam, &iluparam);
}
 
void fasp_fwrapper_dcsr_krylov_amg_(INT*  n,
                                    INT*  nnz,
                                    INT*  ia,
                                    INT*  ja,
                                    REAL* a,
                                    REAL* b,
                                    REAL* u,
                                    REAL* tol,
                                    INT*  maxit,
                                    INT*  ptrlvl)
{
    dCSRmat     mat;      // coefficient matrix
    dvector     rhs, sol; // right-hand-side, solution
    input_param inparam;  // parameters from input files
    AMG_param   amgparam; // parameters for AMG
    ITS_param   itsparam; // parameters for itsolver
    ILU_param   iluparam; // parameters for ILU
 
    char* inputfile = "ini/amg.dat"; // Added for fasp4ns 2022.04.08 --zcs
    fasp_param_input(inputfile, &inparam);
    fasp_param_init(&inparam, &itsparam, &amgparam, &iluparam, NULL);
 
    itsparam.tol         = *tol;
    itsparam.maxit       = *maxit;
    itsparam.print_level = *ptrlvl;
 
    // set up coefficient matrix
    mat.row = *n;
    mat.col = *n;
    mat.nnz = *nnz;
    mat.IA  = ia;
    mat.JA  = ja;
    mat.val = a;
 
    rhs.row = *n;
    rhs.val = b;
    sol.row = *n;
    sol.val = u;
 
    fasp_solver_dcsr_krylov_amg(&mat, &rhs, &sol, &itsparam, &amgparam);
}
 
void fasp_fwrapper_dbsr_krylov_ilu_(INT*  n,
                                    INT*  nnz,
                                    INT*  nb,
                                    INT*  ia,
                                    INT*  ja,
                                    REAL* a,
                                    REAL* b,
                                    REAL* u,
                                    REAL* tol,
                                    INT*  maxit,
                                    INT*  ptrlvl)
{
    dBSRmat mat;        // coefficient matrix in BSR format
    dvector rhs, sol;   // right-hand-side, solution
 
    ILU_param iluparam; // parameters for ILU
    ITS_param itsparam; // parameters for itsolver
 
    // setup ILU parameters
    fasp_param_ilu_init(&iluparam);
    iluparam.ILU_lfil    = 0;
    iluparam.print_level = *ptrlvl;
 
    // setup Krylov method parameters
    fasp_param_solver_init(&itsparam);
 
    itsparam.itsolver_type = SOLVER_VFGMRES;
    itsparam.tol           = *tol;
    itsparam.maxit         = *maxit;
    itsparam.print_level   = *ptrlvl;
 
    // set up coefficient matrix
    mat.ROW = *n;
    mat.COL = *n;
    mat.NNZ = *nnz;
    mat.nb  = *nb;
    mat.IA  = ia;
    mat.JA  = ja;
    mat.val = a;
 
    rhs.row = *n * *nb;
    rhs.val = b;
    sol.row = *n * *nb;
    sol.val = u;
 
    // solve
    fasp_solver_dbsr_krylov_ilu(&mat, &rhs, &sol, &itsparam, &iluparam);
}
 
void fasp_fwrapper_dbsr_krylov_amg_(INT*  n,
                                    INT*  nnz,
                                    INT*  nb,
                                    INT*  ia,
                                    INT*  ja,
                                    REAL* a,
                                    REAL* b,
                                    REAL* u,
                                    REAL* tol,
                                    INT*  maxit,
                                    INT*  ptrlvl)
{
    dBSRmat mat;        // coefficient matrix in CSR format
    dvector rhs, sol;   // right-hand-side, solution
 
    AMG_param amgparam; // parameters for AMG
    ITS_param itsparam; // parameters for itsolver
 
    // setup AMG parameters
    fasp_param_amg_init(&amgparam);
    amgparam.AMG_type    = UA_AMG;
    amgparam.print_level = *ptrlvl;
 
    // setup Krylov method parameters
    fasp_param_solver_init(&itsparam);
    itsparam.tol           = *tol;
    itsparam.print_level   = *ptrlvl;
    itsparam.maxit         = *maxit;
    itsparam.itsolver_type = SOLVER_VFGMRES;
 
    // set up coefficient matrix
    mat.ROW = *n;
    mat.COL = *n;
    mat.NNZ = *nnz;
    mat.nb  = *nb;
    mat.IA  = ia;
    mat.JA  = ja;
    mat.val = a;
 
    rhs.row = *n * *nb;
    rhs.val = b;
    sol.row = *n * *nb;
    sol.val = u;
 
    // solve
    fasp_solver_dbsr_krylov_amg(&mat, &rhs, &sol, &itsparam, &amgparam);
}
 
/*---------------------------------*/
/*--        End of File          --*/
/*---------------------------------*/