Python code coverage for Modules/_ctypes/libffi/generate-ios-source-and-headers.py

#	count	content
1	n/a	#!/usr/bin/env python
2	n/a
3	n/a	import subprocess
4	n/a	import re
5	n/a	import os
6	n/a	import errno
7	n/a	import collections
8	n/a	import sys
9	n/a
10	n/a	class Platform(object):
11	n/a	pass
12	n/a
13	n/a	sdk_re = re.compile(r'.-sdk ([a-zA-Z0-9.])')
14	n/a
15	n/a	def sdkinfo(sdkname):
16	n/a	ret = {}
17	n/a	for line in subprocess.Popen(['xcodebuild', '-sdk', sdkname, '-version'], stdout=subprocess.PIPE).stdout:
18	n/a	kv = line.strip().split(': ', 1)
19	n/a	if len(kv) == 2:
20	n/a	k,v = kv
21	n/a	ret[k] = v
22	n/a	return ret
23	n/a
24	n/a	sim_sdk_info = sdkinfo('iphonesimulator')
25	n/a	device_sdk_info = sdkinfo('iphoneos')
26	n/a
27	n/a	def latest_sdks():
28	n/a	latest_sim = None
29	n/a	latest_device = None
30	n/a	for line in subprocess.Popen(['xcodebuild', '-showsdks'], stdout=subprocess.PIPE).stdout:
31	n/a	match = sdk_re.match(line)
32	n/a	if match:
33	n/a	if 'Simulator' in line:
34	n/a	latest_sim = match.group(1)
35	n/a	elif 'iOS' in line:
36	n/a	latest_device = match.group(1)
37	n/a
38	n/a	return latest_sim, latest_device
39	n/a
40	n/a	sim_sdk, device_sdk = latest_sdks()
41	n/a
42	n/a	class simulator_platform(Platform):
43	n/a	sdk='iphonesimulator'
44	n/a	arch = 'i386'
45	n/a	name = 'simulator'
46	n/a	triple = 'i386-apple-darwin10'
47	n/a	sdkroot = sim_sdk_info['Path']
48	n/a
49	n/a	prefix = "#if !defined(__arm__) && defined(__i386__)\n\n"
50	n/a	suffix = "\n\n#endif"
51	n/a
52	n/a	class device_platform(Platform):
53	n/a	sdk='iphoneos'
54	n/a	name = 'ios'
55	n/a	arch = 'armv7'
56	n/a	triple = 'arm-apple-darwin10'
57	n/a	sdkroot = device_sdk_info['Path']
58	n/a
59	n/a	prefix = "#ifdef __arm__\n\n"
60	n/a	suffix = "\n\n#endif"
61	n/a
62	n/a
63	n/a	def move_file(src_dir, dst_dir, filename, file_suffix=None, prefix='', suffix=''):
64	n/a	if not os.path.exists(dst_dir):
65	n/a	os.makedirs(dst_dir)
66	n/a
67	n/a	out_filename = filename
68	n/a
69	n/a	if file_suffix:
70	n/a	split_name = os.path.splitext(filename)
71	n/a	out_filename = "%s_%s%s" % (split_name[0], file_suffix, split_name[1])
72	n/a
73	n/a	with open(os.path.join(src_dir, filename)) as in_file:
74	n/a	with open(os.path.join(dst_dir, out_filename), 'w') as out_file:
75	n/a	if prefix:
76	n/a	out_file.write(prefix)
77	n/a
78	n/a	out_file.write(in_file.read())
79	n/a
80	n/a	if suffix:
81	n/a	out_file.write(suffix)
82	n/a
83	n/a	headers_seen = collections.defaultdict(set)
84	n/a
85	n/a	def move_source_tree(src_dir, dest_dir, dest_include_dir, arch=None, prefix=None, suffix=None):
86	n/a	for root, dirs, files in os.walk(src_dir, followlinks=True):
87	n/a	relroot = os.path.relpath(root,src_dir)
88	n/a
89	n/a	def move_dir(arch, prefix='', suffix='', files=[]):
90	n/a	for file in files:
91	n/a	file_suffix = None
92	n/a	if file.endswith('.h'):
93	n/a	if dest_include_dir:
94	n/a	file_suffix = arch
95	n/a	if arch:
96	n/a	headers_seen[file].add(arch)
97	n/a	move_file(root, dest_include_dir, file, arch, prefix=prefix, suffix=suffix)
98	n/a
99	n/a	elif dest_dir:
100	n/a	outroot = os.path.join(dest_dir, relroot)
101	n/a	move_file(root, outroot, file, prefix=prefix, suffix=suffix)
102	n/a
103	n/a	if relroot == '.':
104	n/a	move_dir(arch=arch,
105	n/a	files=files,
106	n/a	prefix=prefix,
107	n/a	suffix=suffix)
108	n/a	elif relroot == 'arm':
109	n/a	move_dir(arch='arm',
110	n/a	prefix="#ifdef __arm__\n\n",
111	n/a	suffix="\n\n#endif",
112	n/a	files=files)
113	n/a	elif relroot == 'x86':
114	n/a	move_dir(arch='i386',
115	n/a	prefix="#if !defined(__arm__) && defined(__i386__)\n\n",
116	n/a	suffix="\n\n#endif",
117	n/a	files=files)
118	n/a
119	n/a	def build_target(platform):
120	n/a	def xcrun_cmd(cmd):
121	n/a	return subprocess.check_output(['xcrun', '-sdk', platform.sdkroot, '-find', cmd]).strip()
122	n/a
123	n/a	build_dir = 'build_' + platform.name
124	n/a	if not os.path.exists(build_dir):
125	n/a	os.makedirs(build_dir)
126	n/a	env = dict(CC=xcrun_cmd('clang'),
127	n/a	LD=xcrun_cmd('ld'),
128	n/a	CFLAGS='-arch %s -isysroot %s -miphoneos-version-min=4.0' % (platform.arch, platform.sdkroot))
129	n/a	working_dir=os.getcwd()
130	n/a	try:
131	n/a	os.chdir(build_dir)
132	n/a	subprocess.check_call(['../configure', '-host', platform.triple], env=env)
133	n/a	move_source_tree('.', None, '../ios/include',
134	n/a	arch=platform.arch,
135	n/a	prefix=platform.prefix,
136	n/a	suffix=platform.suffix)
137	n/a	move_source_tree('./include', None, '../ios/include',
138	n/a	arch=platform.arch,
139	n/a	prefix=platform.prefix,
140	n/a	suffix=platform.suffix)
141	n/a	finally:
142	n/a	os.chdir(working_dir)
143	n/a
144	n/a	for header_name, archs in headers_seen.iteritems():
145	n/a	basename, suffix = os.path.splitext(header_name)
146	n/a
147	n/a	def main():
148	n/a	move_source_tree('src', 'ios/src', 'ios/include')
149	n/a	move_source_tree('include', None, 'ios/include')
150	n/a	build_target(simulator_platform)
151	n/a	build_target(device_platform)
152	n/a
153	n/a	for header_name, archs in headers_seen.iteritems():
154	n/a	basename, suffix = os.path.splitext(header_name)
155	n/a	with open(os.path.join('ios/include', header_name), 'w') as header:
156	n/a	for arch in archs:
157	n/a	header.write('#include <%s_%s%s>\n' % (basename, arch, suffix))
158	n/a
159	n/a	if __name__ == '__main__':
160	n/a	main()