FDMLinearSystem3.h

Go to the documentation of this file.

/*************************************************************************
> File Name: FDMLinearSystem3.h
> Project Name: CubbyFlow
> Author: Chan-Ho Chris Ohk
> Purpose: Linear system (Ax=b) for 3-D finite differencing.
> Created Time: 2017/08/12
> Copyright (c) 2018, Chan-Ho Chris Ohk
*************************************************************************/
#ifndef CUBBYFLOW_FDM_LINEAR_SYSTEM3_H
#define CUBBYFLOW_FDM_LINEAR_SYSTEM3_H

#include <Core/Array/Array3.h>
#include <Core/Matrix/MatrixCSR.h>
#include <Core/Size/Size3.h>
#include <Core/Vector/VectorN.h>

namespace CubbyFlow
{
    struct FDMMatrixRow3
    {
        double center = 0.0;

        double right = 0.0;

        double up = 0.0;

        double front = 0.0;
    };

    using FDMVector3 = Array3<double>;

    using FDMMatrix3 = Array3<FDMMatrixRow3>;

    struct FDMLinearSystem3
    {
        FDMMatrix3 A;

        FDMVector3 x;

        FDMVector3 b;

        void Clear();

        void Resize(const Size3& size);
    };

    struct FDMCompressedLinearSystem3
    {
        MatrixCSRD A;

        VectorND x;

        VectorND b;

        void Clear();
    };

    struct FDMBLAS3
    {
        using ScalarType = double;
        using VectorType = FDMVector3;
        using MatrixType = FDMMatrix3;

        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 AXPlusY(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 FDMCompressedBLAS3
    {
        using ScalarType = double;
        using VectorType = VectorND;
        using MatrixType = MatrixCSRD;

        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 AXPlusY(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);
    };
}

#endif