Added gflags switches: duration, tests, save, out.

client/deps/glbench/src/testbase.cc

 // Copyright (c) 2010 The Chromium OS Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
+#include <gflags/gflags.h>
 #include <stdio.h>
 
+#include "base/scoped_ptr.h"
+#include "base/file_util.h"
+
 #include "testbase.h"
+#include "utils.h"
+
+DEFINE_bool(save, false, "save images after each test case");
+DEFINE_string(out, "out", "directory to save images");
 
 namespace glbench {
 
 uint64_t TimeTest(TestBase* test, int iter) {
   SwapBuffers();
   glFinish();
   uint64_t time1 = GetUTime();
   if (!test->TestFunc(iter))
     return ~0;
   glFinish();
@@ skipping 37 matching lines @@
   if (count < 2) {
     *slope = 0.f;
     *bias = 0;
   }
   *slope = static_cast<float>(sum_x * sum_y - count * sum_xy) /
     (sum_x * sum_x - count * sum_x2);
   *bias = (sum_x * sum_xy - sum_x2 * sum_y) / (sum_x * sum_x - count * sum_x2);
   return true;
 }
 
-void RunTest(TestBase* test, const char *name,
+void SaveImage(const char* name) {
+  const int size = g_width * g_height * 4;
+  scoped_array<char> pixels(new char[size]);
+  glReadPixels(0, 0, g_width, g_height, GL_RGBA, GL_UNSIGNED_BYTE,
+               pixels.get());
+  FilePath dirname = GetBasePath().Append(FLAGS_out);
+  file_util::CreateDirectory(dirname);
+  FilePath filename = dirname.Append(name);
+  file_util::WriteFile(filename, pixels.get(), size);
+}
+
+void RunTest(TestBase* test, const char* name,
              float coefficient, bool inverse) {
   float slope;
   int64_t bias;
 
   GLenum err = glGetError();
   if (err != 0) {
     printf("# %s failed, glGetError returned 0x%x.\n", name, err);
     // float() in python will happily parse Nan.
     printf("%s: Nan\n", name);
   } else {
     if (Bench(test, &slope, &bias)) {
+      if (FLAGS_save)
+        SaveImage(name);
       printf("%s: %g\n", name, coefficient * (inverse ? 1.f / slope : slope));
     } else {
       printf("# %s is too slow, returning zero.\n", name);
       printf("%s: 0\n", name);
     }
   }
 }
 
 bool DrawArraysTestFunc::TestFunc(int iter) {
   glDrawArrays(GL_TRIANGLE_STRIP, 0, 4);
@@ skipping 46 matching lines @@
 bool DrawElementsTestFunc::TestFunc(int iter) {
   glDrawElements(GL_TRIANGLES, count_, GL_UNSIGNED_INT, 0);
   glFlush();
   for (int i = 0 ; i < iter-1; ++i) {
     glDrawElements(GL_TRIANGLES, count_, GL_UNSIGNED_INT, 0);
   }
   return true;
 }
 
 } // namespace glbench