SurfaceJacobian.hpp

Go to the documentation of this file.

 
#ifndef OPENSWMM_ENGINE_2D_SURFACE_JACOBIAN_HPP
#define OPENSWMM_ENGINE_2D_SURFACE_JACOBIAN_HPP
 
#include "../data/MeshData.hpp"
#include "../data/SurfaceStateData.hpp"
 
#include <vector>
 
namespace openswmm::twoD {

class SurfaceJacobian {
public:
    void buildSparsity(const MeshData& mesh);

    void assemble(const MeshData& mesh, const SurfaceStateData& state,
                  double gamma, double dh_floor = 1.0e-9);
 
    int           rows()   const noexcept { return n_; }
    int           nnz()    const noexcept { return static_cast<int>(col_idx_.size()); }
    const int*    rowPtr() const noexcept { return row_ptr_.data(); }
    const int*    colIdx() const noexcept { return col_idx_.data(); }
    const double* values() const noexcept { return values_.data(); }
    double*       values()       noexcept { return values_.data(); }
 
private:
    int                 n_ = 0;
    std::vector<int>    row_ptr_;        
    std::vector<int>    col_idx_;        
    std::vector<double> values_;         
    std::vector<int>    diag_pos_;       
    std::vector<int>    edge_pos_;       
};
 
} // namespace openswmm::twoD
 
#endif // OPENSWMM_ENGINE_2D_SURFACE_JACOBIAN_HPP