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
|
import sys, os, shutil
from lib import util, make
def _asymptote(in_path, out_path, asymptote_dir, working_dir):
args = [os.environ['ASYMPTOTE'], '-vv', '-f', 'pdf', '-o', out_path, in_path]
with util.command_context(args, set_env={'ASYMPTOTE_DIR': asymptote_dir}, working_dir=working_dir, use_stderr=True) as process:
def get_loaded_file(line):
if any(line.startswith(j) for j in ['Loading ', 'Including ']):
parts = line.rstrip('\n').split(' ')
if len(parts) == 4:
_, _, from_, path = parts
if from_ == 'from':
return path
return None
def iter_loaded_files():
for i in process.stderr:
loaded_file = get_loaded_file(i)
if loaded_file is not None:
yield loaded_file
elif not any(i.startswith(j) for j in ['cd ', 'Using configuration ']):
print >> sys.stderr, i,
loaded_files = list(iter_loaded_files())
return loaded_files
@util.main
def main(in_path, out_path):
try:
_, out_suffix = os.path.splitext(out_path)
with util.TemporaryDirectory() as temp_dir:
absolute_in_path = os.path.abspath(in_path)
temp_out_path = os.path.join(temp_dir, 'out.pdf')
# Asymptote creates A LOT of temp files (presumably when invoking
# LaTeX) and leaves some of them behind. Thus we run asymptote
# in a temporary directory.
loaded_files = _asymptote(
absolute_in_path,
'out',
os.path.dirname(absolute_in_path),
temp_dir)
if not os.path.exists(temp_out_path):
raise util.UserError('Asymptote did not generate a PDF file.', in_path)
# All dependencies as paths relative to the project root.
dependencies = set(map(os.path.relpath, loaded_files))
# Write output files.
make.write_dependencies(out_path + '.d', out_path, dependencies - {in_path})
shutil.copyfile(temp_out_path, out_path)
except util.UserError as e:
raise util.UserError('While processing {}: {}', in_path, e)
|
