pumpcurve.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 PUMPCURVE_H_
#define PUMPCURVE_H_

class Network;
class Curve;


class PumpCurve
{
  public:

    enum PumpCurveType {
        NO_CURVE,              
        CONST_HP,              
        POWER_FUNC,            
        CUSTOM                 
    };

    // Constructor/Destructor
    PumpCurve();
    ~PumpCurve();

    // Methods
    void   setCurve(Curve* c);
    int    setupCurve(Network* network);
    void   findHeadLoss(
               double speed, double flow, double& headLoss, double& gradient);
    bool   isConstHP() {return curveType == CONST_HP;}

    // Properties
    int    curveType;      
    Curve* curve;          
    double horsepower;     
    double qInit;          
    double qMax;           
    double hMax;           

  private:

    double h0;             
    double r;              
    double n;              
    double qUcf;           
    double hUcf;           

    void   setupConstHpCurve();
    int    setupPowerFuncCurve();
    int    setupCustomCurve();
    void   constHpHeadLoss(
               double speed, double flow, double& headLoss, double& gradient);
    void   powerFuncHeadLoss(
               double speed, double flow, double& headLoss, double& gradient);
    void   customCurveHeadLoss(
               double speed, double flow, double& headLoss, double& gradient);
};

#endif