1 files changed, 34 insertions, 0 deletions
diff --git a/tools/skqp/remove_unneeded_assets b/tools/skqp/remove_unneeded_assets
new file mode 100755
index 0000000000..67bf9cff03
--- /dev/null
+++ b/tools/skqp/remove_unneeded_assets
@@ -0,0 +1,34 @@
+#! /usr/bin/env python
+
+# Copyright 2018 Google Inc.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+import os
+import shutil
+import sys
+
+def gset(path):
+    s = set()
+    if os.path.isfile(path):
+        with open(path, 'r') as f:
+            for line in f:
+                s.add(line.strip())
+    return s
+
+def main():
+    assets = os.path.join('platform_tools', 'android', 'apps', 'skqp', 'src', 'main', 'assets')
+    os.chdir(os.path.join(os.path.dirname(__file__), os.pardir, os.pardir, assets))
+    known = gset('skqp/KnownGMs.txt')
+    nope = gset('skqp/DoNotScoreInCompatibilityTestMode.txt')
+    present = set(os.listdir('gmkb'))
+    to_delete = present & nope
+    if (known):
+        to_delete |= (present - known)
+    for x in to_delete:
+        shutil.rmtree(os.path.join('gmkb', x))
+    sys.stdout.write('%s: %d of %d models removed\n' %(sys.argv[0], len(to_delete), len(present)))
+
+if __name__ == '__main__':
+    main()
+