custom_scalar_field2.h

Go to the documentation of this file.

// Copyright (c) 2018 Doyub Kim
//
// I am making my contributions/submissions to this project solely in my
// personal capacity and am not conveying any rights to any intellectual
// property of any third parties.
 
#ifndef INCLUDE_JET_CUSTOM_SCALAR_FIELD2_H_
#define INCLUDE_JET_CUSTOM_SCALAR_FIELD2_H_
 
#include <jet/scalar_field2.h>
 
namespace jet {
 
class CustomScalarField2 final : public ScalarField2 {
 public:
    class Builder;
 
    CustomScalarField2(
        const std::function<double(const Vector2D&)>& customFunction,
        double derivativeResolution = 1e-3);
 
    CustomScalarField2(
        const std::function<double(const Vector2D&)>& customFunction,
        const std::function<Vector2D(const Vector2D&)>& customGradientFunction,
        double derivativeResolution = 1e-3);
 
    CustomScalarField2(
        const std::function<double(const Vector2D&)>& customFunction,
        const std::function<Vector2D(const Vector2D&)>& customGradientFunction,
        const std::function<double(const Vector2D&)>& customLaplacianFunction);
 
    double sample(const Vector2D& x) const override;
 
    std::function<double(const Vector2D&)> sampler() const override;
 
    Vector2D gradient(const Vector2D& x) const override;
 
    double laplacian(const Vector2D& x) const override;
 
    static Builder builder();
 
 private:
    std::function<double(const Vector2D&)> _customFunction;
    std::function<Vector2D(const Vector2D&)> _customGradientFunction;
    std::function<double(const Vector2D&)> _customLaplacianFunction;
    double _resolution = 1e-3;
};
 
typedef std::shared_ptr<CustomScalarField2> CustomScalarField2Ptr;
 
 
class CustomScalarField2::Builder final {
 public:
    Builder& withFunction(
        const std::function<double(const Vector2D&)>& func);
 
    Builder& withGradientFunction(
        const std::function<Vector2D(const Vector2D&)>& func);
 
    Builder& withLaplacianFunction(
        const std::function<double(const Vector2D&)>& func);
 
    Builder& withDerivativeResolution(double resolution);
 
    CustomScalarField2 build() const;
 
    CustomScalarField2Ptr makeShared() const;
 
 private:
    double _resolution = 1e-3;
    std::function<double(const Vector2D&)> _customFunction;
    std::function<Vector2D(const Vector2D&)> _customGradientFunction;
    std::function<double(const Vector2D&)> _customLaplacianFunction;
};
 
}  // namespace jet
 
#endif  // INCLUDE_JET_CUSTOM_SCALAR_FIELD2_H_