Chris@63: // Copyright (c) 2013-2014 Sandstorm Development Group, Inc. and contributors
Chris@63: // Licensed under the MIT License:
Chris@63: //
Chris@63: // Permission is hereby granted, free of charge, to any person obtaining a copy
Chris@63: // of this software and associated documentation files (the "Software"), to deal
Chris@63: // in the Software without restriction, including without limitation the rights
Chris@63: // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
Chris@63: // copies of the Software, and to permit persons to whom the Software is
Chris@63: // furnished to do so, subject to the following conditions:
Chris@63: //
Chris@63: // The above copyright notice and this permission notice shall be included in
Chris@63: // all copies or substantial portions of the Software.
Chris@63: //
Chris@63: // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
Chris@63: // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
Chris@63: // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
Chris@63: // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
Chris@63: // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
Chris@63: // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
Chris@63: // THE SOFTWARE.
Chris@63: 
Chris@63: // This file is included from all generated headers.
Chris@63: 
Chris@63: #ifndef CAPNP_GENERATED_HEADER_SUPPORT_H_
Chris@63: #define CAPNP_GENERATED_HEADER_SUPPORT_H_
Chris@63: 
Chris@63: #if defined(__GNUC__) && !defined(CAPNP_HEADER_WARNINGS)
Chris@63: #pragma GCC system_header
Chris@63: #endif
Chris@63: 
Chris@63: #include "raw-schema.h"
Chris@63: #include "layout.h"
Chris@63: #include "list.h"
Chris@63: #include "orphan.h"
Chris@63: #include "pointer-helpers.h"
Chris@63: #include "any.h"
Chris@63: #include <kj/string.h>
Chris@63: #include <kj/string-tree.h>
Chris@63: 
Chris@63: namespace capnp {
Chris@63: 
Chris@63: class MessageBuilder;  // So that it can be declared a friend.
Chris@63: 
Chris@63: template <typename T, Kind k = CAPNP_KIND(T)>
Chris@63: struct ToDynamic_;   // Defined in dynamic.h, needs to be declared as everyone's friend.
Chris@63: 
Chris@63: struct DynamicStruct;  // So that it can be declared a friend.
Chris@63: 
Chris@63: struct Capability;  // To declare brandBindingFor<Capability>()
Chris@63: 
Chris@63: namespace _ {  // private
Chris@63: 
Chris@63: #if !CAPNP_LITE
Chris@63: 
Chris@63: template <typename T, typename CapnpPrivate = typename T::_capnpPrivate, bool = false>
Chris@63: inline const RawSchema& rawSchema() {
Chris@63:   return *CapnpPrivate::schema;
Chris@63: }
Chris@63: template <typename T, uint64_t id = schemas::EnumInfo<T>::typeId>
Chris@63: inline const RawSchema& rawSchema() {
Chris@63:   return *schemas::EnumInfo<T>::schema;
Chris@63: }
Chris@63: 
Chris@63: template <typename T, typename CapnpPrivate = typename T::_capnpPrivate>
Chris@63: inline const RawBrandedSchema& rawBrandedSchema() {
Chris@63:   return *CapnpPrivate::brand();
Chris@63: }
Chris@63: template <typename T, uint64_t id = schemas::EnumInfo<T>::typeId>
Chris@63: inline const RawBrandedSchema& rawBrandedSchema() {
Chris@63:   return schemas::EnumInfo<T>::schema->defaultBrand;
Chris@63: }
Chris@63: 
Chris@63: template <typename TypeTag, typename... Params>
Chris@63: struct ChooseBrand;
Chris@63: // If all of `Params` are `AnyPointer`, return the type's default brand. Otherwise, return a
Chris@63: // specific brand instance. TypeTag is the _capnpPrivate struct for the type in question.
Chris@63: 
Chris@63: template <typename TypeTag>
Chris@63: struct ChooseBrand<TypeTag> {
Chris@63:   // All params were AnyPointer. No specific brand needed.
Chris@63:   static constexpr _::RawBrandedSchema const* brand() { return &TypeTag::schema->defaultBrand; }
Chris@63: };
Chris@63: 
Chris@63: template <typename TypeTag, typename... Rest>
Chris@63: struct ChooseBrand<TypeTag, AnyPointer, Rest...>: public ChooseBrand<TypeTag, Rest...> {};
Chris@63: // The first parameter is AnyPointer, so recurse to check the rest.
Chris@63: 
Chris@63: template <typename TypeTag, typename First, typename... Rest>
Chris@63: struct ChooseBrand<TypeTag, First, Rest...> {
Chris@63:   // At least one parameter is not AnyPointer, so use the specificBrand constant.
Chris@63:   static constexpr _::RawBrandedSchema const* brand() { return &TypeTag::specificBrand; }
Chris@63: };
Chris@63: 
Chris@63: template <typename T, Kind k = kind<T>()>
Chris@63: struct BrandBindingFor_;
Chris@63: 
Chris@63: #define HANDLE_TYPE(Type, which) \
Chris@63:   template <> \
Chris@63:   struct BrandBindingFor_<Type, Kind::PRIMITIVE> { \
Chris@63:     static constexpr RawBrandedSchema::Binding get(uint16_t listDepth) { \
Chris@63:       return { which, listDepth, nullptr }; \
Chris@63:     } \
Chris@63:   }
Chris@63: HANDLE_TYPE(Void, 0);
Chris@63: HANDLE_TYPE(bool, 1);
Chris@63: HANDLE_TYPE(int8_t, 2);
Chris@63: HANDLE_TYPE(int16_t, 3);
Chris@63: HANDLE_TYPE(int32_t, 4);
Chris@63: HANDLE_TYPE(int64_t, 5);
Chris@63: HANDLE_TYPE(uint8_t, 6);
Chris@63: HANDLE_TYPE(uint16_t, 7);
Chris@63: HANDLE_TYPE(uint32_t, 8);
Chris@63: HANDLE_TYPE(uint64_t, 9);
Chris@63: HANDLE_TYPE(float, 10);
Chris@63: HANDLE_TYPE(double, 11);
Chris@63: #undef HANDLE_TYPE
Chris@63: 
Chris@63: template <>
Chris@63: struct BrandBindingFor_<Text, Kind::BLOB> {
Chris@63:   static constexpr RawBrandedSchema::Binding get(uint16_t listDepth) {
Chris@63:     return { 12, listDepth, nullptr };
Chris@63:   }
Chris@63: };
Chris@63: 
Chris@63: template <>
Chris@63: struct BrandBindingFor_<Data, Kind::BLOB> {
Chris@63:   static constexpr RawBrandedSchema::Binding get(uint16_t listDepth) {
Chris@63:     return { 13, listDepth, nullptr };
Chris@63:   }
Chris@63: };
Chris@63: 
Chris@63: template <typename T>
Chris@63: struct BrandBindingFor_<List<T>, Kind::LIST> {
Chris@63:   static constexpr RawBrandedSchema::Binding get(uint16_t listDepth) {
Chris@63:     return BrandBindingFor_<T>::get(listDepth + 1);
Chris@63:   }
Chris@63: };
Chris@63: 
Chris@63: template <typename T>
Chris@63: struct BrandBindingFor_<T, Kind::ENUM> {
Chris@63:   static constexpr RawBrandedSchema::Binding get(uint16_t listDepth) {
Chris@63:     return { 15, listDepth, nullptr };
Chris@63:   }
Chris@63: };
Chris@63: 
Chris@63: template <typename T>
Chris@63: struct BrandBindingFor_<T, Kind::STRUCT> {
Chris@63:   static constexpr RawBrandedSchema::Binding get(uint16_t listDepth) {
Chris@63:     return { 16, listDepth, T::_capnpPrivate::brand() };
Chris@63:   }
Chris@63: };
Chris@63: 
Chris@63: template <typename T>
Chris@63: struct BrandBindingFor_<T, Kind::INTERFACE> {
Chris@63:   static constexpr RawBrandedSchema::Binding get(uint16_t listDepth) {
Chris@63:     return { 17, listDepth, T::_capnpPrivate::brand() };
Chris@63:   }
Chris@63: };
Chris@63: 
Chris@63: template <>
Chris@63: struct BrandBindingFor_<AnyPointer, Kind::OTHER> {
Chris@63:   static constexpr RawBrandedSchema::Binding get(uint16_t listDepth) {
Chris@63:     return { 18, listDepth, 0, 0 };
Chris@63:   }
Chris@63: };
Chris@63: 
Chris@63: template <>
Chris@63: struct BrandBindingFor_<AnyStruct, Kind::OTHER> {
Chris@63:   static constexpr RawBrandedSchema::Binding get(uint16_t listDepth) {
Chris@63:     return { 18, listDepth, 0, 1 };
Chris@63:   }
Chris@63: };
Chris@63: 
Chris@63: template <>
Chris@63: struct BrandBindingFor_<AnyList, Kind::OTHER> {
Chris@63:   static constexpr RawBrandedSchema::Binding get(uint16_t listDepth) {
Chris@63:     return { 18, listDepth, 0, 2 };
Chris@63:   }
Chris@63: };
Chris@63: 
Chris@63: template <>
Chris@63: struct BrandBindingFor_<Capability, Kind::OTHER> {
Chris@63:   static constexpr RawBrandedSchema::Binding get(uint16_t listDepth) {
Chris@63:     return { 18, listDepth, 0, 3 };
Chris@63:   }
Chris@63: };
Chris@63: 
Chris@63: template <typename T>
Chris@63: constexpr RawBrandedSchema::Binding brandBindingFor() {
Chris@63:   return BrandBindingFor_<T>::get(0);
Chris@63: }
Chris@63: 
Chris@63: kj::StringTree structString(StructReader reader, const RawBrandedSchema& schema);
Chris@63: kj::String enumString(uint16_t value, const RawBrandedSchema& schema);
Chris@63: // Declared here so that we can declare inline stringify methods on generated types.
Chris@63: // Defined in stringify.c++, which depends on dynamic.c++, which is allowed not to be linked in.
Chris@63: 
Chris@63: template <typename T>
Chris@63: inline kj::StringTree structString(StructReader reader) {
Chris@63:   return structString(reader, rawBrandedSchema<T>());
Chris@63: }
Chris@63: template <typename T>
Chris@63: inline kj::String enumString(T value) {
Chris@63:   return enumString(static_cast<uint16_t>(value), rawBrandedSchema<T>());
Chris@63: }
Chris@63: 
Chris@63: #endif  // !CAPNP_LITE
Chris@63: 
Chris@63: // TODO(cleanup):  Unify ConstStruct and ConstList.
Chris@63: template <typename T>
Chris@63: class ConstStruct {
Chris@63: public:
Chris@63:   ConstStruct() = delete;
Chris@63:   KJ_DISALLOW_COPY(ConstStruct);
Chris@63:   inline explicit constexpr ConstStruct(const word* ptr): ptr(ptr) {}
Chris@63: 
Chris@63:   inline typename T::Reader get() const {
Chris@63:     return AnyPointer::Reader(PointerReader::getRootUnchecked(ptr)).getAs<T>();
Chris@63:   }
Chris@63: 
Chris@63:   inline operator typename T::Reader() const { return get(); }
Chris@63:   inline typename T::Reader operator*() const { return get(); }
Chris@63:   inline TemporaryPointer<typename T::Reader> operator->() const { return get(); }
Chris@63: 
Chris@63: private:
Chris@63:   const word* ptr;
Chris@63: };
Chris@63: 
Chris@63: template <typename T>
Chris@63: class ConstList {
Chris@63: public:
Chris@63:   ConstList() = delete;
Chris@63:   KJ_DISALLOW_COPY(ConstList);
Chris@63:   inline explicit constexpr ConstList(const word* ptr): ptr(ptr) {}
Chris@63: 
Chris@63:   inline typename List<T>::Reader get() const {
Chris@63:     return AnyPointer::Reader(PointerReader::getRootUnchecked(ptr)).getAs<List<T>>();
Chris@63:   }
Chris@63: 
Chris@63:   inline operator typename List<T>::Reader() const { return get(); }
Chris@63:   inline typename List<T>::Reader operator*() const { return get(); }
Chris@63:   inline TemporaryPointer<typename List<T>::Reader> operator->() const { return get(); }
Chris@63: 
Chris@63: private:
Chris@63:   const word* ptr;
Chris@63: };
Chris@63: 
Chris@63: template <size_t size>
Chris@63: class ConstText {
Chris@63: public:
Chris@63:   ConstText() = delete;
Chris@63:   KJ_DISALLOW_COPY(ConstText);
Chris@63:   inline explicit constexpr ConstText(const word* ptr): ptr(ptr) {}
Chris@63: 
Chris@63:   inline Text::Reader get() const {
Chris@63:     return Text::Reader(reinterpret_cast<const char*>(ptr), size);
Chris@63:   }
Chris@63: 
Chris@63:   inline operator Text::Reader() const { return get(); }
Chris@63:   inline Text::Reader operator*() const { return get(); }
Chris@63:   inline TemporaryPointer<Text::Reader> operator->() const { return get(); }
Chris@63: 
Chris@63:   inline kj::StringPtr toString() const {
Chris@63:     return get();
Chris@63:   }
Chris@63: 
Chris@63: private:
Chris@63:   const word* ptr;
Chris@63: };
Chris@63: 
Chris@63: template <size_t size>
Chris@63: inline kj::StringPtr KJ_STRINGIFY(const ConstText<size>& s) {
Chris@63:   return s.get();
Chris@63: }
Chris@63: 
Chris@63: template <size_t size>
Chris@63: class ConstData {
Chris@63: public:
Chris@63:   ConstData() = delete;
Chris@63:   KJ_DISALLOW_COPY(ConstData);
Chris@63:   inline explicit constexpr ConstData(const word* ptr): ptr(ptr) {}
Chris@63: 
Chris@63:   inline Data::Reader get() const {
Chris@63:     return Data::Reader(reinterpret_cast<const byte*>(ptr), size);
Chris@63:   }
Chris@63: 
Chris@63:   inline operator Data::Reader() const { return get(); }
Chris@63:   inline Data::Reader operator*() const { return get(); }
Chris@63:   inline TemporaryPointer<Data::Reader> operator->() const { return get(); }
Chris@63: 
Chris@63: private:
Chris@63:   const word* ptr;
Chris@63: };
Chris@63: 
Chris@63: template <size_t size>
Chris@63: inline auto KJ_STRINGIFY(const ConstData<size>& s) -> decltype(kj::toCharSequence(s.get())) {
Chris@63:   return kj::toCharSequence(s.get());
Chris@63: }
Chris@63: 
Chris@63: }  // namespace _ (private)
Chris@63: 
Chris@63: template <typename T, typename CapnpPrivate = typename T::_capnpPrivate>
Chris@63: inline constexpr uint64_t typeId() { return CapnpPrivate::typeId; }
Chris@63: template <typename T, uint64_t id = schemas::EnumInfo<T>::typeId>
Chris@63: inline constexpr uint64_t typeId() { return id; }
Chris@63: // typeId<MyType>() returns the type ID as defined in the schema.  Works with structs, enums, and
Chris@63: // interfaces.
Chris@63: 
Chris@63: template <typename T>
Chris@63: inline constexpr uint sizeInWords() {
Chris@63:   // Return the size, in words, of a Struct type, if allocated free-standing (not in a list).
Chris@63:   // May be useful for pre-computing space needed in order to precisely allocate messages.
Chris@63: 
Chris@63:   return unbound((upgradeBound<uint>(_::structSize<T>().data) +
Chris@63:       _::structSize<T>().pointers * WORDS_PER_POINTER) / WORDS);
Chris@63: }
Chris@63: 
Chris@63: }  // namespace capnp
Chris@63: 
Chris@63: #if _MSC_VER
Chris@63: // MSVC doesn't understand floating-point constexpr yet.
Chris@63: //
Chris@63: // TODO(msvc): Remove this hack when MSVC is fixed.
Chris@63: #define CAPNP_NON_INT_CONSTEXPR_DECL_INIT(value)
Chris@63: #define CAPNP_NON_INT_CONSTEXPR_DEF_INIT(value) = value
Chris@63: #else
Chris@63: #define CAPNP_NON_INT_CONSTEXPR_DECL_INIT(value) = value
Chris@63: #define CAPNP_NON_INT_CONSTEXPR_DEF_INIT(value)
Chris@63: #endif
Chris@63: 
Chris@63: #if _MSC_VER
Chris@63: // TODO(msvc): A little hack to allow MSVC to use C++14 return type deduction in cases where the
Chris@63: // explicit type exposes bugs in the compiler.
Chris@63: #define CAPNP_AUTO_IF_MSVC(...) auto
Chris@63: #else
Chris@63: #define CAPNP_AUTO_IF_MSVC(...) __VA_ARGS__
Chris@63: #endif
Chris@63: 
Chris@63: #if CAPNP_LITE
Chris@63: 
Chris@63: #define CAPNP_DECLARE_SCHEMA(id) \
Chris@63:     extern ::capnp::word const* const bp_##id
Chris@63: 
Chris@63: #define CAPNP_DECLARE_ENUM(type, id) \
Chris@63:     inline ::kj::String KJ_STRINGIFY(type##_##id value) { \
Chris@63:       return ::kj::str(static_cast<uint16_t>(value)); \
Chris@63:     } \
Chris@63:     template <> struct EnumInfo<type##_##id> { \
Chris@63:       struct IsEnum; \
Chris@63:       static constexpr uint64_t typeId = 0x##id; \
Chris@63:       static inline ::capnp::word const* encodedSchema() { return bp_##id; } \
Chris@63:     }
Chris@63: 
Chris@63: #if _MSC_VER
Chris@63: // TODO(msvc): MSVC dosen't expect constexprs to have definitions.
Chris@63: #define CAPNP_DEFINE_ENUM(type, id)
Chris@63: #else
Chris@63: #define CAPNP_DEFINE_ENUM(type, id) \
Chris@63:     constexpr uint64_t EnumInfo<type>::typeId
Chris@63: #endif
Chris@63: 
Chris@63: #define CAPNP_DECLARE_STRUCT_HEADER(id, dataWordSize_, pointerCount_) \
Chris@63:       struct IsStruct; \
Chris@63:       static constexpr uint64_t typeId = 0x##id; \
Chris@63:       static constexpr uint16_t dataWordSize = dataWordSize_; \
Chris@63:       static constexpr uint16_t pointerCount = pointerCount_; \
Chris@63:       static inline ::capnp::word const* encodedSchema() { return ::capnp::schemas::bp_##id; }
Chris@63: 
Chris@63: #else  // CAPNP_LITE
Chris@63: 
Chris@63: #define CAPNP_DECLARE_SCHEMA(id) \
Chris@63:     extern ::capnp::word const* const bp_##id; \
Chris@63:     extern const ::capnp::_::RawSchema s_##id
Chris@63: 
Chris@63: #define CAPNP_DECLARE_ENUM(type, id) \
Chris@63:     inline ::kj::String KJ_STRINGIFY(type##_##id value) { \
Chris@63:       return ::capnp::_::enumString(value); \
Chris@63:     } \
Chris@63:     template <> struct EnumInfo<type##_##id> { \
Chris@63:       struct IsEnum; \
Chris@63:       static constexpr uint64_t typeId = 0x##id; \
Chris@63:       static inline ::capnp::word const* encodedSchema() { return bp_##id; } \
Chris@63:       static constexpr ::capnp::_::RawSchema const* schema = &s_##id; \
Chris@63:     }
Chris@63: #define CAPNP_DEFINE_ENUM(type, id) \
Chris@63:     constexpr uint64_t EnumInfo<type>::typeId; \
Chris@63:     constexpr ::capnp::_::RawSchema const* EnumInfo<type>::schema
Chris@63: 
Chris@63: #define CAPNP_DECLARE_STRUCT_HEADER(id, dataWordSize_, pointerCount_) \
Chris@63:       struct IsStruct; \
Chris@63:       static constexpr uint64_t typeId = 0x##id; \
Chris@63:       static constexpr ::capnp::Kind kind = ::capnp::Kind::STRUCT; \
Chris@63:       static constexpr uint16_t dataWordSize = dataWordSize_; \
Chris@63:       static constexpr uint16_t pointerCount = pointerCount_; \
Chris@63:       static inline ::capnp::word const* encodedSchema() { return ::capnp::schemas::bp_##id; } \
Chris@63:       static constexpr ::capnp::_::RawSchema const* schema = &::capnp::schemas::s_##id;
Chris@63: 
Chris@63: #define CAPNP_DECLARE_INTERFACE_HEADER(id) \
Chris@63:       struct IsInterface; \
Chris@63:       static constexpr uint64_t typeId = 0x##id; \
Chris@63:       static constexpr ::capnp::Kind kind = ::capnp::Kind::INTERFACE; \
Chris@63:       static inline ::capnp::word const* encodedSchema() { return ::capnp::schemas::bp_##id; } \
Chris@63:       static constexpr ::capnp::_::RawSchema const* schema = &::capnp::schemas::s_##id;
Chris@63: 
Chris@63: #endif  // CAPNP_LITE, else
Chris@63: 
Chris@63: #endif  // CAPNP_GENERATED_HEADER_SUPPORT_H_