protobuf_cpp_type_helpers.h

// Copyright 2014-2023:
//   GobySoft, LLC (2013-)
//   Massachusetts Institute of Technology (2007-2014)
//   Community contributors (see AUTHORS file)
// File authors:
//   Toby Schneider <toby@gobysoft.org>
//
//
// This file is part of the Dynamic Compact Control Language Library
// ("DCCL").
//
// DCCL is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 2.1 of the License, or
// (at your option) any later version.
//
// DCCL is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with DCCL.  If not, see <http://www.gnu.org/licenses/>.
#ifndef DCCLPROTOBUFCPPTYPEHELPERS20110323H
#define DCCLPROTOBUFCPPTYPEHELPERS20110323H
 
#include "../any.h"
 
#include <google/protobuf/descriptor.h>
#include <google/protobuf/descriptor.pb.h>
#include <google/protobuf/message.h>
 
namespace dccl
{
namespace internal
{
class FromProtoTypeBase
{
  public:
    virtual ~FromProtoTypeBase() = default;
    virtual std::string as_str() { return "TYPE_UNKNOWN"; }
};
 
template <google::protobuf::FieldDescriptor::Type t> class FromProtoType : public FromProtoTypeBase
{
  public:
    std::string as_str() override
    {
        return google::protobuf::FieldDescriptorProto::Type_Name(
            static_cast<google::protobuf::FieldDescriptorProto::Type>(t));
    }
};
 
class FromProtoCppTypeBase
{
  public:
    virtual ~FromProtoCppTypeBase() = default;
 
    virtual std::string as_str() { return "CPPTYPE_UNKNOWN"; }
 
    dccl::any get_value(const google::protobuf::FieldDescriptor* field,
                        const google::protobuf::Message& msg)
    {
        // For fields without presence tracking (proto3 singular scalar fields without the
        // `optional` keyword), HasField() is not valid. Such fields always have a value
        // (defaulting to 0/false/empty if not explicitly set), so we always return it.
        if (!field->has_presence())
            return _get_value(field, msg);
        const google::protobuf::Reflection* refl = msg.GetReflection();
        if (!refl->HasField(msg, field))
            return dccl::any();
        else
            return _get_value(field, msg);
    }
 
    dccl::any get_value(const google::protobuf::Message& msg) { return _get_value(nullptr, msg); }
 
    dccl::any get_repeated_value(const google::protobuf::FieldDescriptor* field,
                                 const google::protobuf::Message& msg, int index)
    {
        return _get_repeated_value(field, msg, index);
    }
 
    void set_value(const google::protobuf::FieldDescriptor* field, google::protobuf::Message* msg,
                   dccl::any value)
    {
        if (is_empty(value))
            return;
        else
            _set_value(field, msg, value);
    }
 
    void set_value(google::protobuf::Message* msg, dccl::any value)
    {
        if (is_empty(value))
            return;
        else
            _set_value(nullptr, msg, value);
    }
 
    void add_value(const google::protobuf::FieldDescriptor* field, google::protobuf::Message* msg,
                   dccl::any value)
    {
        if (is_empty(value))
            return;
        else
            _add_value(field, msg, value);
    }
 
    virtual void _set_value(const google::protobuf::FieldDescriptor* /*field*/,
                            google::protobuf::Message* /*msg*/, dccl::any /*value*/)
    {
        return;
    }
 
    virtual void _add_value(const google::protobuf::FieldDescriptor* /*field*/,
                            google::protobuf::Message* /*msg*/, dccl::any /*value*/)
    {
        return;
    }
 
    virtual dccl::any _get_repeated_value(const google::protobuf::FieldDescriptor* /*field*/,
                                          const google::protobuf::Message& /*msg*/, int /*index*/)
    {
        return dccl::any();
    }
 
    virtual dccl::any _get_value(const google::protobuf::FieldDescriptor* /*field*/,
                                 const google::protobuf::Message& /*msg*/)
    {
        return dccl::any();
    }
};
 
template <google::protobuf::FieldDescriptor::CppType> class FromProtoCppType
{
};
template <>
class FromProtoCppType<google::protobuf::FieldDescriptor::CPPTYPE_DOUBLE>
    : public FromProtoCppTypeBase
{
  public:
    using type = double;
    std::string as_str() override { return "CPPTYPE_DOUBLE"; }
 
  private:
    dccl::any _get_value(const google::protobuf::FieldDescriptor* field,
                         const google::protobuf::Message& msg) override
    {
        return msg.GetReflection()->GetDouble(msg, field);
    }
    dccl::any _get_repeated_value(const google::protobuf::FieldDescriptor* field,
                                  const google::protobuf::Message& msg, int index) override
    {
        return msg.GetReflection()->GetRepeatedDouble(msg, field, index);
    }
    void _set_value(const google::protobuf::FieldDescriptor* field, google::protobuf::Message* msg,
                    dccl::any value) override
    {
        msg->GetReflection()->SetDouble(msg, field, dccl::any_cast<type>(value));
    }
    void _add_value(const google::protobuf::FieldDescriptor* field, google::protobuf::Message* msg,
                    dccl::any value) override
    {
        msg->GetReflection()->AddDouble(msg, field, dccl::any_cast<type>(value));
    }
};
 
template <>
class FromProtoCppType<google::protobuf::FieldDescriptor::CPPTYPE_FLOAT>
    : public FromProtoCppTypeBase
{
  public:
    using type = float;
    std::string as_str() override { return "CPPTYPE_FLOAT"; }
 
  private:
    dccl::any _get_value(const google::protobuf::FieldDescriptor* field,
                         const google::protobuf::Message& msg) override
    {
        return msg.GetReflection()->GetFloat(msg, field);
    }
    dccl::any _get_repeated_value(const google::protobuf::FieldDescriptor* field,
                                  const google::protobuf::Message& msg, int index) override
    {
        return msg.GetReflection()->GetRepeatedFloat(msg, field, index);
    }
    void _set_value(const google::protobuf::FieldDescriptor* field, google::protobuf::Message* msg,
                    dccl::any value) override
    {
        msg->GetReflection()->SetFloat(msg, field, dccl::any_cast<type>(value));
    }
    void _add_value(const google::protobuf::FieldDescriptor* field, google::protobuf::Message* msg,
                    dccl::any value) override
    {
        msg->GetReflection()->AddFloat(msg, field, dccl::any_cast<type>(value));
    }
};
template <>
class FromProtoCppType<google::protobuf::FieldDescriptor::CPPTYPE_INT32>
    : public FromProtoCppTypeBase
{
  public:
    using type = google::protobuf::int32;
    std::string as_str() override { return "CPPTYPE_INT32"; }
 
  private:
    dccl::any _get_value(const google::protobuf::FieldDescriptor* field,
                         const google::protobuf::Message& msg) override
    {
        return msg.GetReflection()->GetInt32(msg, field);
    }
    dccl::any _get_repeated_value(const google::protobuf::FieldDescriptor* field,
                                  const google::protobuf::Message& msg, int index) override
    {
        return msg.GetReflection()->GetRepeatedInt32(msg, field, index);
    }
    void _set_value(const google::protobuf::FieldDescriptor* field, google::protobuf::Message* msg,
                    dccl::any value) override
    {
        msg->GetReflection()->SetInt32(msg, field, dccl::any_cast<type>(value));
    }
    void _add_value(const google::protobuf::FieldDescriptor* field, google::protobuf::Message* msg,
                    dccl::any value) override
    {
        msg->GetReflection()->AddInt32(msg, field, dccl::any_cast<type>(value));
    }
};
template <>
class FromProtoCppType<google::protobuf::FieldDescriptor::CPPTYPE_INT64>
    : public FromProtoCppTypeBase
{
  public:
    using type = google::protobuf::int64;
    std::string as_str() override { return "CPPTYPE_INT64"; }
 
  private:
    dccl::any _get_value(const google::protobuf::FieldDescriptor* field,
                         const google::protobuf::Message& msg) override
    {
        return msg.GetReflection()->GetInt64(msg, field);
    }
    dccl::any _get_repeated_value(const google::protobuf::FieldDescriptor* field,
                                  const google::protobuf::Message& msg, int index) override
    {
        return msg.GetReflection()->GetRepeatedInt64(msg, field, index);
    }
    void _set_value(const google::protobuf::FieldDescriptor* field, google::protobuf::Message* msg,
                    dccl::any value) override
    {
        msg->GetReflection()->SetInt64(msg, field, dccl::any_cast<type>(value));
    }
    void _add_value(const google::protobuf::FieldDescriptor* field, google::protobuf::Message* msg,
                    dccl::any value) override
    {
        msg->GetReflection()->AddInt64(msg, field, dccl::any_cast<type>(value));
    }
};
template <>
class FromProtoCppType<google::protobuf::FieldDescriptor::CPPTYPE_UINT32>
    : public FromProtoCppTypeBase
{
  public:
    using type = google::protobuf::uint32;
    std::string as_str() override { return "CPPTYPE_UINT32"; }
 
  private:
    dccl::any _get_value(const google::protobuf::FieldDescriptor* field,
                         const google::protobuf::Message& msg) override
    {
        return msg.GetReflection()->GetUInt32(msg, field);
    }
    dccl::any _get_repeated_value(const google::protobuf::FieldDescriptor* field,
                                  const google::protobuf::Message& msg, int index) override
    {
        return msg.GetReflection()->GetRepeatedUInt32(msg, field, index);
    }
    void _set_value(const google::protobuf::FieldDescriptor* field, google::protobuf::Message* msg,
                    dccl::any value) override
    {
        msg->GetReflection()->SetUInt32(msg, field, dccl::any_cast<type>(value));
    }
    void _add_value(const google::protobuf::FieldDescriptor* field, google::protobuf::Message* msg,
                    dccl::any value) override
    {
        msg->GetReflection()->AddUInt32(msg, field, dccl::any_cast<type>(value));
    }
};
template <>
class FromProtoCppType<google::protobuf::FieldDescriptor::CPPTYPE_UINT64>
    : public FromProtoCppTypeBase
{
  public:
    using type = google::protobuf::uint64;
 
    std::string as_str() override { return "CPPTYPE_UINT64"; }
 
  private:
    dccl::any _get_value(const google::protobuf::FieldDescriptor* field,
                         const google::protobuf::Message& msg) override
    {
        return msg.GetReflection()->GetUInt64(msg, field);
    }
    dccl::any _get_repeated_value(const google::protobuf::FieldDescriptor* field,
                                  const google::protobuf::Message& msg, int index) override
    {
        return msg.GetReflection()->GetRepeatedUInt64(msg, field, index);
    }
    void _set_value(const google::protobuf::FieldDescriptor* field, google::protobuf::Message* msg,
                    dccl::any value) override
    {
        msg->GetReflection()->SetUInt64(msg, field, dccl::any_cast<type>(value));
    }
    void _add_value(const google::protobuf::FieldDescriptor* field, google::protobuf::Message* msg,
                    dccl::any value) override
    {
        msg->GetReflection()->AddUInt64(msg, field, dccl::any_cast<type>(value));
    }
};
template <>
class FromProtoCppType<google::protobuf::FieldDescriptor::CPPTYPE_BOOL>
    : public FromProtoCppTypeBase
{
  public:
    using type = bool;
    std::string as_str() override { return "CPPTYPE_BOOL"; }
 
  private:
    dccl::any _get_value(const google::protobuf::FieldDescriptor* field,
                         const google::protobuf::Message& msg) override
    {
        return msg.GetReflection()->GetBool(msg, field);
    }
    dccl::any _get_repeated_value(const google::protobuf::FieldDescriptor* field,
                                  const google::protobuf::Message& msg, int index) override
    {
        return msg.GetReflection()->GetRepeatedBool(msg, field, index);
    }
    void _set_value(const google::protobuf::FieldDescriptor* field, google::protobuf::Message* msg,
                    dccl::any value) override
    {
        msg->GetReflection()->SetBool(msg, field, dccl::any_cast<type>(value));
    }
    void _add_value(const google::protobuf::FieldDescriptor* field, google::protobuf::Message* msg,
                    dccl::any value) override
    {
        msg->GetReflection()->AddBool(msg, field, dccl::any_cast<type>(value));
    }
};
template <>
class FromProtoCppType<google::protobuf::FieldDescriptor::CPPTYPE_STRING>
    : public FromProtoCppTypeBase
{
  public:
    using type = std::string;
    std::string as_str() override { return "CPPTYPE_STRING"; }
 
  private:
    dccl::any _get_value(const google::protobuf::FieldDescriptor* field,
                         const google::protobuf::Message& msg) override
    {
        return msg.GetReflection()->GetString(msg, field);
    }
    dccl::any _get_repeated_value(const google::protobuf::FieldDescriptor* field,
                                  const google::protobuf::Message& msg, int index) override
    {
        return msg.GetReflection()->GetRepeatedString(msg, field, index);
    }
    void _set_value(const google::protobuf::FieldDescriptor* field, google::protobuf::Message* msg,
                    dccl::any value) override
    {
        msg->GetReflection()->SetString(msg, field, dccl::any_cast<type>(value));
    }
    void _add_value(const google::protobuf::FieldDescriptor* field, google::protobuf::Message* msg,
                    dccl::any value) override
    {
        msg->GetReflection()->AddString(msg, field, dccl::any_cast<type>(value));
    }
};
 
template <>
class FromProtoCppType<google::protobuf::FieldDescriptor::CPPTYPE_ENUM>
    : public FromProtoCppTypeBase
{
  public:
    using const_type = const google::protobuf::EnumValueDescriptor*;
    using mutable_type = google::protobuf::EnumValueDescriptor*;
 
    std::string as_str() override { return "CPPTYPE_ENUM"; }
 
  private:
    dccl::any _get_value(const google::protobuf::FieldDescriptor* field,
                         const google::protobuf::Message& msg) override
    {
        return msg.GetReflection()->GetEnum(msg, field);
    }
    dccl::any _get_repeated_value(const google::protobuf::FieldDescriptor* field,
                                  const google::protobuf::Message& msg, int index) override
    {
        return msg.GetReflection()->GetRepeatedEnum(msg, field, index);
    }
    void _set_value(const google::protobuf::FieldDescriptor* field, google::protobuf::Message* msg,
                    dccl::any value) override
    {
        msg->GetReflection()->SetEnum(msg, field, dccl::any_cast<const_type>(value));
    }
    void _add_value(const google::protobuf::FieldDescriptor* field, google::protobuf::Message* msg,
                    dccl::any value) override
    {
        msg->GetReflection()->AddEnum(msg, field, dccl::any_cast<const_type>(value));
    }
};
 
template <>
class FromProtoCppType<google::protobuf::FieldDescriptor::CPPTYPE_MESSAGE>
    : public FromProtoCppTypeBase
{
  public:
    using const_type = const google::protobuf::Message*;
    using mutable_type = google::protobuf::Message*;
    std::string as_str() override { return "CPPTYPE_MESSAGE"; }
 
  protected:
    dccl::any _get_value(const google::protobuf::FieldDescriptor* field,
                         const google::protobuf::Message& msg) override
    {
        if (field)
            return &(msg.GetReflection()->GetMessage(msg, field));
        else
            return &msg;
    }
 
    dccl::any _get_repeated_value(const google::protobuf::FieldDescriptor* field,
                                  const google::protobuf::Message& msg, int index) override
    {
        return &(msg.GetReflection()->GetRepeatedMessage(msg, field, index));
    }
 
    void _set_value(const google::protobuf::FieldDescriptor* field, google::protobuf::Message* msg,
                    dccl::any value) override
    {
        try
        {
            auto p = dccl::any_cast<const_type>(value);
 
            if (field)
                msg->GetReflection()->MutableMessage(msg, field)->MergeFrom(*p);
            else
                msg->MergeFrom(*p);
        }
        catch (dccl::bad_any_cast& e)
        {
            auto p = dccl::any_cast<mutable_type>(value);
            if (field)
                msg->GetReflection()->MutableMessage(msg, field)->MergeFrom(*p);
            else
                msg->MergeFrom(*p);
        }
    }
    void _add_value(const google::protobuf::FieldDescriptor* field, google::protobuf::Message* msg,
                    dccl::any value) override
    {
        try
        {
            auto p = dccl::any_cast<const_type>(value);
            msg->GetReflection()->AddMessage(msg, field)->MergeFrom(*p);
        }
        catch (dccl::bad_any_cast& e)
        {
            auto p = dccl::any_cast<mutable_type>(value);
            msg->GetReflection()->AddMessage(msg, field)->MergeFrom(*p);
        }
    }
};
 
template <typename CustomMessage>
class FromProtoCustomMessage
    : public FromProtoCppType<google::protobuf::FieldDescriptor::CPPTYPE_MESSAGE>
{
  public:
    using type = CustomMessage;
 
  private:
    using Parent = FromProtoCppType<google::protobuf::FieldDescriptor::CPPTYPE_MESSAGE>;
 
    dccl::any _get_value(const google::protobuf::FieldDescriptor* field,
                         const google::protobuf::Message& msg) override
    {
        auto p = dccl::any_cast<Parent::const_type>(Parent::_get_value(field, msg));
        type r;
        r.CopyFrom(*p);
        return r;
        //                return dynamic_cast<const_type>(*p);
    }
    dccl::any _get_repeated_value(const google::protobuf::FieldDescriptor* field,
                                  const google::protobuf::Message& msg, int index) override
    {
        auto p = dccl::any_cast<Parent::const_type>(Parent::_get_repeated_value(field, msg, index));
        type r;
        r.CopyFrom(*p);
        return r;
        //                return dynamic_cast<const_type>(*p);
    }
    void _set_value(const google::protobuf::FieldDescriptor* field, google::protobuf::Message* msg,
                    dccl::any value) override
    {
        type v = dccl::any_cast<type>(value);
        Parent::const_type p = &v;
        Parent::_set_value(field, msg, p);
    }
    void _add_value(const google::protobuf::FieldDescriptor* field, google::protobuf::Message* msg,
                    dccl::any value) override
    {
        type v = dccl::any_cast<type>(value);
        Parent::const_type p = &v;
        Parent::_add_value(field, msg, p);
    }
};
 
template <typename T> class ToProtoCppType
{
};
template <> class ToProtoCppType<double>
{
  public:
    static constexpr google::protobuf::FieldDescriptor::CppType as_enum()
    {
        return google::protobuf::FieldDescriptor::CPPTYPE_DOUBLE;
    }
};
template <> class ToProtoCppType<float>
{
  public:
    static constexpr google::protobuf::FieldDescriptor::CppType as_enum()
    {
        return google::protobuf::FieldDescriptor::CPPTYPE_FLOAT;
    }
};
template <> class ToProtoCppType<google::protobuf::int32>
{
  public:
    static constexpr google::protobuf::FieldDescriptor::CppType as_enum()
    {
        return google::protobuf::FieldDescriptor::CPPTYPE_INT32;
    }
};
template <> class ToProtoCppType<google::protobuf::uint32>
{
  public:
    static constexpr google::protobuf::FieldDescriptor::CppType as_enum()
    {
        return google::protobuf::FieldDescriptor::CPPTYPE_UINT32;
    }
};
template <> class ToProtoCppType<google::protobuf::int64>
{
  public:
    static constexpr google::protobuf::FieldDescriptor::CppType as_enum()
    {
        return google::protobuf::FieldDescriptor::CPPTYPE_INT64;
    }
};
template <> class ToProtoCppType<google::protobuf::uint64>
{
  public:
    static constexpr google::protobuf::FieldDescriptor::CppType as_enum()
    {
        return google::protobuf::FieldDescriptor::CPPTYPE_UINT64;
    }
};
template <> class ToProtoCppType<std::string>
{
  public:
    static constexpr google::protobuf::FieldDescriptor::CppType as_enum()
    {
        return google::protobuf::FieldDescriptor::CPPTYPE_STRING;
    }
};
template <> class ToProtoCppType<const google::protobuf::EnumValueDescriptor*>
{
  public:
    static constexpr google::protobuf::FieldDescriptor::CppType as_enum()
    {
        return google::protobuf::FieldDescriptor::CPPTYPE_ENUM;
    }
};
 
template <> class ToProtoCppType<bool>
{
  public:
    static constexpr google::protobuf::FieldDescriptor::CppType as_enum()
    {
        return google::protobuf::FieldDescriptor::CPPTYPE_BOOL;
    }
};
 
template <> class ToProtoCppType<google::protobuf::Message>
{
  public:
    static constexpr google::protobuf::FieldDescriptor::CppType as_enum()
    {
        return google::protobuf::FieldDescriptor::CPPTYPE_MESSAGE;
    }
};
} // namespace internal
} // namespace dccl
 
#endif