forked from numpy/numpy
-
Notifications
You must be signed in to change notification settings - Fork 0
/
runcode.py
148 lines (134 loc) · 4.22 KB
/
runcode.py
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
148
# This script takes a lyx file and runs the python code in it.
# Then rewrites the lyx file again.
#
# Each section of code portion is assumed to be in the same namespace
# where a from numpy import * has been applied
#
# If a PYNEW inside a Note is encountered, the name space is restarted
#
# The output (if any) is replaced in the file
# by the output produced during the code run.
#
# Options:
# -n name of code section (default MyCode)
#
import sys
import optparse
import cStringIO
import re
import os
newre = re.compile(r"\\begin_inset Note.*PYNEW\s+\\end_inset", re.DOTALL)
def getoutput(tstr, dic):
print "\n\nRunning..."
print tstr,
tempstr = cStringIO.StringIO()
sys.stdout = tempstr
code = compile(tstr, '<input>', 'exec')
try:
res = eval(tstr, dic)
sys.stdout = sys.__stdout__
except SyntaxError:
try:
res = None
exec code in dic
finally:
sys.stdout = sys.__stdout__
if res is None:
res = tempstr.getvalue()
else:
res = tempstr.getvalue() + '\n' + repr(res)
if res != '':
print "\nOutput is"
print res,
return res
# now find the code in the code segment
def getnewcodestr(substr, dic):
end = substr.find('\\layout ')
lines = substr[:end].split('\\newline')
outlines = []
first = 1
cmd = ''
lines.append('dummy')
for line in lines:
line = line.strip()
if (line[:3]=='>>>') or (line == 'dummy'):
# we have a new output
pyoutstr = getoutput(cmd, dic).strip()
if pyoutstr != '':
pyout = pyoutstr.split('\n')
outlines.extend(pyout)
cmd = line[4:]
elif (line[:3]=='...'):
# continuation output
cmd += "\n%s" % line[4:]
else:
# first line or output
if first:
first = 0
cmd = line
else:
continue
if line != 'dummy':
outlines.append(line)
return "\n\\newline \n".join(outlines), end
def runpycode(lyxstr, name='MyCode'):
schobj = re.compile(r"\\layout %s\s+>>> " % name)
outstr = cStringIO.StringIO()
num = 0
indx = []
for it in schobj.finditer(lyxstr):
indx.extend([it.start(), it.end()])
num += 1
if num == 0:
print "Nothing found for %s" % name
return lyxstr
start = 0
del indx[0]
indx.append(len(lyxstr))
edic = {}
exec 'from numpy import *' in edic
exec 'set_printoptions(linewidth=65)' in edic
# indx now contains [st0,en0, ..., stN,enN]
# where stX is the start of code segment X
# and enX is the start of \layout MyCode for
# the X+1 code section (or string length if X=N)
for k in range(num):
# first write everything up to the start of the code segment
substr = lyxstr[start:indx[2*k]]
outstr.write(substr)
if start > 0:
mat = newre.search(substr)
# if PYNEW found, then start a new namespace
if mat:
edic = {}
exec 'from numpy import *' in edic
exec 'set_printoptions(linewidth=65)' in edic
# now find the code in the code segment
# endoutput will contain the index just past any output
# already present in the lyx string.
substr = lyxstr[indx[2*k]:indx[2*k+1]]
lyxcodestr, endcode = getnewcodestr(substr, edic)
# write the lyx for the input + new output
outstr.write(lyxcodestr)
outstr.write('\n')
start = endcode + indx[2*k]
outstr.write(lyxstr[start:])
return outstr.getvalue()
def main(args):
usage = "%prog {options} filename"
parser = optparse.OptionParser(usage)
parser.add_option('-n','--name', default='MyCode')
options, args = parser.parse_args(args)
if len(args) < 1:
parser.error("incorrect number of arguments")
os.system('cp -f %s %s.bak' % (args[0], args[0]))
fid = file(args[0])
str = fid.read()
fid.close()
print "Processing %s" % options.name
newstr = runpycode(str, options.name)
fid = file(args[0],'w')
fid.write(newstr)
fid.close()
if __name__ == "__main__":
main(sys.argv[1:])