Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
| 2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
| 4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
| 5 | 5 |
| 6 """ | 6 """ |
| 7 Generates shim headers that mirror the directory structure of bundled headers, | 7 Generates shim headers that mirror the directory structure of bundled headers, |
| 8 but just forward to the system ones. | 8 but just forward to the system ones. |
| 9 | 9 |
| 10 This allows seamless compilation against system headers with no changes | 10 This allows seamless compilation against system headers with no changes |
| 11 to our source code. | 11 to our source code. |
| 12 """ | 12 """ |
| 13 | 13 |
| 14 | 14 |
| 15 import optparse | 15 import optparse |
| 16 import os.path | 16 import os.path |
| 17 import sys | 17 import sys |
| 18 | 18 |
| 19 | 19 |
| 20 SHIM_TEMPLATE = """ | 20 SHIM_TEMPLATE = """ |
| 21 #if defined(OFFICIAL_BUILD) | 21 #if defined(OFFICIAL_BUILD) |
| 22 #error shim headers must not be used in official builds! | 22 #error shim headers must not be used in official builds! |
| 23 #endif | 23 #endif |
| 24 | |
| 25 #include <%s> | |
| 26 """ | 24 """ |
| 27 | 25 |
| 28 | 26 |
| 29 def GeneratorMain(argv): | 27 def GeneratorMain(argv): |
| 30 parser = optparse.OptionParser() | 28 parser = optparse.OptionParser() |
| 31 parser.add_option('--headers-root') | 29 parser.add_option('--headers-root') |
| 30 parser.add_option('--additional-headers-root', action='append', default=[]) | |
|
greggman
2013/01/11 02:09:18
why not just make --headers-root be action='append
Paweł Hajdan Jr.
2013/01/11 21:25:15
Done.
| |
| 32 parser.add_option('--output-directory') | 31 parser.add_option('--output-directory') |
| 32 parser.add_option('--use-include-next', action='store_true') | |
| 33 parser.add_option('--outputs', action='store_true') | 33 parser.add_option('--outputs', action='store_true') |
| 34 parser.add_option('--generate', action='store_true') | 34 parser.add_option('--generate', action='store_true') |
| 35 | 35 |
| 36 options, args = parser.parse_args(argv) | 36 options, args = parser.parse_args(argv) |
| 37 | 37 |
| 38 if not options.headers_root: | 38 if not options.headers_root: |
| 39 parser.error('Missing --headers-root parameter.') | 39 parser.error('Missing --headers-root parameter.') |
| 40 if not options.output_directory: | 40 if not options.output_directory: |
| 41 parser.error('Missing --output-directory parameter.') | 41 parser.error('Missing --output-directory parameter.') |
| 42 if not args: | 42 if not args: |
| 43 parser.error('Missing arguments - header file names.') | 43 parser.error('Missing arguments - header file names.') |
| 44 | 44 |
| 45 source_tree_root = os.path.abspath( | 45 source_tree_root = os.path.abspath( |
| 46 os.path.join(os.path.dirname(__file__), '..', '..')) | 46 os.path.join(os.path.dirname(__file__), '..', '..')) |
| 47 | 47 |
| 48 target_directory = os.path.join( | 48 for root in [options.headers_root] + options.additional_headers_root: |
| 49 options.output_directory, | 49 target_directory = os.path.join( |
| 50 os.path.relpath(options.headers_root, source_tree_root)) | 50 options.output_directory, |
| 51 if options.generate and not os.path.exists(target_directory): | 51 os.path.relpath(root, source_tree_root)) |
| 52 os.makedirs(target_directory) | 52 if options.generate and not os.path.exists(target_directory): |
| 53 for header_filename in args: | 53 os.makedirs(target_directory) |
| 54 if options.outputs: | 54 |
| 55 yield os.path.join(target_directory, header_filename) | 55 for header_spec in args: |
| 56 if options.generate: | 56 if ';' in header_spec: |
| 57 with open(os.path.join(target_directory, header_filename), 'w') as f: | 57 (header_filename, |
| 58 f.write(SHIM_TEMPLATE % header_filename) | 58 include_before, |
| 59 include_after) = header_spec.split(';', 2) | |
| 60 else: | |
| 61 header_filename = header_spec | |
| 62 include_before = '' | |
| 63 include_after = '' | |
| 64 if options.outputs: | |
| 65 yield os.path.join(target_directory, header_filename) | |
| 66 if options.generate: | |
| 67 with open(os.path.join(target_directory, header_filename), 'w') as f: | |
| 68 f.write(SHIM_TEMPLATE) | |
| 69 | |
| 70 if include_before: | |
| 71 for header in include_before.split(':'): | |
| 72 f.write('#include %s\n' % header) | |
| 73 | |
| 74 if options.use_include_next: | |
| 75 f.write('#include_next <%s>\n' % header_filename) | |
| 76 else: | |
| 77 f.write('#include <%s>\n' % header_filename) | |
| 78 | |
| 79 if include_after: | |
| 80 for header in include_after.split(':'): | |
| 81 f.write('#include %s\n' % header) | |
| 59 | 82 |
| 60 | 83 |
| 61 def DoMain(argv): | 84 def DoMain(argv): |
| 62 return '\n'.join(GeneratorMain(argv)) | 85 return '\n'.join(GeneratorMain(argv)) |
| 63 | 86 |
| 64 | 87 |
| 65 if __name__ == '__main__': | 88 if __name__ == '__main__': |
| 66 DoMain(sys.argv[1:]) | 89 DoMain(sys.argv[1:]) |
| OLD | NEW |