Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(408)

Side by Side Diff: third_party/cython/src/Cython/Includes/numpy/__init__.pxd

Issue 385073004: Adding cython v0.20.2 in third-party. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Reference cython dev list thread. Created 6 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
OLDNEW
(Empty)
1 # NumPy static imports for Cython
2 #
3 # If any of the PyArray_* functions are called, import_array must be
4 # called first.
5 #
6 # This also defines backwards-compatability buffer acquisition
7 # code for use in Python 2.x (or Python <= 2.5 when NumPy starts
8 # implementing PEP-3118 directly).
9 #
10 # Because of laziness, the format string of the buffer is statically
11 # allocated. Increase the size if this is not enough, or submit a
12 # patch to do this properly.
13 #
14 # Author: Dag Sverre Seljebotn
15 #
16
17 DEF _buffer_format_string_len = 255
18
19 cimport cpython.buffer as pybuf
20 from cpython.ref cimport Py_INCREF, Py_XDECREF
21 from cpython.object cimport PyObject
22 from cpython.type cimport type
23 cimport libc.stdlib as stdlib
24 cimport libc.stdio as stdio
25
26 cdef extern from "Python.h":
27 ctypedef int Py_intptr_t
28
29 cdef extern from "numpy/arrayobject.h":
30 ctypedef Py_intptr_t npy_intp
31 ctypedef size_t npy_uintp
32
33 cdef enum NPY_TYPES:
34 NPY_BOOL
35 NPY_BYTE
36 NPY_UBYTE
37 NPY_SHORT
38 NPY_USHORT
39 NPY_INT
40 NPY_UINT
41 NPY_LONG
42 NPY_ULONG
43 NPY_LONGLONG
44 NPY_ULONGLONG
45 NPY_FLOAT
46 NPY_DOUBLE
47 NPY_LONGDOUBLE
48 NPY_CFLOAT
49 NPY_CDOUBLE
50 NPY_CLONGDOUBLE
51 NPY_OBJECT
52 NPY_STRING
53 NPY_UNICODE
54 NPY_VOID
55 NPY_NTYPES
56 NPY_NOTYPE
57
58 NPY_INT8
59 NPY_INT16
60 NPY_INT32
61 NPY_INT64
62 NPY_INT128
63 NPY_INT256
64 NPY_UINT8
65 NPY_UINT16
66 NPY_UINT32
67 NPY_UINT64
68 NPY_UINT128
69 NPY_UINT256
70 NPY_FLOAT16
71 NPY_FLOAT32
72 NPY_FLOAT64
73 NPY_FLOAT80
74 NPY_FLOAT96
75 NPY_FLOAT128
76 NPY_FLOAT256
77 NPY_COMPLEX32
78 NPY_COMPLEX64
79 NPY_COMPLEX128
80 NPY_COMPLEX160
81 NPY_COMPLEX192
82 NPY_COMPLEX256
83 NPY_COMPLEX512
84
85 NPY_INTP
86
87 ctypedef enum NPY_ORDER:
88 NPY_ANYORDER
89 NPY_CORDER
90 NPY_FORTRANORDER
91
92 ctypedef enum NPY_CLIPMODE:
93 NPY_CLIP
94 NPY_WRAP
95 NPY_RAISE
96
97 ctypedef enum NPY_SCALARKIND:
98 NPY_NOSCALAR,
99 NPY_BOOL_SCALAR,
100 NPY_INTPOS_SCALAR,
101 NPY_INTNEG_SCALAR,
102 NPY_FLOAT_SCALAR,
103 NPY_COMPLEX_SCALAR,
104 NPY_OBJECT_SCALAR
105
106 ctypedef enum NPY_SORTKIND:
107 NPY_QUICKSORT
108 NPY_HEAPSORT
109 NPY_MERGESORT
110
111 ctypedef enum NPY_SEARCHSIDE:
112 NPY_SEARCHLEFT
113 NPY_SEARCHRIGHT
114
115 enum:
116 NPY_C_CONTIGUOUS
117 NPY_F_CONTIGUOUS
118 NPY_CONTIGUOUS
119 NPY_FORTRAN
120 NPY_OWNDATA
121 NPY_FORCECAST
122 NPY_ENSURECOPY
123 NPY_ENSUREARRAY
124 NPY_ELEMENTSTRIDES
125 NPY_ALIGNED
126 NPY_NOTSWAPPED
127 NPY_WRITEABLE
128 NPY_UPDATEIFCOPY
129 NPY_ARR_HAS_DESCR
130
131 NPY_BEHAVED
132 NPY_BEHAVED_NS
133 NPY_CARRAY
134 NPY_CARRAY_RO
135 NPY_FARRAY
136 NPY_FARRAY_RO
137 NPY_DEFAULT
138
139 NPY_IN_ARRAY
140 NPY_OUT_ARRAY
141 NPY_INOUT_ARRAY
142 NPY_IN_FARRAY
143 NPY_OUT_FARRAY
144 NPY_INOUT_FARRAY
145
146 NPY_UPDATE_ALL
147
148 cdef enum:
149 NPY_MAXDIMS
150
151 npy_intp NPY_MAX_ELSIZE
152
153 ctypedef void (*PyArray_VectorUnaryFunc)(void *, void *, npy_intp, void *, void *)
154
155 ctypedef class numpy.dtype [object PyArray_Descr]:
156 # Use PyDataType_* macros when possible, however there are no macros
157 # for accessing some of the fields, so some are defined. Please
158 # ask on cython-dev if you need more.
159 cdef int type_num
160 cdef int itemsize "elsize"
161 cdef char byteorder
162 cdef object fields
163 cdef tuple names
164
165 ctypedef extern class numpy.flatiter [object PyArrayIterObject]:
166 # Use through macros
167 pass
168
169 ctypedef extern class numpy.broadcast [object PyArrayMultiIterObject]:
170 # Use through macros
171 pass
172
173 ctypedef struct PyArrayObject:
174 # For use in situations where ndarray can't replace PyArrayObject*,
175 # like PyArrayObject**.
176 pass
177
178 ctypedef class numpy.ndarray [object PyArrayObject]:
179 cdef __cythonbufferdefaults__ = {"mode": "strided"}
180
181 cdef:
182 # Only taking a few of the most commonly used and stable fields.
183 # One should use PyArray_* macros instead to access the C fields.
184 char *data
185 int ndim "nd"
186 npy_intp *shape "dimensions"
187 npy_intp *strides
188 dtype descr
189 PyObject* base
190
191 # Note: This syntax (function definition in pxd files) is an
192 # experimental exception made for __getbuffer__ and __releasebuffer__
193 # -- the details of this may change.
194 def __getbuffer__(ndarray self, Py_buffer* info, int flags):
195 # This implementation of getbuffer is geared towards Cython
196 # requirements, and does not yet fullfill the PEP.
197 # In particular strided access is always provided regardless
198 # of flags
199
200 if info == NULL: return
201
202 cdef int copy_shape, i, ndim
203 cdef int endian_detector = 1
204 cdef bint little_endian = ((<char*>&endian_detector)[0] != 0)
205
206 ndim = PyArray_NDIM(self)
207
208 if sizeof(npy_intp) != sizeof(Py_ssize_t):
209 copy_shape = 1
210 else:
211 copy_shape = 0
212
213 if ((flags & pybuf.PyBUF_C_CONTIGUOUS == pybuf.PyBUF_C_CONTIGUOUS)
214 and not PyArray_CHKFLAGS(self, NPY_C_CONTIGUOUS)):
215 raise ValueError(u"ndarray is not C contiguous")
216
217 if ((flags & pybuf.PyBUF_F_CONTIGUOUS == pybuf.PyBUF_F_CONTIGUOUS)
218 and not PyArray_CHKFLAGS(self, NPY_F_CONTIGUOUS)):
219 raise ValueError(u"ndarray is not Fortran contiguous")
220
221 info.buf = PyArray_DATA(self)
222 info.ndim = ndim
223 if copy_shape:
224 # Allocate new buffer for strides and shape info.
225 # This is allocated as one block, strides first.
226 info.strides = <Py_ssize_t*>stdlib.malloc(sizeof(Py_ssize_t) * < size_t>ndim * 2)
227 info.shape = info.strides + ndim
228 for i in range(ndim):
229 info.strides[i] = PyArray_STRIDES(self)[i]
230 info.shape[i] = PyArray_DIMS(self)[i]
231 else:
232 info.strides = <Py_ssize_t*>PyArray_STRIDES(self)
233 info.shape = <Py_ssize_t*>PyArray_DIMS(self)
234 info.suboffsets = NULL
235 info.itemsize = PyArray_ITEMSIZE(self)
236 info.readonly = not PyArray_ISWRITEABLE(self)
237
238 cdef int t
239 cdef char* f = NULL
240 cdef dtype descr = self.descr
241 cdef list stack
242 cdef int offset
243
244 cdef bint hasfields = PyDataType_HASFIELDS(descr)
245
246 if not hasfields and not copy_shape:
247 # do not call releasebuffer
248 info.obj = None
249 else:
250 # need to call releasebuffer
251 info.obj = self
252
253 if not hasfields:
254 t = descr.type_num
255 if ((descr.byteorder == c'>' and little_endian) or
256 (descr.byteorder == c'<' and not little_endian)):
257 raise ValueError(u"Non-native byte order not supported")
258 if t == NPY_BYTE: f = "b"
259 elif t == NPY_UBYTE: f = "B"
260 elif t == NPY_SHORT: f = "h"
261 elif t == NPY_USHORT: f = "H"
262 elif t == NPY_INT: f = "i"
263 elif t == NPY_UINT: f = "I"
264 elif t == NPY_LONG: f = "l"
265 elif t == NPY_ULONG: f = "L"
266 elif t == NPY_LONGLONG: f = "q"
267 elif t == NPY_ULONGLONG: f = "Q"
268 elif t == NPY_FLOAT: f = "f"
269 elif t == NPY_DOUBLE: f = "d"
270 elif t == NPY_LONGDOUBLE: f = "g"
271 elif t == NPY_CFLOAT: f = "Zf"
272 elif t == NPY_CDOUBLE: f = "Zd"
273 elif t == NPY_CLONGDOUBLE: f = "Zg"
274 elif t == NPY_OBJECT: f = "O"
275 else:
276 raise ValueError(u"unknown dtype code in numpy.pxd (%d)" % t )
277 info.format = f
278 return
279 else:
280 info.format = <char*>stdlib.malloc(_buffer_format_string_len)
281 info.format[0] = c'^' # Native data types, manual alignment
282 offset = 0
283 f = _util_dtypestring(descr, info.format + 1,
284 info.format + _buffer_format_string_len,
285 &offset)
286 f[0] = c'\0' # Terminate format string
287
288 def __releasebuffer__(ndarray self, Py_buffer* info):
289 if PyArray_HASFIELDS(self):
290 stdlib.free(info.format)
291 if sizeof(npy_intp) != sizeof(Py_ssize_t):
292 stdlib.free(info.strides)
293 # info.shape was stored after info.strides in the same block
294
295
296 ctypedef unsigned char npy_bool
297
298 ctypedef signed char npy_byte
299 ctypedef signed short npy_short
300 ctypedef signed int npy_int
301 ctypedef signed long npy_long
302 ctypedef signed long long npy_longlong
303
304 ctypedef unsigned char npy_ubyte
305 ctypedef unsigned short npy_ushort
306 ctypedef unsigned int npy_uint
307 ctypedef unsigned long npy_ulong
308 ctypedef unsigned long long npy_ulonglong
309
310 ctypedef float npy_float
311 ctypedef double npy_double
312 ctypedef long double npy_longdouble
313
314 ctypedef signed char npy_int8
315 ctypedef signed short npy_int16
316 ctypedef signed int npy_int32
317 ctypedef signed long long npy_int64
318 ctypedef signed long long npy_int96
319 ctypedef signed long long npy_int128
320
321 ctypedef unsigned char npy_uint8
322 ctypedef unsigned short npy_uint16
323 ctypedef unsigned int npy_uint32
324 ctypedef unsigned long long npy_uint64
325 ctypedef unsigned long long npy_uint96
326 ctypedef unsigned long long npy_uint128
327
328 ctypedef float npy_float32
329 ctypedef double npy_float64
330 ctypedef long double npy_float80
331 ctypedef long double npy_float96
332 ctypedef long double npy_float128
333
334 ctypedef struct npy_cfloat:
335 double real
336 double imag
337
338 ctypedef struct npy_cdouble:
339 double real
340 double imag
341
342 ctypedef struct npy_clongdouble:
343 double real
344 double imag
345
346 ctypedef struct npy_complex64:
347 double real
348 double imag
349
350 ctypedef struct npy_complex128:
351 double real
352 double imag
353
354 ctypedef struct npy_complex160:
355 double real
356 double imag
357
358 ctypedef struct npy_complex192:
359 double real
360 double imag
361
362 ctypedef struct npy_complex256:
363 double real
364 double imag
365
366 ctypedef struct PyArray_Dims:
367 npy_intp *ptr
368 int len
369
370 void import_array()
371
372 #
373 # Macros from ndarrayobject.h
374 #
375 bint PyArray_CHKFLAGS(ndarray m, int flags)
376 bint PyArray_ISCONTIGUOUS(ndarray m)
377 bint PyArray_ISWRITEABLE(ndarray m)
378 bint PyArray_ISALIGNED(ndarray m)
379
380 int PyArray_NDIM(ndarray)
381 bint PyArray_ISONESEGMENT(ndarray)
382 bint PyArray_ISFORTRAN(ndarray)
383 int PyArray_FORTRANIF(ndarray)
384
385 void* PyArray_DATA(ndarray)
386 char* PyArray_BYTES(ndarray)
387 npy_intp* PyArray_DIMS(ndarray)
388 npy_intp* PyArray_STRIDES(ndarray)
389 npy_intp PyArray_DIM(ndarray, size_t)
390 npy_intp PyArray_STRIDE(ndarray, size_t)
391
392 # object PyArray_BASE(ndarray) wrong refcount semantics
393 # dtype PyArray_DESCR(ndarray) wrong refcount semantics
394 int PyArray_FLAGS(ndarray)
395 npy_intp PyArray_ITEMSIZE(ndarray)
396 int PyArray_TYPE(ndarray arr)
397
398 object PyArray_GETITEM(ndarray arr, void *itemptr)
399 int PyArray_SETITEM(ndarray arr, void *itemptr, object obj)
400
401 bint PyTypeNum_ISBOOL(int)
402 bint PyTypeNum_ISUNSIGNED(int)
403 bint PyTypeNum_ISSIGNED(int)
404 bint PyTypeNum_ISINTEGER(int)
405 bint PyTypeNum_ISFLOAT(int)
406 bint PyTypeNum_ISNUMBER(int)
407 bint PyTypeNum_ISSTRING(int)
408 bint PyTypeNum_ISCOMPLEX(int)
409 bint PyTypeNum_ISPYTHON(int)
410 bint PyTypeNum_ISFLEXIBLE(int)
411 bint PyTypeNum_ISUSERDEF(int)
412 bint PyTypeNum_ISEXTENDED(int)
413 bint PyTypeNum_ISOBJECT(int)
414
415 bint PyDataType_ISBOOL(dtype)
416 bint PyDataType_ISUNSIGNED(dtype)
417 bint PyDataType_ISSIGNED(dtype)
418 bint PyDataType_ISINTEGER(dtype)
419 bint PyDataType_ISFLOAT(dtype)
420 bint PyDataType_ISNUMBER(dtype)
421 bint PyDataType_ISSTRING(dtype)
422 bint PyDataType_ISCOMPLEX(dtype)
423 bint PyDataType_ISPYTHON(dtype)
424 bint PyDataType_ISFLEXIBLE(dtype)
425 bint PyDataType_ISUSERDEF(dtype)
426 bint PyDataType_ISEXTENDED(dtype)
427 bint PyDataType_ISOBJECT(dtype)
428 bint PyDataType_HASFIELDS(dtype)
429
430 bint PyArray_ISBOOL(ndarray)
431 bint PyArray_ISUNSIGNED(ndarray)
432 bint PyArray_ISSIGNED(ndarray)
433 bint PyArray_ISINTEGER(ndarray)
434 bint PyArray_ISFLOAT(ndarray)
435 bint PyArray_ISNUMBER(ndarray)
436 bint PyArray_ISSTRING(ndarray)
437 bint PyArray_ISCOMPLEX(ndarray)
438 bint PyArray_ISPYTHON(ndarray)
439 bint PyArray_ISFLEXIBLE(ndarray)
440 bint PyArray_ISUSERDEF(ndarray)
441 bint PyArray_ISEXTENDED(ndarray)
442 bint PyArray_ISOBJECT(ndarray)
443 bint PyArray_HASFIELDS(ndarray)
444
445 bint PyArray_ISVARIABLE(ndarray)
446
447 bint PyArray_SAFEALIGNEDCOPY(ndarray)
448 bint PyArray_ISNBO(char) # works on ndarray.byteorder
449 bint PyArray_IsNativeByteOrder(char) # works on ndarray.byteorder
450 bint PyArray_ISNOTSWAPPED(ndarray)
451 bint PyArray_ISBYTESWAPPED(ndarray)
452
453 bint PyArray_FLAGSWAP(ndarray, int)
454
455 bint PyArray_ISCARRAY(ndarray)
456 bint PyArray_ISCARRAY_RO(ndarray)
457 bint PyArray_ISFARRAY(ndarray)
458 bint PyArray_ISFARRAY_RO(ndarray)
459 bint PyArray_ISBEHAVED(ndarray)
460 bint PyArray_ISBEHAVED_RO(ndarray)
461
462
463 bint PyDataType_ISNOTSWAPPED(dtype)
464 bint PyDataType_ISBYTESWAPPED(dtype)
465
466 bint PyArray_DescrCheck(object)
467
468 bint PyArray_Check(object)
469 bint PyArray_CheckExact(object)
470
471 # Cannot be supported due to out arg:
472 # bint PyArray_HasArrayInterfaceType(object, dtype, object, object&)
473 # bint PyArray_HasArrayInterface(op, out)
474
475
476 bint PyArray_IsZeroDim(object)
477 # Cannot be supported due to ## ## in macro:
478 # bint PyArray_IsScalar(object, verbatim work)
479 bint PyArray_CheckScalar(object)
480 bint PyArray_IsPythonNumber(object)
481 bint PyArray_IsPythonScalar(object)
482 bint PyArray_IsAnyScalar(object)
483 bint PyArray_CheckAnyScalar(object)
484 ndarray PyArray_GETCONTIGUOUS(ndarray)
485 bint PyArray_SAMESHAPE(ndarray, ndarray)
486 npy_intp PyArray_SIZE(ndarray)
487 npy_intp PyArray_NBYTES(ndarray)
488
489 object PyArray_FROM_O(object)
490 object PyArray_FROM_OF(object m, int flags)
491 object PyArray_FROM_OT(object m, int type)
492 object PyArray_FROM_OTF(object m, int type, int flags)
493 object PyArray_FROMANY(object m, int type, int min, int max, int flags)
494 object PyArray_ZEROS(int nd, npy_intp* dims, int type, int fortran)
495 object PyArray_EMPTY(int nd, npy_intp* dims, int type, int fortran)
496 void PyArray_FILLWBYTE(object, int val)
497 npy_intp PyArray_REFCOUNT(object)
498 object PyArray_ContiguousFromAny(op, int, int min_depth, int max_depth)
499 unsigned char PyArray_EquivArrTypes(ndarray a1, ndarray a2)
500 bint PyArray_EquivByteorders(int b1, int b2)
501 object PyArray_SimpleNew(int nd, npy_intp* dims, int typenum)
502 object PyArray_SimpleNewFromData(int nd, npy_intp* dims, int typenum, void* data)
503 #object PyArray_SimpleNewFromDescr(int nd, npy_intp* dims, dtype descr)
504 object PyArray_ToScalar(void* data, ndarray arr)
505
506 void* PyArray_GETPTR1(ndarray m, npy_intp i)
507 void* PyArray_GETPTR2(ndarray m, npy_intp i, npy_intp j)
508 void* PyArray_GETPTR3(ndarray m, npy_intp i, npy_intp j, npy_intp k)
509 void* PyArray_GETPTR4(ndarray m, npy_intp i, npy_intp j, npy_intp k, npy_int p l)
510
511 void PyArray_XDECREF_ERR(ndarray)
512 # Cannot be supported due to out arg
513 # void PyArray_DESCR_REPLACE(descr)
514
515
516 object PyArray_Copy(ndarray)
517 object PyArray_FromObject(object op, int type, int min_depth, int max_depth)
518 object PyArray_ContiguousFromObject(object op, int type, int min_depth, int max_depth)
519 object PyArray_CopyFromObject(object op, int type, int min_depth, int max_de pth)
520
521 object PyArray_Cast(ndarray mp, int type_num)
522 object PyArray_Take(ndarray ap, object items, int axis)
523 object PyArray_Put(ndarray ap, object items, object values)
524
525 void PyArray_ITER_RESET(flatiter it) nogil
526 void PyArray_ITER_NEXT(flatiter it) nogil
527 void PyArray_ITER_GOTO(flatiter it, npy_intp* destination) nogil
528 void PyArray_ITER_GOTO1D(flatiter it, npy_intp ind) nogil
529 void* PyArray_ITER_DATA(flatiter it) nogil
530 bint PyArray_ITER_NOTDONE(flatiter it) nogil
531
532 void PyArray_MultiIter_RESET(broadcast multi) nogil
533 void PyArray_MultiIter_NEXT(broadcast multi) nogil
534 void PyArray_MultiIter_GOTO(broadcast multi, npy_intp dest) nogil
535 void PyArray_MultiIter_GOTO1D(broadcast multi, npy_intp ind) nogil
536 void* PyArray_MultiIter_DATA(broadcast multi, npy_intp i) nogil
537 void PyArray_MultiIter_NEXTi(broadcast multi, npy_intp i) nogil
538 bint PyArray_MultiIter_NOTDONE(broadcast multi) nogil
539
540 # Functions from __multiarray_api.h
541
542 # Functions taking dtype and returning object/ndarray are disabled
543 # for now as they steal dtype references. I'm conservative and disable
544 # more than is probably needed until it can be checked further.
545 int PyArray_SetNumericOps (object)
546 object PyArray_GetNumericOps ()
547 int PyArray_INCREF (ndarray)
548 int PyArray_XDECREF (ndarray)
549 void PyArray_SetStringFunction (object, int)
550 dtype PyArray_DescrFromType (int)
551 object PyArray_TypeObjectFromType (int)
552 char * PyArray_Zero (ndarray)
553 char * PyArray_One (ndarray)
554 #object PyArray_CastToType (ndarray, dtype, int)
555 int PyArray_CastTo (ndarray, ndarray)
556 int PyArray_CastAnyTo (ndarray, ndarray)
557 int PyArray_CanCastSafely (int, int)
558 npy_bool PyArray_CanCastTo (dtype, dtype)
559 int PyArray_ObjectType (object, int)
560 dtype PyArray_DescrFromObject (object, dtype)
561 #ndarray* PyArray_ConvertToCommonType (object, int *)
562 dtype PyArray_DescrFromScalar (object)
563 dtype PyArray_DescrFromTypeObject (object)
564 npy_intp PyArray_Size (object)
565 #object PyArray_Scalar (void *, dtype, object)
566 #object PyArray_FromScalar (object, dtype)
567 void PyArray_ScalarAsCtype (object, void *)
568 #int PyArray_CastScalarToCtype (object, void *, dtype)
569 #int PyArray_CastScalarDirect (object, dtype, void *, int)
570 object PyArray_ScalarFromObject (object)
571 #PyArray_VectorUnaryFunc * PyArray_GetCastFunc (dtype, int)
572 object PyArray_FromDims (int, int *, int)
573 #object PyArray_FromDimsAndDataAndDescr (int, int *, dtype, char *)
574 #object PyArray_FromAny (object, dtype, int, int, int, object)
575 object PyArray_EnsureArray (object)
576 object PyArray_EnsureAnyArray (object)
577 #object PyArray_FromFile (stdio.FILE *, dtype, npy_intp, char *)
578 #object PyArray_FromString (char *, npy_intp, dtype, npy_intp, char *)
579 #object PyArray_FromBuffer (object, dtype, npy_intp, npy_intp)
580 #object PyArray_FromIter (object, dtype, npy_intp)
581 object PyArray_Return (ndarray)
582 #object PyArray_GetField (ndarray, dtype, int)
583 #int PyArray_SetField (ndarray, dtype, int, object)
584 object PyArray_Byteswap (ndarray, npy_bool)
585 object PyArray_Resize (ndarray, PyArray_Dims *, int, NPY_ORDER)
586 int PyArray_MoveInto (ndarray, ndarray)
587 int PyArray_CopyInto (ndarray, ndarray)
588 int PyArray_CopyAnyInto (ndarray, ndarray)
589 int PyArray_CopyObject (ndarray, object)
590 object PyArray_NewCopy (ndarray, NPY_ORDER)
591 object PyArray_ToList (ndarray)
592 object PyArray_ToString (ndarray, NPY_ORDER)
593 int PyArray_ToFile (ndarray, stdio.FILE *, char *, char *)
594 int PyArray_Dump (object, object, int)
595 object PyArray_Dumps (object, int)
596 int PyArray_ValidType (int)
597 void PyArray_UpdateFlags (ndarray, int)
598 object PyArray_New (type, int, npy_intp *, int, npy_intp *, void *, int, int , object)
599 #object PyArray_NewFromDescr (type, dtype, int, npy_intp *, npy_intp *, void *, int, object)
600 #dtype PyArray_DescrNew (dtype)
601 dtype PyArray_DescrNewFromType (int)
602 double PyArray_GetPriority (object, double)
603 object PyArray_IterNew (object)
604 object PyArray_MultiIterNew (int, ...)
605
606 int PyArray_PyIntAsInt (object)
607 npy_intp PyArray_PyIntAsIntp (object)
608 int PyArray_Broadcast (broadcast)
609 void PyArray_FillObjectArray (ndarray, object)
610 int PyArray_FillWithScalar (ndarray, object)
611 npy_bool PyArray_CheckStrides (int, int, npy_intp, npy_intp, npy_intp *, npy _intp *)
612 dtype PyArray_DescrNewByteorder (dtype, char)
613 object PyArray_IterAllButAxis (object, int *)
614 #object PyArray_CheckFromAny (object, dtype, int, int, int, object)
615 #object PyArray_FromArray (ndarray, dtype, int)
616 object PyArray_FromInterface (object)
617 object PyArray_FromStructInterface (object)
618 #object PyArray_FromArrayAttr (object, dtype, object)
619 #NPY_SCALARKIND PyArray_ScalarKind (int, ndarray*)
620 int PyArray_CanCoerceScalar (int, int, NPY_SCALARKIND)
621 object PyArray_NewFlagsObject (object)
622 npy_bool PyArray_CanCastScalar (type, type)
623 #int PyArray_CompareUCS4 (npy_ucs4 *, npy_ucs4 *, register size_t)
624 int PyArray_RemoveSmallest (broadcast)
625 int PyArray_ElementStrides (object)
626 void PyArray_Item_INCREF (char *, dtype)
627 void PyArray_Item_XDECREF (char *, dtype)
628 object PyArray_FieldNames (object)
629 object PyArray_Transpose (ndarray, PyArray_Dims *)
630 object PyArray_TakeFrom (ndarray, object, int, ndarray, NPY_CLIPMODE)
631 object PyArray_PutTo (ndarray, object, object, NPY_CLIPMODE)
632 object PyArray_PutMask (ndarray, object, object)
633 object PyArray_Repeat (ndarray, object, int)
634 object PyArray_Choose (ndarray, object, ndarray, NPY_CLIPMODE)
635 int PyArray_Sort (ndarray, int, NPY_SORTKIND)
636 object PyArray_ArgSort (ndarray, int, NPY_SORTKIND)
637 object PyArray_SearchSorted (ndarray, object, NPY_SEARCHSIDE)
638 object PyArray_ArgMax (ndarray, int, ndarray)
639 object PyArray_ArgMin (ndarray, int, ndarray)
640 object PyArray_Reshape (ndarray, object)
641 object PyArray_Newshape (ndarray, PyArray_Dims *, NPY_ORDER)
642 object PyArray_Squeeze (ndarray)
643 #object PyArray_View (ndarray, dtype, type)
644 object PyArray_SwapAxes (ndarray, int, int)
645 object PyArray_Max (ndarray, int, ndarray)
646 object PyArray_Min (ndarray, int, ndarray)
647 object PyArray_Ptp (ndarray, int, ndarray)
648 object PyArray_Mean (ndarray, int, int, ndarray)
649 object PyArray_Trace (ndarray, int, int, int, int, ndarray)
650 object PyArray_Diagonal (ndarray, int, int, int)
651 object PyArray_Clip (ndarray, object, object, ndarray)
652 object PyArray_Conjugate (ndarray, ndarray)
653 object PyArray_Nonzero (ndarray)
654 object PyArray_Std (ndarray, int, int, ndarray, int)
655 object PyArray_Sum (ndarray, int, int, ndarray)
656 object PyArray_CumSum (ndarray, int, int, ndarray)
657 object PyArray_Prod (ndarray, int, int, ndarray)
658 object PyArray_CumProd (ndarray, int, int, ndarray)
659 object PyArray_All (ndarray, int, ndarray)
660 object PyArray_Any (ndarray, int, ndarray)
661 object PyArray_Compress (ndarray, object, int, ndarray)
662 object PyArray_Flatten (ndarray, NPY_ORDER)
663 object PyArray_Ravel (ndarray, NPY_ORDER)
664 npy_intp PyArray_MultiplyList (npy_intp *, int)
665 int PyArray_MultiplyIntList (int *, int)
666 void * PyArray_GetPtr (ndarray, npy_intp*)
667 int PyArray_CompareLists (npy_intp *, npy_intp *, int)
668 #int PyArray_AsCArray (object*, void *, npy_intp *, int, dtype)
669 #int PyArray_As1D (object*, char **, int *, int)
670 #int PyArray_As2D (object*, char ***, int *, int *, int)
671 int PyArray_Free (object, void *)
672 #int PyArray_Converter (object, object*)
673 int PyArray_IntpFromSequence (object, npy_intp *, int)
674 object PyArray_Concatenate (object, int)
675 object PyArray_InnerProduct (object, object)
676 object PyArray_MatrixProduct (object, object)
677 object PyArray_CopyAndTranspose (object)
678 object PyArray_Correlate (object, object, int)
679 int PyArray_TypestrConvert (int, int)
680 #int PyArray_DescrConverter (object, dtype*)
681 #int PyArray_DescrConverter2 (object, dtype*)
682 int PyArray_IntpConverter (object, PyArray_Dims *)
683 #int PyArray_BufferConverter (object, chunk)
684 int PyArray_AxisConverter (object, int *)
685 int PyArray_BoolConverter (object, npy_bool *)
686 int PyArray_ByteorderConverter (object, char *)
687 int PyArray_OrderConverter (object, NPY_ORDER *)
688 unsigned char PyArray_EquivTypes (dtype, dtype)
689 #object PyArray_Zeros (int, npy_intp *, dtype, int)
690 #object PyArray_Empty (int, npy_intp *, dtype, int)
691 object PyArray_Where (object, object, object)
692 object PyArray_Arange (double, double, double, int)
693 #object PyArray_ArangeObj (object, object, object, dtype)
694 int PyArray_SortkindConverter (object, NPY_SORTKIND *)
695 object PyArray_LexSort (object, int)
696 object PyArray_Round (ndarray, int, ndarray)
697 unsigned char PyArray_EquivTypenums (int, int)
698 int PyArray_RegisterDataType (dtype)
699 int PyArray_RegisterCastFunc (dtype, int, PyArray_VectorUnaryFunc *)
700 int PyArray_RegisterCanCast (dtype, int, NPY_SCALARKIND)
701 #void PyArray_InitArrFuncs (PyArray_ArrFuncs *)
702 object PyArray_IntTupleFromIntp (int, npy_intp *)
703 int PyArray_TypeNumFromName (char *)
704 int PyArray_ClipmodeConverter (object, NPY_CLIPMODE *)
705 #int PyArray_OutputConverter (object, ndarray*)
706 object PyArray_BroadcastToShape (object, npy_intp *, int)
707 void _PyArray_SigintHandler (int)
708 void* _PyArray_GetSigintBuf ()
709 #int PyArray_DescrAlignConverter (object, dtype*)
710 #int PyArray_DescrAlignConverter2 (object, dtype*)
711 int PyArray_SearchsideConverter (object, void *)
712 object PyArray_CheckAxis (ndarray, int *, int)
713 npy_intp PyArray_OverflowMultiplyList (npy_intp *, int)
714 int PyArray_CompareString (char *, char *, size_t)
715
716
717 # Typedefs that matches the runtime dtype objects in
718 # the numpy module.
719
720 # The ones that are commented out needs an IFDEF function
721 # in Cython to enable them only on the right systems.
722
723 ctypedef npy_int8 int8_t
724 ctypedef npy_int16 int16_t
725 ctypedef npy_int32 int32_t
726 ctypedef npy_int64 int64_t
727 #ctypedef npy_int96 int96_t
728 #ctypedef npy_int128 int128_t
729
730 ctypedef npy_uint8 uint8_t
731 ctypedef npy_uint16 uint16_t
732 ctypedef npy_uint32 uint32_t
733 ctypedef npy_uint64 uint64_t
734 #ctypedef npy_uint96 uint96_t
735 #ctypedef npy_uint128 uint128_t
736
737 ctypedef npy_float32 float32_t
738 ctypedef npy_float64 float64_t
739 #ctypedef npy_float80 float80_t
740 #ctypedef npy_float128 float128_t
741
742 ctypedef float complex complex64_t
743 ctypedef double complex complex128_t
744
745 # The int types are mapped a bit surprising --
746 # numpy.int corresponds to 'l' and numpy.long to 'q'
747 ctypedef npy_long int_t
748 ctypedef npy_longlong long_t
749 ctypedef npy_longlong longlong_t
750
751 ctypedef npy_ulong uint_t
752 ctypedef npy_ulonglong ulong_t
753 ctypedef npy_ulonglong ulonglong_t
754
755 ctypedef npy_intp intp_t
756 ctypedef npy_uintp uintp_t
757
758 ctypedef npy_double float_t
759 ctypedef npy_double double_t
760 ctypedef npy_longdouble longdouble_t
761
762 ctypedef npy_cfloat cfloat_t
763 ctypedef npy_cdouble cdouble_t
764 ctypedef npy_clongdouble clongdouble_t
765
766 ctypedef npy_cdouble complex_t
767
768 cdef inline object PyArray_MultiIterNew1(a):
769 return PyArray_MultiIterNew(1, <void*>a)
770
771 cdef inline object PyArray_MultiIterNew2(a, b):
772 return PyArray_MultiIterNew(2, <void*>a, <void*>b)
773
774 cdef inline object PyArray_MultiIterNew3(a, b, c):
775 return PyArray_MultiIterNew(3, <void*>a, <void*>b, <void*> c)
776
777 cdef inline object PyArray_MultiIterNew4(a, b, c, d):
778 return PyArray_MultiIterNew(4, <void*>a, <void*>b, <void*>c, <void*> d)
779
780 cdef inline object PyArray_MultiIterNew5(a, b, c, d, e):
781 return PyArray_MultiIterNew(5, <void*>a, <void*>b, <void*>c, <void*> d, <voi d*> e)
782
783 cdef inline char* _util_dtypestring(dtype descr, char* f, char* end, int* offset ) except NULL:
784 # Recursive utility function used in __getbuffer__ to get format
785 # string. The new location in the format string is returned.
786
787 cdef dtype child
788 cdef int delta_offset
789 cdef tuple i
790 cdef int endian_detector = 1
791 cdef bint little_endian = ((<char*>&endian_detector)[0] != 0)
792 cdef tuple fields
793
794 for childname in descr.names:
795 fields = descr.fields[childname]
796 child, new_offset = fields
797
798 if (end - f) - <int>(new_offset - offset[0]) < 15:
799 raise RuntimeError(u"Format string allocated too short, see comment in numpy.pxd")
800
801 if ((child.byteorder == c'>' and little_endian) or
802 (child.byteorder == c'<' and not little_endian)):
803 raise ValueError(u"Non-native byte order not supported")
804 # One could encode it in the format string and have Cython
805 # complain instead, BUT: < and > in format strings also imply
806 # standardized sizes for datatypes, and we rely on native in
807 # order to avoid reencoding data types based on their size.
808 #
809 # A proper PEP 3118 exporter for other clients than Cython
810 # must deal properly with this!
811
812 # Output padding bytes
813 while offset[0] < new_offset:
814 f[0] = 120 # "x"; pad byte
815 f += 1
816 offset[0] += 1
817
818 offset[0] += child.itemsize
819
820 if not PyDataType_HASFIELDS(child):
821 t = child.type_num
822 if end - f < 5:
823 raise RuntimeError(u"Format string allocated too short.")
824
825 # Until ticket #99 is fixed, use integers to avoid warnings
826 if t == NPY_BYTE: f[0] = 98 #"b"
827 elif t == NPY_UBYTE: f[0] = 66 #"B"
828 elif t == NPY_SHORT: f[0] = 104 #"h"
829 elif t == NPY_USHORT: f[0] = 72 #"H"
830 elif t == NPY_INT: f[0] = 105 #"i"
831 elif t == NPY_UINT: f[0] = 73 #"I"
832 elif t == NPY_LONG: f[0] = 108 #"l"
833 elif t == NPY_ULONG: f[0] = 76 #"L"
834 elif t == NPY_LONGLONG: f[0] = 113 #"q"
835 elif t == NPY_ULONGLONG: f[0] = 81 #"Q"
836 elif t == NPY_FLOAT: f[0] = 102 #"f"
837 elif t == NPY_DOUBLE: f[0] = 100 #"d"
838 elif t == NPY_LONGDOUBLE: f[0] = 103 #"g"
839 elif t == NPY_CFLOAT: f[0] = 90; f[1] = 102; f += 1 # Zf
840 elif t == NPY_CDOUBLE: f[0] = 90; f[1] = 100; f += 1 # Zd
841 elif t == NPY_CLONGDOUBLE: f[0] = 90; f[1] = 103; f += 1 # Zg
842 elif t == NPY_OBJECT: f[0] = 79 #"O"
843 else:
844 raise ValueError(u"unknown dtype code in numpy.pxd (%d)" % t)
845 f += 1
846 else:
847 # Cython ignores struct boundary information ("T{...}"),
848 # so don't output it
849 f = _util_dtypestring(child, f, end, offset)
850 return f
851
852
853 #
854 # ufunc API
855 #
856
857 cdef extern from "numpy/ufuncobject.h":
858
859 ctypedef void (*PyUFuncGenericFunction) (char **, npy_intp *, npy_intp *, vo id *)
860
861 ctypedef extern class numpy.ufunc [object PyUFuncObject]:
862 cdef:
863 int nin, nout, nargs
864 int identity
865 PyUFuncGenericFunction *functions
866 void **data
867 int ntypes
868 int check_return
869 char *name
870 char *types
871 char *doc
872 void *ptr
873 PyObject *obj
874 PyObject *userloops
875
876 cdef enum:
877 PyUFunc_Zero
878 PyUFunc_One
879 PyUFunc_None
880 UFUNC_ERR_IGNORE
881 UFUNC_ERR_WARN
882 UFUNC_ERR_RAISE
883 UFUNC_ERR_CALL
884 UFUNC_ERR_PRINT
885 UFUNC_ERR_LOG
886 UFUNC_MASK_DIVIDEBYZERO
887 UFUNC_MASK_OVERFLOW
888 UFUNC_MASK_UNDERFLOW
889 UFUNC_MASK_INVALID
890 UFUNC_SHIFT_DIVIDEBYZERO
891 UFUNC_SHIFT_OVERFLOW
892 UFUNC_SHIFT_UNDERFLOW
893 UFUNC_SHIFT_INVALID
894 UFUNC_FPE_DIVIDEBYZERO
895 UFUNC_FPE_OVERFLOW
896 UFUNC_FPE_UNDERFLOW
897 UFUNC_FPE_INVALID
898 UFUNC_ERR_DEFAULT
899 UFUNC_ERR_DEFAULT2
900
901 object PyUFunc_FromFuncAndData(PyUFuncGenericFunction *,
902 void **, char *, int, int, int, int, char *, char *, int)
903 int PyUFunc_RegisterLoopForType(ufunc, int,
904 PyUFuncGenericFunction, int *, void *)
905 int PyUFunc_GenericFunction \
906 (ufunc, PyObject *, PyObject *, PyArrayObject **)
907 void PyUFunc_f_f_As_d_d \
908 (char **, npy_intp *, npy_intp *, void *)
909 void PyUFunc_d_d \
910 (char **, npy_intp *, npy_intp *, void *)
911 void PyUFunc_f_f \
912 (char **, npy_intp *, npy_intp *, void *)
913 void PyUFunc_g_g \
914 (char **, npy_intp *, npy_intp *, void *)
915 void PyUFunc_F_F_As_D_D \
916 (char **, npy_intp *, npy_intp *, void *)
917 void PyUFunc_F_F \
918 (char **, npy_intp *, npy_intp *, void *)
919 void PyUFunc_D_D \
920 (char **, npy_intp *, npy_intp *, void *)
921 void PyUFunc_G_G \
922 (char **, npy_intp *, npy_intp *, void *)
923 void PyUFunc_O_O \
924 (char **, npy_intp *, npy_intp *, void *)
925 void PyUFunc_ff_f_As_dd_d \
926 (char **, npy_intp *, npy_intp *, void *)
927 void PyUFunc_ff_f \
928 (char **, npy_intp *, npy_intp *, void *)
929 void PyUFunc_dd_d \
930 (char **, npy_intp *, npy_intp *, void *)
931 void PyUFunc_gg_g \
932 (char **, npy_intp *, npy_intp *, void *)
933 void PyUFunc_FF_F_As_DD_D \
934 (char **, npy_intp *, npy_intp *, void *)
935 void PyUFunc_DD_D \
936 (char **, npy_intp *, npy_intp *, void *)
937 void PyUFunc_FF_F \
938 (char **, npy_intp *, npy_intp *, void *)
939 void PyUFunc_GG_G \
940 (char **, npy_intp *, npy_intp *, void *)
941 void PyUFunc_OO_O \
942 (char **, npy_intp *, npy_intp *, void *)
943 void PyUFunc_O_O_method \
944 (char **, npy_intp *, npy_intp *, void *)
945 void PyUFunc_OO_O_method \
946 (char **, npy_intp *, npy_intp *, void *)
947 void PyUFunc_On_Om \
948 (char **, npy_intp *, npy_intp *, void *)
949 int PyUFunc_GetPyValues \
950 (char *, int *, int *, PyObject **)
951 int PyUFunc_checkfperr \
952 (int, PyObject *, int *)
953 void PyUFunc_clearfperr()
954 int PyUFunc_getfperr()
955 int PyUFunc_handlefperr \
956 (int, PyObject *, int, int *)
957 int PyUFunc_ReplaceLoopBySignature \
958 (ufunc, PyUFuncGenericFunction, int *, PyUFuncGenericFunction *)
959 object PyUFunc_FromFuncAndDataAndSignature \
960 (PyUFuncGenericFunction *, void **, char *, int, int, int,
961 int, char *, char *, int, char *)
962
963 void import_ufunc()
964
965
966 cdef inline void set_array_base(ndarray arr, object base):
967 cdef PyObject* baseptr
968 if base is None:
969 baseptr = NULL
970 else:
971 Py_INCREF(base) # important to do this before decref below!
972 baseptr = <PyObject*>base
973 Py_XDECREF(arr.base)
974 arr.base = baseptr
975
976 cdef inline object get_array_base(ndarray arr):
977 if arr.base is NULL:
978 return None
979 else:
980 return <object>arr.base
OLDNEW
« no previous file with comments | « third_party/cython/src/Cython/Includes/libcpp/vector.pxd ('k') | third_party/cython/src/Cython/Includes/numpy/math.pxd » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698