summaryrefslogtreecommitdiff
path: root/pokemontools/scan_includes.py
blob: 53ff091c83ba3099e5c424a1d76b7a592c24bd3a (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
#!/bin/python
# coding: utf-8

"""
Recursively scan an asm file for dependencies.
"""

import sys
import argparse
import os.path

includes = set()

def scan_file(filename):
	for line in open(filename):
		if 'INC' not in line:
			continue
		line = line.split(';')[0]
		if 'INCLUDE' in line:
			include = line.split('"')[1]
			if os.path.exists("src/"):
				includes.add("src/" + include)
				scan_file("src/" + include)
			else:
				includes.add(include)
				scan_file(include)
		elif 'INCBIN' in line:
			include = line.split('"')[1]
			if 'baserom.gbc' not in line and os.path.exists("src/"):
				includes.add("src/" + include)
			else:
				includes.add(include)

def main():
	ap = argparse.ArgumentParser()
	ap.add_argument('filenames', nargs='*')
	args = ap.parse_args()
	for filename in set(args.filenames):
		scan_file(filename)
	sys.stdout.write(' '.join(includes))

if __name__ == '__main__':
	main()