Add adb support to skpbench

BUG=skia:
GOLD_TRYBOT_URL= https://gold.skia.org/search?issue=2350003002

Review-Url: https://codereview.chromium.org/2350003002

1 files changed, 33 insertions, 0 deletions

diff --git a/tools/skpbench/_adb_path.py b/tools/skpbench/_adb_path.py
new file mode 100644
index 0000000000..377ba12490
--- /dev/null
+++ b/tools/skpbench/_adb_path.py
@@ -0,0 +1,33 @@
+# Copyright 2016 Google Inc.
+#
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+import _adb
+import re
+import subprocess
+
+__ADB_DEVICE_SERIAL = None
+
+def set_device_serial(device_serial):
+  global __ADB_DEVICE_SERIAL
+  __ADB_DEVICE_SERIAL = device_serial
+
+def join(*pathnames):
+  return '/'.join(pathnames)
+
+def basename(pathname):
+  return pathname.rsplit('/', maxsplit=1)[-1]
+
+def find_skps(skps):
+  escapedskps = [re.sub(r'([^a-zA-Z0-9_\*\?\[\!\]])', r'\\\1', x) # Keep globs.
+                 for x in skps]
+  pathnames = _adb.check('''
+    for PATHNAME in %s; do
+      if [ -d "$PATHNAME" ]; then
+        ls "$PATHNAME"/*.skp
+      else
+        echo "$PATHNAME"
+      fi
+    done''' % ' '.join(escapedskps), device_serial=__ADB_DEVICE_SERIAL)
+  return re.split('[\r\n]+', pathnames)