RDII.hpp

Go to the documentation of this file.

 
#ifndef OPENSWMM_RDII_HPP
#define OPENSWMM_RDII_HPP
 
#include <vector>
#include <string>
#include <unordered_map>
 
namespace openswmm {
 
struct SimulationContext;
 
namespace rdii {
 
struct UnitHydParams {
    double r[12][3]     = {};  
    double tPeak[12][3] = {};  
    double tBase[12][3] = {};  
    double iaMax[12][3] = {};  
    double iaRecov[12][3] = {};
    double iaInit[12][3] = {}; 
};
 
struct RDIIGroupSoA {
    int count = 0;
    std::vector<int>    node_idx;       
    std::vector<int>    uh_idx;         
    std::vector<double> area;           
 
    // Circular buffer of past rainfall (per group)
    std::vector<std::vector<double>> past_rain;   
    std::vector<std::vector<int>>    past_month;  
    std::vector<int>    period;         
    std::vector<int>    max_periods;    
 
    void resize(int n);
};
 
class RDIISolver {
public:
    void init(SimulationContext& ctx);
 
    int addUnitHydParams(const std::string& name, const UnitHydParams& params);
 
    int findUnitHyd(const std::string& name) const;
 
    void computeAll(SimulationContext& ctx, double rainfall, int month, double dt);
 
    std::vector<UnitHydParams> uh_params;
 
private:
    RDIIGroupSoA groups_;
    std::unordered_map<std::string, int> uh_name_to_idx_;
 
    double uhOrdinate(const UnitHydParams& uh, int month, int response, double t) const;
 
    static int getRainInterval(const UnitHydParams& uh, double wet_step);
 
    static int getMaxPeriods(const UnitHydParams& uh, int response, int rainInterval);
};
 
} // namespace rdii
} // namespace openswmm
 
#endif // OPENSWMM_RDII_HPP