| #!/usr/bin/env python3 |
| # |
| # Copyright (C) 2021 The Android Open Source Project |
| # |
| # Licensed under the Apache License, Version 2.0 (the "License"); |
| # you may not use this file except in compliance with the License. |
| # You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| |
| """Copies ninja file information to another directory for processing. |
| |
| Usage: |
| ./collect.py [ninja_file] [dest_directory] |
| |
| This script should be used as in preparation for further analysis |
| using difftool.py. See directory-level README for details. |
| """ |
| |
| import argparse |
| import os |
| import pathlib |
| import shutil |
| |
| |
| COLLECTION_INFO_FILENAME = "collection_info" |
| |
| |
| def subninja_files(ninja_file_path): |
| result = [] |
| with ninja_file_path.open() as f: |
| for line in f: |
| if line.startswith("subninja "): |
| result += [line[len("subninja "):].strip()] |
| return result |
| |
| |
| def main(): |
| parser = argparse.ArgumentParser(description="") |
| parser.add_argument("ninja_file", |
| help="the path to the root ninja file of the build " + |
| "to be analyzed. Ex: out/combined-aosp_flame.ninja") |
| parser.add_argument("dest_directory", |
| help="directory to copy build-related information for " + |
| "later difftool comparison. Ex: /tmp/buildArtifacts") |
| # TODO(usta): enable multiple files or even a glob to be specified |
| parser.add_argument("--file", dest="output_file", default=None, |
| help="the path to the output artifact to be analyzed. " + |
| "Ex: out/path/to/foo.so") |
| args = parser.parse_args() |
| dest = args.dest_directory |
| |
| if not os.path.isdir(dest): |
| raise Exception("invalid destination directory " + dest) |
| |
| collection_info_filepath = "" |
| if args.output_file is not None: |
| output_file = pathlib.Path(args.output_file) |
| if not output_file.is_file(): |
| raise Exception("Expected file %s was not found. " % output_file) |
| output_file_dest = pathlib.Path(dest).joinpath(output_file) |
| output_file_dest.parent.mkdir(parents=True, exist_ok=True) |
| shutil.copy2(output_file, output_file_dest) |
| collection_info_filepath = str(output_file) |
| |
| ninja_file = pathlib.Path(args.ninja_file) |
| main_ninja_basename = ninja_file.name |
| shutil.copy2(args.ninja_file, os.path.join(dest, main_ninja_basename)) |
| |
| for subninja_file in subninja_files(ninja_file): |
| parent_dir = pathlib.Path(subninja_file).parent |
| dest_dir = os.path.join(dest, parent_dir) |
| pathlib.Path(dest_dir).mkdir(parents=True, exist_ok=True) |
| shutil.copy2(subninja_file, os.path.join(dest, subninja_file)) |
| |
| collection_info = main_ninja_basename + "\n" + collection_info_filepath |
| pathlib.Path(dest).joinpath(COLLECTION_INFO_FILENAME).write_text(collection_info) |
| |
| |
| if __name__ == "__main__": |
| main() |