// Copyright (C) 2018-2024 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//

#pragma once

#include "openvino/core/extension.hpp"
#include "openvino/frontend/node_context.hpp"
#include "openvino/frontend/visibility.hpp"

namespace ov {
namespace frontend {

class FRONTEND_API ConversionExtensionBase : public ov::Extension {
public:
    using Ptr = std::shared_ptr<ConversionExtensionBase>;
    explicit ConversionExtensionBase(const std::string& op_type) : m_op_type(op_type) {}

    const std::string& get_op_type() const {
        return m_op_type;
    }

    ~ConversionExtensionBase() override = default;

private:
    std::string m_op_type;
};

class FRONTEND_API ConversionExtension : public ConversionExtensionBase {
public:
    using Ptr = std::shared_ptr<ConversionExtension>;
    ConversionExtension(const std::string& op_type, const CreatorFunction& converter)
        : ConversionExtensionBase(op_type),
          m_converter(converter) {}

    ConversionExtension(const std::string& op_type, const CreatorFunctionNamed& converter)
        : ConversionExtensionBase(op_type),
          m_converter_named(converter) {}

    ConversionExtension(const std::string& op_type, const CreatorFunctionNamedAndIndexed& converter)
        : ConversionExtensionBase(op_type),
          m_converter_named_and_indexed(converter) {}

    const CreatorFunction& get_converter() const {
        return m_converter;
    };

    const CreatorFunctionNamed& get_converter_named() const {
        return m_converter_named;
    };

    const CreatorFunctionNamedAndIndexed& get_converter_named_and_indexed() const {
        return m_converter_named_and_indexed;
    };

    ~ConversionExtension() override;

private:
    CreatorFunction m_converter;
    CreatorFunctionNamed m_converter_named;
    CreatorFunctionNamedAndIndexed m_converter_named_and_indexed;
};

}  // namespace frontend
}  // namespace ov
