Add benchmarkstream to visualbench

BUG=skia:

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

1 files changed, 37 insertions, 0 deletions

diff --git a/tools/VisualBench/VisualBenchmarkStream.h b/tools/VisualBench/VisualBenchmarkStream.h
new file mode 100644
index 0000000000..4c6a1eb262
--- /dev/null
+++ b/tools/VisualBench/VisualBenchmarkStream.h
@@ -0,0 +1,37 @@
+/*
+ * Copyright 2015 Google Inc.
+ *
+ * Use of this source code is governed by a BSD-style license that can be
+ * found in the LICENSE file.
+ *
+ */
+
+#ifndef VisualBenchmarkStream_DEFINED
+#define VisualBenchmarkStream_DEFINED
+
+#include "Benchmark.h"
+#include "gm.h"
+#include "SkCommandLineFlags.h"
+#include "SkPicture.h"
+
+DECLARE_string(match);
+
+class VisualBenchmarkStream {
+public:
+    VisualBenchmarkStream();
+
+    static bool ReadPicture(const char* path, SkAutoTUnref<SkPicture>* pic);
+
+    Benchmark* next();
+
+private:
+    const BenchRegistry* fBenches;
+    const skiagm::GMRegistry* fGMs;
+    SkTArray<SkString> fSKPs;
+
+    const char* fSourceType;  // What we're benching: bench, GM, SKP, ...
+    const char* fBenchType;   // How we bench it: micro, playback, ...
+    int fCurrentSKP;
+};
+
+#endif