fdm_linear_system2.h

Go to the documentation of this file.

// Copyright (c) 2018 Doyub Kim
//
// I am making my contributions/submissions to this project solely in my
// personal capacity and am not conveying any rights to any intellectual
// property of any third parties.
 
#ifndef INCLUDE_JET_FDM_LINEAR_SYSTEM2_H_
#define INCLUDE_JET_FDM_LINEAR_SYSTEM2_H_
 
#include <jet/array1.h>
#include <jet/array2.h>
#include <jet/matrix_csr.h>
#include <jet/vector_n.h>
 
namespace jet {
 
struct FdmMatrixRow2 {
    double center = 0.0;
 
    double right = 0.0;
 
    double up = 0.0;
};
 
typedef Array2<double> FdmVector2;
 
typedef Array2<FdmMatrixRow2> FdmMatrix2;
 
struct FdmLinearSystem2 {
    FdmMatrix2 A;
 
    FdmVector2 x;
 
    FdmVector2 b;
 
    void clear();
 
    void resize(const Size2& size);
};
 
struct FdmCompressedLinearSystem2 {
    MatrixCsrD A;
 
    VectorND x;
 
    VectorND b;
 
    void clear();
};
 
struct FdmBlas2 {
    typedef double ScalarType;
    typedef FdmVector2 VectorType;
    typedef FdmMatrix2 MatrixType;
 
    static void set(ScalarType s, VectorType* result);
 
    static void set(const VectorType& v, VectorType* result);
 
    static void set(ScalarType s, MatrixType* result);
 
    static void set(const MatrixType& m, MatrixType* result);
 
    static double dot(const VectorType& a, const VectorType& b);
 
    static void axpy(double a, const VectorType& x, const VectorType& y,
                     VectorType* result);
 
    static void mvm(const MatrixType& m, const VectorType& v,
                    VectorType* result);
 
    static void residual(const MatrixType& a, const VectorType& x,
                         const VectorType& b, VectorType* result);
 
    static ScalarType l2Norm(const VectorType& v);
 
    static ScalarType lInfNorm(const VectorType& v);
};
 
struct FdmCompressedBlas2 {
    typedef double ScalarType;
    typedef VectorND VectorType;
    typedef MatrixCsrD MatrixType;
 
    static void set(ScalarType s, VectorType* result);
 
    static void set(const VectorType& v, VectorType* result);
 
    static void set(ScalarType s, MatrixType* result);
 
    static void set(const MatrixType& m, MatrixType* result);
 
    static double dot(const VectorType& a, const VectorType& b);
 
    static void axpy(double a, const VectorType& x, const VectorType& y,
                     VectorType* result);
 
    static void mvm(const MatrixType& m, const VectorType& v,
                    VectorType* result);
 
    static void residual(const MatrixType& a, const VectorType& x,
                         const VectorType& b, VectorType* result);
 
    static ScalarType l2Norm(const VectorType& v);
 
    static ScalarType lInfNorm(const VectorType& v);
};
 
}  // namespace jet
 
#endif  // INCLUDE_JET_FDM_LINEAR_SYSTEM2_H_