UnaryScalarFunc.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
#include "UnaryVectorFunc.hpp"
 
AST_NAMESPACE_BEGIN
 
 
class UnaryScalarFunc: public UnaryVectorFunc
{
public:
    virtual ~UnaryScalarFunc() = default;
 
    virtual void evaluate(double x, double* out) const override = 0;
 
    int getDimension() const final{ return 1; }
};
 
 
template<typename Func>
class UnaryScalarGenericFunc : public UnaryScalarFunc {
public:
    explicit UnaryScalarGenericFunc(Func func) 
        : func_(std::move(func)) 
    {}
    
    void evaluate(double x, double* out) const override {
        *out = func_(x);
    }
    
private:
    Func func_;
};
 
AST_NAMESPACE_END