1#!/usr/bin/env python
2import subprocess
3import os
4import errno
5import collections
6import glob
7import argparse
8
9class Platform(object):
10    pass
11
12class simulator_platform(Platform):
13    directory = 'darwin_ios'
14    sdk = 'iphonesimulator'
15    arch = 'i386'
16    triple = 'i386-apple-darwin11'
17    version_min = '-miphoneos-version-min=5.1.1'
18
19    prefix = "#ifdef __i386__\n\n"
20    suffix = "\n\n#endif"
21    src_dir = 'x86'
22    src_files = ['darwin.S', 'win32.S', 'ffi.c']
23
24
25class simulator64_platform(Platform):
26    directory = 'darwin_ios'
27    sdk = 'iphonesimulator'
28    arch = 'x86_64'
29    triple = 'x86_64-apple-darwin13'
30    version_min = '-miphoneos-version-min=7.0'
31
32    prefix = "#ifdef __x86_64__\n\n"
33    suffix = "\n\n#endif"
34    src_dir = 'x86'
35    src_files = ['darwin64.S', 'ffi64.c']
36
37
38class device_platform(Platform):
39    directory = 'darwin_ios'
40    sdk = 'iphoneos'
41    arch = 'armv7'
42    triple = 'arm-apple-darwin11'
43    version_min = '-miphoneos-version-min=5.1.1'
44
45    prefix = "#ifdef __arm__\n\n"
46    suffix = "\n\n#endif"
47    src_dir = 'arm'
48    src_files = ['sysv.S', 'trampoline.S', 'ffi.c']
49
50
51class device64_platform(Platform):
52    directory = 'darwin_ios'
53    sdk = 'iphoneos'
54    arch = 'arm64'
55    triple = 'aarch64-apple-darwin13'
56    version_min = '-miphoneos-version-min=7.0'
57
58    prefix = "#ifdef __arm64__\n\n"
59    suffix = "\n\n#endif"
60    src_dir = 'aarch64'
61    src_files = ['sysv.S', 'ffi.c']
62
63
64class desktop32_platform(Platform):
65    directory = 'darwin_osx'
66    sdk = 'macosx'
67    arch = 'i386'
68    triple = 'i386-apple-darwin10'
69    version_min = '-mmacosx-version-min=10.6'
70    src_dir = 'x86'
71    src_files = ['darwin.S', 'win32.S', 'ffi.c']
72
73    prefix = "#ifdef __i386__\n\n"
74    suffix = "\n\n#endif"
75
76
77class desktop64_platform(Platform):
78    directory = 'darwin_osx'
79    sdk = 'macosx'
80    arch = 'x86_64'
81    triple = 'x86_64-apple-darwin10'
82    version_min = '-mmacosx-version-min=10.6'
83
84    prefix = "#ifdef __x86_64__\n\n"
85    suffix = "\n\n#endif"
86    src_dir = 'x86'
87    src_files = ['darwin64.S', 'ffi64.c']
88
89
90def mkdir_p(path):
91    try:
92        os.makedirs(path)
93    except OSError as exc:  # Python >2.5
94        if exc.errno == errno.EEXIST:
95            pass
96        else:
97            raise
98
99
100def move_file(src_dir, dst_dir, filename, file_suffix=None, prefix='', suffix=''):
101    mkdir_p(dst_dir)
102    out_filename = filename
103
104    if file_suffix:
105        split_name = os.path.splitext(filename)
106        out_filename = "%s_%s%s" % (split_name[0], file_suffix, split_name[1])
107
108    with open(os.path.join(src_dir, filename)) as in_file:
109        with open(os.path.join(dst_dir, out_filename), 'w') as out_file:
110            if prefix:
111                out_file.write(prefix)
112
113            out_file.write(in_file.read())
114
115            if suffix:
116                out_file.write(suffix)
117
118
119def list_files(src_dir, pattern=None, filelist=None):
120    if pattern: filelist = glob.iglob(os.path.join(src_dir, pattern))
121    for file in filelist:
122        yield os.path.basename(file)
123
124
125def copy_files(src_dir, dst_dir, pattern=None, filelist=None, file_suffix=None, prefix=None, suffix=None):
126    for filename in list_files(src_dir, pattern=pattern, filelist=filelist):
127        move_file(src_dir, dst_dir, filename, file_suffix=file_suffix, prefix=prefix, suffix=suffix)
128
129
130def copy_src_platform_files(platform):
131    src_dir = os.path.join('src', platform.src_dir)
132    dst_dir = os.path.join(platform.directory, 'src', platform.src_dir)
133    copy_files(src_dir, dst_dir, filelist=platform.src_files, file_suffix=platform.arch, prefix=platform.prefix, suffix=platform.suffix)
134
135
136def build_target(platform, platform_headers):
137    def xcrun_cmd(cmd):
138        return 'xcrun -sdk %s %s -arch %s' % (platform.sdk, cmd, platform.arch)
139
140    tag='%s-%s' % (platform.sdk, platform.arch)
141    build_dir = 'build_%s' % tag
142    mkdir_p(build_dir)
143    env = dict(CC=xcrun_cmd('clang'),
144               LD=xcrun_cmd('ld'),
145               CFLAGS='%s' % (platform.version_min))
146    working_dir = os.getcwd()
147    try:
148        os.chdir(build_dir)
149        subprocess.check_call(['../configure', '-host', platform.triple], env=env)
150    finally:
151        os.chdir(working_dir)
152
153    for src_dir in [build_dir, os.path.join(build_dir, 'include')]:
154        copy_files(src_dir,
155                   os.path.join(platform.directory, 'include'),
156                   pattern='*.h',
157                   file_suffix=platform.arch,
158                   prefix=platform.prefix,
159                   suffix=platform.suffix)
160
161        for filename in list_files(src_dir, pattern='*.h'):
162            platform_headers[filename].add((platform.prefix, platform.arch, platform.suffix))
163
164
165def make_tramp():
166    with open('src/arm/trampoline.S', 'w') as tramp_out:
167        p = subprocess.Popen(['bash', 'src/arm/gentramp.sh'], stdout=tramp_out)
168        p.wait()
169
170
171def generate_source_and_headers(generate_osx=True, generate_ios=True):
172    copy_files('src', 'darwin_common/src', pattern='*.c')
173    copy_files('include', 'darwin_common/include', pattern='*.h')
174
175    if generate_ios:
176        make_tramp()
177        copy_src_platform_files(simulator_platform)
178        copy_src_platform_files(simulator64_platform)
179        copy_src_platform_files(device_platform)
180        copy_src_platform_files(device64_platform)
181    if generate_osx:
182        copy_src_platform_files(desktop32_platform)
183        copy_src_platform_files(desktop64_platform)
184
185    platform_headers = collections.defaultdict(set)
186
187    if generate_ios:
188        build_target(simulator_platform, platform_headers)
189        build_target(simulator64_platform, platform_headers)
190        build_target(device_platform, platform_headers)
191        build_target(device64_platform, platform_headers)
192    if generate_osx:
193        build_target(desktop32_platform, platform_headers)
194        build_target(desktop64_platform, platform_headers)
195
196    mkdir_p('darwin_common/include')
197    for header_name, tag_tuples in platform_headers.iteritems():
198        basename, suffix = os.path.splitext(header_name)
199        with open(os.path.join('darwin_common/include', header_name), 'w') as header:
200            for tag_tuple in tag_tuples:
201                header.write('%s#include <%s_%s%s>\n%s\n' % (tag_tuple[0], basename, tag_tuple[1], suffix, tag_tuple[2]))
202
203if __name__ == '__main__':
204    parser = argparse.ArgumentParser()
205    parser.add_argument('--only-ios', action='store_true', default=False)
206    parser.add_argument('--only-osx', action='store_true', default=False)
207    args = parser.parse_args()
208
209    generate_source_and_headers(generate_osx=not args.only_ios, generate_ios=not args.only_osx)
210