headlossmodel.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 HEADLOSSMODEL_H_
#define HEADLOSSMODEL_H_

#include <string>

class Pipe;


class HeadLossModel
{
  public:

    HeadLossModel(double viscos);
    virtual ~HeadLossModel() = 0;

    static HeadLossModel* factory(const std::string model, double viscos);

    static void findClosedHeadLoss(double flow, double& headLoss, double& gradient);
    static void addCVHeadLoss(double flow, double& headLoss, double& gradient);

    void    setViscosity(double v) { viscosity = v;}
    virtual void setResistance(Pipe* pipe) = 0;

    virtual void findHeadLoss(
                     Pipe* pipe, double flow, double& headLoss, double& gradient) = 0;

  protected:
    double  viscosity;         
};


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

class HW_HeadLossModel : public HeadLossModel
{
  public:
    HW_HeadLossModel(double viscos);
    void   setResistance(Pipe* pipe);
    void   findHeadLoss(Pipe* pipe, double flow, double& headLoss, double& gradient);
};


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

class DW_HeadLossModel : public HeadLossModel
{
  public:
    DW_HeadLossModel(double viscos);
    void   setResistance(Pipe* pipe);
    void   findHeadLoss(Pipe* pipe, double flow, double& headLoss, double& gradient);
};


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

class CM_HeadLossModel : public HeadLossModel
{
  public:
    CM_HeadLossModel(double viscos);
    void   setResistance(Pipe* pipe);
    void   findHeadLoss(Pipe* pipe, double flow, double& headLoss, double& gradient);
};

#endif