1 // Main binary for DM.
2 // For a high-level overview, please see dm/README.
3
4 #include "GrContext.h"
5 #include "GrContextFactory.h"
6 #include "SkCommandLineFlags.h"
7 #include "SkForceLinking.h"
8 #include "SkGraphics.h"
9 #include "SkString.h"
10 #include "gm.h"
11
12 #include "DMReporter.h"
13 #include "DMTask.h"
14 #include "DMTaskRunner.h"
15 #include "DMCpuTask.h"
16 #include "DMGpuTask.h"
17 #include "DMWriteTask.h"
18
19 #include <string.h>
20
21 using skiagm::GM;
22 using skiagm::GMRegistry;
23
24 DEFINE_int32(cpuThreads, -1, "Threads for CPU work. Default NUM_CPUS.");
25 DEFINE_int32(gpuThreads, 1, "Threads for GPU work.");
26 DEFINE_string2(expectations, r, "",
27 "If a directory, compare generated images against images under this path. "
28 "If a file, compare generated images against JSON expectations at this path.");
29 DEFINE_string(resources, "resources", "Path to resources directory.");
30 DEFINE_string(match, "", "[~][^]substring[$] [...] of GM name to run.\n"
31 "Multiple matches may be separated by spaces.\n"
32 "~ causes a matching GM to always be skipped\n"
33 "^ requires the start of the GM to match\n"
34 "$ requires the end of the GM to match\n"
35 "^ and $ requires an exact match\n"
36 "If a GM does not match any list entry,\n"
37 "it is skipped unless some list entry starts with ~");
38 DEFINE_string(config, "8888 gpu",
39 "Options: 565 8888 gpu msaa4 msaa16 gpunull gpudebug angle mesa"); // TODO(mtklein): pdf
40
41 __SK_FORCE_IMAGE_DECODER_LINKING;
42
43 // "FooBar" -> "foobar". Obviously, ASCII only.
lowercase(SkString s)44 static SkString lowercase(SkString s) {
45 for (size_t i = 0; i < s.size(); i++) {
46 s[i] = tolower(s[i]);
47 }
48 return s;
49 }
50
kick_off_tasks(const SkTDArray<GMRegistry::Factory> & gms,const SkTArray<SkString> & configs,const DM::Expectations & expectations,DM::Reporter * reporter,DM::TaskRunner * tasks)51 static void kick_off_tasks(const SkTDArray<GMRegistry::Factory>& gms,
52 const SkTArray<SkString>& configs,
53 const DM::Expectations& expectations,
54 DM::Reporter* reporter,
55 DM::TaskRunner* tasks) {
56 const SkBitmap::Config _565 = SkBitmap::kRGB_565_Config;
57 const SkBitmap::Config _8888 = SkBitmap::kARGB_8888_Config;
58 const GrContextFactory::GLContextType native = GrContextFactory::kNative_GLContextType;
59 const GrContextFactory::GLContextType null = GrContextFactory::kNull_GLContextType;
60 const GrContextFactory::GLContextType debug = GrContextFactory::kDebug_GLContextType;
61 const GrContextFactory::GLContextType angle =
62 #if SK_ANGLE
63 GrContextFactory::kANGLE_GLContextType;
64 #else
65 native;
66 #endif
67 const GrContextFactory::GLContextType mesa =
68 #if SK_MESA
69 GLContextFactory::kMESA_GLContextType;
70 #else
71 native;
72 #endif
73
74 for (int i = 0; i < gms.count(); i++) {
75 #define START(name, type, ...) \
76 if (lowercase(configs[j]).equals(name)) { \
77 tasks->add(SkNEW_ARGS(DM::type, \
78 (name, reporter, tasks, expectations, gms[i], __VA_ARGS__))); \
79 }
80 for (int j = 0; j < configs.count(); j++) {
81 START("565", CpuTask, _565);
82 START("8888", CpuTask, _8888);
83 START("gpu", GpuTask, _8888, native, 0);
84 START("msaa4", GpuTask, _8888, native, 4);
85 START("msaa16", GpuTask, _8888, native, 16);
86 START("gpunull", GpuTask, _8888, null, 0);
87 START("gpudebug", GpuTask, _8888, debug, 0);
88 START("angle", GpuTask, _8888, angle, 0);
89 START("mesa", GpuTask, _8888, mesa, 0);
90 //START("pdf", PdfTask, _8888);
91 }
92 }
93 #undef START
94 }
95
report_failures(const DM::Reporter & reporter)96 static void report_failures(const DM::Reporter& reporter) {
97 SkTArray<SkString> failures;
98 reporter.getFailures(&failures);
99
100 if (failures.count() == 0) {
101 return;
102 }
103
104 SkDebugf("Failures:\n");
105 for (int i = 0; i < failures.count(); i++) {
106 SkDebugf(" %s\n", failures[i].c_str());
107 }
108 }
109
110 int tool_main(int argc, char** argv);
tool_main(int argc,char ** argv)111 int tool_main(int argc, char** argv) {
112 SkGraphics::Init();
113
114 SkCommandLineFlags::Parse(argc, argv);
115 GM::SetResourcePath(FLAGS_resources[0]);
116 SkTArray<SkString> configs;
117 for (int i = 0; i < FLAGS_config.count(); i++) {
118 SkStrSplit(FLAGS_config[i], ", ", &configs);
119 }
120
121 SkTDArray<GMRegistry::Factory> gms;
122 for (const GMRegistry* reg = GMRegistry::Head(); reg != NULL; reg = reg->next()) {
123 SkAutoTDelete<GM> gmForName(reg->factory()(NULL));
124 if (!SkCommandLineFlags::ShouldSkip(FLAGS_match, gmForName->shortName())) {
125 *gms.append() = reg->factory();
126 }
127 }
128 SkDebugf("%d GMs x %d configs\n", gms.count(), configs.count());
129
130 SkAutoTDelete<DM::Expectations> expectations(SkNEW(DM::NoExpectations));
131 if (FLAGS_expectations.count() > 0) {
132 const char* path = FLAGS_expectations[0];
133 if (sk_isdir(path)) {
134 expectations.reset(SkNEW_ARGS(DM::WriteTask::Expectations, (path)));
135 } else {
136 expectations.reset(SkNEW_ARGS(DM::JsonExpectations, (path)));
137 }
138 }
139
140 DM::Reporter reporter;
141 DM::TaskRunner tasks(FLAGS_cpuThreads, FLAGS_gpuThreads);
142 kick_off_tasks(gms, configs, *expectations, &reporter, &tasks);
143 tasks.wait();
144
145 reporter.updateStatusLine();
146 SkDebugf("\n");
147 report_failures(reporter);
148
149 SkGraphics::Term();
150
151 return reporter.failed() > 0;
152 }
153
154 #if !defined(SK_BUILD_FOR_IOS) && !defined(SK_BUILD_FOR_NACL)
main(int argc,char ** argv)155 int main(int argc, char** argv) {
156 return tool_main(argc, argv);
157 }
158 #endif
159