364 lines
14 KiB
C++
364 lines
14 KiB
C++
// 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__
|