2 * Copyright 2014 Google Inc.
4 * Use of this source code is governed by a BSD-style license that can be
5 * found in the LICENSE file.
8 #include "SkCommandLineFlags.h"
9 #include "SkForceLinking.h"
10 #include "SkGraphics.h"
12 #include "SkPicture.h"
13 #include "SkPictureRecorder.h"
17 #include "LazyDecodeBitmap.h"
21 __SK_FORCE_IMAGE_DECODER_LINKING;
23 DEFINE_string2(skps, r, "skps", "Directory containing SKPs to read and re-record.");
24 DEFINE_int32(samples, 10, "Number of times to re-record each SKP.");
25 DEFINE_int32(tileGridSize, 512, "Set the tile grid size. Has no effect if bbh is not set to tilegrid.");
26 DEFINE_string(bbh, "", "Turn on the bbh and select the type, one of rtree, tilegrid, quadtree");
27 DEFINE_bool(skr, false, "Record SKR instead of SKP.");
28 DEFINE_string(match, "", "The usual filters on file names of SKPs to bench.");
29 DEFINE_string(timescale, "us", "Print times in ms, us, or ns");
30 DEFINE_double(overheadGoal, 0.0001,
31 "Try to make timer overhead at most this fraction of our sample measurements.");
32 DEFINE_int32(verbose, 0, "0: print min sample; "
33 "1: print min, mean, max and noise indication "
34 "2: print all samples");
36 static double timescale() {
37 if (FLAGS_timescale.contains("us")) return 1000;
38 if (FLAGS_timescale.contains("ns")) return 1000000;
42 static SkBBHFactory* parse_FLAGS_bbh() {
43 if (FLAGS_bbh.isEmpty()) {
47 if (FLAGS_bbh.contains("rtree")) {
48 return SkNEW(SkRTreeFactory);
50 if (FLAGS_bbh.contains("tilegrid")) {
51 SkTileGridFactory::TileGridInfo info;
52 info.fTileInterval.set(FLAGS_tileGridSize, FLAGS_tileGridSize);
53 info.fMargin.setEmpty();
54 info.fOffset.setZero();
55 return SkNEW_ARGS(SkTileGridFactory, (info));
57 if (FLAGS_bbh.contains("quadtree")) {
58 return SkNEW(SkQuadTreeFactory);
60 SkDebugf("Invalid bbh type %s, must be one of rtree, tilegrid, quadtree.\n", FLAGS_bbh[0]);
64 static void rerecord(const SkPicture& src, SkBBHFactory* bbhFactory) {
65 SkPictureRecorder recorder;
67 src.draw(recorder.EXPERIMENTAL_beginRecording(src.width(), src.height(), bbhFactory));
69 src.draw(recorder.beginRecording(src.width(), src.height(), bbhFactory));
71 SkAutoTUnref<SkPicture> pic(recorder.endRecording());
74 static void bench_record(const SkPicture& src,
75 const double timerOverhead,
77 SkBBHFactory* bbhFactory) {
78 // Rerecord once to warm up any caches. Otherwise the first sample can be very noisy.
79 rerecord(src, bbhFactory);
81 // Rerecord once to see how many times we should loop to make timer overhead insignificant.
83 const double scale = timescale();
86 rerecord(src, bbhFactory);
88 } while (timer.fWall * scale < timerOverhead); // Loop just in case something bizarre happens.
90 // We want (timer overhead / measurement) to be less than FLAGS_overheadGoal.
91 // So in each sample, we'll loop enough times to have made that true for our first measurement.
92 const int loops = (int)ceil(timerOverhead / timer.fWall / FLAGS_overheadGoal);
94 SkAutoTMalloc<double> samples(FLAGS_samples);
95 for (int i = 0; i < FLAGS_samples; i++) {
97 for (int j = 0; j < loops; j++) {
98 rerecord(src, bbhFactory);
101 samples[i] = timer.fWall * scale / loops;
104 Stats stats(samples.get(), FLAGS_samples);
105 if (FLAGS_verbose == 0) {
106 printf("%g\t%s\n", stats.min, name);
107 } else if (FLAGS_verbose == 1) {
108 // Get a rough idea of how noisy the measurements were.
109 const double noisePercent = 100 * sqrt(stats.var) / stats.mean;
110 printf("%g\t%g\t%g\t±%.0f%%\t%s\n", stats.min, stats.mean, stats.max, noisePercent, name);
111 } else if (FLAGS_verbose == 2) {
113 for (int i = 0; i < FLAGS_samples; i++) {
114 printf("\t%g", samples[i]);
120 int tool_main(int argc, char** argv);
121 int tool_main(int argc, char** argv) {
122 SkCommandLineFlags::Parse(argc, argv);
123 SkAutoGraphics autoGraphics;
125 if (FLAGS_bbh.count() > 1) {
126 SkDebugf("Multiple bbh arguments supplied.\n");
130 SkAutoTDelete<SkBBHFactory> bbhFactory(parse_FLAGS_bbh());
132 // Each run will use this timer overhead estimate to guess how many times it should run.
133 static const int kOverheadLoops = 10000000;
135 double overheadEstimate = 0.0;
136 const double scale = timescale();
137 for (int i = 0; i < kOverheadLoops; i++) {
140 overheadEstimate += timer.fWall * scale;
142 overheadEstimate /= kOverheadLoops;
144 SkOSFile::Iter it(FLAGS_skps[0], ".skp");
147 while (it.next(&filename)) {
148 if (SkCommandLineFlags::ShouldSkip(FLAGS_match, filename.c_str())) {
152 const SkString path = SkOSPath::SkPathJoin(FLAGS_skps[0], filename.c_str());
154 SkAutoTUnref<SkStream> stream(SkStream::NewFromFile(path.c_str()));
156 SkDebugf("Could not read %s.\n", path.c_str());
160 SkAutoTUnref<SkPicture> src(
161 SkPicture::CreateFromStream(stream, sk_tools::LazyDecodeBitmap));
163 SkDebugf("Could not read %s as an SkPicture.\n", path.c_str());
167 bench_record(*src, overheadEstimate, filename.c_str(), bbhFactory.get());
169 return failed ? 1 : 0;
172 #if !defined SK_BUILD_FOR_IOS
173 int main(int argc, char * const argv[]) {
174 return tool_main(argc, (char**) argv);