QualityRouting.hpp

Go to the documentation of this file.

 
#ifndef OPENSWMM_QUALITY_ROUTING_HPP
#define OPENSWMM_QUALITY_ROUTING_HPP
 
#include <vector>
 
namespace openswmm {
 
struct SimulationContext;
 
namespace quality {
 
// ============================================================================
// Constants
// ============================================================================
 
constexpr double ZERO_VOLUME = 0.0353147;  
constexpr double ZERO_DEPTH  = 0.003281;   
 
// ============================================================================
// Quality solver
// ============================================================================
 
class QualitySolver {
public:
    void init(int n_nodes, int n_links, int n_pollutants);
 
    void execute(SimulationContext& ctx, double dt);
 
    void addWetWeatherLoads(SimulationContext& ctx, double dt);
 
private:
    int n_pollutants_ = 0;
 
    // Per-node quality working arrays: [node * n_pollutants + pollutant]
    std::vector<double> node_mass_in_;   
    std::vector<double> node_vol_in_;    
 
    void accumulateLinkLoads(SimulationContext& ctx, double dt);
 
    void mixAtNodes(SimulationContext& ctx, double dt);
 
    void applyDecay(SimulationContext& ctx, double dt);
 
    void updateLinkQuality(SimulationContext& ctx);
};
 
} // namespace quality
} // namespace openswmm
 
#endif // OPENSWMM_QUALITY_ROUTING_HPP