FDMLinearSystem2.hpp

Go to the documentation of this file.

// This code is based on Jet framework.
// Copyright (c) 2018 Doyub Kim
// CubbyFlow is voxel-based fluid simulation engine for computer games.
// Copyright (c) 2020 CubbyFlow Team
// Core Part: Chris Ohk, Junwoo Hwang, Jihong Sin, Seungwoo Yoo
// AI Part: Dongheon Cho, Minseo Kim
// We are making my contributions/submissions to this project solely in our
// personal capacity and are not conveying any rights to any intellectual
// property of any third parties.

#ifndef CUBBYFLOW_FDM_LINEAR_SYSTEM2_HPP
#define CUBBYFLOW_FDM_LINEAR_SYSTEM2_HPP

#include <Core/Array/Array.hpp>
#include <Core/Matrix/Matrix.hpp>
#include <Core/Matrix/MatrixCSR.hpp>

namespace CubbyFlow
{
struct FDMMatrixRow2
{
    double center = 0.0;

    double right = 0.0;

    double up = 0.0;
};

using FDMVector2 = Array2<double>;

using FDMMatrix2 = Array2<FDMMatrixRow2>;

struct FDMLinearSystem2
{
    void Clear();

    void Resize(const Vector2UZ& size);

    FDMMatrix2 A;

    FDMVector2 x;

    FDMVector2 b;
};

struct FDMCompressedLinearSystem2
{
    void Clear();

    MatrixCSRD A;

    VectorND x;

    VectorND b;
};

struct FDMBLAS2
{
    using ScalarType = double;
    using VectorType = FDMVector2;
    using MatrixType = FDMMatrix2;

    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);

    [[nodiscard]] static ScalarType L2Norm(const VectorType& v);

    [[nodiscard]] static ScalarType LInfNorm(const VectorType& v);
};

struct FDMCompressedBLAS2
{
    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);

    [[nodiscard]] static ScalarType L2Norm(const VectorType& v);

    [[nodiscard]] static ScalarType LInfNorm(const VectorType& v);
};
}  // namespace CubbyFlow

#endif