peak.hpp

#ifndef ABLATELIBRARY_PEAK_FUNTION_HPP
#define ABLATELIBRARY_PEAK_FUNTION_HPP
#include <muParser.h>
#include "formulaBase.hpp"
 
namespace ablate::mathFunctions {
class Peak : public MathFunction {
   private:
    const std::vector<double> startValue;
    const std::vector<double> peakValue;
    const std::vector<double> endValue;
    const double start;
    const double peak;
    const double end;
    const int dir;
 
    static PetscErrorCode PeakPetscFunction(PetscInt dim, PetscReal time, const PetscReal x[], PetscInt Nf, PetscScalar* u, void* ctx);
 
    inline static double Interpolate(double x, double xS, double xP, double xE, double yS, double yP, double yE) {
        if (x < xS) {
            return yS;
        } else if (x > xE) {
            return yE;
        } else if (x < xP) {
            return yS + (x - xS) * (yP - yS) / (xP - xS);
        } else {
            return yP + (x - xP) * (yE - yP) / (xE - xP);
        }
    }
 
    inline double DetermineDirectionValue(double x, double y, double z) const {
        switch (dir) {
            case 0:
                return x;
            case 1:
                return y;
            case 2:
                return z;
            default:
                return x;
        }
    }
 
   public:
    Peak(const Peak&) = delete;
    void operator=(const Peak&) = delete;
 
    explicit Peak(std::vector<double> startValue, std::vector<double> peakValue, std::vector<double> endValue, double start, double peak, double end, int dir);
 
    double Eval(const double& x, const double& y, const double& z, const double& t) const override;
 
    double Eval(const double* xyz, const int& ndims, const double& t) const override;
 
    void Eval(const double& x, const double& y, const double& z, const double& t, std::vector<double>& result) const override;
 
    void Eval(const double* xyz, const int& ndims, const double& t, std::vector<double>& result) const override;
 
    void* GetContext() override { return this; }
 
    PetscFunction GetPetscFunction() override { return PeakPetscFunction; }
};
}  // namespace ablate::mathFunctions
 
#endif  // ABLATELIBRARY_SIMPLEFORMULA_HPP