OLD | NEW |
(Empty) | |
| 1 #!/usr/bin/env python |
| 2 |
| 3 import json |
| 4 import os |
| 5 import re |
| 6 import subprocess |
| 7 import warnings |
| 8 |
| 9 EMSCRIPTEN_PATH = '../emscripten' |
| 10 SOURCE_DIR = './compiled' |
| 11 SOURCE_FILES = [ |
| 12 os.path.join(SOURCE_DIR, f) |
| 13 for f in os.listdir(SOURCE_DIR) |
| 14 if f.endswith('.cpp') |
| 15 ] |
| 16 API_FILE = os.path.join(SOURCE_DIR, 'api.cpp') |
| 17 API_OUTPUT = os.path.join(SOURCE_DIR, 'api.js') |
| 18 COMPILER_OUTPUT = './lib/compiled.js' |
| 19 GENERATION_PARAMS = { |
| 20 'SHELL_FILE': "'%s'" % os.path.abspath(os.path.join(SOURCE_DIR, 'shell.js')), |
| 21 'TOTAL_MEMORY': 16*1024*1024, |
| 22 'TOTAL_STACK': 1*1024*1024, |
| 23 'ALLOW_MEMORY_GROWTH': 1, |
| 24 'NO_EXIT_RUNTIME': 1, |
| 25 'DISABLE_EXCEPTION_CATCHING': 0, |
| 26 'NO_DYNAMIC_EXECUTION': 1, |
| 27 'NO_BROWSER': 1, |
| 28 'NO_FILESYSTEM': 1, |
| 29 'INVOKE_RUN': 0, |
| 30 'NODE_STDOUT_FLUSH_WORKAROUND': 0, |
| 31 } |
| 32 DEFINES = ['DEBUG'] |
| 33 ADDITIONAL_PARAMS = ['-O3', '-m32', '-std=gnu++11', '--memory-init-file', '0', |
| 34 '--emit-symbol-map'] |
| 35 |
| 36 def getenv(): |
| 37 path = [] |
| 38 env = {} |
| 39 output = subprocess.check_output([ |
| 40 '/bin/sh', '-c', os.path.join(EMSCRIPTEN_PATH, 'emsdk_env.sh')]) |
| 41 for line in output.splitlines(): |
| 42 match = re.search(r'^\s*PATH\s*\+=\s*(.*)', line) |
| 43 if match: |
| 44 path.append(match.group(1)) |
| 45 match = re.search(r'^\s*(\w+)\s*=\s*(.*)', line) |
| 46 if match: |
| 47 env[match.group(1)] = match.group(2) |
| 48 env['PATH'] = ':'.join([os.environ['PATH']] + path) |
| 49 return env |
| 50 |
| 51 def generate_api(env): |
| 52 params = [os.path.join(env['EMSCRIPTEN'], 'emcc'), '-E', API_FILE] |
| 53 params.extend(ADDITIONAL_PARAMS) |
| 54 output = subprocess.check_output(params, env=env) |
| 55 |
| 56 differentiators = {} |
| 57 differentiator = None |
| 58 |
| 59 cls = None |
| 60 method = None |
| 61 |
| 62 def wrap_call(func, is_instance, result_type, string_args, arg_count=10): |
| 63 params = ['arg%i' % i for i in range(arg_count)] |
| 64 prefix = '''\ |
| 65 function(%s) |
| 66 { |
| 67 ''' % ', '.join(params) |
| 68 suffix = '''\ |
| 69 return result; |
| 70 }''' |
| 71 |
| 72 if result_type == 'string' or len(string_args): |
| 73 prefix += ' var sp = Runtime.stackSave();\n' |
| 74 suffix = ' Runtime.stackRestore(sp);\n' + suffix |
| 75 |
| 76 for pos in string_args: |
| 77 params[pos] = 'createString(%s)' % params[pos] |
| 78 |
| 79 if is_instance: |
| 80 params.insert(0, 'this._pointer') |
| 81 |
| 82 if result_type == 'primitive': |
| 83 prefix += ' var result = _%s(%s);\n' % (func, ', '.join(params)) |
| 84 elif result_type == 'string': |
| 85 prefix += ' var result = createString();\n' |
| 86 params.insert(0, 'result') |
| 87 prefix += ' _%s(%s);\n' % (func, ', '.join(params)) |
| 88 prefix += ' result = getStringData(result);\n' |
| 89 else: |
| 90 prefix += ' var pointer = _%s(%s);\n' % (func, ', '.join(params)) |
| 91 if result_type in differentiators: |
| 92 prefix += ' var type = _%s(pointer);\n' % differentiator['func'] |
| 93 prefix += ' if (type in %s_mapping)\n' % result_type |
| 94 prefix += ' var result = new (exports[%s_mapping[type]])(pointer);\n'
% result_type |
| 95 prefix += ' else\n' |
| 96 prefix += ' throw new Error("Unexpected %s type: " + type);\n' % resu
lt_type |
| 97 else: |
| 98 prefix += ' var result = %s(pointer);\n' % result_type |
| 99 |
| 100 return prefix + suffix |
| 101 |
| 102 def property_descriptor(property): |
| 103 if property['type'] == 'static': |
| 104 return 'value: %s' % property['getter'] |
| 105 |
| 106 result = 'get: %s' % wrap_call( |
| 107 property['getter'], is_instance=True, result_type=property['type'], |
| 108 string_args=[], arg_count=0 |
| 109 ) |
| 110 if property['setter'] is not None: |
| 111 result += ', set: %s' % wrap_call( |
| 112 property['setter'], is_instance=True, result_type='primitive', |
| 113 string_args=[0] if property['type'] == 'string' else [], |
| 114 arg_count=1 |
| 115 ) |
| 116 return result |
| 117 |
| 118 def method_descriptor(method): |
| 119 return wrap_call( |
| 120 method['func'], is_instance=(method['type'] == 'instance'), |
| 121 result_type=method['result'], string_args=method['string_args'] |
| 122 ) |
| 123 |
| 124 def write_class(file, cls): |
| 125 name = cls['name'] |
| 126 if name in differentiators: |
| 127 print >>file, 'var %s_mapping = %s;' % ( |
| 128 name, json.dumps(differentiators[name]['mapping'], sort_keys=True)) |
| 129 |
| 130 print >>file, 'exports.%s = createClass(%s);' % ( |
| 131 name, 'exports.' + cls['superclass'] if cls['superclass'] else '') |
| 132 for property in cls['properties']: |
| 133 print >>file, 'Object.defineProperty(exports.%s.prototype, "%s", {%s});' %
( |
| 134 name, property['name'], property_descriptor(property)) |
| 135 for method in cls['methods']: |
| 136 obj = ('exports.%s' if method['type'] == 'class' else 'exports.%s.prototyp
e') % name |
| 137 print >>file, '%s.%s = %s;' % ( |
| 138 obj, method['name'], method_descriptor(method)) |
| 139 for initializer in cls['class_initializers']: |
| 140 print >>file, '_%s();' % initializer |
| 141 |
| 142 def handle_class(name): |
| 143 if cls is not None: |
| 144 write_class(file, cls) |
| 145 differentiator = None |
| 146 return { |
| 147 'name': command[1], |
| 148 'superclass': None, |
| 149 'properties': [], |
| 150 'methods': [], |
| 151 'class_initializers': [], |
| 152 } |
| 153 |
| 154 def handle_superclass(name): |
| 155 if cls is None: |
| 156 warnings.warn('Superclass declared outside a class: ' + name) |
| 157 return |
| 158 differentiator = None |
| 159 cls['superclass'] = name |
| 160 |
| 161 def handle_class_init(func): |
| 162 if cls is None: |
| 163 warnings.warn('Class initializer declared outside a class: ' + func) |
| 164 return |
| 165 differentiator = None |
| 166 method = None |
| 167 cls['class_initializers'].append(func) |
| 168 |
| 169 def handle_differentiator(cls, func): |
| 170 differentiator = {'func': func, 'mapping': {}} |
| 171 differentiators[cls] = differentiator |
| 172 return differentiator |
| 173 |
| 174 def handle_differentiator_mapping(value, subclass): |
| 175 if differentiator is None: |
| 176 warnings.warn('Differentiator mapping declared without a differentiator: '
+ subclass) |
| 177 return |
| 178 differentiator['mapping'][value] = subclass |
| 179 |
| 180 def handle_property(type, name, getter, setter): |
| 181 if cls is None: |
| 182 warnings.warn('Property declared outside a class: ' + name) |
| 183 return |
| 184 method = None |
| 185 differentiator = None |
| 186 cls['properties'].append({ |
| 187 'type': type, |
| 188 'name': name, |
| 189 'getter': getter, |
| 190 'setter': setter, |
| 191 }) |
| 192 |
| 193 def handle_method(type, name, func): |
| 194 if cls is None: |
| 195 warnings.warn('Method declared outside a class: ' + name) |
| 196 return |
| 197 differentiator = None |
| 198 method = { |
| 199 'type': type, |
| 200 'name': name, |
| 201 'func': func, |
| 202 'result': 'primitive', |
| 203 'string_args': [], |
| 204 } |
| 205 cls['methods'].append(method) |
| 206 return method |
| 207 |
| 208 def handle_method_result(type): |
| 209 if method is None: |
| 210 warnings.warn('Method result declared without a method definition') |
| 211 return |
| 212 method['result'] = type |
| 213 |
| 214 def handle_string_arg(pos): |
| 215 if method is None: |
| 216 warnings.warn('Argument type declared without a method definition') |
| 217 return |
| 218 method['string_args'].append(int(pos)) |
| 219 |
| 220 with open(API_OUTPUT, 'w') as file: |
| 221 for line in output.splitlines(): |
| 222 match = re.search(r'#pragma\s+comment\((.+?)\)', line) |
| 223 if match: |
| 224 command = match.group(1).strip().split() |
| 225 if command[0] == 'class': |
| 226 cls = handle_class(command[1]) |
| 227 elif command[0] == 'augments': |
| 228 handle_superclass(command[1]) |
| 229 elif command[0] == 'class_init': |
| 230 handle_class_init(command[1]) |
| 231 elif command[0] == 'differentiator': |
| 232 differentiator = handle_differentiator(command[1], command[2]) |
| 233 elif command[0] == 'differentiator_mapping': |
| 234 handle_differentiator_mapping(command[1], command[2]) |
| 235 elif command[0] == 'property': |
| 236 handle_property('primitive', command[1], command[2], command[3] if len
(command) > 3 else None) |
| 237 elif command[0] == 'string_property': |
| 238 handle_property('string', command[1], command[2], command[3] if len(co
mmand) > 3 else None) |
| 239 elif command[0] == 'static_property': |
| 240 handle_property('static', command[1], command[2], None) |
| 241 elif command[0] == 'method': |
| 242 method = handle_method('instance', command[1], command[2]) |
| 243 elif command[0] == 'class_method': |
| 244 method = handle_method('class', command[1], command[2]) |
| 245 elif command[0] == 'string_result': |
| 246 handle_method_result('string') |
| 247 elif command[0] == 'pointer_result': |
| 248 handle_method_result(command[1]) |
| 249 elif command[0] == 'string_arg': |
| 250 handle_string_arg(command[1]) |
| 251 else: |
| 252 warnings.warn('Unknown declaration: ' + str(command)) |
| 253 |
| 254 if cls is not None: |
| 255 write_class(file, cls) |
| 256 |
| 257 def run_compiler(env): |
| 258 params = [ |
| 259 os.path.join(env['EMSCRIPTEN'], 'emcc'), |
| 260 '-o', COMPILER_OUTPUT, |
| 261 '--post-js', API_OUTPUT, |
| 262 ] |
| 263 params.extend(SOURCE_FILES) |
| 264 params.extend('-D' + flag for flag in DEFINES) |
| 265 for key, value in GENERATION_PARAMS.iteritems(): |
| 266 params.extend(['-s', '%s=%s' % (key, str(value))]) |
| 267 params.extend(ADDITIONAL_PARAMS) |
| 268 subprocess.check_call(params, env=env) |
| 269 |
| 270 if __name__ == '__main__': |
| 271 env = getenv() |
| 272 generate_api(env) |
| 273 run_compiler(env) |
OLD | NEW |