HydroUnit.h

#ifndef HYDROBRICKS_HYDRO_UNIT_H
#define HYDROBRICKS_HYDRO_UNIT_H
 
#include "Brick.h"
#include "Forcing.h"
#include "HydroUnitLateralConnection.h"
#include "HydroUnitProperty.h"
#include "Includes.h"
#include "LandCover.h"
#include "Splitter.h"
 
struct HydroUnitSettings;
 
class HydroUnit : public wxObject {
  public:
    enum Types {
        Distributed,
        SemiDistributed,
        Lumped,
        Undefined
    };
 
    HydroUnit(double area = UNDEFINED, Types type = Undefined);
 
    ~HydroUnit() override;
 
    void Reset();
 
    void SaveAsInitialState();
 
    void SetProperties(HydroUnitSettings& unitSettings);
 
    void AddProperty(HydroUnitProperty* property);
 
    double GetPropertyDouble(const string& name, const string& unit = "");
 
    string GetPropertyString(const string& name);
 
    void AddBrick(Brick* brick);
 
    void AddSplitter(Splitter* splitter);
 
    bool HasForcing(VariableType type);
 
    void AddForcing(Forcing* forcing);
 
    Forcing* GetForcing(VariableType type);
 
    void AddLateralConnection(HydroUnit* receiver, double fraction, const string& type = "");
 
    int GetBricksCount();
 
    int GetSplittersCount();
 
    Brick* GetBrick(int index);
 
    bool HasBrick(const string& name);
 
    Brick* GetBrick(const string& name);
 
    vector<Brick*> GetSnowpacks() const {
        vector<Brick*> snowBricks;
        for (auto& brick : _bricks) {
            if (brick->IsSnowpack()) {
                snowBricks.push_back(brick);
            }
        }
        return snowBricks;
    }
 
    LandCover* GetLandCover(const string& name);
 
    Splitter* GetSplitter(int index);
 
    bool HasSplitter(const string& name);
 
    Splitter* GetSplitter(const string& name);
 
    bool IsOk();
 
    bool ChangeLandCoverAreaFraction(const string& name, double fraction);
 
    bool FixLandCoverFractionsTotal();
 
    Types GetType() {
        return _type;
    }
 
    void SetId(int id) {
        _id = id;
    }
 
    double GetArea() const {
        return _area;
    }
 
    int GetId() const {
        return _id;
    }
 
    vector<HydroUnitLateralConnection*> GetLateralConnections() const {
        return _lateralConnections;
    }
 
  protected:
    Types _type;
    int _id;
    double _area;  // m2
    vector<HydroUnitProperty*> _properties;
    vector<HydroUnitLateralConnection*> _lateralConnections;
    vector<Brick*> _bricks;
    vector<LandCover*> _landCoverBricks;
    vector<Splitter*> _splitters;
    vector<Forcing*> _forcing;
};
 
#endif