HeadTool.h

#ifndef AMO_TOOLS_SUITE_HEADTOOL_H
#define AMO_TOOLS_SUITE_HEADTOOL_H

#include <string>
#include <unordered_map>

class HeadToolBase
{
  protected:
    struct Output
    {
        Output(double elevationHead, double pressureHead, double velocityHeadDifferential, double suctionHead, double dischargeHead, double pumpHead)
            : elevationHead(elevationHead), pressureHead(pressureHead), velocityHeadDifferential(velocityHeadDifferential), suctionHead(suctionHead), dischargeHead(dischargeHead), pumpHead(pumpHead) {}

        Output() = default;
        double elevationHead = 0, pressureHead = 0, velocityHeadDifferential = 0, suctionHead = 0, dischargeHead = 0, pumpHead = 0;
    };
    HeadToolBase(
        const double specificGravity,
        const double flowRate,
        const double suctionPipeDiameter,
        const double suctionLineLossCoefficients,
        const double dischargePipeDiameter,
        const double dischargeGaugePressure,
        const double dischargeGaugeElevation,
        const double dischargeLineLossCoefficients) : specificGravity(specificGravity),
                                                      flowRate(flowRate),
                                                      suctionPipeDiameter(suctionPipeDiameter),
                                                      suctionLineLossCoefficients(suctionLineLossCoefficients),
                                                      dischargePipeDiameter(dischargePipeDiameter),
                                                      dischargeGaugePressure(dischargeGaugePressure),
                                                      dischargeGaugeElevation(dischargeGaugeElevation),
                                                      dischargeLineLossCoefficients(dischargeLineLossCoefficients)
    {
    }

    virtual HeadToolBase::Output calculate() = 0;

    double velocity(double diameter, double flow);

    double velocityHead(double velocity, double gravity);

    const double specificGravity;
    const double flowRate;
    const double suctionPipeDiameter;
    const double suctionLineLossCoefficients;
    const double dischargePipeDiameter;
    const double dischargeGaugePressure;
    const double dischargeGaugeElevation;
    const double dischargeLineLossCoefficients;
    const double gravity = 32.1740;
    const double PI = 3.141592653589793238463;
};

class HeadToolSuctionTank : private HeadToolBase
{
  public:
    HeadToolSuctionTank(
        const double specificGravity,
        const double flowRate,
        const double suctionPipeDiameter,
        const double suctionTankGasOverPressure,
        const double suctionTankFluidSurfaceElevation,
        const double suctionLineLossCoefficients,
        const double dischargePipeDiameter,
        const double dischargeGaugePressure,
        const double dischargeGaugeElevation,
        const double dischargeLineLossCoefficients) : HeadToolBase(specificGravity,
                                                                   flowRate,
                                                                   suctionPipeDiameter,
                                                                   suctionLineLossCoefficients,
                                                                   dischargePipeDiameter,
                                                                   dischargeGaugePressure,
                                                                   dischargeGaugeElevation,
                                                                   dischargeLineLossCoefficients),
                                                      suctionTankGasOverPressure(suctionTankGasOverPressure),
                                                      suctionTankFluidSurfaceElevation(suctionTankFluidSurfaceElevation)
    {
    }

    HeadToolBase::Output calculate() override;

  private:
    const double suctionTankGasOverPressure, suctionTankFluidSurfaceElevation;
    HeadToolBase::Output output;
};

class HeadTool : private HeadToolBase
{
  public:
    HeadTool(
        const double specificGravity,
        const double flowRate,
        const double suctionPipeDiameter,
        const double suctionGaugePressure,
        const double suctionGaugeElevation,
        const double suctionLineLossCoefficients,
        const double dischargePipeDiameter,
        const double dischargeGaugePressure,
        const double dischargeGaugeElevation,
        const double dischargeLineLossCoefficients) : HeadToolBase(specificGravity,
                                                                   flowRate,
                                                                   suctionPipeDiameter,
                                                                   suctionLineLossCoefficients,
                                                                   dischargePipeDiameter,
                                                                   dischargeGaugePressure,
                                                                   dischargeGaugeElevation,
                                                                   dischargeLineLossCoefficients),
                                                      suctionGaugePressure(suctionGaugePressure),
                                                      suctionGaugeElevation(suctionGaugeElevation)
    {
    }
    HeadToolBase::Output calculate() override;
    // std::unordered_map<std::string, double> calculate() override;

  private:
    const double suctionGaugePressure, suctionGaugeElevation;
    HeadToolBase::Output output;
};

#endif //AMO_TOOLS_SUITE_HEADTOOL_H