CustomScalarField2.h

Go to the documentation of this file.

/*************************************************************************
> File Name: CustomScalarField2.h
> Project Name: CubbyFlow
> Author: Chan-Ho Chris Ohk
> Purpose: 2-D scalar field with custom field function.
> Created Time: 2017/03/15
> Copyright (c) 2018, Chan-Ho Chris Ohk
*************************************************************************/
#ifndef CUBBYFLOW_CUSTOM_SCALAR_FIELD2_H
#define CUBBYFLOW_CUSTOM_SCALAR_FIELD2_H

#include <Core/Field/ScalarField2.h>

namespace CubbyFlow
{
    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 GetBuilder();

    private:
        std::function<double(const Vector2D&)> m_customFunction;
        std::function<Vector2D(const Vector2D&)> m_customGradientFunction;
        std::function<double(const Vector2D&)> m_customLaplacianFunction;
        double m_resolution = 1e-3;
    };

    using CustomScalarField2Ptr = std::shared_ptr<CustomScalarField2>;

    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 m_resolution = 1e-3;
        std::function<double(const Vector2D&)> m_customFunction;
        std::function<Vector2D(const Vector2D&)> m_customGradientFunction;
        std::function<double(const Vector2D&)> m_customLaplacianFunction;
    };
}

#endif