Lines Matching full:ndim
52 #define ND_SCALAR 0x008 /* scalar: ndim = 0 */
151 base->ndim = 1; in ndbuf_new()
264 if (ndbuf->base.ndim == 0) in init_flags()
470 copy_rec(const Py_ssize_t *shape, Py_ssize_t ndim, Py_ssize_t itemsize, in copy_rec() argument
477 assert(ndim >= 1); in copy_rec()
479 if (ndim == 1) { in copy_rec()
503 copy_rec(shape+1, ndim-1, itemsize, in copy_rec()
517 dest->ndim != src->ndim) in cmp_structure()
520 for (i = 0; i < dest->ndim; i++) { in cmp_structure()
531 ndim and shape. Copying is atomic, the function never fails with
538 assert(dest->ndim > 0); in copy_buffer()
546 if ((dest->suboffsets && dest->suboffsets[dest->ndim-1] >= 0) || in copy_buffer()
547 (src->suboffsets && src->suboffsets[src->ndim-1] >= 0) || in copy_buffer()
548 dest->strides[dest->ndim-1] != dest->itemsize || in copy_buffer()
549 src->strides[src->ndim-1] != src->itemsize) { in copy_buffer()
550 mem = PyMem_Malloc(dest->shape[dest->ndim-1] * dest->itemsize); in copy_buffer()
557 copy_rec(dest->shape, dest->ndim, dest->itemsize, in copy_buffer()
605 for ndim = 0. */
609 const Py_ssize_t *suboffsets, Py_ssize_t ndim, Py_ssize_t itemsize) in unpack_rec() argument
614 assert(ndim >= 0); in unpack_rec()
618 if (ndim == 0) { in unpack_rec()
641 ndim-1, itemsize); in unpack_rec()
679 assert(base->ndim <= 1); in ndarray_as_list()
717 base->ndim, base->itemsize); in ndarray_as_list()
769 | base.ndim | 1 | 1 | OK |
850 /* ndim = len <= ND_MAX_NDIM, so PyMem_New() is actually not needed. */ in seq_as_ssize_array()
889 s = PyMem_Malloc(base->ndim * (sizeof *s)); in strides_from_shape()
897 for (i = 1; i < base->ndim; i++) in strides_from_shape()
901 s[base->ndim-1] = base->itemsize; in strides_from_shape()
902 for (i = base->ndim-2; i >= 0; i--) in strides_from_shape()
928 Py_ssize_t ndim) in verify_structure() argument
933 assert(ndim >= 0); in verify_structure()
935 if (ndim == 0 && (offset < 0 || offset+itemsize > len)) in verify_structure()
938 for (n = 0; n < ndim; n++) in verify_structure()
945 for (n = 0; n < ndim; n++) in verify_structure()
950 for (n = 0; n < ndim; n++) in verify_structure()
970 the first dimension. Requirements: ndim > 0.
1036 assert(base->ndim > 0); in init_suboffsets()
1060 same as calculating -imin' for a sub-array of ndim-1. */ in init_suboffsets()
1062 for (n = 0; n < base->ndim; n++) { in init_suboffsets()
1080 base->suboffsets = PyMem_Malloc(base->ndim * (sizeof *base->suboffsets)); in init_suboffsets()
1086 for (n = 1; n < base->ndim; n++) in init_suboffsets()
1111 for (i = 0; i < base->ndim; i++) in init_len()
1118 Py_ssize_t ndim) in init_structure() argument
1122 base->ndim = (int)ndim; in init_structure()
1123 if (ndim == 0) { in init_structure()
1126 "ndim = 0 cannot be used in conjunction with ND_PIL"); in init_structure()
1134 base->shape = seq_as_ssize_array(shape, ndim, 1); in init_structure()
1140 base->strides = seq_as_ssize_array(strides, ndim, 0); in init_structure()
1148 base->shape, base->strides, ndim) < 0) in init_structure()
1178 Py_ssize_t ndim; in init_ndbuf() local
1182 /* ndim = len(shape) */ in init_ndbuf()
1184 ndim = PySequence_Fast_GET_SIZE(shape); in init_ndbuf()
1185 if (ndim > ND_MAX_NDIM) { in init_ndbuf()
1187 "ndim must not exceed %d", ND_MAX_NDIM); in init_ndbuf()
1201 else if (PySequence_Fast_GET_SIZE(strides) != ndim) { in init_ndbuf()
1219 if (ndim == 0) { in init_ndbuf()
1247 if (init_structure(ndbuf, shape, strides, ndim) < 0) in init_ndbuf()
1433 if (base->ndim != 0 && in ndarray_getbuf()
1507 view->ndim = 1; in ndarray_getbuf()
1559 assert(base->ndim == 1 && SIMPLE_FORMAT(base->format)); in ptr_from_index()
1590 if (base->ndim == 0) { in ndarray_item()
1599 if (base->ndim == 1) { in ndarray_item()
1620 subview->ndim--; in ndarray_item()
1642 indices[ndim-1] * strides[ndim-1]
1648 c = start[0] * strides[0] + ... + start[ndim-1] * strides[ndim-1]
1670 to sub-arrays of ndim-1. Striding in the first dimension is done by
1688 c = start[1] * strides[1] + ... + start[ndim-1] * strides[ndim-1]
1702 sub-arrays of ndim-2:
1742 shape = PyMem_Malloc(base->ndim * (sizeof *shape)); in copy_structure()
1743 strides = PyMem_Malloc(base->ndim * (sizeof *strides)); in copy_structure()
1749 suboffsets = PyMem_Malloc(base->ndim * (sizeof *suboffsets)); in copy_structure()
1754 for (i = 0; i < base->ndim; i++) { in copy_structure()
1782 if (base->ndim == 0) { in ndarray_subscript()
1877 if (dest->ndim == 0) { in ndarray_ass_subscript()
1888 if (dest->ndim == 1 && PyIndex_Check(key)) { in ndarray_ass_subscript()
2055 return PyLong_FromSsize_t(base->ndim); in ndarray_get_ndim()
2062 return ssize_array_as_tuple(base->shape, base->ndim); in ndarray_get_shape()
2069 return ssize_array_as_tuple(base->strides, base->ndim); in ndarray_get_strides()
2076 return ssize_array_as_tuple(base->suboffsets, base->ndim); in ndarray_get_suboffsets()
2133 { "ndim", (getter)ndarray_get_ndim, NULL, NULL, NULL},
2163 assert(src->ndim > 0); in ndarray_tobytes()
2207 base->suboffsets = PyMem_Malloc(base->ndim * (sizeof *base->suboffsets)); in ndarray_add_suboffsets()
2213 for (i = 0; i < base->ndim; i++) in ndarray_add_suboffsets()
2275 if (view->ndim > ND_MAX_NDIM) { in ndarray_memoryview_from_buffer()
2277 "memoryview_from_buffer: ndim is limited to %d", ND_MAX_NDIM); in ndarray_memoryview_from_buffer()
2281 memcpy(shape, view->shape, view->ndim * sizeof(Py_ssize_t)); in ndarray_memoryview_from_buffer()
2285 memcpy(strides, view->strides, view->ndim * sizeof(Py_ssize_t)); in ndarray_memoryview_from_buffer()
2289 memcpy(suboffsets, view->suboffsets, view->ndim * sizeof(Py_ssize_t)); in ndarray_memoryview_from_buffer()
2316 if (view.ndim > ND_MAX_NDIM) { in get_pointer()
2318 "get_pointer(): ndim > %d", ND_MAX_NDIM); in get_pointer()
2321 if (PySequence_Fast_GET_SIZE(seq) != view.ndim) { in get_pointer()
2323 "get_pointer(): len(indices) != ndim"); in get_pointer()
2327 for (i = 0; i < view.ndim; i++) { in get_pointer()
2477 Py_ssize_t ndim) in arraycmp() argument
2482 for (i = 0; i < ndim; i++) { in arraycmp()
2530 v1.ndim != v2.ndim || in cmp_contig()
2538 if ((v1.shape && !arraycmp(v1.shape, v2.shape, NULL, v1.ndim)) || in cmp_contig()
2539 (v1.strides && !arraycmp(v1.strides, v2.strides, v1.shape, v1.ndim)) || in cmp_contig()
2541 v1.ndim))) { in cmp_contig()
2696 1, /* ndim */