InterfaceFile.hpp

Go to the documentation of this file.

 
#ifndef OPENSWMM_INTERFACE_FILE_HPP
#define OPENSWMM_INTERFACE_FILE_HPP
 
#include <vector>
#include <string>
#include <cstdio>
 
namespace openswmm {
 
struct SimulationContext;
 
namespace iface {
 
// ============================================================================
// SoA for interface node data (flows + quality per node)
// ============================================================================
 
struct InterfaceFileSoA {
    int count = 0;
    std::vector<int>    node_idx;       
    std::vector<double> old_values;     
    std::vector<double> new_values;     
 
    void resize(int n);
};
 
// ============================================================================
// InterfaceManager — manages reading/writing of routing interface files
// ============================================================================
 
class InterfaceManager {
public:
    void init(SimulationContext& ctx);

    int openFiles(const std::string& infile_path, const std::string& outfile_path);

    int readFileHeader(SimulationContext& ctx);

    void writeFileHeader(const SimulationContext& ctx);

    void readInflows(SimulationContext& ctx, double current_time);

    void writeOutfallResults(const SimulationContext& ctx, double current_time);

    void closeFiles();

    double getFlow(int node_idx, double frac) const;

    double getQuality(int node_idx, int pollut_idx, double frac) const;

    int getNumIfaceNodes() const { return data_.count; }

    int getIfaceNode(int index) const;

    double getFrac() const { return iface_frac_; }
 
private:
    InterfaceFileSoA data_;
    FILE* infile_  = nullptr;
    FILE* outfile_ = nullptr;
 
    double old_time_ = 0.0;         
    double new_time_ = 0.0;         
    int n_iface_polluts_ = 0;       
    int n_values_per_node_ = 1;     
    int iface_flow_units_ = 0;      
    int iface_step_ = 0;            
    double iface_frac_ = 0.0;       
 
    std::vector<int> pollut_map_;   

    bool readNextPeriod();

    void setOldValues();

    static bool isOutletNode(const SimulationContext& ctx, int node_idx);
};
 
} // namespace iface
} // namespace openswmm
 
#endif // OPENSWMM_INTERFACE_FILE_HPP