| #!/usr/bin/env python |
| # Copyright (c) 2015 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. |
| |
| """This script is meant to be run on a Swarming bot.""" |
| |
| import argparse |
| import os |
| import subprocess |
| import sys |
| |
| |
| PARENT_DIR = os.path.dirname(os.path.realpath(__file__)) |
| |
| REPOS_BASE_DIR = os.path.normpath(os.path.join( |
| PARENT_DIR, os.pardir, os.pardir, os.pardir, os.pardir)) |
| |
| SKIA_SRC_DIR = os.path.join(REPOS_BASE_DIR, 'skia') |
| |
| |
| def main(): |
| parser = argparse.ArgumentParser() |
| parser.add_argument('-s', '--slave_num', required=True, type=int, |
| help='The slave num of this CT run.') |
| parser.add_argument('-t', '--tool', required=True, |
| choices=['dm', 'nanobench', 'get_images_from_skps'], |
| help='The tool to run on the SKPs.') |
| parser.add_argument('-g', '--git_hash', required=True, |
| help='The Skia hash the tool was built at.') |
| parser.add_argument('-c', '--configuration', required=True, |
| help='The build configuration to use.') |
| parser.add_argument('-i', '--isolated_outdir', required=True, |
| help='Swarming will automatically upload to ' |
| 'isolateserver all artifacts in this dir.') |
| parser.add_argument('-b', '--builder', required=True, |
| help='The name of the builder.') |
| args = parser.parse_args() |
| |
| tool_path = os.path.join(PARENT_DIR, args.tool) |
| skps_dir = os.path.join(REPOS_BASE_DIR, 'skps', args.builder, |
| 'slave%d' % args.slave_num) |
| resource_path = os.path.join(SKIA_SRC_DIR, 'resources') |
| |
| cmd = [tool_path] |
| if args.tool == 'dm': |
| # Add DM specific arguments. |
| cmd.extend([ |
| '--src', 'skp', |
| '--skps', skps_dir, |
| '--resourcePath', resource_path, |
| '--config', '8888', |
| '--verbose', |
| ]) |
| elif args.tool == 'nanobench': |
| # Add Nanobench specific arguments. |
| config = '8888' |
| cpu_or_gpu = 'CPU' |
| cpu_or_gpu_value = 'AVX2' |
| if 'GPU' in args.builder: |
| config = 'gpu' |
| cpu_or_gpu = 'GPU' |
| cpu_or_gpu_value = 'GT610' |
| |
| out_results_file = os.path.join( |
| args.isolated_outdir, 'nanobench_%s_%s_slave%d.json' % ( |
| args.git_hash, config, args.slave_num)) |
| cmd.extend([ |
| '--skps', skps_dir, |
| '--match', 'skp', |
| '--resourcePath', resource_path, |
| '--config', config, |
| '--outResultsFile', out_results_file, |
| '--properties', 'gitHash', args.git_hash, |
| '--key', 'arch', 'x86_64', |
| 'compiler', 'GCC', |
| 'cpu_or_gpu', cpu_or_gpu, |
| 'cpu_or_gpu_value', cpu_or_gpu_value, |
| 'model', 'SWARM', |
| 'os', 'Ubuntu', |
| '--verbose', |
| ]) |
| elif args.tool == 'get_images_from_skps': |
| # Add get_images_from_skps specific arguments. |
| img_out = os.path.join(args.isolated_outdir, 'img_out') |
| os.makedirs(img_out) |
| failures_json_out = os.path.join(args.isolated_outdir, 'failures.json') |
| cmd.extend([ |
| '--out', img_out, |
| '--skps', skps_dir, |
| '--failuresJsonPath', failures_json_out, |
| '--writeImages', 'false', |
| '--testDecode', |
| ]) |
| |
| return subprocess.call(cmd) |
| |
| |
| if __name__ == '__main__': |
| sys.exit(main()) |