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
122
123
124
125
126
127
128
129
130
131
|
# This module converts code written for Numeric to run with scipy.base
# Makes the following changes:
# * Converts typecharacters
# * Changes import statements (warns of use of from Numeric import *)
# * Changes import statements (using numerix) ...
# * Makes search and replace changes to:
# - .typecode()
# - .iscontiguous()
# - .byteswapped()
# - .itemsize()
# * Converts useage of .flat.xxx to .ravel().xxx
# * Prints warning of other usage of flat.
# * Prints warning for use of bool, int, float, copmlex, object, and unicode
import sys
import os
import re
import warnings
flatindex_re = re.compile('([.]flat(\s*?[[=]))')
int_re = re.compile('int\s*[(][^)]*[)]')
bool_re = re.compile('bool\s*[(][^)]*[)]')
float_re = re.compile('float\s*[(][^)]*[)]')
complex_re = re.compile('complex\s*[(][^)]*[)]')
unicode_re = re.compile('unicode\s*[(][^)]*[)]')
def replacetypechars(astr):
astr = astr.replace("'s'","'h'")
astr = astr.replace("'c'","'S1'")
astr = astr.replace("'b'","'B'")
astr = astr.replace("'1'","'b'")
astr = astr.replace("'s'","'h'")
astr = astr.replace("'w'","'H'")
astr = astr.replace("'u'","'I'")
return astr
def changeimports(fstr, name):
importstr = 'import %s' % name
importasstr = 'import %s as ' % name
fromstr = 'from %s import ' % name
fromallstr = 'from %s import *' % name
fromall=0
fstr = fstr.replace(importasstr, 'import scipy.base as ')
fstr = fstr.replace(importstr, 'import scipy.base as %s' % name)
if (fstr.find(fromallstr) >= 0):
warnings.warn('Usage of %s found.' % fromallstr)
fstr = fstr.replace(fromallstr, 'from scipy.base import *')
fromall=1
ind = 0
Nlen = len(fromstr)
Nlen2 = len("from scipy.base import ")
while 1:
found = fstr.find(fromstr,ind)
if (found < 0):
break
ind = found + Nlen
if fstr[ind] == '*':
continue
fstr = "%sfrom scipy.base import %s" % (fstr[:found], fstr[ind:])
ind += Nlen2 - Nlen
return fstr, fromall
def replaceattr(astr):
astr = astr.replace(".typecode()",".dtypechar")
astr = astr.replace(".iscontiguous()",".flags['CONTIGUOUS']")
astr = astr.replace(".byteswapped()",".byteswap()")
astr = astr.replace(".itemsize()",".itemsize")
# preserve uses of flat that should be o.k.
tmpstr = flatindex_re.sub("@@@@\\2",astr)
# replace other uses of flat
tmpstr = tmpstr.replace(".flat",".ravel()")
# put back .flat where it was valid
astr = tmpstr.replace("@@@@", ".flat")
return astr
def warnofnewtypes(filestr):
if int_re.search(filestr) or \
float_re.search(filestr) or \
complex_re.search(filestr) or \
unicode_re.search(filestr) or \
bool_re.search(filestr):
warnings.warn("Use of builtin bool, int, float, complex, or unicode\n" \
"found when import * used -- these will be handled by\n" \
"new array scalars under scipy.base")
return
def process(filestr):
filestr = replacetypechars(filestr)
filestr, fromall1 = changeimports(filestr, 'Numeric')
filestr, fromall2 = changeimports(filestr, 'numerix')
filestr, fromall3 = changeimports(filestr, 'scipy_base')
fromall = fromall1 or fromall2 or fromall3
filestr = replaceattr(filestr)
#filestr = convertflat(filestr)
if fromall:
warnofnewtypes(filestr)
return filestr
def makenewfile(name, filestr):
fid = file(name, 'w')
fid.write(filestr)
fid.close()
def getandcopy(name):
fid = file(name)
filestr = fid.read()
fid.close()
base, ext = os.path.splitext(name)
makenewfile(base+'.orig', filestr)
return filestr
def main(args):
filename = args[1]
filestr = getandcopy(filename)
filestr = process(filestr)
makenewfile(filename, filestr)
if __name__ == '__main__':
main(sys.argv)
|
