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
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
|
# 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 .flat to .ravel() except for .flat = xxx or .flat[xxx]
# * Change typecode= to dtype=
# * Eliminates savespace=xxx
# * Replace xxx.spacesaver() with True
# * Convert xx.savespace(?) to pass + ## xx.savespace(?)
# #### -- not * Convert a.shape = ? to a.reshape(?)
# * Prints warning for use of bool, int, float, copmlex, object, and unicode
#
__all__ = ['fromfile', 'fromstr']
import sys
import os
import re
import glob
flatindex_re = re.compile('([.]flat(\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("'w'","'H'")
astr = astr.replace("'u'","'I'")
return astr
def changeimports(fstr, name, newname):
importstr = 'import %s' % name
importasstr = 'import %s as ' % name
fromstr = 'from %s import ' % name
fromall=0
fstr = fstr.replace(importasstr, 'import %s as ' % newname)
fstr = fstr.replace(importstr, 'import %s as %s' % (newname,name))
ind = 0
Nlen = len(fromstr)
Nlen2 = len("from %s import " % newname)
while 1:
found = fstr.find(fromstr,ind)
if (found < 0):
break
ind = found + Nlen
if fstr[ind] == '*':
continue
fstr = "%sfrom %s import %s" % (fstr[:found], newname, 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(".toscalar()", ".item()")
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
svspc = re.compile(r'(\S+\s*[(].+),\s*savespace\s*=.+\s*[)]')
svspc2 = re.compile(r'([^,(\s]+[.]spacesaver[(][)])')
svspc3 = re.compile(r'(\S+[.]savespace[(].*[)])')
#shpe = re.compile(r'(\S+\s*)[.]shape\s*=[^=]\s*(.+)')
def replaceother(astr):
astr = astr.replace("typecode=","dtype=")
astr = astr.replace("UserArray","ndarray")
astr = svspc.sub('\\1)',astr)
astr = svspc2.sub('True',astr)
astr = svspc3.sub('pass ## \\1', astr)
#astr = shpe.sub('\\1=\\1.reshape(\\2)', astr)
return astr
import datetime
def fromstr(filestr):
filestr = replacetypechars(filestr)
filestr, fromall1 = changeimports(filestr, 'Numeric', 'scipy')
filestr, fromall1 = changeimports(filestr, 'multiarray',
'scipy.base.multiarray')
filestr, fromall1 = changeimports(filestr, 'umath',
'scipy.base.umath')
filestr, fromall1 = changeimports(filestr, 'Precision', 'scipy.base')
filestr, fromall2 = changeimports(filestr, 'numerix', 'scipy.base')
filestr, fromall3 = changeimports(filestr, 'scipy_base', 'scipy.base')
filestr, fromall3 = changeimports(filestr, 'MLab', 'scipy.base.mlab')
filestr, fromall3 = changeimports(filestr, 'LinearAlgebra', 'scipy.corelinalg')
filestr, fromall3 = changeimports(filestr, 'RNG', 'scipy.random')
filestr, fromall3 = changeimports(filestr, 'RandomArray', 'scipy.random')
filestr, fromall3 = changeimports(filestr, 'FFT', 'scipy.corefft')
filestr, fromall3 = changeimports(filestr, 'MA', 'scipy.base.ma')
fromall = fromall1 or fromall2 or fromall3
filestr = replaceattr(filestr)
filestr = replaceother(filestr)
today = datetime.date.today().strftime('%b %d, %Y')
name = os.path.split(sys.argv[0])[-1]
filestr = '## Automatically adapted for '\
'scipy %s by %s\n\n%s' % (today, name, 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 fromfile(filename):
filestr = getandcopy(filename)
filestr = fromstr(filestr)
makenewfile(filename, filestr)
def fromargs(args):
filename = args[1]
fromfile(filename)
def convertall(direc=''):
files = glob.glob(os.path.join(direc,'*.py'))
for afile in files:
fromfile(afile)
if __name__ == '__main__':
fromargs(sys.argv)
|
