sparspaksolver.h

Go to the documentation of this file.

/* EPANET 3
 *
 * Copyright (c) 2016 Open Water Analytics
 * Licensed under the terms of the MIT License (see the LICENSE file for details).
 *
 */


#ifndef SPARSPAKSOLVER_H_
#define SPARSPAKSOLVER_H_

#include "matrixsolver.h"


class SparspakSolver: public MatrixSolver
{
  public:

    // Constructor/Destructor

    SparspakSolver(std::ostream& logger);
    ~SparspakSolver();

    // Methods

    int    init(int nrows, int nnz, int* xrow, int* xcol);
    void   reset();

    double getDiag(int i);
    double getOffDiag(int i);
    double getRhs(int i);

    void   setDiag(int i, double a);
    void   setRhs(int i, double b);
    void   addToDiag(int i, double a);
    void   addToOffDiag(int j, double a);
    void   addToRhs(int i, double b);
    int    solve(int n, double x[]);

  private:

    int     nrows;    // number of rows in system Ax = b
    int     nnz;      // number of non-zero off-diag. coeffs. in A
    int     nnzl;     // number of non-zero off-diag. coeffs. in factorized matrix L
    int*    perm;     // permutation of rows in A
    int*    invp;     // inverse row permutation
    int*    xlnz;     // index vector for non-zero entries in L
    int*    xnzsub;   // index vector for entries of nzsub
    int*    nzsub;    // column indexes for non-zero entries in each row of L
    int*    xaij;     // maps off-diag. coeffs. of A to lnz
    int*    link;     // work array
    int*    first;    // work array
    double* lnz;      // off-diag. coeffs. of factorized matrix L
    double* diag;     // diagonal coeffs. of A
    double* rhs;      // right hand side vector
    double* temp;     // work array
    std::ostream& msgLog;
};

#endif