demandmodel.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 DEMANDMODEL_H_
#define DEMANDMODEL_H_

#include <string>

class Junction;


class DemandModel
{
  public:
    DemandModel();
    DemandModel(double expon_);
    virtual ~DemandModel() = 0;
    static  DemandModel* factory(const std::string model, double expon_);

    virtual double findDemand(Junction* junc, double h, double& dqdh);

    virtual bool isPressureDeficient(Junction* junc) { return false; }

  protected:
    double expon;
};


//-----------------------------------------------------------------------------
//-----------------------------------------------------------------------------

class FixedDemandModel : public DemandModel
{
  public:
    FixedDemandModel();
};


//-----------------------------------------------------------------------------
//-----------------------------------------------------------------------------

class ConstrainedDemandModel : public DemandModel
{
  public:
    ConstrainedDemandModel();
    bool isPressureDeficient(Junction* junc);
    double findDemand(Junction* junc, double p, double& dqdh);
};


//-----------------------------------------------------------------------------
//-----------------------------------------------------------------------------

class PowerDemandModel : public DemandModel
{
  public:
    PowerDemandModel(double expon_);
    double findDemand(Junction* junc, double p, double& dqdh);
};


//-----------------------------------------------------------------------------
//-----------------------------------------------------------------------------

class LogisticDemandModel : public DemandModel
{
  public:
    LogisticDemandModel(double expon_);
    double findDemand(Junction* junc, double p, double& dqdh);

  private:
    double a, b;  // logistic function coefficients
    void  setCoeffs(double pMin, double pFull);
};

#endif