/* Array object implementation */ /* An array is a uniform list -- all items have the same type. The item type is restricted to simple C types like int or float */ #ifndef Py_BUILD_CORE_BUILTIN #define Py_BUILD_CORE_MODULE … #endif #include "Python.h" #include "pycore_bytesobject.h" // _PyBytes_Repeat #include "pycore_call.h" // _PyObject_CallMethod() #include "pycore_ceval.h" // _PyEval_GetBuiltin() #include "pycore_modsupport.h" // _PyArg_NoKeywords() #include "pycore_moduleobject.h" // _PyModule_GetState() #include <stddef.h> // offsetof() #include <stdbool.h> /*[clinic input] module array [clinic start generated code]*/ /*[clinic end generated code: output=da39a3ee5e6b4b0d input=7d1b8d7f5958fd83]*/ struct arrayobject; /* Forward */ static struct PyModuleDef arraymodule; /* All possible arraydescr values are defined in the vector "descriptors" * below. That's defined later because the appropriate get and set * functions aren't visible yet. */ struct arraydescr { … }; arrayobject; arrayiterobject; array_state; static array_state * get_array_state(PyObject *module) { … } #define find_array_state_by_type(tp) … #define get_array_state_by_class(cls) … enum machine_format_code { … }; #define MACHINE_FORMAT_CODE_MIN … #define MACHINE_FORMAT_CODE_MAX … /* * Must come after arrayobject, arrayiterobject, * and enum machine_code_type definitions. */ #include "clinic/arraymodule.c.h" #define array_Check(op, state) … static int array_resize(arrayobject *self, Py_ssize_t newsize) { … } /**************************************************************************** Get and Set functions for each type. A Get function takes an arrayobject* and an integer index, returning the array value at that index wrapped in an appropriate PyObject*. A Set function takes an arrayobject, integer index, and PyObject*; sets the array value at that index to the raw C data extracted from the PyObject*, and returns 0 if successful, else nonzero on failure (PyObject* not of an appropriate type or value). Note that the basic Get and Set functions do NOT check that the index is in bounds; that's the responsibility of the caller. ****************************************************************************/ static PyObject * b_getitem(arrayobject *ap, Py_ssize_t i) { … } static int b_setitem(arrayobject *ap, Py_ssize_t i, PyObject *v) { … } static PyObject * BB_getitem(arrayobject *ap, Py_ssize_t i) { … } static int BB_setitem(arrayobject *ap, Py_ssize_t i, PyObject *v) { … } static PyObject * u_getitem(arrayobject *ap, Py_ssize_t i) { … } static int u_setitem(arrayobject *ap, Py_ssize_t i, PyObject *v) { … } static PyObject * w_getitem(arrayobject *ap, Py_ssize_t i) { … } static int w_setitem(arrayobject *ap, Py_ssize_t i, PyObject *v) { … } static PyObject * h_getitem(arrayobject *ap, Py_ssize_t i) { … } static int h_setitem(arrayobject *ap, Py_ssize_t i, PyObject *v) { … } static PyObject * HH_getitem(arrayobject *ap, Py_ssize_t i) { … } static int HH_setitem(arrayobject *ap, Py_ssize_t i, PyObject *v) { … } static PyObject * i_getitem(arrayobject *ap, Py_ssize_t i) { … } static int i_setitem(arrayobject *ap, Py_ssize_t i, PyObject *v) { … } static PyObject * II_getitem(arrayobject *ap, Py_ssize_t i) { … } static int II_setitem(arrayobject *ap, Py_ssize_t i, PyObject *v) { … } static PyObject * l_getitem(arrayobject *ap, Py_ssize_t i) { … } static int l_setitem(arrayobject *ap, Py_ssize_t i, PyObject *v) { … } static PyObject * LL_getitem(arrayobject *ap, Py_ssize_t i) { … } static int LL_setitem(arrayobject *ap, Py_ssize_t i, PyObject *v) { … } static PyObject * q_getitem(arrayobject *ap, Py_ssize_t i) { … } static int q_setitem(arrayobject *ap, Py_ssize_t i, PyObject *v) { … } static PyObject * QQ_getitem(arrayobject *ap, Py_ssize_t i) { … } static int QQ_setitem(arrayobject *ap, Py_ssize_t i, PyObject *v) { … } static PyObject * f_getitem(arrayobject *ap, Py_ssize_t i) { … } static int f_setitem(arrayobject *ap, Py_ssize_t i, PyObject *v) { … } static PyObject * d_getitem(arrayobject *ap, Py_ssize_t i) { … } static int d_setitem(arrayobject *ap, Py_ssize_t i, PyObject *v) { … } #define DEFINE_COMPAREITEMS(code, type) … DEFINE_COMPAREITEMS(b, signed char) DEFINE_COMPAREITEMS(BB, unsigned char) DEFINE_COMPAREITEMS(u, wchar_t) DEFINE_COMPAREITEMS(w, Py_UCS4) DEFINE_COMPAREITEMS(h, short) DEFINE_COMPAREITEMS(HH, unsigned short) DEFINE_COMPAREITEMS(i, int) DEFINE_COMPAREITEMS(II, unsigned int) DEFINE_COMPAREITEMS(l, long) DEFINE_COMPAREITEMS(LL, unsigned long) DEFINE_COMPAREITEMS(q, long long) DEFINE_COMPAREITEMS(QQ, unsigned long long) /* Description of types. * * Don't forget to update typecode_to_mformat_code() if you add a new * typecode. */ static const struct arraydescr descriptors[] = …; /**************************************************************************** Implementations of array object methods. ****************************************************************************/ /*[clinic input] class array.array "arrayobject *" "ArrayType" [clinic start generated code]*/ /*[clinic end generated code: output=da39a3ee5e6b4b0d input=a5c29edf59f176a3]*/ static PyObject * newarrayobject(PyTypeObject *type, Py_ssize_t size, const struct arraydescr *descr) { … } static PyObject * getarrayitem(PyObject *op, Py_ssize_t i) { … } static int ins1(arrayobject *self, Py_ssize_t where, PyObject *v) { … } /* Methods */ static int array_tp_traverse(arrayobject *op, visitproc visit, void *arg) { … } static void array_dealloc(arrayobject *op) { … } static PyObject * array_richcompare(PyObject *v, PyObject *w, int op) { … } static Py_ssize_t array_length(arrayobject *a) { … } static PyObject * array_item(arrayobject *a, Py_ssize_t i) { … } static PyObject * array_slice(arrayobject *a, Py_ssize_t ilow, Py_ssize_t ihigh) { … } /*[clinic input] array.array.clear Remove all items from the array. [clinic start generated code]*/ static PyObject * array_array_clear_impl(arrayobject *self) /*[clinic end generated code: output=5efe0417062210a9 input=5dffa30e94e717a4]*/ { … } /*[clinic input] array.array.__copy__ Return a copy of the array. [clinic start generated code]*/ static PyObject * array_array___copy___impl(arrayobject *self) /*[clinic end generated code: output=dec7c3f925d9619e input=ad1ee5b086965f09]*/ { … } /*[clinic input] array.array.__deepcopy__ unused: object / Return a copy of the array. [clinic start generated code]*/ static PyObject * array_array___deepcopy__(arrayobject *self, PyObject *unused) /*[clinic end generated code: output=1ec748d8e14a9faa input=2405ecb4933748c4]*/ { … } static PyObject * array_concat(arrayobject *a, PyObject *bb) { … } static PyObject * array_repeat(arrayobject *a, Py_ssize_t n) { … } static int array_del_slice(arrayobject *a, Py_ssize_t ilow, Py_ssize_t ihigh) { … } static int array_ass_item(arrayobject *a, Py_ssize_t i, PyObject *v) { … } static int setarrayitem(PyObject *a, Py_ssize_t i, PyObject *v) { … } static int array_iter_extend(arrayobject *self, PyObject *bb) { … } static int array_do_extend(array_state *state, arrayobject *self, PyObject *bb) { … } static PyObject * array_inplace_concat(arrayobject *self, PyObject *bb) { … } static PyObject * array_inplace_repeat(arrayobject *self, Py_ssize_t n) { … } static PyObject * ins(arrayobject *self, Py_ssize_t where, PyObject *v) { … } /*[clinic input] array.array.count v: object / Return number of occurrences of v in the array. [clinic start generated code]*/ static PyObject * array_array_count(arrayobject *self, PyObject *v) /*[clinic end generated code: output=3dd3624bf7135a3a input=d9bce9d65e39d1f5]*/ { … } /*[clinic input] array.array.index v: object start: slice_index(accept={int}) = 0 stop: slice_index(accept={int}, c_default="PY_SSIZE_T_MAX") = sys.maxsize / Return index of first occurrence of v in the array. Raise ValueError if the value is not present. [clinic start generated code]*/ static PyObject * array_array_index_impl(arrayobject *self, PyObject *v, Py_ssize_t start, Py_ssize_t stop) /*[clinic end generated code: output=c45e777880c99f52 input=089dff7baa7e5a7e]*/ { … } static int array_contains(arrayobject *self, PyObject *v) { … } /*[clinic input] array.array.remove v: object / Remove the first occurrence of v in the array. [clinic start generated code]*/ static PyObject * array_array_remove(arrayobject *self, PyObject *v) /*[clinic end generated code: output=bef06be9fdf9dceb input=0b1e5aed25590027]*/ { … } /*[clinic input] array.array.pop i: Py_ssize_t = -1 / Return the i-th element and delete it from the array. i defaults to -1. [clinic start generated code]*/ static PyObject * array_array_pop_impl(arrayobject *self, Py_ssize_t i) /*[clinic end generated code: output=bc1f0c54fe5308e4 input=8e5feb4c1a11cd44]*/ { … } /*[clinic input] array.array.extend cls: defining_class bb: object / Append items to the end of the array. [clinic start generated code]*/ static PyObject * array_array_extend_impl(arrayobject *self, PyTypeObject *cls, PyObject *bb) /*[clinic end generated code: output=e65eb7588f0bc266 input=8eb6817ec4d2cb62]*/ { … } /*[clinic input] array.array.insert i: Py_ssize_t v: object / Insert a new item v into the array before position i. [clinic start generated code]*/ static PyObject * array_array_insert_impl(arrayobject *self, Py_ssize_t i, PyObject *v) /*[clinic end generated code: output=5a3648e278348564 input=5577d1b4383e9313]*/ { … } /*[clinic input] array.array.buffer_info Return a tuple (address, length) giving the current memory address and the length in items of the buffer used to hold array's contents. The length should be multiplied by the itemsize attribute to calculate the buffer length in bytes. [clinic start generated code]*/ static PyObject * array_array_buffer_info_impl(arrayobject *self) /*[clinic end generated code: output=9b2a4ec3ae7e98e7 input=a58bae5c6e1ac6a6]*/ { … } /*[clinic input] array.array.append v: object / Append new value v to the end of the array. [clinic start generated code]*/ static PyObject * array_array_append(arrayobject *self, PyObject *v) /*[clinic end generated code: output=745a0669bf8db0e2 input=0b98d9d78e78f0fa]*/ { … } /*[clinic input] array.array.byteswap Byteswap all items of the array. If the items in the array are not 1, 2, 4, or 8 bytes in size, RuntimeError is raised. [clinic start generated code]*/ static PyObject * array_array_byteswap_impl(arrayobject *self) /*[clinic end generated code: output=5f8236cbdf0d90b5 input=6a85591b950a0186]*/ { … } /*[clinic input] array.array.reverse Reverse the order of the items in the array. [clinic start generated code]*/ static PyObject * array_array_reverse_impl(arrayobject *self) /*[clinic end generated code: output=c04868b36f6f4089 input=cd904f01b27d966a]*/ { … } /*[clinic input] array.array.fromfile cls: defining_class f: object n: Py_ssize_t / Read n objects from the file object f and append them to the end of the array. [clinic start generated code]*/ static PyObject * array_array_fromfile_impl(arrayobject *self, PyTypeObject *cls, PyObject *f, Py_ssize_t n) /*[clinic end generated code: output=83a667080b345ebc input=3822e907c1c11f1a]*/ { … } /*[clinic input] array.array.tofile cls: defining_class f: object / Write all items (as machine values) to the file object f. [clinic start generated code]*/ static PyObject * array_array_tofile_impl(arrayobject *self, PyTypeObject *cls, PyObject *f) /*[clinic end generated code: output=4560c628d9c18bc2 input=5a24da7a7b407b52]*/ { … } /*[clinic input] array.array.fromlist list: object / Append items to array from list. [clinic start generated code]*/ static PyObject * array_array_fromlist(arrayobject *self, PyObject *list) /*[clinic end generated code: output=26411c2d228a3e3f input=be2605a96c49680f]*/ { … } /*[clinic input] array.array.tolist Convert array to an ordinary list with the same items. [clinic start generated code]*/ static PyObject * array_array_tolist_impl(arrayobject *self) /*[clinic end generated code: output=00b60cc9eab8ef89 input=a8d7784a94f86b53]*/ { … } static PyObject * frombytes(arrayobject *self, Py_buffer *buffer) { … } /*[clinic input] array.array.frombytes buffer: Py_buffer / Appends items from the string, interpreting it as an array of machine values, as if it had been read from a file using the fromfile() method. [clinic start generated code]*/ static PyObject * array_array_frombytes_impl(arrayobject *self, Py_buffer *buffer) /*[clinic end generated code: output=d9842c8f7510a516 input=378db226dfac949e]*/ { … } /*[clinic input] array.array.tobytes Convert the array to an array of machine values and return the bytes representation. [clinic start generated code]*/ static PyObject * array_array_tobytes_impl(arrayobject *self) /*[clinic end generated code: output=87318e4edcdc2bb6 input=90ee495f96de34f5]*/ { … } /*[clinic input] array.array.fromunicode ustr: unicode / Extends this array with data from the unicode string ustr. The array must be a unicode type array; otherwise a ValueError is raised. Use array.frombytes(ustr.encode(...)) to append Unicode data to an array of some other type. [clinic start generated code]*/ static PyObject * array_array_fromunicode_impl(arrayobject *self, PyObject *ustr) /*[clinic end generated code: output=24359f5e001a7f2b input=025db1fdade7a4ce]*/ { … } /*[clinic input] array.array.tounicode Extends this array with data from the unicode string ustr. Convert the array to a unicode string. The array must be a unicode type array; otherwise a ValueError is raised. Use array.tobytes().decode() to obtain a unicode string from an array of some other type. [clinic start generated code]*/ static PyObject * array_array_tounicode_impl(arrayobject *self) /*[clinic end generated code: output=08e442378336e1ef input=127242eebe70b66d]*/ { … } /*[clinic input] array.array.__sizeof__ Size of the array in memory, in bytes. [clinic start generated code]*/ static PyObject * array_array___sizeof___impl(arrayobject *self) /*[clinic end generated code: output=d8e1c61ebbe3eaed input=805586565bf2b3c6]*/ { … } /*********************** Pickling support ************************/ static const struct mformatdescr { … } mformat_descriptors[] = …; /* * Internal: This function is used to find the machine format of a given * array type code. This returns UNKNOWN_FORMAT when the machine format cannot * be found. */ static enum machine_format_code typecode_to_mformat_code(char typecode) { … } /* Forward declaration. */ static PyObject *array_new(PyTypeObject *type, PyObject *args, PyObject *kwds); /* * Internal: This function wraps the array constructor--i.e., array_new()--to * allow the creation of array objects from C code without having to deal * directly the tuple argument of array_new(). The typecode argument is a * Unicode character value, like 'i' or 'f' for example, representing an array * type code. The items argument is a bytes or a list object from which * contains the initial value of the array. * * On success, this functions returns the array object created. Otherwise, * NULL is returned to indicate a failure. */ static PyObject * make_array(PyTypeObject *arraytype, char typecode, PyObject *items) { … } /* * This functions is a special constructor used when unpickling an array. It * provides a portable way to rebuild an array from its memory representation. */ /*[clinic input] array._array_reconstructor arraytype: object(type="PyTypeObject *") typecode: int(accept={str}) mformat_code: int(type="enum machine_format_code") items: object / Internal. Used for pickling support. [clinic start generated code]*/ static PyObject * array__array_reconstructor_impl(PyObject *module, PyTypeObject *arraytype, int typecode, enum machine_format_code mformat_code, PyObject *items) /*[clinic end generated code: output=e05263141ba28365 input=2464dc8f4c7736b5]*/ { … } /*[clinic input] array.array.__reduce_ex__ cls: defining_class value: object / Return state information for pickling. [clinic start generated code]*/ static PyObject * array_array___reduce_ex___impl(arrayobject *self, PyTypeObject *cls, PyObject *value) /*[clinic end generated code: output=4958ee5d79452ad5 input=19968cf0f91d3eea]*/ { … } static PyObject * array_get_typecode(arrayobject *a, void *closure) { … } static PyObject * array_get_itemsize(arrayobject *a, void *closure) { … } static PyGetSetDef array_getsets [] = …; static PyMethodDef array_methods[] = …; static PyObject * array_repr(arrayobject *a) { … } static PyObject* array_subscr(arrayobject* self, PyObject* item) { … } static int array_ass_subscr(arrayobject* self, PyObject* item, PyObject* value) { … } static const void *emptybuf = …; static int array_buffer_getbuf(arrayobject *self, Py_buffer *view, int flags) { … } static void array_buffer_relbuf(arrayobject *self, Py_buffer *view) { … } static PyObject * array_new(PyTypeObject *type, PyObject *args, PyObject *kwds) { … } PyDoc_STRVAR(module_doc, "This module defines an object type which can efficiently represent\n\ an array of basic values: characters, integers, floating-point\n\ numbers. Arrays are sequence types and behave very much like lists,\n\ except that the type of objects stored in them is constrained.\n"); PyDoc_STRVAR(arraytype_doc, "array(typecode [, initializer]) -> array\n\ \n\ Return a new array whose items are restricted by typecode, and\n\ initialized from the optional initializer value, which must be a list,\n\ string or iterable over elements of the appropriate type.\n\ \n\ Arrays represent basic values and behave very much like lists, except\n\ the type of objects stored in them is constrained. The type is specified\n\ at object creation time by using a type code, which is a single character.\n\ The following type codes are defined:\n\ \n\ Type code C Type Minimum size in bytes\n\ 'b' signed integer 1\n\ 'B' unsigned integer 1\n\ 'u' Unicode character 2 (see note)\n\ 'h' signed integer 2\n\ 'H' unsigned integer 2\n\ 'i' signed integer 2\n\ 'I' unsigned integer 2\n\ 'l' signed integer 4\n\ 'L' unsigned integer 4\n\ 'q' signed integer 8 (see note)\n\ 'Q' unsigned integer 8 (see note)\n\ 'f' floating-point 4\n\ 'd' floating-point 8\n\ \n\ NOTE: The 'u' typecode corresponds to Python's unicode character. On\n\ narrow builds this is 2-bytes on wide builds this is 4-bytes.\n\ \n\ NOTE: The 'q' and 'Q' type codes are only available if the platform\n\ C compiler used to build Python supports 'long long', or, on Windows,\n\ '__int64'.\n\ \n\ Methods:\n\ \n\ append() -- append a new item to the end of the array\n\ buffer_info() -- return information giving the current memory info\n\ byteswap() -- byteswap all the items of the array\n\ count() -- return number of occurrences of an object\n\ extend() -- extend array by appending multiple elements from an iterable\n\ fromfile() -- read items from a file object\n\ fromlist() -- append items from the list\n\ frombytes() -- append items from the string\n\ index() -- return index of first occurrence of an object\n\ insert() -- insert a new item into the array at a provided position\n\ pop() -- remove and return item (default last)\n\ remove() -- remove first occurrence of an object\n\ reverse() -- reverse the order of the items in the array\n\ tofile() -- write all items to a file object\n\ tolist() -- return the array converted to an ordinary list\n\ tobytes() -- return the array converted to a string\n\ \n\ Attributes:\n\ \n\ typecode -- the typecode character used to create the array\n\ itemsize -- the length in bytes of one array item\n\ "); static PyObject *array_iter(arrayobject *ao); static struct PyMemberDef array_members[] = …; static PyType_Slot array_slots[] = …; static PyType_Spec array_spec = …; /*********************** Array Iterator **************************/ /*[clinic input] class array.arrayiterator "arrayiterobject *" "find_array_state_by_type(type)->ArrayIterType" [clinic start generated code]*/ /*[clinic end generated code: output=da39a3ee5e6b4b0d input=fb46d5ef98dd95ff]*/ static PyObject * array_iter(arrayobject *ao) { … } static PyObject * arrayiter_next(arrayiterobject *it) { … } static void arrayiter_dealloc(arrayiterobject *it) { … } static int arrayiter_traverse(arrayiterobject *it, visitproc visit, void *arg) { … } /*[clinic input] array.arrayiterator.__reduce__ cls: defining_class / Return state information for pickling. [clinic start generated code]*/ static PyObject * array_arrayiterator___reduce___impl(arrayiterobject *self, PyTypeObject *cls) /*[clinic end generated code: output=4b032417a2c8f5e6 input=ac64e65a87ad452e]*/ { … } /*[clinic input] array.arrayiterator.__setstate__ state: object / Set state information for unpickling. [clinic start generated code]*/ static PyObject * array_arrayiterator___setstate__(arrayiterobject *self, PyObject *state) /*[clinic end generated code: output=397da9904e443cbe input=f47d5ceda19e787b]*/ { … } static PyMethodDef arrayiter_methods[] = …; static PyType_Slot arrayiter_slots[] = …; static PyType_Spec arrayiter_spec = …; /*********************** Install Module **************************/ static int array_traverse(PyObject *module, visitproc visit, void *arg) { … } static int array_clear(PyObject *module) { … } static void array_free(void *module) { … } /* No functions in array module. */ static PyMethodDef a_methods[] = …; #define CREATE_TYPE(module, type, spec) … #define ADD_INTERNED(state, string) … static int array_modexec(PyObject *m) { … } static PyModuleDef_Slot arrayslots[] = …; static struct PyModuleDef arraymodule = …; PyMODINIT_FUNC PyInit_array(void) { … }
Codebase Browser
cpython