GageData.hpp

Go to the documentation of this file.

 
#ifndef OPENSWMM_ENGINE_GAGE_DATA_HPP
#define OPENSWMM_ENGINE_GAGE_DATA_HPP
 
#include <vector>
#include <string>
#include <cstdint>
 
namespace openswmm {
 
// ============================================================================
// Rain source enumeration
// ============================================================================
 
enum class RainSource : int8_t {
    TIMESERIES = 0,  
    FILE_RAIN  = 1   
};
 
enum class RainFileFormat : int8_t {
    UNKNOWN  = -1,
    NWS_15   = 0,   
    NWS_HOURLY = 1, 
    DSI_3240 = 2,   
    DSI_3260 = 3,   
    HLY_PRCP = 4,   
    STAN_PRCP = 5,  
    USER_CSV = 6    
};
 
// ============================================================================
// GageData — SoA layout
// ============================================================================
 
struct GageData {
 
    // -----------------------------------------------------------------------
    // Static properties — set at parse time
    // -----------------------------------------------------------------------
 
    std::vector<int>            rain_type;
 
    std::vector<RainSource>     source;
 
    std::vector<int>            ts_index;
 
    std::vector<std::string>    file_path;
 
    std::vector<std::string>    ts_name;
 
    std::vector<std::string>    col_name;
 
    std::vector<RainFileFormat> file_format;
 
    std::vector<int>            interval_sec;
 
    std::vector<double>         snow_factor;
 
    // -----------------------------------------------------------------------
    // State variables — updated each timestep
    // -----------------------------------------------------------------------
 
    std::vector<double>         rainfall;
 
    std::vector<double>         next_rainfall;
 
    std::vector<double>         api_rainfall;
 
    std::vector<double>         next_rain_date;
 
    std::vector<bool>           is_raining;
 
    // -----------------------------------------------------------------------
    // Past-rain history (for control rules — GAGE_RAIN_PAST)
    // -----------------------------------------------------------------------
 
    static constexpr int MAXPASTRAIN = 48; 
 
    std::vector<double>         past_rain;
 
    std::vector<double>         past_rain_accum;
 
    std::vector<double>         past_rain_time;
 
    // -----------------------------------------------------------------------
    // Capacity management
    // -----------------------------------------------------------------------
 
    int count() const noexcept { return static_cast<int>(source.size()); }
 
    void resize(int n) {
        const auto un = static_cast<std::size_t>(n);
 
        rain_type.assign(un, 0);
        source.assign(un, RainSource::TIMESERIES);
        ts_index.assign(un, -1);
        ts_name.assign(un, std::string{});
        file_path.assign(un, std::string{});
        col_name.assign(un, std::string{});
        file_format.assign(un, RainFileFormat::UNKNOWN);
        interval_sec.assign(un, 3600);
        snow_factor.assign(un, 1.0);
 
        rainfall.assign(un, 0.0);
        next_rainfall.assign(un, 0.0);
        api_rainfall.assign(un, -1.0);  // -1.0 means no API override
        next_rain_date.assign(un, 0.0);
        is_raining.assign(un, false);
 
        past_rain.assign(un * MAXPASTRAIN, 0.0);
        past_rain_accum.assign(un, 0.0);
        past_rain_time.assign(un, 0.0);
    }
 
    void reset_state() noexcept {
        std::fill(rainfall.begin(),      rainfall.end(),      0.0);
        std::fill(next_rainfall.begin(), next_rainfall.end(), 0.0);
        std::fill(api_rainfall.begin(),  api_rainfall.end(),  -1.0); // -1 = no override
        std::fill(is_raining.begin(),    is_raining.end(),    false);
        std::fill(past_rain.begin(),     past_rain.end(),     0.0);
        std::fill(past_rain_accum.begin(), past_rain_accum.end(), 0.0);
        std::fill(past_rain_time.begin(),  past_rain_time.end(),  0.0);
    }
};
 
} /* namespace openswmm */
 
#endif /* OPENSWMM_ENGINE_GAGE_DATA_HPP */