KinematicWave.hpp

Go to the documentation of this file.

 
#ifndef OPENSWMM_KINEMATIC_WAVE_HPP
#define OPENSWMM_KINEMATIC_WAVE_HPP
 
#include "XSectBatch.hpp"
#include <vector>
 
namespace openswmm {
 
struct SimulationContext;
 
namespace kinwave {
 
// ============================================================================
// Constants (matching legacy)
// ============================================================================
 
constexpr double WX    = 0.6;     
constexpr double WT    = 0.6;     
constexpr double EPSIL = 0.001;   
 
// ============================================================================
// KW solver — operates on entire conduit set
// ============================================================================
 
class KWSolver {
public:
    void init(int n_conduits, const XSectGroups& groups);
 
    void setLinkOrder(const std::vector<int>& sorted_links) {
        sorted_links_ = sorted_links;
    }
 
    int execute(SimulationContext& ctx, double dt);
 
    std::vector<int> sorted_links_;
 
    int solveConduit(int idx, const XSectParams& xs,
                     double q_full, double a_full, double s_full,
                     double beta, double length, double dt,
                     double loss_rate);
 
    // Per-conduit SoA state (indexed by conduit-link index)
    // Public so that unit tests can set up / inspect working arrays directly.
    std::vector<double> q1_;    
    std::vector<double> a1_;    
    std::vector<double> q2_;    
    std::vector<double> a2_;    
 
    // Working buffers (reused each timestep)
    std::vector<double> q_in_;      
    std::vector<double> a_in_;      
    std::vector<double> q_out_;     
    std::vector<double> a_out_;     
    std::vector<double> sf_in_;     
 
private:
    int n_conduits_ = 0;
};
 
} // namespace kinwave
} // namespace openswmm
 
#endif // OPENSWMM_KINEMATIC_WAVE_HPP