Provide an option in bench_pictures to count pixel ram.

tools/bench_pictures_main.cpp

 /*
  * Copyright 2012 Google Inc.
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 
 #include "BenchTimer.h"
 #include "CopyTilesRenderer.h"
 #include "PictureBenchmark.h"
 #include "PictureRenderingFlags.h"
 #include "SkBenchLogger.h"
 #include "SkBitmapFactory.h"
 #include "SkCanvas.h"
 #include "SkFlags.h"
 #include "SkGraphics.h"
 #include "SkImageDecoder.h"
 #include "SkMath.h"
 #include "SkOSFile.h"
 #include "SkPicture.h"
 #include "SkStream.h"
 #include "SkTArray.h"
 #include "picture_utils.h"
 
 
 SkBenchLogger gLogger;
 
 // Flags used by this file, in alphabetical order.
+DEFINE_bool(countRAM, false, "Count the RAM used for bitmap pixels in each skp file");
 DECLARE_bool(deferImageDecoding);
 DEFINE_string(filter, "",
         "type:flag : Enable canvas filtering to disable a paint flag, "
         "use no blur or low quality blur, or use no hinting or "
         "slight hinting. For all flags except AAClip, specify the "
         "type of primitive to effect, or choose all. for AAClip "
         "alone, the filter affects all clips independent of type. "
         "Specific flags are listed above.");
 DEFINE_string(logFile, "", "Destination for writing log output, in addition to stdout.");
 DEFINE_bool(logPerIter, false, "Log each repeat timer instead of mean.");
@@ skipping 118 matching lines @@
     SkFILEStream inputStream;
 
     inputStream.setPath(inputPath.c_str());
     if (!inputStream.isValid()) {
         SkString err;
         err.printf("Could not open file %s\n", inputPath.c_str());
         gLogger.logError(err);
         return false;
     }
 
+    // Since the old picture has been deleted, all pixels should be cleared.
+    SkASSERT(gLruImageCache.getRamUsed() == 0);
+    if (FLAGS_countRAM) {
+        // Set the budget to zero, so all pixels will be kept
+        gLruImageCache.setBudget(0);
+    }
+
     bool success = false;
     SkPicture* picture;
     if (FLAGS_deferImageDecoding) {
         picture = SkNEW_ARGS(SkPicture, (&inputStream, &success, &lazy_decode_bitmap));
     } else {
         picture = SkNEW_ARGS(SkPicture, (&inputStream, &success, &SkImageDecoder::DecodeMemory));
     }
     SkAutoTDelete<SkPicture> ad(picture);
 
     if (!success) {
         SkString err;
         err.printf("Could not read an SkPicture from %s\n", inputPath.c_str());
         gLogger.logError(err);
         return false;
     }
 
     SkString filename;
     sk_tools::get_basename(&filename, inputPath);
 
     SkString result;
     result.printf("running bench [%i %i] %s ", picture->width(), picture->height(),
                   filename.c_str());
     gLogger.logProgress(result);
 
     benchmark.run(picture);
+
+    if (FLAGS_countRAM) {
+        SkDebugf("RAM used for bitmaps: ");
+        size_t bytes = gLruImageCache.getRamUsed();
+        if (bytes > 1024) {
+            size_t kb = bytes / 1024;
+            if (kb > 1024) {
+                size_t mb = kb / 1024;
+                SkDebugf("%i MB\n", mb);
+            } else {
+                SkDebugf("%i KB\n", kb);
+            }
+        } else {
+            SkDebugf("%i bytes\n", bytes);
+        }
+    }
+
     return true;
 }
 
 static void setup_benchmark(sk_tools::PictureBenchmark* benchmark) {
     sk_tools::PictureRenderer::DrawFilterFlags drawFilters[SkDrawFilter::kTypeCount];
     sk_bzero(drawFilters, sizeof(drawFilters));
 
     if (FLAGS_filter.count() > 0) {
         const char* filters = FLAGS_filter[0];
         const char* colon = strchr(filters, ':');
@@ skipping 195 matching lines @@
         return 1;
     }
     return 0;
 }
 
 #if !defined SK_BUILD_FOR_IOS
 int main(int argc, char * const argv[]) {
     return tool_main(argc, (char**) argv);
 }
 #endif