1 /*
2 * Copyright 2014 Google Inc.
3 *
4 * Use of this source code is governed by a BSD-style license that can be
5 * found in the LICENSE file.
6 */
7
8 #include "SkCommandLineFlags.h"
9 #include "SkForceLinking.h"
10 #include "SkGraphics.h"
11 #include "SkOSFile.h"
12 #include "SkPicture.h"
13 #include "SkPictureRecorder.h"
14 #include "SkStream.h"
15 #include "SkString.h"
16
17 #include "LazyDecodeBitmap.h"
18 #include "Stats.h"
19 #include "Timer.h"
20
21 __SK_FORCE_IMAGE_DECODER_LINKING;
22
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");
35
timescale()36 static double timescale() {
37 if (FLAGS_timescale.contains("us")) return 1000;
38 if (FLAGS_timescale.contains("ns")) return 1000000;
39 return 1;
40 }
41
parse_FLAGS_bbh()42 static SkBBHFactory* parse_FLAGS_bbh() {
43 if (FLAGS_bbh.isEmpty()) {
44 return NULL;
45 }
46
47 if (FLAGS_bbh.contains("rtree")) {
48 return SkNEW(SkRTreeFactory);
49 }
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));
56 }
57 if (FLAGS_bbh.contains("quadtree")) {
58 return SkNEW(SkQuadTreeFactory);
59 }
60 SkDebugf("Invalid bbh type %s, must be one of rtree, tilegrid, quadtree.\n", FLAGS_bbh[0]);
61 return NULL;
62 }
63
rerecord(const SkPicture & src,SkBBHFactory * bbhFactory)64 static void rerecord(const SkPicture& src, SkBBHFactory* bbhFactory) {
65 SkPictureRecorder recorder;
66 if (FLAGS_skr) {
67 src.draw(recorder.EXPERIMENTAL_beginRecording(src.width(), src.height(), bbhFactory));
68 } else {
69 src.draw(recorder.beginRecording(src.width(), src.height(), bbhFactory));
70 }
71 SkAutoTUnref<SkPicture> pic(recorder.endRecording());
72 }
73
bench_record(const SkPicture & src,const double timerOverhead,const char * name,SkBBHFactory * bbhFactory)74 static void bench_record(const SkPicture& src,
75 const double timerOverhead,
76 const char* name,
77 SkBBHFactory* bbhFactory) {
78 // Rerecord once to warm up any caches. Otherwise the first sample can be very noisy.
79 rerecord(src, bbhFactory);
80
81 // Rerecord once to see how many times we should loop to make timer overhead insignificant.
82 WallTimer timer;
83 const double scale = timescale();
84 do {
85 timer.start();
86 rerecord(src, bbhFactory);
87 timer.end();
88 } while (timer.fWall * scale < timerOverhead); // Loop just in case something bizarre happens.
89
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);
93
94 SkAutoTMalloc<double> samples(FLAGS_samples);
95 for (int i = 0; i < FLAGS_samples; i++) {
96 timer.start();
97 for (int j = 0; j < loops; j++) {
98 rerecord(src, bbhFactory);
99 }
100 timer.end();
101 samples[i] = timer.fWall * scale / loops;
102 }
103
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) {
112 printf("%s", name);
113 for (int i = 0; i < FLAGS_samples; i++) {
114 printf("\t%g", samples[i]);
115 }
116 printf("\n");
117 }
118 }
119
120 int tool_main(int argc, char** argv);
tool_main(int argc,char ** argv)121 int tool_main(int argc, char** argv) {
122 SkCommandLineFlags::Parse(argc, argv);
123 SkAutoGraphics autoGraphics;
124
125 if (FLAGS_bbh.count() > 1) {
126 SkDebugf("Multiple bbh arguments supplied.\n");
127 return 1;
128 }
129
130 SkAutoTDelete<SkBBHFactory> bbhFactory(parse_FLAGS_bbh());
131
132 // Each run will use this timer overhead estimate to guess how many times it should run.
133 static const int kOverheadLoops = 10000000;
134 WallTimer timer;
135 double overheadEstimate = 0.0;
136 const double scale = timescale();
137 for (int i = 0; i < kOverheadLoops; i++) {
138 timer.start();
139 timer.end();
140 overheadEstimate += timer.fWall * scale;
141 }
142 overheadEstimate /= kOverheadLoops;
143
144 SkOSFile::Iter it(FLAGS_skps[0], ".skp");
145 SkString filename;
146 bool failed = false;
147 while (it.next(&filename)) {
148 if (SkCommandLineFlags::ShouldSkip(FLAGS_match, filename.c_str())) {
149 continue;
150 }
151
152 const SkString path = SkOSPath::SkPathJoin(FLAGS_skps[0], filename.c_str());
153
154 SkAutoTUnref<SkStream> stream(SkStream::NewFromFile(path.c_str()));
155 if (!stream) {
156 SkDebugf("Could not read %s.\n", path.c_str());
157 failed = true;
158 continue;
159 }
160 SkAutoTUnref<SkPicture> src(
161 SkPicture::CreateFromStream(stream, sk_tools::LazyDecodeBitmap));
162 if (!src) {
163 SkDebugf("Could not read %s as an SkPicture.\n", path.c_str());
164 failed = true;
165 continue;
166 }
167 bench_record(*src, overheadEstimate, filename.c_str(), bbhFactory.get());
168 }
169 return failed ? 1 : 0;
170 }
171
172 #if !defined SK_BUILD_FOR_IOS
main(int argc,char * const argv[])173 int main(int argc, char * const argv[]) {
174 return tool_main(argc, (char**) argv);
175 }
176 #endif
177