project.h

Go to the documentation of this file.

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



#ifndef PROJECT_H_
#define PROJECT_H_

#include "Core/network.h"
#include "Core/hydengine.h"
#include "Core/qualengine.h"
#include "Output/outputfile.h"

#include <string>
#include <fstream>

namespace Epanet
{

    class Project
    {
      public:

        Project();
        ~Project();

        int   load(const char* fname);
        int   save(const char* fname);
        void  clear();

        int   initSolver(bool initFlows);
        int   runSolver(int* t);
        int   advanceSolver(int* dt);

        int   openOutput(const char* fname);
        int   saveOutput();

        int   openReport(const char* fname);
        void  writeSummary();
        void  writeResults(int t);
        int   writeReport();

        void  writeMsg(const std::string& msg);
        void  writeMsgLog(std::ostream& out);
        void  writeMsgLog();
        Network* getNetwork() { return &network; }

      private:

        Network        network;        
        HydEngine      hydEngine;      
        QualEngine     qualEngine;     
        OutputFile     outputFile;     
        std::string    inpFileName;    
        std::string    outFileName;    
        std::string    tmpFileName;    
        std::string    rptFileName;    
        std::ofstream  rptFile;        

        // Project status conditions
        bool           networkEmpty;
        bool           hydEngineOpened;
        bool           qualEngineOpened;
        bool           outputFileOpened;
        bool           solverInitialized;
        bool           runQuality;

        void           finalizeSolver();
        void           closeReport();
    };
}
#endif