| // Protocol Buffers - Google's data interchange format |
| // Copyright 2008 Google Inc. All rights reserved. |
| // https://developers.google.com/protocol-buffers/ |
| // |
| // Redistribution and use in source and binary forms, with or without |
| // modification, are permitted provided that the following conditions are |
| // met: |
| // |
| // * Redistributions of source code must retain the above copyright |
| // notice, this list of conditions and the following disclaimer. |
| // * Redistributions in binary form must reproduce the above |
| // copyright notice, this list of conditions and the following disclaimer |
| // in the documentation and/or other materials provided with the |
| // distribution. |
| // * Neither the name of Google Inc. nor the names of its |
| // contributors may be used to endorse or promote products derived from |
| // this software without specific prior written permission. |
| // |
| // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| |
| #ifndef GOOGLE_PROTOBUF_REFLECTION_INTERNAL_H__ |
| #define GOOGLE_PROTOBUF_REFLECTION_INTERNAL_H__ |
| |
| #include <google/protobuf/map_field.h> |
| #include <google/protobuf/reflection.h> |
| #include <google/protobuf/repeated_field.h> |
| |
| namespace google { |
| namespace protobuf { |
| namespace internal { |
| // A base class for RepeatedFieldAccessor implementations that can support |
| // random-access efficiently. All iterator methods delegates the work to |
| // corresponding random-access methods. |
| class RandomAccessRepeatedFieldAccessor : public RepeatedFieldAccessor { |
| public: |
| Iterator* BeginIterator(const Field* data) const override { |
| return PositionToIterator(0); |
| } |
| Iterator* EndIterator(const Field* data) const override { |
| return PositionToIterator(this->Size(data)); |
| } |
| Iterator* CopyIterator(const Field* data, |
| const Iterator* iterator) const override { |
| return const_cast<Iterator*>(iterator); |
| } |
| Iterator* AdvanceIterator(const Field* data, |
| Iterator* iterator) const override { |
| return PositionToIterator(IteratorToPosition(iterator) + 1); |
| } |
| bool EqualsIterator(const Field* data, const Iterator* a, |
| const Iterator* b) const override { |
| return a == b; |
| } |
| void DeleteIterator(const Field* data, Iterator* iterator) const override {} |
| const Value* GetIteratorValue(const Field* data, const Iterator* iterator, |
| Value* scratch_space) const override { |
| return Get(data, static_cast<int>(IteratorToPosition(iterator)), |
| scratch_space); |
| } |
| |
| protected: |
| ~RandomAccessRepeatedFieldAccessor() = default; |
| |
| private: |
| static intptr_t IteratorToPosition(const Iterator* iterator) { |
| return reinterpret_cast<intptr_t>(iterator); |
| } |
| static Iterator* PositionToIterator(intptr_t position) { |
| return reinterpret_cast<Iterator*>(position); |
| } |
| }; |
| |
| // Base class for RepeatedFieldAccessor implementations that manipulates |
| // RepeatedField<T>. |
| template <typename T> |
| class RepeatedFieldWrapper : public RandomAccessRepeatedFieldAccessor { |
| public: |
| RepeatedFieldWrapper() {} |
| bool IsEmpty(const Field* data) const override { |
| return GetRepeatedField(data)->empty(); |
| } |
| int Size(const Field* data) const override { |
| return GetRepeatedField(data)->size(); |
| } |
| const Value* Get(const Field* data, int index, |
| Value* scratch_space) const override { |
| return ConvertFromT(GetRepeatedField(data)->Get(index), scratch_space); |
| } |
| void Clear(Field* data) const override { |
| MutableRepeatedField(data)->Clear(); |
| } |
| void Set(Field* data, int index, const Value* value) const override { |
| MutableRepeatedField(data)->Set(index, ConvertToT(value)); |
| } |
| void Add(Field* data, const Value* value) const override { |
| MutableRepeatedField(data)->Add(ConvertToT(value)); |
| } |
| void RemoveLast(Field* data) const override { |
| MutableRepeatedField(data)->RemoveLast(); |
| } |
| void SwapElements(Field* data, int index1, int index2) const override { |
| MutableRepeatedField(data)->SwapElements(index1, index2); |
| } |
| |
| protected: |
| ~RepeatedFieldWrapper() = default; |
| typedef RepeatedField<T> RepeatedFieldType; |
| static const RepeatedFieldType* GetRepeatedField(const Field* data) { |
| return reinterpret_cast<const RepeatedFieldType*>(data); |
| } |
| static RepeatedFieldType* MutableRepeatedField(Field* data) { |
| return reinterpret_cast<RepeatedFieldType*>(data); |
| } |
| |
| // Convert an object received by this accessor to an object to be stored in |
| // the underlying RepeatedField. |
| virtual T ConvertToT(const Value* value) const = 0; |
| |
| // Convert an object stored in RepeatedPtrField to an object that will be |
| // returned by this accessor. If the two objects have the same type (true for |
| // string fields with ctype=STRING), a pointer to the source object can be |
| // returned directly. Otherwise, data should be copied from value to |
| // scratch_space and scratch_space should be returned. |
| virtual const Value* ConvertFromT(const T& value, |
| Value* scratch_space) const = 0; |
| }; |
| |
| // Base class for RepeatedFieldAccessor implementations that manipulates |
| // RepeatedPtrField<T>. |
| template <typename T> |
| class RepeatedPtrFieldWrapper : public RandomAccessRepeatedFieldAccessor { |
| public: |
| bool IsEmpty(const Field* data) const override { |
| return GetRepeatedField(data)->empty(); |
| } |
| int Size(const Field* data) const override { |
| return GetRepeatedField(data)->size(); |
| } |
| const Value* Get(const Field* data, int index, |
| Value* scratch_space) const override { |
| return ConvertFromT(GetRepeatedField(data)->Get(index), scratch_space); |
| } |
| void Clear(Field* data) const override { |
| MutableRepeatedField(data)->Clear(); |
| } |
| void Set(Field* data, int index, const Value* value) const override { |
| ConvertToT(value, MutableRepeatedField(data)->Mutable(index)); |
| } |
| void Add(Field* data, const Value* value) const override { |
| T* allocated = New(value); |
| ConvertToT(value, allocated); |
| MutableRepeatedField(data)->AddAllocated(allocated); |
| } |
| void RemoveLast(Field* data) const override { |
| MutableRepeatedField(data)->RemoveLast(); |
| } |
| void SwapElements(Field* data, int index1, int index2) const override { |
| MutableRepeatedField(data)->SwapElements(index1, index2); |
| } |
| |
| protected: |
| ~RepeatedPtrFieldWrapper() = default; |
| typedef RepeatedPtrField<T> RepeatedFieldType; |
| static const RepeatedFieldType* GetRepeatedField(const Field* data) { |
| return reinterpret_cast<const RepeatedFieldType*>(data); |
| } |
| static RepeatedFieldType* MutableRepeatedField(Field* data) { |
| return reinterpret_cast<RepeatedFieldType*>(data); |
| } |
| |
| // Create a new T instance. For repeated message fields, T can be specified |
| // as google::protobuf::Message so we can't use "new T()" directly. In that case, value |
| // should be a message of the same type (it's ensured by the caller) and a |
| // new message object will be created using it. |
| virtual T* New(const Value* value) const = 0; |
| |
| // Convert an object received by this accessor to an object that will be |
| // stored in the underlying RepeatedPtrField. |
| virtual void ConvertToT(const Value* value, T* result) const = 0; |
| |
| // Convert an object stored in RepeatedPtrField to an object that will be |
| // returned by this accessor. If the two objects have the same type (true for |
| // string fields with ctype=STRING), a pointer to the source object can be |
| // returned directly. Otherwise, data should be copied from value to |
| // scratch_space and scratch_space should be returned. |
| virtual const Value* ConvertFromT(const T& value, |
| Value* scratch_space) const = 0; |
| }; |
| |
| // An implementation of RandomAccessRepeatedFieldAccessor that manipulates |
| // MapFieldBase. |
| class MapFieldAccessor final : public RandomAccessRepeatedFieldAccessor { |
| public: |
| MapFieldAccessor() {} |
| virtual ~MapFieldAccessor() {} |
| bool IsEmpty(const Field* data) const override { |
| return GetRepeatedField(data)->empty(); |
| } |
| int Size(const Field* data) const override { |
| return GetRepeatedField(data)->size(); |
| } |
| const Value* Get(const Field* data, int index, |
| Value* scratch_space) const override { |
| return ConvertFromEntry(GetRepeatedField(data)->Get(index), scratch_space); |
| } |
| void Clear(Field* data) const override { |
| MutableRepeatedField(data)->Clear(); |
| } |
| void Set(Field* data, int index, const Value* value) const override { |
| ConvertToEntry(value, MutableRepeatedField(data)->Mutable(index)); |
| } |
| void Add(Field* data, const Value* value) const override { |
| Message* allocated = New(value); |
| ConvertToEntry(value, allocated); |
| MutableRepeatedField(data)->AddAllocated(allocated); |
| } |
| void RemoveLast(Field* data) const override { |
| MutableRepeatedField(data)->RemoveLast(); |
| } |
| void SwapElements(Field* data, int index1, int index2) const override { |
| MutableRepeatedField(data)->SwapElements(index1, index2); |
| } |
| void Swap(Field* data, const internal::RepeatedFieldAccessor* other_mutator, |
| Field* other_data) const override { |
| GOOGLE_CHECK(this == other_mutator); |
| MutableRepeatedField(data)->Swap(MutableRepeatedField(other_data)); |
| } |
| |
| protected: |
| typedef RepeatedPtrField<Message> RepeatedFieldType; |
| static const RepeatedFieldType* GetRepeatedField(const Field* data) { |
| return reinterpret_cast<const RepeatedFieldType*>( |
| (&reinterpret_cast<const MapFieldBase*>(data)->GetRepeatedField())); |
| } |
| static RepeatedFieldType* MutableRepeatedField(Field* data) { |
| return reinterpret_cast<RepeatedFieldType*>( |
| reinterpret_cast<MapFieldBase*>(data)->MutableRepeatedField()); |
| } |
| virtual Message* New(const Value* value) const { |
| return static_cast<const Message*>(value)->New(); |
| } |
| // Convert an object received by this accessor to an MapEntry message to be |
| // stored in the underlying MapFieldBase. |
| virtual void ConvertToEntry(const Value* value, Message* result) const { |
| result->CopyFrom(*static_cast<const Message*>(value)); |
| } |
| // Convert a MapEntry message stored in the underlying MapFieldBase to an |
| // object that will be returned by this accessor. |
| virtual const Value* ConvertFromEntry(const Message& value, |
| Value* scratch_space) const { |
| return static_cast<const Value*>(&value); |
| } |
| }; |
| |
| // Default implementations of RepeatedFieldAccessor for primitive types. |
| template <typename T> |
| class RepeatedFieldPrimitiveAccessor final : public RepeatedFieldWrapper<T> { |
| typedef void Field; |
| typedef void Value; |
| using RepeatedFieldWrapper<T>::MutableRepeatedField; |
| |
| public: |
| RepeatedFieldPrimitiveAccessor() {} |
| void Swap(Field* data, const internal::RepeatedFieldAccessor* other_mutator, |
| Field* other_data) const override { |
| // Currently RepeatedFieldPrimitiveAccessor is the only implementation of |
| // RepeatedFieldAccessor for primitive types. As we are using singletons |
| // for these accessors, here "other_mutator" must be "this". |
| GOOGLE_CHECK(this == other_mutator); |
| MutableRepeatedField(data)->Swap(MutableRepeatedField(other_data)); |
| } |
| |
| protected: |
| T ConvertToT(const Value* value) const override { |
| return *static_cast<const T*>(value); |
| } |
| const Value* ConvertFromT(const T& value, |
| Value* scratch_space) const override { |
| return static_cast<const Value*>(&value); |
| } |
| }; |
| |
| // Default implementation of RepeatedFieldAccessor for string fields with |
| // ctype=STRING. |
| class RepeatedPtrFieldStringAccessor final |
| : public RepeatedPtrFieldWrapper<std::string> { |
| typedef void Field; |
| typedef void Value; |
| using RepeatedFieldAccessor::Add; |
| |
| public: |
| RepeatedPtrFieldStringAccessor() {} |
| void Swap(Field* data, const internal::RepeatedFieldAccessor* other_mutator, |
| Field* other_data) const override { |
| if (this == other_mutator) { |
| MutableRepeatedField(data)->Swap(MutableRepeatedField(other_data)); |
| } else { |
| RepeatedPtrField<std::string> tmp; |
| tmp.Swap(MutableRepeatedField(data)); |
| int other_size = other_mutator->Size(other_data); |
| for (int i = 0; i < other_size; ++i) { |
| Add<std::string>(data, other_mutator->Get<std::string>(other_data, i)); |
| } |
| int size = Size(data); |
| other_mutator->Clear(other_data); |
| for (int i = 0; i < size; ++i) { |
| other_mutator->Add<std::string>(other_data, tmp.Get(i)); |
| } |
| } |
| } |
| |
| protected: |
| std::string* New(const Value*) const override { return new std::string(); } |
| void ConvertToT(const Value* value, std::string* result) const override { |
| *result = *static_cast<const std::string*>(value); |
| } |
| const Value* ConvertFromT(const std::string& value, |
| Value* scratch_space) const override { |
| return static_cast<const Value*>(&value); |
| } |
| }; |
| |
| |
| class RepeatedPtrFieldMessageAccessor final |
| : public RepeatedPtrFieldWrapper<Message> { |
| typedef void Field; |
| typedef void Value; |
| |
| public: |
| RepeatedPtrFieldMessageAccessor() {} |
| void Swap(Field* data, const internal::RepeatedFieldAccessor* other_mutator, |
| Field* other_data) const override { |
| GOOGLE_CHECK(this == other_mutator); |
| MutableRepeatedField(data)->Swap(MutableRepeatedField(other_data)); |
| } |
| |
| protected: |
| Message* New(const Value* value) const override { |
| return static_cast<const Message*>(value)->New(); |
| } |
| void ConvertToT(const Value* value, Message* result) const override { |
| result->CopyFrom(*static_cast<const Message*>(value)); |
| } |
| const Value* ConvertFromT(const Message& value, |
| Value* scratch_space) const override { |
| return static_cast<const Value*>(&value); |
| } |
| }; |
| } // namespace internal |
| } // namespace protobuf |
| } // namespace google |
| |
| #endif // GOOGLE_PROTOBUF_REFLECTION_INTERNAL_H__ |