outputfile.h

/* EPANET 3
 *
 * Copyright (c) 2016 Open Water Analytics
 * Licensed under the terms of the MIT License (see the LICENSE file for details).
 *
 */


#ifndef OUTPUTFILE_H_
#define OUTPUTFILE_H_

#include <fstream>
#include <string>

class Network;
class ReportWriter;

const    int   IntSize = sizeof(int);
const    int   FloatSize = sizeof(float);
const    int   NumSysVars = 21;
const    int   NumNodeVars = 6;
const    int   NumLinkVars = 7;
const    int   NumPumpVars = 6;


class OutputFile
{
  public:
    OutputFile();
    ~OutputFile();

    int    open(const std::string fileName, Network* nw);
    void   close();

    int    initWriter();
    int    writeEnergyResults(double totalHrs, double peakKwatts);
    int    writeNetworkResults();

    int    initReader();
    void   seekEnergyOffset();
    void   readEnergyResults(int* pumpIndex);
    void   readEnergyDemandCharge(float* demandCharge);
    void   seekNetworkOffset();
    void   readNodeResults();
    void   readLinkResults();
    void   skipNodeResults();
    void   skipLinkResults();

    friend ReportWriter;

  private:
    std::string   fname;                    
    std::ofstream fwriter;                  
    std::ifstream freader;                  
    Network*      network;                  
    int           nodeCount;                
    int           linkCount;                
    int           pumpCount;                
    int           timePeriodCount;          
    int           reportStart;              
    int           reportStep;               
    int           energyResultsOffset;      
    int           networkResultsOffset;     
    float         nodeResults[NumNodeVars]; 
    float         linkResults[NumLinkVars]; 
    float         pumpResults[NumPumpVars]; 
    void          writeNodeResults();
    void          writeLinkResults();
};

#endif