qualengine.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 QUALENGINE_H_
#define QUALENGINE_H_

#include <vector>

class Network;
class QualSolver;
//class JuncMixer;
//class TankMixer;


class QualEngine
{
  public:

    // Constructor/Destructor

    QualEngine();
    ~QualEngine();

    // Public Methods

    void   open(Network* nw);
    void   init();
    void   solve(int tstep);
    void   close();

private:

    // Engine state

    enum EngineState {CLOSED, OPENED, INITIALIZED};
    EngineState engineState;

    // Engine components

    Network*    network;            
    QualSolver* qualSolver;         

    // Engine properties

    int         nodeCount;          
    int         linkCount;          
    int         qualTime;           
    int         qualStep;           
    std::vector<int>  sortedLinks;      
    std::vector<char> flowDirection;    

    // Simulation sub-tasks

    bool        flowDirectionsChanged();
    void        setFlowDirections();
    void        sortLinks();
    void        setSourceQuality();
};

#endif