| Index: third_party/protobuf/python/google/protobuf/pyext/repeated_scalar_container.cc
|
| diff --git a/third_party/protobuf/python/google/protobuf/pyext/repeated_scalar_container.cc b/third_party/protobuf/python/google/protobuf/pyext/repeated_scalar_container.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..fd19683604747439b0c8f0c2ce54365703d89621
|
| --- /dev/null
|
| +++ b/third_party/protobuf/python/google/protobuf/pyext/repeated_scalar_container.cc
|
| @@ -0,0 +1,811 @@
|
| +// 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.
|
| +
|
| +// Author: anuraag@google.com (Anuraag Agrawal)
|
| +// Author: tibell@google.com (Johan Tibell)
|
| +
|
| +#include <google/protobuf/pyext/repeated_scalar_container.h>
|
| +
|
| +#include <memory>
|
| +#ifndef _SHARED_PTR_H
|
| +#include <google/protobuf/stubs/shared_ptr.h>
|
| +#endif
|
| +
|
| +#include <google/protobuf/stubs/common.h>
|
| +#include <google/protobuf/descriptor.h>
|
| +#include <google/protobuf/dynamic_message.h>
|
| +#include <google/protobuf/message.h>
|
| +#include <google/protobuf/pyext/descriptor.h>
|
| +#include <google/protobuf/pyext/message.h>
|
| +#include <google/protobuf/pyext/scoped_pyobject_ptr.h>
|
| +
|
| +#if PY_MAJOR_VERSION >= 3
|
| + #define PyInt_FromLong PyLong_FromLong
|
| + #if PY_VERSION_HEX < 0x03030000
|
| + #error "Python 3.0 - 3.2 are not supported."
|
| + #else
|
| + #define PyString_AsString(ob) \
|
| + (PyUnicode_Check(ob)? PyUnicode_AsUTF8(ob): PyBytes_AsString(ob))
|
| + #endif
|
| +#endif
|
| +
|
| +namespace google {
|
| +namespace protobuf {
|
| +namespace python {
|
| +
|
| +namespace repeated_scalar_container {
|
| +
|
| +static int InternalAssignRepeatedField(
|
| + RepeatedScalarContainer* self, PyObject* list) {
|
| + self->message->GetReflection()->ClearField(self->message,
|
| + self->parent_field_descriptor);
|
| + for (Py_ssize_t i = 0; i < PyList_GET_SIZE(list); ++i) {
|
| + PyObject* value = PyList_GET_ITEM(list, i);
|
| + if (Append(self, value) == NULL) {
|
| + return -1;
|
| + }
|
| + }
|
| + return 0;
|
| +}
|
| +
|
| +static Py_ssize_t Len(RepeatedScalarContainer* self) {
|
| + Message* message = self->message;
|
| + return message->GetReflection()->FieldSize(*message,
|
| + self->parent_field_descriptor);
|
| +}
|
| +
|
| +static int AssignItem(RepeatedScalarContainer* self,
|
| + Py_ssize_t index,
|
| + PyObject* arg) {
|
| + cmessage::AssureWritable(self->parent);
|
| + Message* message = self->message;
|
| + const FieldDescriptor* field_descriptor = self->parent_field_descriptor;
|
| +
|
| + const Reflection* reflection = message->GetReflection();
|
| + int field_size = reflection->FieldSize(*message, field_descriptor);
|
| + if (index < 0) {
|
| + index = field_size + index;
|
| + }
|
| + if (index < 0 || index >= field_size) {
|
| + PyErr_Format(PyExc_IndexError,
|
| + "list assignment index (%d) out of range",
|
| + static_cast<int>(index));
|
| + return -1;
|
| + }
|
| +
|
| + if (arg == NULL) {
|
| + ScopedPyObjectPtr py_index(PyLong_FromLong(index));
|
| + return cmessage::InternalDeleteRepeatedField(self->parent, field_descriptor,
|
| + py_index, NULL);
|
| + }
|
| +
|
| + if (PySequence_Check(arg) && !(PyBytes_Check(arg) || PyUnicode_Check(arg))) {
|
| + PyErr_SetString(PyExc_TypeError, "Value must be scalar");
|
| + return -1;
|
| + }
|
| +
|
| + switch (field_descriptor->cpp_type()) {
|
| + case FieldDescriptor::CPPTYPE_INT32: {
|
| + GOOGLE_CHECK_GET_INT32(arg, value, -1);
|
| + reflection->SetRepeatedInt32(message, field_descriptor, index, value);
|
| + break;
|
| + }
|
| + case FieldDescriptor::CPPTYPE_INT64: {
|
| + GOOGLE_CHECK_GET_INT64(arg, value, -1);
|
| + reflection->SetRepeatedInt64(message, field_descriptor, index, value);
|
| + break;
|
| + }
|
| + case FieldDescriptor::CPPTYPE_UINT32: {
|
| + GOOGLE_CHECK_GET_UINT32(arg, value, -1);
|
| + reflection->SetRepeatedUInt32(message, field_descriptor, index, value);
|
| + break;
|
| + }
|
| + case FieldDescriptor::CPPTYPE_UINT64: {
|
| + GOOGLE_CHECK_GET_UINT64(arg, value, -1);
|
| + reflection->SetRepeatedUInt64(message, field_descriptor, index, value);
|
| + break;
|
| + }
|
| + case FieldDescriptor::CPPTYPE_FLOAT: {
|
| + GOOGLE_CHECK_GET_FLOAT(arg, value, -1);
|
| + reflection->SetRepeatedFloat(message, field_descriptor, index, value);
|
| + break;
|
| + }
|
| + case FieldDescriptor::CPPTYPE_DOUBLE: {
|
| + GOOGLE_CHECK_GET_DOUBLE(arg, value, -1);
|
| + reflection->SetRepeatedDouble(message, field_descriptor, index, value);
|
| + break;
|
| + }
|
| + case FieldDescriptor::CPPTYPE_BOOL: {
|
| + GOOGLE_CHECK_GET_BOOL(arg, value, -1);
|
| + reflection->SetRepeatedBool(message, field_descriptor, index, value);
|
| + break;
|
| + }
|
| + case FieldDescriptor::CPPTYPE_STRING: {
|
| + if (!CheckAndSetString(
|
| + arg, message, field_descriptor, reflection, false, index)) {
|
| + return -1;
|
| + }
|
| + break;
|
| + }
|
| + case FieldDescriptor::CPPTYPE_ENUM: {
|
| + GOOGLE_CHECK_GET_INT32(arg, value, -1);
|
| + if (reflection->SupportsUnknownEnumValues()) {
|
| + reflection->SetRepeatedEnumValue(message, field_descriptor, index,
|
| + value);
|
| + } else {
|
| + const EnumDescriptor* enum_descriptor = field_descriptor->enum_type();
|
| + const EnumValueDescriptor* enum_value =
|
| + enum_descriptor->FindValueByNumber(value);
|
| + if (enum_value != NULL) {
|
| + reflection->SetRepeatedEnum(message, field_descriptor, index,
|
| + enum_value);
|
| + } else {
|
| + ScopedPyObjectPtr s(PyObject_Str(arg));
|
| + if (s != NULL) {
|
| + PyErr_Format(PyExc_ValueError, "Unknown enum value: %s",
|
| + PyString_AsString(s));
|
| + }
|
| + return -1;
|
| + }
|
| + }
|
| + break;
|
| + }
|
| + default:
|
| + PyErr_Format(
|
| + PyExc_SystemError, "Adding value to a field of unknown type %d",
|
| + field_descriptor->cpp_type());
|
| + return -1;
|
| + }
|
| + return 0;
|
| +}
|
| +
|
| +static PyObject* Item(RepeatedScalarContainer* self, Py_ssize_t index) {
|
| + Message* message = self->message;
|
| + const FieldDescriptor* field_descriptor = self->parent_field_descriptor;
|
| + const Reflection* reflection = message->GetReflection();
|
| +
|
| + int field_size = reflection->FieldSize(*message, field_descriptor);
|
| + if (index < 0) {
|
| + index = field_size + index;
|
| + }
|
| + if (index < 0 || index >= field_size) {
|
| + PyErr_Format(PyExc_IndexError,
|
| + "list index (%zd) out of range",
|
| + index);
|
| + return NULL;
|
| + }
|
| +
|
| + PyObject* result = NULL;
|
| + switch (field_descriptor->cpp_type()) {
|
| + case FieldDescriptor::CPPTYPE_INT32: {
|
| + int32 value = reflection->GetRepeatedInt32(
|
| + *message, field_descriptor, index);
|
| + result = PyInt_FromLong(value);
|
| + break;
|
| + }
|
| + case FieldDescriptor::CPPTYPE_INT64: {
|
| + int64 value = reflection->GetRepeatedInt64(
|
| + *message, field_descriptor, index);
|
| + result = PyLong_FromLongLong(value);
|
| + break;
|
| + }
|
| + case FieldDescriptor::CPPTYPE_UINT32: {
|
| + uint32 value = reflection->GetRepeatedUInt32(
|
| + *message, field_descriptor, index);
|
| + result = PyLong_FromLongLong(value);
|
| + break;
|
| + }
|
| + case FieldDescriptor::CPPTYPE_UINT64: {
|
| + uint64 value = reflection->GetRepeatedUInt64(
|
| + *message, field_descriptor, index);
|
| + result = PyLong_FromUnsignedLongLong(value);
|
| + break;
|
| + }
|
| + case FieldDescriptor::CPPTYPE_FLOAT: {
|
| + float value = reflection->GetRepeatedFloat(
|
| + *message, field_descriptor, index);
|
| + result = PyFloat_FromDouble(value);
|
| + break;
|
| + }
|
| + case FieldDescriptor::CPPTYPE_DOUBLE: {
|
| + double value = reflection->GetRepeatedDouble(
|
| + *message, field_descriptor, index);
|
| + result = PyFloat_FromDouble(value);
|
| + break;
|
| + }
|
| + case FieldDescriptor::CPPTYPE_BOOL: {
|
| + bool value = reflection->GetRepeatedBool(
|
| + *message, field_descriptor, index);
|
| + result = PyBool_FromLong(value ? 1 : 0);
|
| + break;
|
| + }
|
| + case FieldDescriptor::CPPTYPE_ENUM: {
|
| + const EnumValueDescriptor* enum_value =
|
| + message->GetReflection()->GetRepeatedEnum(
|
| + *message, field_descriptor, index);
|
| + result = PyInt_FromLong(enum_value->number());
|
| + break;
|
| + }
|
| + case FieldDescriptor::CPPTYPE_STRING: {
|
| + string value = reflection->GetRepeatedString(
|
| + *message, field_descriptor, index);
|
| + result = ToStringObject(field_descriptor, value);
|
| + break;
|
| + }
|
| + case FieldDescriptor::CPPTYPE_MESSAGE: {
|
| + PyObject* py_cmsg = PyObject_CallObject(reinterpret_cast<PyObject*>(
|
| + &CMessage_Type), NULL);
|
| + if (py_cmsg == NULL) {
|
| + return NULL;
|
| + }
|
| + CMessage* cmsg = reinterpret_cast<CMessage*>(py_cmsg);
|
| + const Message& msg = reflection->GetRepeatedMessage(
|
| + *message, field_descriptor, index);
|
| + cmsg->owner = self->owner;
|
| + cmsg->parent = self->parent;
|
| + cmsg->message = const_cast<Message*>(&msg);
|
| + cmsg->read_only = false;
|
| + result = reinterpret_cast<PyObject*>(py_cmsg);
|
| + break;
|
| + }
|
| + default:
|
| + PyErr_Format(
|
| + PyExc_SystemError,
|
| + "Getting value from a repeated field of unknown type %d",
|
| + field_descriptor->cpp_type());
|
| + }
|
| +
|
| + return result;
|
| +}
|
| +
|
| +static PyObject* Subscript(RepeatedScalarContainer* self, PyObject* slice) {
|
| + Py_ssize_t from;
|
| + Py_ssize_t to;
|
| + Py_ssize_t step;
|
| + Py_ssize_t length;
|
| + Py_ssize_t slicelength;
|
| + bool return_list = false;
|
| +#if PY_MAJOR_VERSION < 3
|
| + if (PyInt_Check(slice)) {
|
| + from = to = PyInt_AsLong(slice);
|
| + } else // NOLINT
|
| +#endif
|
| + if (PyLong_Check(slice)) {
|
| + from = to = PyLong_AsLong(slice);
|
| + } else if (PySlice_Check(slice)) {
|
| + length = Len(self);
|
| +#if PY_MAJOR_VERSION >= 3
|
| + if (PySlice_GetIndicesEx(slice,
|
| +#else
|
| + if (PySlice_GetIndicesEx(reinterpret_cast<PySliceObject*>(slice),
|
| +#endif
|
| + length, &from, &to, &step, &slicelength) == -1) {
|
| + return NULL;
|
| + }
|
| + return_list = true;
|
| + } else {
|
| + PyErr_SetString(PyExc_TypeError, "list indices must be integers");
|
| + return NULL;
|
| + }
|
| +
|
| + if (!return_list) {
|
| + return Item(self, from);
|
| + }
|
| +
|
| + PyObject* list = PyList_New(0);
|
| + if (list == NULL) {
|
| + return NULL;
|
| + }
|
| + if (from <= to) {
|
| + if (step < 0) {
|
| + return list;
|
| + }
|
| + for (Py_ssize_t index = from; index < to; index += step) {
|
| + if (index < 0 || index >= length) {
|
| + break;
|
| + }
|
| + ScopedPyObjectPtr s(Item(self, index));
|
| + PyList_Append(list, s);
|
| + }
|
| + } else {
|
| + if (step > 0) {
|
| + return list;
|
| + }
|
| + for (Py_ssize_t index = from; index > to; index += step) {
|
| + if (index < 0 || index >= length) {
|
| + break;
|
| + }
|
| + ScopedPyObjectPtr s(Item(self, index));
|
| + PyList_Append(list, s);
|
| + }
|
| + }
|
| + return list;
|
| +}
|
| +
|
| +PyObject* Append(RepeatedScalarContainer* self, PyObject* item) {
|
| + cmessage::AssureWritable(self->parent);
|
| + Message* message = self->message;
|
| + const FieldDescriptor* field_descriptor = self->parent_field_descriptor;
|
| +
|
| + const Reflection* reflection = message->GetReflection();
|
| + switch (field_descriptor->cpp_type()) {
|
| + case FieldDescriptor::CPPTYPE_INT32: {
|
| + GOOGLE_CHECK_GET_INT32(item, value, NULL);
|
| + reflection->AddInt32(message, field_descriptor, value);
|
| + break;
|
| + }
|
| + case FieldDescriptor::CPPTYPE_INT64: {
|
| + GOOGLE_CHECK_GET_INT64(item, value, NULL);
|
| + reflection->AddInt64(message, field_descriptor, value);
|
| + break;
|
| + }
|
| + case FieldDescriptor::CPPTYPE_UINT32: {
|
| + GOOGLE_CHECK_GET_UINT32(item, value, NULL);
|
| + reflection->AddUInt32(message, field_descriptor, value);
|
| + break;
|
| + }
|
| + case FieldDescriptor::CPPTYPE_UINT64: {
|
| + GOOGLE_CHECK_GET_UINT64(item, value, NULL);
|
| + reflection->AddUInt64(message, field_descriptor, value);
|
| + break;
|
| + }
|
| + case FieldDescriptor::CPPTYPE_FLOAT: {
|
| + GOOGLE_CHECK_GET_FLOAT(item, value, NULL);
|
| + reflection->AddFloat(message, field_descriptor, value);
|
| + break;
|
| + }
|
| + case FieldDescriptor::CPPTYPE_DOUBLE: {
|
| + GOOGLE_CHECK_GET_DOUBLE(item, value, NULL);
|
| + reflection->AddDouble(message, field_descriptor, value);
|
| + break;
|
| + }
|
| + case FieldDescriptor::CPPTYPE_BOOL: {
|
| + GOOGLE_CHECK_GET_BOOL(item, value, NULL);
|
| + reflection->AddBool(message, field_descriptor, value);
|
| + break;
|
| + }
|
| + case FieldDescriptor::CPPTYPE_STRING: {
|
| + if (!CheckAndSetString(
|
| + item, message, field_descriptor, reflection, true, -1)) {
|
| + return NULL;
|
| + }
|
| + break;
|
| + }
|
| + case FieldDescriptor::CPPTYPE_ENUM: {
|
| + GOOGLE_CHECK_GET_INT32(item, value, NULL);
|
| + if (reflection->SupportsUnknownEnumValues()) {
|
| + reflection->AddEnumValue(message, field_descriptor, value);
|
| + } else {
|
| + const EnumDescriptor* enum_descriptor = field_descriptor->enum_type();
|
| + const EnumValueDescriptor* enum_value =
|
| + enum_descriptor->FindValueByNumber(value);
|
| + if (enum_value != NULL) {
|
| + reflection->AddEnum(message, field_descriptor, enum_value);
|
| + } else {
|
| + ScopedPyObjectPtr s(PyObject_Str(item));
|
| + if (s != NULL) {
|
| + PyErr_Format(PyExc_ValueError, "Unknown enum value: %s",
|
| + PyString_AsString(s));
|
| + }
|
| + return NULL;
|
| + }
|
| + }
|
| + break;
|
| + }
|
| + default:
|
| + PyErr_Format(
|
| + PyExc_SystemError, "Adding value to a field of unknown type %d",
|
| + field_descriptor->cpp_type());
|
| + return NULL;
|
| + }
|
| +
|
| + Py_RETURN_NONE;
|
| +}
|
| +
|
| +static int AssSubscript(RepeatedScalarContainer* self,
|
| + PyObject* slice,
|
| + PyObject* value) {
|
| + Py_ssize_t from;
|
| + Py_ssize_t to;
|
| + Py_ssize_t step;
|
| + Py_ssize_t length;
|
| + Py_ssize_t slicelength;
|
| + bool create_list = false;
|
| +
|
| + cmessage::AssureWritable(self->parent);
|
| + Message* message = self->message;
|
| + const FieldDescriptor* field_descriptor =
|
| + self->parent_field_descriptor;
|
| +
|
| +#if PY_MAJOR_VERSION < 3
|
| + if (PyInt_Check(slice)) {
|
| + from = to = PyInt_AsLong(slice);
|
| + } else
|
| +#endif
|
| + if (PyLong_Check(slice)) {
|
| + from = to = PyLong_AsLong(slice);
|
| + } else if (PySlice_Check(slice)) {
|
| + const Reflection* reflection = message->GetReflection();
|
| + length = reflection->FieldSize(*message, field_descriptor);
|
| +#if PY_MAJOR_VERSION >= 3
|
| + if (PySlice_GetIndicesEx(slice,
|
| +#else
|
| + if (PySlice_GetIndicesEx(reinterpret_cast<PySliceObject*>(slice),
|
| +#endif
|
| + length, &from, &to, &step, &slicelength) == -1) {
|
| + return -1;
|
| + }
|
| + create_list = true;
|
| + } else {
|
| + PyErr_SetString(PyExc_TypeError, "list indices must be integers");
|
| + return -1;
|
| + }
|
| +
|
| + if (value == NULL) {
|
| + return cmessage::InternalDeleteRepeatedField(
|
| + self->parent, field_descriptor, slice, NULL);
|
| + }
|
| +
|
| + if (!create_list) {
|
| + return AssignItem(self, from, value);
|
| + }
|
| +
|
| + ScopedPyObjectPtr full_slice(PySlice_New(NULL, NULL, NULL));
|
| + if (full_slice == NULL) {
|
| + return -1;
|
| + }
|
| + ScopedPyObjectPtr new_list(Subscript(self, full_slice));
|
| + if (new_list == NULL) {
|
| + return -1;
|
| + }
|
| + if (PySequence_SetSlice(new_list, from, to, value) < 0) {
|
| + return -1;
|
| + }
|
| +
|
| + return InternalAssignRepeatedField(self, new_list);
|
| +}
|
| +
|
| +PyObject* Extend(RepeatedScalarContainer* self, PyObject* value) {
|
| + cmessage::AssureWritable(self->parent);
|
| +
|
| + // TODO(ptucker): Deprecate this behavior. b/18413862
|
| + if (value == Py_None) {
|
| + Py_RETURN_NONE;
|
| + }
|
| + if ((Py_TYPE(value)->tp_as_sequence == NULL) && PyObject_Not(value)) {
|
| + Py_RETURN_NONE;
|
| + }
|
| +
|
| + ScopedPyObjectPtr iter(PyObject_GetIter(value));
|
| + if (iter == NULL) {
|
| + PyErr_SetString(PyExc_TypeError, "Value must be iterable");
|
| + return NULL;
|
| + }
|
| + ScopedPyObjectPtr next;
|
| + while ((next.reset(PyIter_Next(iter))) != NULL) {
|
| + if (Append(self, next) == NULL) {
|
| + return NULL;
|
| + }
|
| + }
|
| + if (PyErr_Occurred()) {
|
| + return NULL;
|
| + }
|
| + Py_RETURN_NONE;
|
| +}
|
| +
|
| +static PyObject* Insert(RepeatedScalarContainer* self, PyObject* args) {
|
| + Py_ssize_t index;
|
| + PyObject* value;
|
| + if (!PyArg_ParseTuple(args, "lO", &index, &value)) {
|
| + return NULL;
|
| + }
|
| + ScopedPyObjectPtr full_slice(PySlice_New(NULL, NULL, NULL));
|
| + ScopedPyObjectPtr new_list(Subscript(self, full_slice));
|
| + if (PyList_Insert(new_list, index, value) < 0) {
|
| + return NULL;
|
| + }
|
| + int ret = InternalAssignRepeatedField(self, new_list);
|
| + if (ret < 0) {
|
| + return NULL;
|
| + }
|
| + Py_RETURN_NONE;
|
| +}
|
| +
|
| +static PyObject* Remove(RepeatedScalarContainer* self, PyObject* value) {
|
| + Py_ssize_t match_index = -1;
|
| + for (Py_ssize_t i = 0; i < Len(self); ++i) {
|
| + ScopedPyObjectPtr elem(Item(self, i));
|
| + if (PyObject_RichCompareBool(elem, value, Py_EQ)) {
|
| + match_index = i;
|
| + break;
|
| + }
|
| + }
|
| + if (match_index == -1) {
|
| + PyErr_SetString(PyExc_ValueError, "remove(x): x not in container");
|
| + return NULL;
|
| + }
|
| + if (AssignItem(self, match_index, NULL) < 0) {
|
| + return NULL;
|
| + }
|
| + Py_RETURN_NONE;
|
| +}
|
| +
|
| +static PyObject* RichCompare(RepeatedScalarContainer* self,
|
| + PyObject* other,
|
| + int opid) {
|
| + if (opid != Py_EQ && opid != Py_NE) {
|
| + Py_INCREF(Py_NotImplemented);
|
| + return Py_NotImplemented;
|
| + }
|
| +
|
| + // Copy the contents of this repeated scalar container, and other if it is
|
| + // also a repeated scalar container, into Python lists so we can delegate
|
| + // to the list's compare method.
|
| +
|
| + ScopedPyObjectPtr full_slice(PySlice_New(NULL, NULL, NULL));
|
| + if (full_slice == NULL) {
|
| + return NULL;
|
| + }
|
| +
|
| + ScopedPyObjectPtr other_list_deleter;
|
| + if (PyObject_TypeCheck(other, &RepeatedScalarContainer_Type)) {
|
| + other_list_deleter.reset(Subscript(
|
| + reinterpret_cast<RepeatedScalarContainer*>(other), full_slice));
|
| + other = other_list_deleter.get();
|
| + }
|
| +
|
| + ScopedPyObjectPtr list(Subscript(self, full_slice));
|
| + if (list == NULL) {
|
| + return NULL;
|
| + }
|
| + return PyObject_RichCompare(list, other, opid);
|
| +}
|
| +
|
| +PyObject* Reduce(RepeatedScalarContainer* unused_self) {
|
| + PyErr_Format(
|
| + PickleError_class,
|
| + "can't pickle repeated message fields, convert to list first");
|
| + return NULL;
|
| +}
|
| +
|
| +static PyObject* Sort(RepeatedScalarContainer* self,
|
| + PyObject* args,
|
| + PyObject* kwds) {
|
| + // Support the old sort_function argument for backwards
|
| + // compatibility.
|
| + if (kwds != NULL) {
|
| + PyObject* sort_func = PyDict_GetItemString(kwds, "sort_function");
|
| + if (sort_func != NULL) {
|
| + // Must set before deleting as sort_func is a borrowed reference
|
| + // and kwds might be the only thing keeping it alive.
|
| + if (PyDict_SetItemString(kwds, "cmp", sort_func) == -1)
|
| + return NULL;
|
| + if (PyDict_DelItemString(kwds, "sort_function") == -1)
|
| + return NULL;
|
| + }
|
| + }
|
| +
|
| + ScopedPyObjectPtr full_slice(PySlice_New(NULL, NULL, NULL));
|
| + if (full_slice == NULL) {
|
| + return NULL;
|
| + }
|
| + ScopedPyObjectPtr list(Subscript(self, full_slice));
|
| + if (list == NULL) {
|
| + return NULL;
|
| + }
|
| + ScopedPyObjectPtr m(PyObject_GetAttrString(list, "sort"));
|
| + if (m == NULL) {
|
| + return NULL;
|
| + }
|
| + ScopedPyObjectPtr res(PyObject_Call(m, args, kwds));
|
| + if (res == NULL) {
|
| + return NULL;
|
| + }
|
| + int ret = InternalAssignRepeatedField(self, list);
|
| + if (ret < 0) {
|
| + return NULL;
|
| + }
|
| + Py_RETURN_NONE;
|
| +}
|
| +
|
| +static PyObject* Pop(RepeatedScalarContainer* self,
|
| + PyObject* args) {
|
| + Py_ssize_t index = -1;
|
| + if (!PyArg_ParseTuple(args, "|n", &index)) {
|
| + return NULL;
|
| + }
|
| + PyObject* item = Item(self, index);
|
| + if (item == NULL) {
|
| + PyErr_Format(PyExc_IndexError,
|
| + "list index (%zd) out of range",
|
| + index);
|
| + return NULL;
|
| + }
|
| + if (AssignItem(self, index, NULL) < 0) {
|
| + return NULL;
|
| + }
|
| + return item;
|
| +}
|
| +
|
| +// The private constructor of RepeatedScalarContainer objects.
|
| +PyObject *NewContainer(
|
| + CMessage* parent, const FieldDescriptor* parent_field_descriptor) {
|
| + if (!CheckFieldBelongsToMessage(parent_field_descriptor, parent->message)) {
|
| + return NULL;
|
| + }
|
| +
|
| + RepeatedScalarContainer* self = reinterpret_cast<RepeatedScalarContainer*>(
|
| + PyType_GenericAlloc(&RepeatedScalarContainer_Type, 0));
|
| + if (self == NULL) {
|
| + return NULL;
|
| + }
|
| +
|
| + self->message = parent->message;
|
| + self->parent = parent;
|
| + self->parent_field_descriptor = parent_field_descriptor;
|
| + self->owner = parent->owner;
|
| +
|
| + return reinterpret_cast<PyObject*>(self);
|
| +}
|
| +
|
| +// Initializes the underlying Message object of "to" so it becomes a new parent
|
| +// repeated scalar, and copies all the values from "from" to it. A child scalar
|
| +// container can be released by passing it as both from and to (e.g. making it
|
| +// the recipient of the new parent message and copying the values from itself).
|
| +static int InitializeAndCopyToParentContainer(
|
| + RepeatedScalarContainer* from,
|
| + RepeatedScalarContainer* to) {
|
| + ScopedPyObjectPtr full_slice(PySlice_New(NULL, NULL, NULL));
|
| + if (full_slice == NULL) {
|
| + return -1;
|
| + }
|
| + ScopedPyObjectPtr values(Subscript(from, full_slice));
|
| + if (values == NULL) {
|
| + return -1;
|
| + }
|
| + Message* new_message = cmessage::GetMessageFactory()->GetPrototype(
|
| + from->message->GetDescriptor())->New();
|
| + to->parent = NULL;
|
| + to->parent_field_descriptor = from->parent_field_descriptor;
|
| + to->message = new_message;
|
| + to->owner.reset(new_message);
|
| + if (InternalAssignRepeatedField(to, values) < 0) {
|
| + return -1;
|
| + }
|
| + return 0;
|
| +}
|
| +
|
| +int Release(RepeatedScalarContainer* self) {
|
| + return InitializeAndCopyToParentContainer(self, self);
|
| +}
|
| +
|
| +PyObject* DeepCopy(RepeatedScalarContainer* self, PyObject* arg) {
|
| + RepeatedScalarContainer* clone = reinterpret_cast<RepeatedScalarContainer*>(
|
| + PyType_GenericAlloc(&RepeatedScalarContainer_Type, 0));
|
| + if (clone == NULL) {
|
| + return NULL;
|
| + }
|
| +
|
| + if (InitializeAndCopyToParentContainer(self, clone) < 0) {
|
| + Py_DECREF(clone);
|
| + return NULL;
|
| + }
|
| + return reinterpret_cast<PyObject*>(clone);
|
| +}
|
| +
|
| +static void Dealloc(RepeatedScalarContainer* self) {
|
| + self->owner.reset();
|
| + Py_TYPE(self)->tp_free(reinterpret_cast<PyObject*>(self));
|
| +}
|
| +
|
| +void SetOwner(RepeatedScalarContainer* self,
|
| + const shared_ptr<Message>& new_owner) {
|
| + self->owner = new_owner;
|
| +}
|
| +
|
| +static PySequenceMethods SqMethods = {
|
| + (lenfunc)Len, /* sq_length */
|
| + 0, /* sq_concat */
|
| + 0, /* sq_repeat */
|
| + (ssizeargfunc)Item, /* sq_item */
|
| + 0, /* sq_slice */
|
| + (ssizeobjargproc)AssignItem /* sq_ass_item */
|
| +};
|
| +
|
| +static PyMappingMethods MpMethods = {
|
| + (lenfunc)Len, /* mp_length */
|
| + (binaryfunc)Subscript, /* mp_subscript */
|
| + (objobjargproc)AssSubscript, /* mp_ass_subscript */
|
| +};
|
| +
|
| +static PyMethodDef Methods[] = {
|
| + { "__deepcopy__", (PyCFunction)DeepCopy, METH_VARARGS,
|
| + "Makes a deep copy of the class." },
|
| + { "__reduce__", (PyCFunction)Reduce, METH_NOARGS,
|
| + "Outputs picklable representation of the repeated field." },
|
| + { "append", (PyCFunction)Append, METH_O,
|
| + "Appends an object to the repeated container." },
|
| + { "extend", (PyCFunction)Extend, METH_O,
|
| + "Appends objects to the repeated container." },
|
| + { "insert", (PyCFunction)Insert, METH_VARARGS,
|
| + "Appends objects to the repeated container." },
|
| + { "pop", (PyCFunction)Pop, METH_VARARGS,
|
| + "Removes an object from the repeated container and returns it." },
|
| + { "remove", (PyCFunction)Remove, METH_O,
|
| + "Removes an object from the repeated container." },
|
| + { "sort", (PyCFunction)Sort, METH_VARARGS | METH_KEYWORDS,
|
| + "Sorts the repeated container."},
|
| + { NULL, NULL }
|
| +};
|
| +
|
| +} // namespace repeated_scalar_container
|
| +
|
| +PyTypeObject RepeatedScalarContainer_Type = {
|
| + PyVarObject_HEAD_INIT(&PyType_Type, 0)
|
| + FULL_MODULE_NAME ".RepeatedScalarContainer", // tp_name
|
| + sizeof(RepeatedScalarContainer), // tp_basicsize
|
| + 0, // tp_itemsize
|
| + (destructor)repeated_scalar_container::Dealloc, // tp_dealloc
|
| + 0, // tp_print
|
| + 0, // tp_getattr
|
| + 0, // tp_setattr
|
| + 0, // tp_compare
|
| + 0, // tp_repr
|
| + 0, // tp_as_number
|
| + &repeated_scalar_container::SqMethods, // tp_as_sequence
|
| + &repeated_scalar_container::MpMethods, // tp_as_mapping
|
| + 0, // tp_hash
|
| + 0, // tp_call
|
| + 0, // tp_str
|
| + 0, // tp_getattro
|
| + 0, // tp_setattro
|
| + 0, // tp_as_buffer
|
| + Py_TPFLAGS_DEFAULT, // tp_flags
|
| + "A Repeated scalar container", // tp_doc
|
| + 0, // tp_traverse
|
| + 0, // tp_clear
|
| + (richcmpfunc)repeated_scalar_container::RichCompare, // tp_richcompare
|
| + 0, // tp_weaklistoffset
|
| + 0, // tp_iter
|
| + 0, // tp_iternext
|
| + repeated_scalar_container::Methods, // tp_methods
|
| + 0, // tp_members
|
| + 0, // tp_getset
|
| + 0, // tp_base
|
| + 0, // tp_dict
|
| + 0, // tp_descr_get
|
| + 0, // tp_descr_set
|
| + 0, // tp_dictoffset
|
| + 0, // tp_init
|
| +};
|
| +
|
| +} // namespace python
|
| +} // namespace protobuf
|
| +} // namespace google
|
|
|