NodeCoupling2DData.hpp

Go to the documentation of this file.

 
#ifndef OPENSWMM_ENGINE_NODE_COUPLING_2D_DATA_HPP
#define OPENSWMM_ENGINE_NODE_COUPLING_2D_DATA_HPP
 
#include <array>
#include <cstddef>
#include <cstdint>
#include <limits>
#include <vector>
 
namespace openswmm {

enum class NodeCoupling2DType : int8_t {
    NONE = 0,
    MANHOLE = 1,
    GRATE = 2,
    CURB = 3,
    OPEN = 4,   
    OUTFALL = 5 
};

enum class NodeCoupling2DKey : int {
    D = 0,             
    PERIMETER,         
    AREA,              
    CHAMBER_AREA,      
    LID_OPEN_FRACTION, 
    CREST,             
    COEFFS,            
    CW,                
    CSW,               
    COI,               
    COO,               
    LAW,               
    SPREAD,            
    TAILWATER_RELAX,   
    COUNT
};
 
constexpr int kNodeCoupling2DKeys = static_cast<int>(NodeCoupling2DKey::COUNT);

enum class NodeCoupling2DCrest : int { NODE_RIM = 0, MESH_BED = 1, MAX = 2, MIN = 3 };
enum class NodeCoupling2DCoeffs : int {
    DEFAULT = 0,
    RUBINATO2017 = 1,
    RUBINATO2017_LOWER = 2,
    RUBINATO2017_UPPER = 3
};

enum class NodeCoupling2DLaw : int { ORIFICE = 0, WEIR_ORIFICE = 1 };
 
using NodeCoupling2DParams = std::array<double, kNodeCoupling2DKeys>;

inline NodeCoupling2DParams unsetNodeCoupling2DParams() noexcept {
    NodeCoupling2DParams p;
    p.fill(std::numeric_limits<double>::quiet_NaN());
    return p;
}

struct NodeCoupling2DRows {
    std::vector<int> node; 
    std::vector<NodeCoupling2DType> type;
    std::vector<NodeCoupling2DParams> params; 

    int count() const noexcept { return static_cast<int>(node.size()); }

    int row_of(int node_idx) const noexcept {
        for (std::size_t r = 0; r < node.size(); ++r)
            if (node[r] == node_idx)
                return static_cast<int>(r);
        return -1;
    }

    int ensure_row(int node_idx, NodeCoupling2DType t) {
        const int r = row_of(node_idx);
        if (r >= 0)
            return r;
        node.push_back(node_idx);
        type.push_back(t);
        params.push_back(unsetNodeCoupling2DParams());
        return count() - 1;
    }

    void erase_row(int r) {
        const auto d = static_cast<std::ptrdiff_t>(r);
        node.erase(node.begin() + d);
        type.erase(type.begin() + d);
        params.erase(params.begin() + d);
    }

    void erase_node(int idx) {
        const int r = row_of(idx);
        if (r >= 0)
            erase_row(r);
        for (auto& n : node)
            if (n > idx)
                --n;
    }

    void clear() {
        node.clear();
        type.clear();
        params.clear();
    }
};
 
} // namespace openswmm
 
#endif // OPENSWMM_ENGINE_NODE_COUPLING_2D_DATA_HPP