LinearSystem.hpp

Go to the documentation of this file.

 
#ifndef __LINEARSYSTEM_HEADER__
#define __LINEARSYSTEM_HEADER__
 
// Standard header files
#include <cmath>
#include <fstream>
#include <iostream>
#include <string>
 
// OpenCAEPoro header files
#include "FaspSolver.hpp"
#include "OCPConst.hpp"
 
using namespace std;
 
//  Note: The matrix is stored in the form of row-segmented CSRx internaly, whose
//  sparsity pattern is almost the same as neighbor in BulkConn.
class LinearSystem
{
    friend class OpenCAEPoro;
    friend class BulkConn;
    friend class Well;
 
public:
    void AllocateRowMem(const OCP_USI& dimMax, const USI& nb);
    void AllocateColMem();
    void AllocateColMem(const OCP_USI& colnum);
    void EnlargeRowCap(const OCP_USI& row, const USI& n) { rowCapacity[row] += n; }
    void AssembleRhs(const vector<OCP_DBL>& rhs);
    void ClearData();
 
    
    vector<OCP_DBL>& GetSolution() { return u; }
    void CheckEquation() const;
    void CheckSolution() const;
    void OutputLinearSystem(const string& fileA, const string& fileb) const;
    void OutputSolution(const string& filename) const;
 
    // Linear Solver
    void SetupLinearSolver(const USI& i, const string& dir, const string& file);
    void AssembleMatLinearSolver() { LS->AssembleMat(colId, val, dim, blockDim, b, u); }
    OCP_INT Solve() { return LS->Solve(); }
 
    USI GetNumIters() { return LS->GetNumIters(); }
 
private:
    // Used for internal mat structure.
    USI blockDim;  
    USI blockSize; 
 
    // maxDim: fixed and used to allocate memory at the beginning of simulation;
    // dim:    might change during simulation but always less than maxDim.
    OCP_USI maxDim; 
    OCP_USI dim;    
 
    // The following values are stored for each row. Among them, rowCapacity is the max
    // possible capacity of each row of the matrix. It is just a little bigger than the
    // actual size and is used to allocate memory at the beginning of simulation.
    // diagVal is an auxiliary variable used to help setup entries in diagnal line and
    // it will only be used when matrix is assembled.
    vector<USI>             rowCapacity; 
    vector<vector<OCP_USI>> colId;       
    vector<USI>             diagPtr;     
    vector<vector<OCP_DBL>> val;         
    vector<OCP_DBL>         diagVal;     
    vector<OCP_DBL>         b;           
    vector<OCP_DBL>         u;           
 
    string solveDir; 
 
    LinearSolver* LS;
};
 
#endif /* end if __LINEARSOLVER_HEADER__ */
 
/*----------------------------------------------------------------------------*/
/*  Brief Change History of This File                                         */
/*----------------------------------------------------------------------------*/
/*  Author              Date             Actions                              */
/*----------------------------------------------------------------------------*/
/*  Shizhe Li           Oct/01/2021      Create file                          */
/*  Chensong Zhang      Oct/15/2021      Format file                          */
/*  Chensong Zhang      Nov/09/2021      Remove decoupling methods            */
/*  Chensong Zhang      Nov/22/2021      renamed to LinearSystem              */
/*----------------------------------------------------------------------------*/