commit fc37ad168090637973bf21452aa2685cfcbb273c
author halcanary <halcanary@google.com> Fri Jan 30 07:31:19 2015 -0800
committer Commit bot <commit-bot@chromium.org> Fri Jan 30 07:31:19 2015 -0800
tree 9178b8c62e9c04420a85ee11f7ac52f1bb8461a5
parent 5d2fd4465b270ec42d123202e50c33543f9c2083

dm: allow multiple --skp flags, allow single files

Review URL: https://codereview.chromium.org/885353002

dm/DM.cpp

diff --git a/dm/DM.cpp b/dm/DM.cpp
index 900eb58..dd4e48f 100644
--- a/dm/DM.cpp
+++ b/dm/DM.cpp
@@ -122,10 +122,15 @@
     for (const skiagm::GMRegistry* r = skiagm::GMRegistry::Head(); r; r = r->next()) {
         push_src("gm", new GMSrc(r->factory()));
     }
-    if (!FLAGS_skps.isEmpty()) {
-        SkOSFile::Iter it(FLAGS_skps[0], "skp");
-        for (SkString file; it.next(&file); ) {
-            push_src("skp", new SKPSrc(SkOSPath::Join(FLAGS_skps[0], file.c_str())));
+    for (int i = 0; i < FLAGS_skps.count(); i++) {
+        const char* path = FLAGS_skps[i];
+        if (sk_isdir(path)) {
+            SkOSFile::Iter it(path, "skp");
+            for (SkString file; it.next(&file); ) {
+                push_src("skp", new SKPSrc(SkOSPath::Join(path, file.c_str())));
+            }
+        } else {
+            push_src("skp", new SKPSrc(SkString(path)));
         }
     }
     if (!FLAGS_images.isEmpty()) {