cannam@149: // Copyright (c) 2013-2014 Sandstorm Development Group, Inc. and contributors cannam@149: // Licensed under the MIT License: cannam@149: // cannam@149: // Permission is hereby granted, free of charge, to any person obtaining a copy cannam@149: // of this software and associated documentation files (the "Software"), to deal cannam@149: // in the Software without restriction, including without limitation the rights cannam@149: // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell cannam@149: // copies of the Software, and to permit persons to whom the Software is cannam@149: // furnished to do so, subject to the following conditions: cannam@149: // cannam@149: // The above copyright notice and this permission notice shall be included in cannam@149: // all copies or substantial portions of the Software. cannam@149: // cannam@149: // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR cannam@149: // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, cannam@149: // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE cannam@149: // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER cannam@149: // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, cannam@149: // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN cannam@149: // THE SOFTWARE. cannam@149: cannam@149: #ifndef CAPNP_LIST_H_ cannam@149: #define CAPNP_LIST_H_ cannam@149: cannam@149: #if defined(__GNUC__) && !defined(CAPNP_HEADER_WARNINGS) cannam@149: #pragma GCC system_header cannam@149: #endif cannam@149: cannam@149: #include "layout.h" cannam@149: #include "orphan.h" cannam@149: #include cannam@149: #ifdef KJ_STD_COMPAT cannam@149: #include cannam@149: #endif // KJ_STD_COMPAT cannam@149: cannam@149: namespace capnp { cannam@149: namespace _ { // private cannam@149: cannam@149: template cannam@149: class TemporaryPointer { cannam@149: // This class is a little hack which lets us define operator->() in cases where it needs to cannam@149: // return a pointer to a temporary value. We instead construct a TemporaryPointer and return that cannam@149: // (by value). The compiler then invokes operator->() on the TemporaryPointer, which itself is cannam@149: // able to return a real pointer to its member. cannam@149: cannam@149: public: cannam@149: TemporaryPointer(T&& value): value(kj::mv(value)) {} cannam@149: TemporaryPointer(const T& value): value(value) {} cannam@149: cannam@149: inline T* operator->() { return &value; } cannam@149: private: cannam@149: T value; cannam@149: }; cannam@149: cannam@149: template cannam@149: class IndexingIterator { cannam@149: public: cannam@149: IndexingIterator() = default; cannam@149: cannam@149: inline Element operator*() const { return (*container)[index]; } cannam@149: inline TemporaryPointer operator->() const { cannam@149: return TemporaryPointer((*container)[index]); cannam@149: } cannam@149: inline Element operator[]( int off) const { return (*container)[index]; } cannam@149: inline Element operator[](uint off) const { return (*container)[index]; } cannam@149: cannam@149: inline IndexingIterator& operator++() { ++index; return *this; } cannam@149: inline IndexingIterator operator++(int) { IndexingIterator other = *this; ++index; return other; } cannam@149: inline IndexingIterator& operator--() { --index; return *this; } cannam@149: inline IndexingIterator operator--(int) { IndexingIterator other = *this; --index; return other; } cannam@149: cannam@149: inline IndexingIterator operator+(uint amount) const { return IndexingIterator(container, index + amount); } cannam@149: inline IndexingIterator operator-(uint amount) const { return IndexingIterator(container, index - amount); } cannam@149: inline IndexingIterator operator+( int amount) const { return IndexingIterator(container, index + amount); } cannam@149: inline IndexingIterator operator-( int amount) const { return IndexingIterator(container, index - amount); } cannam@149: cannam@149: inline int operator-(const IndexingIterator& other) const { return index - other.index; } cannam@149: cannam@149: inline IndexingIterator& operator+=(uint amount) { index += amount; return *this; } cannam@149: inline IndexingIterator& operator-=(uint amount) { index -= amount; return *this; } cannam@149: inline IndexingIterator& operator+=( int amount) { index += amount; return *this; } cannam@149: inline IndexingIterator& operator-=( int amount) { index -= amount; return *this; } cannam@149: cannam@149: // STL says comparing iterators of different containers is not allowed, so we only compare cannam@149: // indices here. cannam@149: inline bool operator==(const IndexingIterator& other) const { return index == other.index; } cannam@149: inline bool operator!=(const IndexingIterator& other) const { return index != other.index; } cannam@149: inline bool operator<=(const IndexingIterator& other) const { return index <= other.index; } cannam@149: inline bool operator>=(const IndexingIterator& other) const { return index >= other.index; } cannam@149: inline bool operator< (const IndexingIterator& other) const { return index < other.index; } cannam@149: inline bool operator> (const IndexingIterator& other) const { return index > other.index; } cannam@149: cannam@149: private: cannam@149: Container* container; cannam@149: uint index; cannam@149: cannam@149: friend Container; cannam@149: inline IndexingIterator(Container* container, uint index) cannam@149: : container(container), index(index) {} cannam@149: }; cannam@149: cannam@149: } // namespace _ (private) cannam@149: cannam@149: template cannam@149: struct List { cannam@149: // List of primitives. cannam@149: cannam@149: List() = delete; cannam@149: cannam@149: class Reader { cannam@149: public: cannam@149: typedef List Reads; cannam@149: cannam@149: inline Reader(): reader(_::elementSizeForType()) {} cannam@149: inline explicit Reader(_::ListReader reader): reader(reader) {} cannam@149: cannam@149: inline uint size() const { return unbound(reader.size() / ELEMENTS); } cannam@149: inline T operator[](uint index) const { cannam@149: KJ_IREQUIRE(index < size()); cannam@149: return reader.template getDataElement(bounded(index) * ELEMENTS); cannam@149: } cannam@149: cannam@149: typedef _::IndexingIterator Iterator; cannam@149: inline Iterator begin() const { return Iterator(this, 0); } cannam@149: inline Iterator end() const { return Iterator(this, size()); } cannam@149: cannam@149: private: cannam@149: _::ListReader reader; cannam@149: template cannam@149: friend struct _::PointerHelpers; cannam@149: template cannam@149: friend struct List; cannam@149: friend class Orphanage; cannam@149: template cannam@149: friend struct ToDynamic_; cannam@149: }; cannam@149: cannam@149: class Builder { cannam@149: public: cannam@149: typedef List Builds; cannam@149: cannam@149: inline Builder(): builder(_::elementSizeForType()) {} cannam@149: inline Builder(decltype(nullptr)): Builder() {} cannam@149: inline explicit Builder(_::ListBuilder builder): builder(builder) {} cannam@149: cannam@149: inline operator Reader() const { return Reader(builder.asReader()); } cannam@149: inline Reader asReader() const { return Reader(builder.asReader()); } cannam@149: cannam@149: inline uint size() const { return unbound(builder.size() / ELEMENTS); } cannam@149: inline T operator[](uint index) { cannam@149: KJ_IREQUIRE(index < size()); cannam@149: return builder.template getDataElement(bounded(index) * ELEMENTS); cannam@149: } cannam@149: inline void set(uint index, T value) { cannam@149: // Alas, it is not possible to make operator[] return a reference to which you can assign, cannam@149: // since the encoded representation does not necessarily match the compiler's representation cannam@149: // of the type. We can't even return a clever class that implements operator T() and cannam@149: // operator=() because it will lead to surprising behavior when using type inference (e.g. cannam@149: // calling a template function with inferred argument types, or using "auto" or "decltype"). cannam@149: cannam@149: builder.template setDataElement(bounded(index) * ELEMENTS, value); cannam@149: } cannam@149: cannam@149: typedef _::IndexingIterator Iterator; cannam@149: inline Iterator begin() { return Iterator(this, 0); } cannam@149: inline Iterator end() { return Iterator(this, size()); } cannam@149: cannam@149: private: cannam@149: _::ListBuilder builder; cannam@149: template cannam@149: friend struct _::PointerHelpers; cannam@149: friend class Orphanage; cannam@149: template cannam@149: friend struct ToDynamic_; cannam@149: }; cannam@149: cannam@149: class Pipeline {}; cannam@149: cannam@149: private: cannam@149: inline static _::ListBuilder initPointer(_::PointerBuilder builder, uint size) { cannam@149: return builder.initList(_::elementSizeForType(), bounded(size) * ELEMENTS); cannam@149: } cannam@149: inline static _::ListBuilder getFromPointer(_::PointerBuilder builder, const word* defaultValue) { cannam@149: return builder.getList(_::elementSizeForType(), defaultValue); cannam@149: } cannam@149: inline static _::ListReader getFromPointer( cannam@149: const _::PointerReader& reader, const word* defaultValue) { cannam@149: return reader.getList(_::elementSizeForType(), defaultValue); cannam@149: } cannam@149: cannam@149: template cannam@149: friend struct List; cannam@149: template cannam@149: friend struct _::PointerHelpers; cannam@149: }; cannam@149: cannam@149: template cannam@149: struct List: public List {}; cannam@149: cannam@149: template cannam@149: struct List { cannam@149: // List of structs. cannam@149: cannam@149: List() = delete; cannam@149: cannam@149: class Reader { cannam@149: public: cannam@149: typedef List Reads; cannam@149: cannam@149: inline Reader(): reader(ElementSize::INLINE_COMPOSITE) {} cannam@149: inline explicit Reader(_::ListReader reader): reader(reader) {} cannam@149: cannam@149: inline uint size() const { return unbound(reader.size() / ELEMENTS); } cannam@149: inline typename T::Reader operator[](uint index) const { cannam@149: KJ_IREQUIRE(index < size()); cannam@149: return typename T::Reader(reader.getStructElement(bounded(index) * ELEMENTS)); cannam@149: } cannam@149: cannam@149: typedef _::IndexingIterator Iterator; cannam@149: inline Iterator begin() const { return Iterator(this, 0); } cannam@149: inline Iterator end() const { return Iterator(this, size()); } cannam@149: cannam@149: private: cannam@149: _::ListReader reader; cannam@149: template cannam@149: friend struct _::PointerHelpers; cannam@149: template cannam@149: friend struct List; cannam@149: friend class Orphanage; cannam@149: template cannam@149: friend struct ToDynamic_; cannam@149: }; cannam@149: cannam@149: class Builder { cannam@149: public: cannam@149: typedef List Builds; cannam@149: cannam@149: inline Builder(): builder(ElementSize::INLINE_COMPOSITE) {} cannam@149: inline Builder(decltype(nullptr)): Builder() {} cannam@149: inline explicit Builder(_::ListBuilder builder): builder(builder) {} cannam@149: cannam@149: inline operator Reader() const { return Reader(builder.asReader()); } cannam@149: inline Reader asReader() const { return Reader(builder.asReader()); } cannam@149: cannam@149: inline uint size() const { return unbound(builder.size() / ELEMENTS); } cannam@149: inline typename T::Builder operator[](uint index) { cannam@149: KJ_IREQUIRE(index < size()); cannam@149: return typename T::Builder(builder.getStructElement(bounded(index) * ELEMENTS)); cannam@149: } cannam@149: cannam@149: inline void adoptWithCaveats(uint index, Orphan&& orphan) { cannam@149: // Mostly behaves like you'd expect `adopt` to behave, but with two caveats originating from cannam@149: // the fact that structs in a struct list are allocated inline rather than by pointer: cannam@149: // * This actually performs a shallow copy, effectively adopting each of the orphan's cannam@149: // children rather than adopting the orphan itself. The orphan ends up being discarded, cannam@149: // possibly wasting space in the message object. cannam@149: // * If the orphan is larger than the target struct -- say, because the orphan was built cannam@149: // using a newer version of the schema that has additional fields -- it will be truncated, cannam@149: // losing data. cannam@149: cannam@149: KJ_IREQUIRE(index < size()); cannam@149: cannam@149: // We pass a zero-valued StructSize to asStruct() because we do not want the struct to be cannam@149: // expanded under any circumstances. We're just going to throw it away anyway, and cannam@149: // transferContentFrom() already carefully compares the struct sizes before transferring. cannam@149: builder.getStructElement(bounded(index) * ELEMENTS).transferContentFrom( cannam@149: orphan.builder.asStruct(_::StructSize(ZERO * WORDS, ZERO * POINTERS))); cannam@149: } cannam@149: inline void setWithCaveats(uint index, const typename T::Reader& reader) { cannam@149: // Mostly behaves like you'd expect `set` to behave, but with a caveat originating from cannam@149: // the fact that structs in a struct list are allocated inline rather than by pointer: cannam@149: // If the source struct is larger than the target struct -- say, because the source was built cannam@149: // using a newer version of the schema that has additional fields -- it will be truncated, cannam@149: // losing data. cannam@149: // cannam@149: // Note: If you are trying to concatenate some lists, use Orphanage::newOrphanConcat() to cannam@149: // do it without losing any data in case the source lists come from a newer version of the cannam@149: // protocol. (Plus, it's easier to use anyhow.) cannam@149: cannam@149: KJ_IREQUIRE(index < size()); cannam@149: builder.getStructElement(bounded(index) * ELEMENTS).copyContentFrom(reader._reader); cannam@149: } cannam@149: cannam@149: // There are no init(), set(), adopt(), or disown() methods for lists of structs because the cannam@149: // elements of the list are inlined and are initialized when the list is initialized. This cannam@149: // means that init() would be redundant, and set() would risk data loss if the input struct cannam@149: // were from a newer version of the protocol. cannam@149: cannam@149: typedef _::IndexingIterator Iterator; cannam@149: inline Iterator begin() { return Iterator(this, 0); } cannam@149: inline Iterator end() { return Iterator(this, size()); } cannam@149: cannam@149: private: cannam@149: _::ListBuilder builder; cannam@149: template cannam@149: friend struct _::PointerHelpers; cannam@149: friend class Orphanage; cannam@149: template cannam@149: friend struct ToDynamic_; cannam@149: }; cannam@149: cannam@149: class Pipeline {}; cannam@149: cannam@149: private: cannam@149: inline static _::ListBuilder initPointer(_::PointerBuilder builder, uint size) { cannam@149: return builder.initStructList(bounded(size) * ELEMENTS, _::structSize()); cannam@149: } cannam@149: inline static _::ListBuilder getFromPointer(_::PointerBuilder builder, const word* defaultValue) { cannam@149: return builder.getStructList(_::structSize(), defaultValue); cannam@149: } cannam@149: inline static _::ListReader getFromPointer( cannam@149: const _::PointerReader& reader, const word* defaultValue) { cannam@149: return reader.getList(ElementSize::INLINE_COMPOSITE, defaultValue); cannam@149: } cannam@149: cannam@149: template cannam@149: friend struct List; cannam@149: template cannam@149: friend struct _::PointerHelpers; cannam@149: }; cannam@149: cannam@149: template cannam@149: struct List, Kind::LIST> { cannam@149: // List of lists. cannam@149: cannam@149: List() = delete; cannam@149: cannam@149: class Reader { cannam@149: public: cannam@149: typedef List> Reads; cannam@149: cannam@149: inline Reader(): reader(ElementSize::POINTER) {} cannam@149: inline explicit Reader(_::ListReader reader): reader(reader) {} cannam@149: cannam@149: inline uint size() const { return unbound(reader.size() / ELEMENTS); } cannam@149: inline typename List::Reader operator[](uint index) const { cannam@149: KJ_IREQUIRE(index < size()); cannam@149: return typename List::Reader(_::PointerHelpers>::get( cannam@149: reader.getPointerElement(bounded(index) * ELEMENTS))); cannam@149: } cannam@149: cannam@149: typedef _::IndexingIterator::Reader> Iterator; cannam@149: inline Iterator begin() const { return Iterator(this, 0); } cannam@149: inline Iterator end() const { return Iterator(this, size()); } cannam@149: cannam@149: private: cannam@149: _::ListReader reader; cannam@149: template cannam@149: friend struct _::PointerHelpers; cannam@149: template cannam@149: friend struct List; cannam@149: friend class Orphanage; cannam@149: template cannam@149: friend struct ToDynamic_; cannam@149: }; cannam@149: cannam@149: class Builder { cannam@149: public: cannam@149: typedef List> Builds; cannam@149: cannam@149: inline Builder(): builder(ElementSize::POINTER) {} cannam@149: inline Builder(decltype(nullptr)): Builder() {} cannam@149: inline explicit Builder(_::ListBuilder builder): builder(builder) {} cannam@149: cannam@149: inline operator Reader() const { return Reader(builder.asReader()); } cannam@149: inline Reader asReader() const { return Reader(builder.asReader()); } cannam@149: cannam@149: inline uint size() const { return unbound(builder.size() / ELEMENTS); } cannam@149: inline typename List::Builder operator[](uint index) { cannam@149: KJ_IREQUIRE(index < size()); cannam@149: return typename List::Builder(_::PointerHelpers>::get( cannam@149: builder.getPointerElement(bounded(index) * ELEMENTS))); cannam@149: } cannam@149: inline typename List::Builder init(uint index, uint size) { cannam@149: KJ_IREQUIRE(index < this->size()); cannam@149: return typename List::Builder(_::PointerHelpers>::init( cannam@149: builder.getPointerElement(bounded(index) * ELEMENTS), size)); cannam@149: } cannam@149: inline void set(uint index, typename List::Reader value) { cannam@149: KJ_IREQUIRE(index < size()); cannam@149: builder.getPointerElement(bounded(index) * ELEMENTS).setList(value.reader); cannam@149: } cannam@149: void set(uint index, std::initializer_list> value) { cannam@149: KJ_IREQUIRE(index < size()); cannam@149: auto l = init(index, value.size()); cannam@149: uint i = 0; cannam@149: for (auto& element: value) { cannam@149: l.set(i++, element); cannam@149: } cannam@149: } cannam@149: inline void adopt(uint index, Orphan&& value) { cannam@149: KJ_IREQUIRE(index < size()); cannam@149: builder.getPointerElement(bounded(index) * ELEMENTS).adopt(kj::mv(value.builder)); cannam@149: } cannam@149: inline Orphan disown(uint index) { cannam@149: KJ_IREQUIRE(index < size()); cannam@149: return Orphan(builder.getPointerElement(bounded(index) * ELEMENTS).disown()); cannam@149: } cannam@149: cannam@149: typedef _::IndexingIterator::Builder> Iterator; cannam@149: inline Iterator begin() { return Iterator(this, 0); } cannam@149: inline Iterator end() { return Iterator(this, size()); } cannam@149: cannam@149: private: cannam@149: _::ListBuilder builder; cannam@149: template cannam@149: friend struct _::PointerHelpers; cannam@149: friend class Orphanage; cannam@149: template cannam@149: friend struct ToDynamic_; cannam@149: }; cannam@149: cannam@149: class Pipeline {}; cannam@149: cannam@149: private: cannam@149: inline static _::ListBuilder initPointer(_::PointerBuilder builder, uint size) { cannam@149: return builder.initList(ElementSize::POINTER, bounded(size) * ELEMENTS); cannam@149: } cannam@149: inline static _::ListBuilder getFromPointer(_::PointerBuilder builder, const word* defaultValue) { cannam@149: return builder.getList(ElementSize::POINTER, defaultValue); cannam@149: } cannam@149: inline static _::ListReader getFromPointer( cannam@149: const _::PointerReader& reader, const word* defaultValue) { cannam@149: return reader.getList(ElementSize::POINTER, defaultValue); cannam@149: } cannam@149: cannam@149: template cannam@149: friend struct List; cannam@149: template cannam@149: friend struct _::PointerHelpers; cannam@149: }; cannam@149: cannam@149: template cannam@149: struct List { cannam@149: List() = delete; cannam@149: cannam@149: class Reader { cannam@149: public: cannam@149: typedef List Reads; cannam@149: cannam@149: inline Reader(): reader(ElementSize::POINTER) {} cannam@149: inline explicit Reader(_::ListReader reader): reader(reader) {} cannam@149: cannam@149: inline uint size() const { return unbound(reader.size() / ELEMENTS); } cannam@149: inline typename T::Reader operator[](uint index) const { cannam@149: KJ_IREQUIRE(index < size()); cannam@149: return reader.getPointerElement(bounded(index) * ELEMENTS) cannam@149: .template getBlob(nullptr, ZERO * BYTES); cannam@149: } cannam@149: cannam@149: typedef _::IndexingIterator Iterator; cannam@149: inline Iterator begin() const { return Iterator(this, 0); } cannam@149: inline Iterator end() const { return Iterator(this, size()); } cannam@149: cannam@149: private: cannam@149: _::ListReader reader; cannam@149: template cannam@149: friend struct _::PointerHelpers; cannam@149: template cannam@149: friend struct List; cannam@149: friend class Orphanage; cannam@149: template cannam@149: friend struct ToDynamic_; cannam@149: }; cannam@149: cannam@149: class Builder { cannam@149: public: cannam@149: typedef List Builds; cannam@149: cannam@149: inline Builder(): builder(ElementSize::POINTER) {} cannam@149: inline Builder(decltype(nullptr)): Builder() {} cannam@149: inline explicit Builder(_::ListBuilder builder): builder(builder) {} cannam@149: cannam@149: inline operator Reader() const { return Reader(builder.asReader()); } cannam@149: inline Reader asReader() const { return Reader(builder.asReader()); } cannam@149: cannam@149: inline uint size() const { return unbound(builder.size() / ELEMENTS); } cannam@149: inline typename T::Builder operator[](uint index) { cannam@149: KJ_IREQUIRE(index < size()); cannam@149: return builder.getPointerElement(bounded(index) * ELEMENTS) cannam@149: .template getBlob(nullptr, ZERO * BYTES); cannam@149: } cannam@149: inline void set(uint index, typename T::Reader value) { cannam@149: KJ_IREQUIRE(index < size()); cannam@149: builder.getPointerElement(bounded(index) * ELEMENTS).template setBlob(value); cannam@149: } cannam@149: inline typename T::Builder init(uint index, uint size) { cannam@149: KJ_IREQUIRE(index < this->size()); cannam@149: return builder.getPointerElement(bounded(index) * ELEMENTS) cannam@149: .template initBlob(bounded(size) * BYTES); cannam@149: } cannam@149: inline void adopt(uint index, Orphan&& value) { cannam@149: KJ_IREQUIRE(index < size()); cannam@149: builder.getPointerElement(bounded(index) * ELEMENTS).adopt(kj::mv(value.builder)); cannam@149: } cannam@149: inline Orphan disown(uint index) { cannam@149: KJ_IREQUIRE(index < size()); cannam@149: return Orphan(builder.getPointerElement(bounded(index) * ELEMENTS).disown()); cannam@149: } cannam@149: cannam@149: typedef _::IndexingIterator Iterator; cannam@149: inline Iterator begin() { return Iterator(this, 0); } cannam@149: inline Iterator end() { return Iterator(this, size()); } cannam@149: cannam@149: private: cannam@149: _::ListBuilder builder; cannam@149: template cannam@149: friend struct _::PointerHelpers; cannam@149: friend class Orphanage; cannam@149: template cannam@149: friend struct ToDynamic_; cannam@149: }; cannam@149: cannam@149: class Pipeline {}; cannam@149: cannam@149: private: cannam@149: inline static _::ListBuilder initPointer(_::PointerBuilder builder, uint size) { cannam@149: return builder.initList(ElementSize::POINTER, bounded(size) * ELEMENTS); cannam@149: } cannam@149: inline static _::ListBuilder getFromPointer(_::PointerBuilder builder, const word* defaultValue) { cannam@149: return builder.getList(ElementSize::POINTER, defaultValue); cannam@149: } cannam@149: inline static _::ListReader getFromPointer( cannam@149: const _::PointerReader& reader, const word* defaultValue) { cannam@149: return reader.getList(ElementSize::POINTER, defaultValue); cannam@149: } cannam@149: cannam@149: template cannam@149: friend struct List; cannam@149: template cannam@149: friend struct _::PointerHelpers; cannam@149: }; cannam@149: cannam@149: } // namespace capnp cannam@149: cannam@149: #ifdef KJ_STD_COMPAT cannam@149: namespace std { cannam@149: cannam@149: template cannam@149: struct iterator_traits> cannam@149: : public std::iterator {}; cannam@149: cannam@149: } // namespace std cannam@149: #endif // KJ_STD_COMPAT cannam@149: cannam@149: #endif // CAPNP_LIST_H_