Move ct_skps recipe from tools repo to Skia repo

infra/bots/ct/run_ct_skps.py

Index: infra/bots/ct/run_ct_skps.py
diff --git a/infra/bots/ct/run_ct_skps.py b/infra/bots/ct/run_ct_skps.py
new file mode 100755
index 0000000000000000000000000000000000000000..48f2e7d3d9b91f013c8eff1bdb66b5be9a3f320c
--- /dev/null
+++ b/infra/bots/ct/run_ct_skps.py
@@ -0,0 +1,100 @@
+#!/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())