summaryrefslogtreecommitdiff
path: root/support/openscad/__main__.py
blob: 3c103e5d533ef50cd10160ad36620568e401db2c (plain)
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
import os, sys
from lib import util


def _openscad(in_path, out_path, deps_path):
	util.command([os.environ['OPENSCAD'], '-o', out_path, '-d', deps_path, in_path])


def _write_dependencies(path, target, dependencies):
	util.write_file(path, '{}: {}\n'.format(target, ' '.join(dependencies)).encode())


def main(in_path, out_path, deps_path):
	cwd = os.getcwd()
	
	def relpath(path):
		return os.path.relpath(path, cwd)
	
	with util.TemporaryDirectory() as temp_dir:
		temp_deps_path = os.path.join(temp_dir, 'deps')
		temp_mk_path = os.path.join(temp_dir, 'mk')
		temp_files_path = os.path.join(temp_dir, 'files')
		
		# OpenSCAD requires the output file name to end in .stl
		temp_stl_path = os.path.join(temp_dir, 'out.stl')
		
		_openscad(in_path, temp_stl_path, temp_deps_path)
		
		mk_content = '%:; echo "$@" >> {}'.format(util.bash_escape_string(temp_files_path))
		
		# Use make to parse the dependency makefile written by OpenSCAD
		util.write_file(temp_mk_path, mk_content.encode())
		util.command(['make', '-s', '-B', '-f', temp_mk_path, '-f', temp_deps_path])
		
		# All dependencies as paths relative to the project root.
		deps = set(map(relpath, util.read_file(temp_files_path).decode().splitlines()))
		
		# Relative paths to all files that should not appear in the dependency makefile.
		ignored_files = set(map(relpath, [in_path, temp_deps_path, temp_mk_path, temp_stl_path]))
		
		# Write output files.
		_write_dependencies(deps_path, relpath(out_path), deps - ignored_files)
		os.rename(temp_stl_path, out_path)


try:
	main(*sys.argv[1:])
except util.UserError as e:
	print 'Error:', e
	sys.exit(1)
except KeyboardInterrupt:
	sys.exit(2)