SubBasin.h

#ifndef HYDROBRICKS_SUBBASIN_H
#define HYDROBRICKS_SUBBASIN_H
 
#include "Connector.h"
#include "HydroUnit.h"
#include "Includes.h"
#include "SettingsBasin.h"
#include "TimeMachine.h"
 
class SubBasin : public wxObject {
  public:
    SubBasin();
 
    ~SubBasin() override;
 
    bool Initialize(SettingsBasin& basinSettings);
 
    void BuildBasin(SettingsBasin& basinSettings);
 
    bool AssignFractions(SettingsBasin& basinSettings);
 
    void Reset();
 
    void SaveAsInitialState();
 
    bool IsOk();
 
    void AddBrick(Brick* brick);
 
    void AddSplitter(Splitter* splitter);
 
    void AddHydroUnit(HydroUnit* unit);
 
    int GetHydroUnitsNb();
 
    HydroUnit* GetHydroUnit(int index);
 
    vector<HydroUnit*> GetHydroUnits() {
        return _hydroUnits;
    }
 
    HydroUnit* GetHydroUnitById(int id);
 
    vecInt GetHydroUnitIds();
 
    vecDouble GetHydroUnitAreas();
 
    int GetBricksCount();
 
    int GetSplittersCount();
 
    Brick* GetBrick(int index);
 
    bool HasBrick(const string& name);
 
    Brick* GetBrick(const string& name);
 
    Splitter* GetSplitter(int index);
 
    bool HasSplitter(const string& name);
 
    Splitter* GetSplitter(const string& name);
 
    bool HasIncomingFlow();
 
    void AddInputConnector(Connector* connector);
 
    void AddOutputConnector(Connector* connector);
 
    void AttachOutletFlux(Flux* pFlux);
 
    double* GetValuePointer(const string& name);
 
    bool ComputeOutletDischarge();
 
    double GetArea() {
        return _area;
    }
 
  protected:
    double _area;  // m2
    double _outletTotal;
    bool _needsCleanup;
    vector<Brick*> _bricks;
    vector<Splitter*> _splitters;
    vector<HydroUnit*> _hydroUnits;
    vector<Connector*> _inConnectors;
    vector<Connector*> _outConnectors;
    vector<Flux*> _outletFluxes;
};
 
#endif