HydStructures.hpp

Go to the documentation of this file.

 
#ifndef OPENSWMM_HYD_STRUCTURES_HPP
#define OPENSWMM_HYD_STRUCTURES_HPP
 
#include "../data/LinkData.hpp"
#include "../data/NodeData.hpp"
#include "../data/TableData.hpp"
#include <vector>
 
namespace openswmm {
 
struct SimulationContext;
 
namespace hydstruct {
 
// ============================================================================
// Constants
// ============================================================================
 
constexpr double GRAVITY = 32.2;
constexpr double FUDGE   = 0.0001;
 
// ============================================================================
// Per-type SoA groups (like XSectGroups but for structure types)
// ============================================================================
 
struct PumpGroup {
    int count = 0;
    std::vector<int>    link_idx;      
    std::vector<int>    curve_idx;     
    std::vector<int>    curve_type;    
    std::vector<double> speed;         
    std::vector<double> y_on;          
    std::vector<double> y_off;         
    void resize(int n);
};
 
struct OrificeGroup {
    int count = 0;
    std::vector<int>    link_idx;
    std::vector<int>    shape;         
    std::vector<double> c_orifice;     
    std::vector<double> c_weir;        
    std::vector<double> h_crit;        
    std::vector<bool>   has_flap;      
    void resize(int n);
};
 
struct WeirGroup {
    int count = 0;
    std::vector<int>    link_idx;
    std::vector<int>    weir_type;     
    std::vector<double> c_disch1;      
    std::vector<double> c_disch2;      
    std::vector<double> end_con;       
    std::vector<double> slope;         
    std::vector<int>    cd_curve;      
    std::vector<bool>   has_flap;
    void resize(int n);
};
 
struct OutletGroup {
    int count = 0;
    std::vector<int>    link_idx;
    std::vector<int>    curve_idx;     
    std::vector<double> q_coeff;       
    std::vector<double> q_expon;       
    void resize(int n);
};
 
// ============================================================================
// Structure flow solver
// ============================================================================
 
class StructureSolver {
public:
    void init(SimulationContext& ctx);
 
    void computeAllFlows(SimulationContext& ctx, double dt);
 
private:
    PumpGroup    pumps_;
    OrificeGroup orifices_;
    WeirGroup    weirs_;
    OutletGroup  outlets_;
 
    void computePumpFlows(SimulationContext& ctx);
 
    void computeOrificeFlows(SimulationContext& ctx);
 
    void computeWeirFlows(SimulationContext& ctx);
 
    void computeOutletFlows(SimulationContext& ctx);
};
 
} // namespace hydstruct
} // namespace openswmm
 
#endif // OPENSWMM_HYD_STRUCTURES_HPP