InitialQualitySeeds.hpp

Go to the documentation of this file.

// SPDX-License-Identifier: Apache-2.0
//
// Copyright 2026 Caleb Buahin
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

 
#ifndef OPENSWMM_ENGINE_INITIAL_QUALITY_SEEDS_HPP
#define OPENSWMM_ENGINE_INITIAL_QUALITY_SEEDS_HPP
 
#include "../core/SimulationContext.hpp"
#include "../data/InitialQualityData.hpp"
 
namespace openswmm::transport {

inline double initialAgeSecondsFor(const SimulationContext& ctx, bool is_link, int idx,
                                   double fallback) {
    double v = fallback;
    const auto& iq = ctx.initial_quality;
    for (int r = 0; r < iq.count(); ++r) {
        const auto ur = static_cast<std::size_t>(r);
        if (iq.kind[ur] != InitialQualityData::kKindWaterAge)
            continue;
        if ((iq.is_link[ur] != 0) != is_link)
            continue;
        if (iq.elem_idx[ur] != idx)
            continue;
        v = iq.value[ur] * 3600.0; // hours -> seconds
    }
    return v;
}

inline double initialTempFor(const SimulationContext& ctx, bool is_link, int idx, double fallback) {
    double v = fallback;
    const auto& iq = ctx.initial_quality;
    for (int r = 0; r < iq.count(); ++r) {
        const auto ur = static_cast<std::size_t>(r);
        if (iq.kind[ur] != InitialQualityData::kKindTemperature)
            continue;
        if ((iq.is_link[ur] != 0) != is_link)
            continue;
        if (iq.elem_idx[ur] != idx)
            continue;
        v = iq.value[ur];
    }
    return v;
}

inline void applyInitialAgeOverrides(SimulationContext& ctx) {
    auto& ws = ctx.water_age_state;
    if (ws.hotstart_loaded)
        return;
    const auto& iq = ctx.initial_quality;
    for (int r = 0; r < iq.count(); ++r) {
        const auto ur = static_cast<std::size_t>(r);
        if (iq.kind[ur] != InitialQualityData::kKindWaterAge)
            continue;
        const int ei = iq.elem_idx[ur];
        if (ei < 0)
            continue;
        auto& arr = iq.is_link[ur] ? ws.link_age : ws.node_age;
        if (static_cast<std::size_t>(ei) < arr.size())
            arr[static_cast<std::size_t>(ei)] = iq.value[ur] * 3600.0;
    }
}

inline void applyInitialTempOverrides(SimulationContext& ctx) {
    auto& hs = ctx.heat_state;
    const auto& iq = ctx.initial_quality;
    for (int r = 0; r < iq.count(); ++r) {
        const auto ur = static_cast<std::size_t>(r);
        if (iq.kind[ur] != InitialQualityData::kKindTemperature)
            continue;
        const int ei = iq.elem_idx[ur];
        if (ei < 0)
            continue;
        auto& arr = iq.is_link[ur] ? hs.link_temp : hs.node_temp;
        if (static_cast<std::size_t>(ei) < arr.size())
            arr[static_cast<std::size_t>(ei)] = iq.value[ur];
    }
}
 
} // namespace openswmm::transport
 
#endif // OPENSWMM_ENGINE_INITIAL_QUALITY_SEEDS_HPP