1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
from pypy.interpreter.error import OperationError, oefmt
from pypy.module.micronumpy import constants as NPY
def byteorder_converter(space, new_order):
endian = new_order[0]
if endian not in (NPY.BIG, NPY.LITTLE, NPY.NATIVE, NPY.IGNORE, NPY.SWAP):
ch = endian
if ch in ('b', 'B'):
endian = NPY.BIG
elif ch in ('l', 'L'):
endian = NPY.LITTLE
elif ch in ('n', 'N'):
endian = NPY.NATIVE
elif ch in ('i', 'I'):
endian = NPY.IGNORE
elif ch in ('s', 'S'):
endian = NPY.SWAP
else:
raise oefmt(space.w_ValueError,
"%s is an unrecognized byteorder", new_order)
return endian
def clipmode_converter(space, w_mode):
if space.is_none(w_mode):
return NPY.RAISE
if space.isinstance_w(w_mode, space.w_text):
mode = space.text_w(w_mode)
if mode.startswith('C') or mode.startswith('c'):
return NPY.CLIP
if mode.startswith('W') or mode.startswith('w'):
return NPY.WRAP
if mode.startswith('R') or mode.startswith('r'):
return NPY.RAISE
elif space.isinstance_w(w_mode, space.w_int):
mode = space.int_w(w_mode)
if NPY.CLIP <= mode <= NPY.RAISE:
return mode
raise oefmt(space.w_TypeError, "clipmode not understood")
def searchside_converter(space, w_obj):
try:
s = space.text_w(w_obj)
except OperationError:
s = None
if not s:
raise oefmt(space.w_ValueError,
"expected nonempty string for keyword 'side'")
if s[0] == 'l' or s[0] == 'L':
return NPY.SEARCHLEFT
elif s[0] == 'r' or s[0] == 'R':
return NPY.SEARCHRIGHT
else:
raise oefmt(space.w_ValueError,
"'%s' is an invalid value for keyword 'side'", s)
def order_converter(space, w_order, default):
if space.is_none(w_order):
return default
if not space.isinstance_w(w_order, space.w_text):
if space.is_true(w_order):
return NPY.FORTRANORDER
else:
return NPY.CORDER
else:
order = space.text_w(w_order)
if order.startswith('C') or order.startswith('c'):
return NPY.CORDER
elif order.startswith('F') or order.startswith('f'):
return NPY.FORTRANORDER
elif order.startswith('A') or order.startswith('a'):
return NPY.ANYORDER
elif order.startswith('K') or order.startswith('k'):
return NPY.KEEPORDER
else:
raise oefmt(space.w_TypeError, "Unknown order: '%s'", order)
return -1
def multi_axis_converter(space, w_axis, ndim):
if space.is_none(w_axis):
return [True] * ndim
out = [False] * ndim
if not space.isinstance_w(w_axis, space.w_tuple):
w_axis = space.newtuple([w_axis])
for w_item in space.fixedview(w_axis):
item = space.int_w(w_item)
axis = item
if axis < 0:
axis += ndim
if axis < 0 or axis >= ndim:
raise oefmt(space.w_ValueError,
"'axis' entry %d is out of bounds [-%d, %d)",
item, ndim, ndim)
if out[axis]:
raise oefmt(space.w_ValueError, "duplicate value in 'axis'")
out[axis] = True
return out
def shape_converter(space, w_size, dtype):
if space.is_none(w_size):
return []
if space.isinstance_w(w_size, space.w_int):
return [space.int_w(w_size)]
shape = []
for w_item in space.fixedview(w_size):
shape.append(space.int_w(w_item))
shape += dtype.shape
return shape[:]
def out_converter(space, w_out):
from .ndarray import W_NDimArray
if space.is_none(w_out):
return None
elif not isinstance(w_out, W_NDimArray):
raise oefmt(space.w_TypeError, 'output must be an array')
else:
return w_out
|
