Chromium Code Reviews| Index: build/android/gyp/emma_instr.py |
| diff --git a/build/android/gyp/emma_instr.py b/build/android/gyp/emma_instr.py |
| new file mode 100755 |
| index 0000000000000000000000000000000000000000..bb5fe3eecbf7f45ef57a8de5f1dc77b566b0236e |
| --- /dev/null |
| +++ b/build/android/gyp/emma_instr.py |
| @@ -0,0 +1,185 @@ |
| +#!/usr/bin/env python |
| +# |
| +# Copyright 2013 The Chromium Authors. All rights reserved. |
| +# Use of this source code is governed by a BSD-style license that can be |
| +# found in the LICENSE file. |
| + |
| +"""Instruments classes and jar files. |
| + |
| +This script corresponds to the 'emma_instr' action in the java build process. |
| +Depending on whether emma_instrument is set, the 'emma_instr' action will either |
| +call one of the instrument commands, or the copy command. |
| + |
| +Possible commands are: |
| +- instrument_jar: Accepts a jar and instruments it using emma.jar. |
| +- instrument_classes: Accepts a directory contains java classes and instruments |
| + it using emma.jar. |
| +- copy: Triggered instead of an instrumentation command when we don't have EMMA |
| + coverage enabled. This allows us to make this a required step without |
| + necessarily instrumenting on every build. |
| +""" |
| + |
| +import collections |
| +import json |
| +import os |
| +import shutil |
| +import sys |
| + |
| +sys.path.append(os.path.join(os.path.dirname(__file__), os.pardir)) |
| +from pylib.utils import command_option_parser |
| + |
| +from util import build_utils |
| + |
| + |
| +def _AddCommonOptions(option_parser): |
| + """Adds common options to |option_parser|.""" |
| + option_parser.add_option('--input-path', |
| + help=('Path to input file(s). Either the classes ' |
| + 'directory, or the path to a jar.')) |
| + option_parser.add_option('--output-path', |
| + help=('Path to output final file(s) to. Either the ' |
| + 'final classes directory, or the directory in ' |
| + 'which to place the instrumented/copied jar.')) |
| + option_parser.add_option('--stamp', help='Path to touch when done.') |
| + |
| + |
| +def _AddInstrumentOptions(option_parser): |
| + """Adds options related to instrumentation to |option_parser|.""" |
| + _AddCommonOptions(option_parser) |
| + option_parser.add_option('--coverage-file', |
| + help='File to create with coverage metadata') |
| + option_parser.add_option('--sources-file', |
| + help='File to create with the list of sources.') |
| + option_parser.add_option('--sources', |
| + help='Space separated list of sources.') |
| + option_parser.add_option('--src-root', |
| + help='Root of the src repository.') |
| + option_parser.add_option('--sdk-root', |
| + help='Root of the Android SDK.') |
| + option_parser.add_option('--intermediate-dir', |
| + help=('Intermediate directory in which the ' |
| + 'instrumented jar will be produced in.')) |
| + |
| + |
| +def _RunCopyCommand(command, options, args, option_parser): |
| + """Just copies the jar from input to output locations. |
| + |
| + Args: |
| + command: String indicating the command that was received to trigger |
| + this function. |
| + options: optparse options dictionary. |
| + args: List of extra args from optparse. |
| + option_parser: optparse.OptionParser object. |
| + |
| + Returns: |
| + An exit code. |
| + """ |
| + if not (options.input_path and options.output_path): |
| + option_parser.error('All arguments are required.') |
| + |
| + if os.path.isdir(options.input_path): |
| + shutil.rmtree(options.output_path, ignore_errors=True) |
| + shutil.copytree(options.input_path, options.output_path) |
| + else: |
| + shutil.copy(options.input_path, options.output_path) |
| + |
| + if options.stamp: |
| + build_utils.Touch(options.stamp) |
| + |
| + |
| +def _CreateSourcesFile(sources_string, sources_file, src_root): |
| + """Adds all normalized source directories to |sources_file|. |
| + |
| + Args: |
| + sources_string: String generated from gyp containing the list of sources. |
| + sources_file: File into which to write the JSON list of sources. |
| + src_root: Root which sources added to the file should be relative to. |
| + |
| + Returns: |
| + An exit code. |
| + """ |
| + src_root = os.path.abspath(src_root) |
| + sources = build_utils.ParseGypList(sources_string) |
| + relative_sources = [] |
| + for s in sources: |
| + abs_source = os.path.abspath(s) |
| + if abs_source[:len(src_root)] != src_root: |
| + print ('Error: found source directory not under repository root: %s %s' |
| + % (abs_source, src_root)) |
| + return 1 |
| + rel_source = os.path.relpath(abs_source, src_root) |
| + |
| + relative_sources.append(rel_source) |
| + |
| + with open(sources_file, 'w') as f: |
| + json.dump(relative_sources, f) |
| + |
| + |
| +def _RunInstrumentCommand(command, options, args, option_parser): |
| + """Instruments the classes/jar files using EMMA. |
| + |
| + Args: |
| + command: 'instrument_jar' or 'instrument_classes'. This distinguishes |
| + whether we copy the output from the created lib/ directory, or classes/ |
| + directory. |
| + options: optparse options dictionary. |
| + args: List of extra args from optparse. |
| + option_parser: optparse.OptionParser object. |
| + |
| + Returns: |
| + An exit code. |
| + """ |
| + if not (options.input_path and options.output_path and |
| + options.intermediate_dir and options.coverage_file and |
| + options.sources_file and options.sources and options.src_root and |
| + options.sdk_root): |
| + option_parser.error('All arguments are required.') |
| + |
| + # TODO(gkanwar): Add '-ix' option to filter out useless classes. |
| + build_utils.CheckCallDie(['java', '-cp', |
| + os.path.join(options.sdk_root, 'tools', |
| + 'lib', 'emma.jar'), |
|
cjhopman
2013/08/21 17:37:48
Nit: It would be better if this path were construc
gkanwar1
2013/08/21 18:28:32
Done.
|
| + 'emma', 'instr', |
| + '-ip', options.input_path, |
| + '-d', options.intermediate_dir, |
| + '-out', options.coverage_file, |
| + '-m', 'fullcopy'], suppress_output=True) |
| + |
| + if command == 'instrument_jar': |
|
cjhopman
2013/08/21 17:37:48
It looks like the intermediate dir here does not n
gkanwar1
2013/08/21 18:28:32
Done.
|
| + for jar in os.listdir(os.path.join(options.intermediate_dir, 'lib')): |
| + shutil.copy(os.path.join(options.intermediate_dir, 'lib', jar), |
| + options.output_path) |
| + else: # 'instrument_classes' |
| + if os.path.isdir(options.output_path): |
| + shutil.rmtree(options.output_path, ignore_errors=True) |
| + shutil.copytree(os.path.join(options.intermediate_dir, 'classes'), |
| + options.output_path) |
| + |
| + _CreateSourcesFile(options.sources, options.sources_file, options.src_root) |
| + |
| + if options.stamp: |
| + build_utils.Touch(options.stamp) |
| + |
| + return 0 |
| + |
| + |
| +CommandFunctionTuple = collections.namedtuple( |
| + 'CommandFunctionTuple', ['add_options_func', 'run_command_func']) |
| +VALID_COMMANDS = { |
| + 'copy': CommandFunctionTuple(_AddCommonOptions, |
| + _RunCopyCommand), |
| + 'instrument_jar': CommandFunctionTuple(_AddInstrumentOptions, |
| + _RunInstrumentCommand), |
| + 'instrument_classes': CommandFunctionTuple(_AddInstrumentOptions, |
| + _RunInstrumentCommand), |
| +} |
| + |
| + |
| +def main(argv): |
| + option_parser = command_option_parser.CommandOptionParser( |
| + commands_dict=VALID_COMMANDS) |
| + command_option_parser.ParseAndExecute(option_parser) |
| + |
| + |
| +if __name__ == '__main__': |
| + sys.exit(main(sys.argv)) |