UnarySolver.hpp

 
 
#pragma once
 
#include "AstGlobal.h"
#include "AstMath/Function.hpp"
#include "AstMath/SolverStats.h"
#include <type_traits>
 
 
AST_NAMESPACE_BEGIN
 
 
constexpr const double kDefaultRelTol = 1e-14;
constexpr const double kDefaultAbsTol = 1e-12;
constexpr const int kDefaultMaxIter = 100;
 
AST_MATH_CAPI bool aIsClose(double a, double b, double relTol = kDefaultRelTol, double absTol = kDefaultAbsTol);
 
 
 
class AST_MATH_API IUnarySolver
{
public:
    virtual ~IUnarySolver() = default;
 
    virtual errc_t solve(UnaryScalarFunc& func, double min, double max, double& result) = 0;
};
 
 
class AST_MATH_API UnarySolver: public IUnarySolver
{
public:
    using IUnarySolver::solve;
 
    UnarySolver();
    UnarySolver(double relTol, double absTol, int maxIter);
    
    UnarySolver(double absTol);
 
    ~UnarySolver() override = default;
 
    double getRelTol() const{return relTol_;};
    double getAbsTol() const{return absTol_;};
    int getMaxIter() const{return maxIter_;};
 
    void setRelTol(double relTol){relTol_ = relTol;};
    void setAbsTol(double absTol){absTol_ = absTol;};
    void setMaxIter(int maxIter){maxIter_ = maxIter;};
 
    const SolverStats& getStats() const{return stats_;};
 
    template<typename Func>
    typename std::enable_if<!std::is_base_of<UnaryScalarFunc, typename std::remove_pointer<Func>::type>::value, errc_t>::type
    solve(Func func, double min, double max, double& result) {
        UnaryScalarGenericFunc<Func> adapter(std::move(func));
        return this->solve(adapter, min, max, result);
    }
protected:
    static double unarycfunc(double x, void* params);
protected:
    double relTol_;         
    double absTol_;         
    int maxIter_;           
    SolverStats stats_;     
};
 
using UnaryScalarSolver = UnarySolver;
 
 
AST_NAMESPACE_END