Routing.hpp

Go to the documentation of this file.

 
#ifndef OPENSWMM_ROUTING_HPP
#define OPENSWMM_ROUTING_HPP
 
#include "XSectBatch.hpp"
#include "KinematicWave.hpp"
#include "DynamicWave.hpp"
#include "Divider.hpp"
 
namespace openswmm {
 
struct SimulationContext;
 
// ============================================================================
// Routing model enum
// ============================================================================
 
enum class RouteModel : int {
    STEADY    = 0,   
    KINWAVE   = 1,   
    DYNWAVE   = 2    
};
 
// ============================================================================
// Routing orchestrator
// ============================================================================
 
class Router {
public:
    void init(SimulationContext& ctx, RouteModel model);
 
    int step(SimulationContext& ctx, double dt);
 
    double getAdaptiveStep(const SimulationContext& ctx,
                           double fixed_step, double courant) const;
 
    const XSectGroups& xsectGroups() const { return groups_; }
 
    void setDWNumThreads(int n) { dw_solver_.setNumThreads(n); }
 
private:
    RouteModel model_ = RouteModel::DYNWAVE;
    XSectGroups groups_;
    kinwave::KWSolver kw_solver_;
    dynwave::DWSolver dw_solver_;
    divider::DividerSoA dividers_;
 
    void saveOldStates(SimulationContext& ctx);
 
    void initNodeFlows(SimulationContext& ctx, double dt);
 
    void updateLinkStates(SimulationContext& ctx);
};
 
} // namespace openswmm
 
#endif // OPENSWMM_ROUTING_HPP