math_function.h

Go to the documentation of this file.

/*
 * Copyright 2026 MusicScience37 (Kenta Kabashima)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
#pragma once
 
#include <string_view>
#include <utility>
#include <vector>
 
#include "func_sketch/common_types.h"
#include "func_sketch/math/math_function_type.h"
#include "func_sketch/utilities/any.h"
 
namespace func_sketch::math {

class MathFunction {
public:
    template <MathFunctionType T>
    explicit MathFunction(T function)
        : function_(std::move(function)),
          get_name_([](const utilities::Any& obj) -> std::string_view {
              return obj.get<T>().name();
          }),
          scalar_operator_(
              [](const utilities::Any& obj, const std::vector<Number>& args,
                  Number& result) { obj.get<T>()(args, result); }) {}

    [[nodiscard]] std::string_view name() const { return get_name_(function_); }

    void operator()(const std::vector<Number>& args, Number& result) const {
        scalar_operator_(function_, args, result);
    }
 
private:
    using GetNameSignature = std::string_view(
        const utilities::Any& /*function_object*/);

    using ScalarOperatorSignature = void(
        const utilities::Any& /*function_object*/,
        const std::vector<Number>& /*args*/, Number& /*result*/);

    utilities::Any function_;

    GetNameSignature* get_name_;

    ScalarOperatorSignature* scalar_operator_;
};
 
}  // namespace func_sketch::math