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
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
|
#! /usr/bin/env python3
"""compileallcl.py: Load all the packages in IRAF and compile all
the CL scripts
The results are stored in the user cache. The old user cache
is moved to ~/iraf/pyraf/clcache.old.
Set the -r flag to recompile (default is to close the system cache
and move the user cache so that everything gets compiled from scratch.)
"""
import os
import sys
import traceback
import time
def printcenter(s, length=70, char="-"):
l1 = (length - len(s)) // 2
l2 = length - l1 - len(s)
print(l1 * char, s, l2 * char)
sys.stdout.flush()
def compileall():
"""Compile all CL procedures & packages"""
# start the timer
t0 = time.time()
# now do the IRAF startup
from pyraf import iraf, cl2py, clcache, irafglobals
from pyraf.iraftask import IrafCLTask, IrafPkg
# close the old code cache and reopen without system cache
cl2py.codeCache.close()
del clcache.codeCache
userCacheDir = os.path.join(irafglobals.userIrafHome, 'pyraf')
if not os.path.exists(userCacheDir):
try:
os.mkdir(userCacheDir)
print('Created directory %s for cache' % userCacheDir)
except OSError:
print('Could not create directory %s' % userCacheDir)
dbfile = 'clcache'
clcache.codeCache = clcache._CodeCache(
[os.path.join(userCacheDir, dbfile)])
cl2py.codeCache = clcache.codeCache
iraf.setVerbose()
pkgs_tried = {}
tasks_tried = {}
npkg_total = 0
ntask_total = 0
ntask_failed = 0
# main loop -- keep loading packages as long as there are
# new ones to try, and after loading each package look for
# and initialize any CL tasks
npass = 0
pkg_list = iraf.getPkgList()
keepGoing = 1
while keepGoing and (npkg_total < len(pkg_list)):
npass = npass + 1
pkg_list.sort()
npkg_new = 0
printcenter("pass %d: %d packages (%d new)" %
(npass, len(pkg_list), len(pkg_list) - npkg_total),
char="=")
for pkg in pkg_list:
if pkg not in pkgs_tried:
pkgs_tried[pkg] = 1
npkg_new = npkg_new + 1
printcenter(pkg)
if pkg in ["newimred", "digiphotx"]:
print("""
Working around bugs in newimred, digiphotx.
They screw up subsequent loading of imred/digiphot tasks.
(It happens in IRAF too.)""")
sys.stdout.flush()
else:
try:
# redirect stdin in case the package tries to
# prompt for parameters (this aborts but keeps
# going)
iraf.load(pkg, kw={'Stdin': 'dev$null'})
except KeyboardInterrupt:
print('Interrupt')
sys.stdout.flush()
keepGoing = 0
break
except Exception as e:
sys.stdout.flush()
traceback.print_exc()
if isinstance(e, MemoryError):
keepGoing = 0
break
print("...continuing...\n")
sys.stdout.flush()
# load tasks after each package
task_list = sorted(iraf.getTaskList())
for taskname in task_list:
if taskname not in tasks_tried:
tasks_tried[taskname] = 1
taskobj = iraf.getTask(taskname)
if isinstance(taskobj, IrafCLTask) and \
not isinstance(taskobj, IrafPkg):
ntask_total = ntask_total + 1
print("%d: %s" % (ntask_total, taskname))
sys.stdout.flush()
try:
taskobj.initTask()
except KeyboardInterrupt:
print('Interrupt')
sys.stdout.flush()
keepGoing = 0
break
except Exception as e:
sys.stdout.flush()
traceback.print_exc(10)
if isinstance(e, MemoryError):
keepGoing = 0
break
print("...continuing...\n")
sys.stdout.flush()
ntask_failed = ntask_failed + 1
if not keepGoing:
break
npkg_total = npkg_total + npkg_new
if not keepGoing:
break
printcenter(
"Finished pass %d new pkgs %d total pkgs %d total tasks %d" %
(npass, npkg_new, npkg_total, ntask_total),
char="=")
pkg_list = iraf.getPkgList()
t1 = time.time()
print("Finished package and task loading ({:f} seconds)".format(t1 - t0))
print("Compiled %d CL tasks -- %d failed" % (ntask_total, ntask_failed))
sys.stdout.flush()
def usage():
print("""Usage: %s [-r] [-h]
-r recompiles only out-of-date routines. Default is to
force recompilation of all CL scripts and packages.
-h prints this message.
""")
sys.stdout.flush()
sys.exit()
if __name__ == '__main__':
# command-line options
import getopt
try:
optlist, args = getopt.getopt(sys.argv[1:], "rh")
if len(args) > 0:
print("Error: no positional arguments accepted")
usage()
except getopt.error as e:
print(str(e))
usage()
clearcache = 1
for opt, value in optlist:
if opt == "-r":
clearcache = 0
elif opt == "-h":
usage()
else:
print("Program bug, unknown option", opt)
raise SystemExit()
# find pyraf directory
for p in sys.path:
absPyrafDir = os.path.join(p, 'pyraf')
if os.path.exists(os.path.join(absPyrafDir, '__init__.py')):
break
else:
raise Exception('pyraf module not found')
usrCache = os.path.expanduser('~/iraf/pyraf/clcache')
if clearcache:
# move the old user cache
if os.path.exists(usrCache):
os.rename(usrCache, usrCache + '.old')
print("Pyraf user cache %s renamed to %s.old" %
(usrCache, usrCache))
else:
print("Warning: pyraf user cache %s was not found" % usrCache)
compileall()
|
