QualityData.hpp

Go to the documentation of this file.

 
#ifndef OPENSWMM_ENGINE_QUALITY_DATA_HPP
#define OPENSWMM_ENGINE_QUALITY_DATA_HPP
 
#include <vector>
#include <string>
 
namespace openswmm {
 
// ============================================================================
// Land use definitions
// ============================================================================
 
struct LanduseData {
    int count() const { return static_cast<int>(sweep_interval.size()); }
 
    std::vector<double> sweep_interval;  
    std::vector<double> sweep_removal;   
    std::vector<double> last_swept;      
 
    void resize(int n) {
        auto un = static_cast<std::size_t>(n);
        sweep_interval.assign(un, 0.0);
        sweep_removal.assign(un, 0.0);
        last_swept.assign(un, 0.0);
    }
};
 
// ============================================================================
// Buildup function per (landuse x pollutant)
// ============================================================================
 
struct BuildupData {
    std::vector<int>    func_type;   
    std::vector<double> coeff1;
    std::vector<double> coeff2;
    std::vector<double> coeff3;
    std::vector<int>    normalizer;  
 
    int n_landuses = 0;
    int n_pollutants = 0;
 
    void resize(int nlu, int npoll) {
        n_landuses = nlu; n_pollutants = npoll;
        auto total = static_cast<std::size_t>(nlu * npoll);
        func_type.assign(total, 0);
        coeff1.assign(total, 0.0);
        coeff2.assign(total, 0.0);
        coeff3.assign(total, 0.0);
        normalizer.assign(total, 0);
    }
};
 
// ============================================================================
// Washoff function per (landuse x pollutant)
// ============================================================================
 
struct WashoffData {
    std::vector<int>    func_type;   
    std::vector<double> coeff;
    std::vector<double> expon;
    std::vector<double> sweep_effic; 
    std::vector<double> bmp_effic;   
 
    int n_landuses = 0;
    int n_pollutants = 0;
 
    void resize(int nlu, int npoll) {
        n_landuses = nlu; n_pollutants = npoll;
        auto total = static_cast<std::size_t>(nlu * npoll);
        func_type.assign(total, 0);
        coeff.assign(total, 0.0);
        expon.assign(total, 0.0);
        sweep_effic.assign(total, 0.0);
        bmp_effic.assign(total, 0.0);
    }
};
 
// ============================================================================
// Treatment expression per (node x pollutant)
// ============================================================================
 
struct TreatmentData {
    std::vector<std::string> expressions;  
 
    int n_nodes = 0;
    int n_pollutants = 0;
 
    void resize(int nn, int npoll) {
        n_nodes = nn; n_pollutants = npoll;
        expressions.assign(static_cast<std::size_t>(nn * npoll), "");
    }
 
    bool hasAny() const {
        for (const auto& e : expressions) if (!e.empty()) return true;
        return false;
    }
};
 
} // namespace openswmm
 
#endif // OPENSWMM_ENGINE_QUALITY_DATA_HPP