NodeCoupling.hpp

Go to the documentation of this file.

 
#ifndef OPENSWMM_ENGINE_2D_NODE_COUPLING_HPP
#define OPENSWMM_ENGINE_2D_NODE_COUPLING_HPP
 
#include <unordered_map>
 
#include "../data/MeshData.hpp"
#include "../data/SurfaceStateData.hpp"
#include "../data/SolverOptions2D.hpp"
 
// Forward declaration — avoid pulling in full SimulationContext
namespace openswmm {
struct SimulationContext;
struct NodeData;
}
 
namespace openswmm::twoD {

struct CouplingPoint {
    int cell_idx;       
    int vertex_idx;     
    int node_idx;       
    double cd;          
    double area;        
    bool is_outfall;    
    bool has_flap_gate; 
};

std::vector<CouplingPoint> buildCouplingPoints(const MeshData& mesh,
                                                const SimulationContext& ctx);

void computeCouplingExchange(const std::vector<CouplingPoint>& cps,
                              const MeshData& mesh,
                              SurfaceStateData& state,
                              SimulationContext& ctx,
                              const SolverOptions2D& opts,
                              double dt);

double computeNodeCouplingQ(const CouplingPoint& cp,
                            const MeshData& mesh,
                            const SurfaceStateData& state,
                            const NodeData& nodes,
                            const SolverOptions2D& opts) noexcept;

void scatterCouplingToYdot(const MeshData& mesh,
                           const SurfaceStateData& state,
                           const CouplingPoint& cp,
                           double Q,
                           double* ydot) noexcept;

void updateOutfallBoundaries(const std::vector<CouplingPoint>& cps,
                              const MeshData& mesh,
                              const SurfaceStateData& state,
                              SimulationContext& ctx,
                              const SolverOptions2D& opts);

int transferOutfallDischarges(const std::vector<CouplingPoint>& cps,
                                const MeshData& mesh,
                                SurfaceStateData& state,
                                const SimulationContext& ctx,
                                const SolverOptions2D& opts,
                                double dt,
                                std::unordered_map<int, double>& applied_q);
 
} // namespace openswmm::twoD
 
#endif // OPENSWMM_ENGINE_2D_NODE_COUPLING_HPP