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
149
150
151
|
"""
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)
"""
from __future__ import division, absolute_import, print_function
import sys
import optparse
import io
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, end=' ')
tempstr = io.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, 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, end=' ')
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 = io.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 *', edic)
exec('set_printoptions(linewidth=65)', 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 *', edic)
exec('set_printoptions(linewidth=65)', 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:])
|
