aboutsummaryrefslogtreecommitdiff
path: root/orderfile/post_process_orderfile.py
diff options
context:
space:
mode:
Diffstat (limited to 'orderfile/post_process_orderfile.py')
-rwxr-xr-xorderfile/post_process_orderfile.py96
1 files changed, 49 insertions, 47 deletions
diff --git a/orderfile/post_process_orderfile.py b/orderfile/post_process_orderfile.py
index 3db0b3b8..5f52aa6f 100755
--- a/orderfile/post_process_orderfile.py
+++ b/orderfile/post_process_orderfile.py
@@ -1,6 +1,6 @@
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
-# Copyright 2019 The Chromium OS Authors. All rights reserved.
+# Copyright 2019 The ChromiumOS Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
@@ -16,10 +16,9 @@ Builtin functions and put them after the input symbols.
in begin and end of the file.
The results of the file is intended to be uploaded and consumed when linking
-Chrome in Chrome OS.
+Chrome in ChromeOS.
"""
-from __future__ import division, print_function
import argparse
import os
@@ -27,65 +26,68 @@ import sys
def _parse_nm_output(stream):
- for line in (line.rstrip() for line in stream):
- if not line:
- continue
+ for line in (line.rstrip() for line in stream):
+ if not line:
+ continue
- pieces = line.split()
- if len(pieces) != 3:
- continue
+ pieces = line.split()
+ if len(pieces) != 3:
+ continue
- _, ty, symbol = pieces
- if ty not in 'tT':
- continue
+ _, ty, symbol = pieces
+ if ty not in "tT":
+ continue
- # We'll sometimes see synthesized symbols that start with $. There isn't
- # much we can do about or with them, regrettably.
- if symbol.startswith('$'):
- continue
+ # We'll sometimes see synthesized symbols that start with $. There isn't
+ # much we can do about or with them, regrettably.
+ if symbol.startswith("$"):
+ continue
- yield symbol
+ yield symbol
def _remove_duplicates(iterable):
- seen = set()
- for item in iterable:
- if item in seen:
- continue
- seen.add(item)
- yield item
+ seen = set()
+ for item in iterable:
+ if item in seen:
+ continue
+ seen.add(item)
+ yield item
def run(c3_ordered_stream, chrome_nm_stream, output_stream):
- head_marker = 'chrome_begin_ordered_code'
- tail_marker = 'chrome_end_ordered_code'
-
- c3_ordered_syms = [x.strip() for x in c3_ordered_stream.readlines()]
- all_chrome_syms = set(_parse_nm_output(chrome_nm_stream))
- # Sort by name, so it's predictable. Otherwise, these should all land in the
- # same hugepage anyway, so order doesn't matter as much.
- builtin_syms = sorted(s for s in all_chrome_syms if s.startswith('Builtins_'))
- output = _remove_duplicates([head_marker] + c3_ordered_syms + builtin_syms +
- [tail_marker])
- output_stream.write('\n'.join(output))
+ head_marker = "chrome_begin_ordered_code"
+ tail_marker = "chrome_end_ordered_code"
+
+ c3_ordered_syms = [x.strip() for x in c3_ordered_stream.readlines()]
+ all_chrome_syms = set(_parse_nm_output(chrome_nm_stream))
+ # Sort by name, so it's predictable. Otherwise, these should all land in the
+ # same hugepage anyway, so order doesn't matter as much.
+ builtin_syms = sorted(
+ s for s in all_chrome_syms if s.startswith("Builtins_")
+ )
+ output = _remove_duplicates(
+ [head_marker] + c3_ordered_syms + builtin_syms + [tail_marker]
+ )
+ output_stream.write("\n".join(output))
def main(argv):
- parser = argparse.ArgumentParser()
- parser.add_argument('--chrome_nm', required=True, dest='chrome_nm')
- parser.add_argument('--input', required=True, dest='input_file')
- parser.add_argument('--output', required=True, dest='output_file')
+ parser = argparse.ArgumentParser()
+ parser.add_argument("--chrome_nm", required=True, dest="chrome_nm")
+ parser.add_argument("--input", required=True, dest="input_file")
+ parser.add_argument("--output", required=True, dest="output_file")
- options = parser.parse_args(argv)
+ options = parser.parse_args(argv)
- if not os.path.exists(options.input_file):
- sys.exit("Input orderfile doesn\'t exist.")
+ if not os.path.exists(options.input_file):
+ sys.exit("Input orderfile doesn't exist.")
- with open(options.input_file) as in_stream, \
- open(options.chrome_nm) as chrome_nm_stream, \
- open(options.output_file, 'w') as out_stream:
- run(in_stream, chrome_nm_stream, out_stream)
+ with open(options.input_file) as in_stream, open(
+ options.chrome_nm
+ ) as chrome_nm_stream, open(options.output_file, "w") as out_stream:
+ run(in_stream, chrome_nm_stream, out_stream)
-if __name__ == '__main__':
- sys.exit(main(sys.argv[1:]))
+if __name__ == "__main__":
+ sys.exit(main(sys.argv[1:]))