mtklein | 114c3cd | 2015-01-15 10:15:02 -0800 | [diff] [blame^] | 1 | // Main binary for DM. |
| 2 | // For a high-level overview, please see dm/README. |
| 3 | |
jcgregorio | 3b27ade | 2014-11-13 08:06:40 -0800 | [diff] [blame] | 4 | #include "CrashHandler.h" |
mtklein | 114c3cd | 2015-01-15 10:15:02 -0800 | [diff] [blame^] | 5 | #include "LazyDecodeBitmap.h" |
caryclark | 17f0b6d | 2014-07-22 10:15:34 -0700 | [diff] [blame] | 6 | #include "SkCommonFlags.h" |
mtklein@google.com | d36522d | 2013-10-16 13:02:15 +0000 | [diff] [blame] | 7 | #include "SkForceLinking.h" |
| 8 | #include "SkGraphics.h" |
mtklein | 1d0f164 | 2014-09-08 08:05:18 -0700 | [diff] [blame] | 9 | #include "SkOSFile.h" |
mtklein | 114c3cd | 2015-01-15 10:15:02 -0800 | [diff] [blame^] | 10 | #include "SkPicture.h" |
| 11 | #include "SkString.h" |
mtklein | 406654b | 2014-09-03 15:34:37 -0700 | [diff] [blame] | 12 | #include "SkTaskGroup.h" |
commit-bot@chromium.org | 0dc5bd1 | 2014-02-26 16:31:22 +0000 | [diff] [blame] | 13 | #include "Test.h" |
mtklein | 114c3cd | 2015-01-15 10:15:02 -0800 | [diff] [blame^] | 14 | #include "gm.h" |
| 15 | #include "sk_tool_utils.h" |
| 16 | #include "sk_tool_utils_flags.h" |
mtklein@google.com | d36522d | 2013-10-16 13:02:15 +0000 | [diff] [blame] | 17 | |
mtklein | 114c3cd | 2015-01-15 10:15:02 -0800 | [diff] [blame^] | 18 | #include "DMCpuGMTask.h" |
| 19 | #include "DMGpuGMTask.h" |
| 20 | #include "DMGpuSupport.h" |
| 21 | #include "DMImageTask.h" |
| 22 | #include "DMJsonWriter.h" |
| 23 | #include "DMPDFTask.h" |
| 24 | #include "DMPDFRasterizeTask.h" |
| 25 | #include "DMReporter.h" |
| 26 | #include "DMSKPTask.h" |
| 27 | #include "DMTask.h" |
| 28 | #include "DMTaskRunner.h" |
| 29 | #include "DMTestTask.h" |
| 30 | |
| 31 | #ifdef SK_BUILD_POPPLER |
| 32 | # include "SkPDFRasterizer.h" |
| 33 | # define RASTERIZE_PDF_PROC SkPopplerRasterizePDF |
| 34 | #elif defined(SK_BUILD_FOR_MAC) || defined(SK_BUILD_FOR_IOS) |
| 35 | # include "SkCGUtils.h" |
| 36 | # define RASTERIZE_PDF_PROC SkPDFDocumentToBitmap |
| 37 | #else |
| 38 | # define RASTERIZE_PDF_PROC NULL |
| 39 | #endif |
| 40 | |
| 41 | #include <ctype.h> |
| 42 | |
| 43 | using skiagm::GM; |
| 44 | using skiagm::GMRegistry; |
| 45 | using skiatest::Test; |
| 46 | using skiatest::TestRegistry; |
| 47 | |
| 48 | static const char kGpuAPINameGL[] = "gl"; |
| 49 | static const char kGpuAPINameGLES[] = "gles"; |
| 50 | |
| 51 | DEFINE_bool(gms, true, "Run GMs?"); |
commit-bot@chromium.org | 0dc5bd1 | 2014-02-26 16:31:22 +0000 | [diff] [blame] | 52 | DEFINE_bool(tests, true, "Run tests?"); |
mtklein | 114c3cd | 2015-01-15 10:15:02 -0800 | [diff] [blame^] | 53 | DEFINE_bool(reportUsedChars, false, "Output test font construction data to be pasted into" |
| 54 | " create_test_font.cpp."); |
| 55 | DEFINE_string(images, "resources", "Path to directory containing images to decode."); |
| 56 | DEFINE_bool(rasterPDF, true, "Rasterize PDFs?"); |
mtklein@google.com | d36522d | 2013-10-16 13:02:15 +0000 | [diff] [blame] | 57 | |
| 58 | __SK_FORCE_IMAGE_DECODER_LINKING; |
| 59 | |
mtklein | 114c3cd | 2015-01-15 10:15:02 -0800 | [diff] [blame^] | 60 | static DM::RasterizePdfProc get_pdf_rasterizer_proc() { |
| 61 | return reinterpret_cast<DM::RasterizePdfProc>( |
| 62 | FLAGS_rasterPDF ? RASTERIZE_PDF_PROC : NULL); |
halcanary | 9e398f7 | 2015-01-10 11:18:04 -0800 | [diff] [blame] | 63 | } |
| 64 | |
mtklein | 114c3cd | 2015-01-15 10:15:02 -0800 | [diff] [blame^] | 65 | // "FooBar" -> "foobar". Obviously, ASCII only. |
| 66 | static SkString lowercase(SkString s) { |
| 67 | for (size_t i = 0; i < s.size(); i++) { |
| 68 | s[i] = tolower(s[i]); |
| 69 | } |
| 70 | return s; |
mtklein@google.com | d36522d | 2013-10-16 13:02:15 +0000 | [diff] [blame] | 71 | } |
| 72 | |
mtklein | 114c3cd | 2015-01-15 10:15:02 -0800 | [diff] [blame^] | 73 | static const GrContextFactory::GLContextType native = GrContextFactory::kNative_GLContextType; |
| 74 | static const GrContextFactory::GLContextType nvpr = GrContextFactory::kNVPR_GLContextType; |
| 75 | static const GrContextFactory::GLContextType null = GrContextFactory::kNull_GLContextType; |
| 76 | static const GrContextFactory::GLContextType debug = GrContextFactory::kDebug_GLContextType; |
| 77 | #if SK_ANGLE |
| 78 | static const GrContextFactory::GLContextType angle = GrContextFactory::kANGLE_GLContextType; |
mtklein | 709d2c3 | 2015-01-15 08:30:25 -0800 | [diff] [blame] | 79 | #endif |
mtklein | 114c3cd | 2015-01-15 10:15:02 -0800 | [diff] [blame^] | 80 | #if SK_MESA |
| 81 | static const GrContextFactory::GLContextType mesa = GrContextFactory::kMESA_GLContextType; |
| 82 | #endif |
| 83 | |
| 84 | static void kick_off_gms(const SkTDArray<GMRegistry::Factory>& gms, |
| 85 | const SkTArray<SkString>& configs, |
| 86 | GrGLStandard gpuAPI, |
| 87 | DM::Reporter* reporter, |
| 88 | DM::TaskRunner* tasks) { |
| 89 | #define START(name, type, ...) \ |
| 90 | if (lowercase(configs[j]).equals(name)) { \ |
| 91 | tasks->add(SkNEW_ARGS(DM::type, (name, reporter, tasks, gms[i], ## __VA_ARGS__))); \ |
| 92 | } |
| 93 | for (int i = 0; i < gms.count(); i++) { |
| 94 | for (int j = 0; j < configs.count(); j++) { |
| 95 | |
| 96 | START("565", CpuGMTask, kRGB_565_SkColorType); |
| 97 | START("8888", CpuGMTask, kN32_SkColorType); |
| 98 | START("gpu", GpuGMTask, native, gpuAPI, 0, false); |
| 99 | START("msaa4", GpuGMTask, native, gpuAPI, 4, false); |
| 100 | START("msaa16", GpuGMTask, native, gpuAPI, 16, false); |
| 101 | START("nvprmsaa4", GpuGMTask, nvpr, gpuAPI, 4, false); |
| 102 | START("nvprmsaa16", GpuGMTask, nvpr, gpuAPI, 16, false); |
| 103 | START("gpudft", GpuGMTask, native, gpuAPI, 0, true); |
| 104 | START("gpunull", GpuGMTask, null, gpuAPI, 0, false); |
| 105 | START("gpudebug", GpuGMTask, debug, gpuAPI, 0, false); |
| 106 | #if SK_ANGLE |
| 107 | START("angle", GpuGMTask, angle, gpuAPI, 0, false); |
| 108 | #endif |
| 109 | #if SK_MESA |
| 110 | START("mesa", GpuGMTask, mesa, gpuAPI, 0, false); |
| 111 | #endif |
| 112 | START("pdf", PDFTask, get_pdf_rasterizer_proc()); |
| 113 | } |
| 114 | } |
| 115 | #undef START |
mtklein | 709d2c3 | 2015-01-15 08:30:25 -0800 | [diff] [blame] | 116 | } |
| 117 | |
mtklein | 114c3cd | 2015-01-15 10:15:02 -0800 | [diff] [blame^] | 118 | static void kick_off_tests(const SkTDArray<TestRegistry::Factory>& tests, |
| 119 | DM::Reporter* reporter, |
| 120 | DM::TaskRunner* tasks) { |
| 121 | for (int i = 0; i < tests.count(); i++) { |
| 122 | SkAutoTDelete<Test> test(tests[i](NULL)); |
| 123 | if (test->isGPUTest()) { |
| 124 | tasks->add(SkNEW_ARGS(DM::GpuTestTask, (reporter, tasks, tests[i]))); |
mtklein | 709d2c3 | 2015-01-15 08:30:25 -0800 | [diff] [blame] | 125 | } else { |
mtklein | 114c3cd | 2015-01-15 10:15:02 -0800 | [diff] [blame^] | 126 | tasks->add(SkNEW_ARGS(DM::CpuTestTask, (reporter, tasks, tests[i]))); |
mtklein | 709d2c3 | 2015-01-15 08:30:25 -0800 | [diff] [blame] | 127 | } |
mtklein | 709d2c3 | 2015-01-15 08:30:25 -0800 | [diff] [blame] | 128 | } |
| 129 | } |
| 130 | |
mtklein | 114c3cd | 2015-01-15 10:15:02 -0800 | [diff] [blame^] | 131 | static void find_files(const char* dir, |
| 132 | const char* suffixes[], |
| 133 | size_t num_suffixes, |
| 134 | SkTArray<SkString>* files) { |
| 135 | if (0 == strcmp(dir, "")) { |
mtklein | 709d2c3 | 2015-01-15 08:30:25 -0800 | [diff] [blame] | 136 | return; |
| 137 | } |
mtklein | 114c3cd | 2015-01-15 10:15:02 -0800 | [diff] [blame^] | 138 | |
| 139 | SkString filename; |
| 140 | for (size_t i = 0; i < num_suffixes; i++) { |
| 141 | SkOSFile::Iter it(dir, suffixes[i]); |
| 142 | while (it.next(&filename)) { |
| 143 | if (!SkCommandLineFlags::ShouldSkip(FLAGS_match, filename.c_str())) { |
| 144 | files->push_back(SkOSPath::Join(dir, filename.c_str())); |
| 145 | } |
mtklein | 709d2c3 | 2015-01-15 08:30:25 -0800 | [diff] [blame] | 146 | } |
mtklein | 709d2c3 | 2015-01-15 08:30:25 -0800 | [diff] [blame] | 147 | } |
| 148 | } |
| 149 | |
mtklein | 114c3cd | 2015-01-15 10:15:02 -0800 | [diff] [blame^] | 150 | static void kick_off_skps(const SkTArray<SkString>& skps, |
| 151 | DM::Reporter* reporter, |
| 152 | DM::TaskRunner* tasks) { |
| 153 | for (int i = 0; i < skps.count(); ++i) { |
| 154 | SkAutoTUnref<SkStream> stream(SkStream::NewFromFile(skps[i].c_str())); |
| 155 | if (stream.get() == NULL) { |
| 156 | SkDebugf("Could not read %s.\n", skps[i].c_str()); |
| 157 | exit(1); |
mtklein | 709d2c3 | 2015-01-15 08:30:25 -0800 | [diff] [blame] | 158 | } |
mtklein | 114c3cd | 2015-01-15 10:15:02 -0800 | [diff] [blame^] | 159 | SkAutoTUnref<SkPicture> pic( |
| 160 | SkPicture::CreateFromStream(stream.get(), &sk_tools::LazyDecodeBitmap)); |
| 161 | if (pic.get() == NULL) { |
| 162 | SkDebugf("Could not read %s as an SkPicture.\n", skps[i].c_str()); |
| 163 | exit(1); |
| 164 | } |
| 165 | |
| 166 | SkString filename = SkOSPath::Basename(skps[i].c_str()); |
| 167 | tasks->add(SkNEW_ARGS(DM::SKPTask, (reporter, tasks, pic, filename))); |
| 168 | tasks->add(SkNEW_ARGS(DM::PDFTask, (reporter, tasks, pic, filename, |
| 169 | get_pdf_rasterizer_proc()))); |
mtklein | 709d2c3 | 2015-01-15 08:30:25 -0800 | [diff] [blame] | 170 | } |
mtklein | 709d2c3 | 2015-01-15 08:30:25 -0800 | [diff] [blame] | 171 | } |
| 172 | |
mtklein | 114c3cd | 2015-01-15 10:15:02 -0800 | [diff] [blame^] | 173 | static void kick_off_images(const SkTArray<SkString>& images, |
| 174 | DM::Reporter* reporter, |
| 175 | DM::TaskRunner* tasks) { |
| 176 | for (int i = 0; i < images.count(); i++) { |
| 177 | SkAutoTUnref<SkData> image(SkData::NewFromFileName(images[i].c_str())); |
| 178 | if (!image) { |
| 179 | SkDebugf("Could not read %s.\n", images[i].c_str()); |
| 180 | exit(1); |
| 181 | } |
| 182 | SkString filename = SkOSPath::Basename(images[i].c_str()); |
| 183 | tasks->add(SkNEW_ARGS(DM::ImageTask, (reporter, tasks, image, filename))); |
| 184 | tasks->add(SkNEW_ARGS(DM::ImageTask, (reporter, tasks, image, filename, 5/*subsets*/))); |
| 185 | } |
| 186 | } |
| 187 | |
| 188 | |
| 189 | static void report_failures(const SkTArray<SkString>& failures) { |
| 190 | if (failures.count() == 0) { |
| 191 | return; |
| 192 | } |
| 193 | |
| 194 | SkDebugf("Failures:\n"); |
| 195 | for (int i = 0; i < failures.count(); i++) { |
| 196 | SkDebugf(" %s\n", failures[i].c_str()); |
| 197 | } |
| 198 | SkDebugf("%d failures.\n", failures.count()); |
| 199 | } |
| 200 | |
| 201 | static GrGLStandard get_gl_standard() { |
| 202 | if (FLAGS_gpuAPI.contains(kGpuAPINameGL)) { |
| 203 | return kGL_GrGLStandard; |
| 204 | } |
| 205 | if (FLAGS_gpuAPI.contains(kGpuAPINameGLES)) { |
| 206 | return kGLES_GrGLStandard; |
| 207 | } |
| 208 | return kNone_GrGLStandard; |
| 209 | } |
| 210 | |
| 211 | template <typename T, typename Registry> |
| 212 | static void append_matching_factories(Registry* head, SkTDArray<typename Registry::Factory>* out) { |
| 213 | for (const Registry* reg = head; reg != NULL; reg = reg->next()) { |
| 214 | SkAutoTDelete<T> forName(reg->factory()(NULL)); |
| 215 | if (!SkCommandLineFlags::ShouldSkip(FLAGS_match, forName->getName())) { |
| 216 | *out->append() = reg->factory(); |
| 217 | } |
| 218 | } |
| 219 | } |
mtklein | 709d2c3 | 2015-01-15 08:30:25 -0800 | [diff] [blame] | 220 | |
jcgregorio | 3b27ade | 2014-11-13 08:06:40 -0800 | [diff] [blame] | 221 | int dm_main(); |
caryclark | 17f0b6d | 2014-07-22 10:15:34 -0700 | [diff] [blame] | 222 | int dm_main() { |
jcgregorio | 3b27ade | 2014-11-13 08:06:40 -0800 | [diff] [blame] | 223 | SetupCrashHandler(); |
commit-bot@chromium.org | 39e8d93 | 2014-05-29 20:14:48 +0000 | [diff] [blame] | 224 | SkAutoGraphics ag; |
mtklein | 406654b | 2014-09-03 15:34:37 -0700 | [diff] [blame] | 225 | SkTaskGroup::Enabler enabled(FLAGS_threads); |
commit-bot@chromium.org | a65e2fd | 2014-05-30 17:23:31 +0000 | [diff] [blame] | 226 | |
mtklein | 114c3cd | 2015-01-15 10:15:02 -0800 | [diff] [blame^] | 227 | if (FLAGS_dryRun || FLAGS_veryVerbose) { |
| 228 | FLAGS_verbose = true; |
| 229 | } |
| 230 | #if SK_ENABLE_INST_COUNT |
| 231 | gPrintInstCount = FLAGS_leaks; |
| 232 | #endif |
commit-bot@chromium.org | 0dc5bd1 | 2014-02-26 16:31:22 +0000 | [diff] [blame] | 233 | |
mtklein | 114c3cd | 2015-01-15 10:15:02 -0800 | [diff] [blame^] | 234 | SkTArray<SkString> configs; |
| 235 | for (int i = 0; i < FLAGS_config.count(); i++) { |
| 236 | SkStrSplit(FLAGS_config[i], ", ", &configs); |
commit-bot@chromium.org | 38aeb0f | 2014-02-26 23:01:57 +0000 | [diff] [blame] | 237 | } |
| 238 | |
mtklein | 114c3cd | 2015-01-15 10:15:02 -0800 | [diff] [blame^] | 239 | GrGLStandard gpuAPI = get_gl_standard(); |
kkinnunen | 80549fc | 2014-06-30 06:36:31 -0700 | [diff] [blame] | 240 | |
mtklein | 114c3cd | 2015-01-15 10:15:02 -0800 | [diff] [blame^] | 241 | SkTDArray<GMRegistry::Factory> gms; |
| 242 | if (FLAGS_gms) { |
| 243 | append_matching_factories<GM>(GMRegistry::Head(), &gms); |
mtklein@google.com | d36522d | 2013-10-16 13:02:15 +0000 | [diff] [blame] | 244 | } |
| 245 | |
mtklein | 114c3cd | 2015-01-15 10:15:02 -0800 | [diff] [blame^] | 246 | SkTDArray<TestRegistry::Factory> tests; |
| 247 | if (FLAGS_tests) { |
| 248 | append_matching_factories<Test>(TestRegistry::Head(), &tests); |
| 249 | } |
| 250 | |
| 251 | |
| 252 | SkTArray<SkString> skps; |
| 253 | if (!FLAGS_skps.isEmpty()) { |
| 254 | const char* suffixes[] = { "skp" }; |
| 255 | find_files(FLAGS_skps[0], suffixes, SK_ARRAY_COUNT(suffixes), &skps); |
| 256 | } |
| 257 | |
| 258 | SkTArray<SkString> images; |
| 259 | if (!FLAGS_images.isEmpty()) { |
| 260 | const char* suffixes[] = { |
| 261 | "bmp", "gif", "jpg", "jpeg", "png", "webp", "ktx", "astc", "wbmp", "ico", |
| 262 | "BMP", "GIF", "JPG", "JPEG", "PNG", "WEBP", "KTX", "ASTC", "WBMP", "ICO", |
| 263 | }; |
| 264 | find_files(FLAGS_images[0], suffixes, SK_ARRAY_COUNT(suffixes), &images); |
| 265 | } |
| 266 | |
| 267 | SkDebugf("%d GMs x %d configs, %d tests, %d pictures, %d images\n", |
| 268 | gms.count(), configs.count(), tests.count(), skps.count(), images.count()); |
| 269 | DM::Reporter reporter; |
| 270 | |
| 271 | DM::TaskRunner tasks; |
| 272 | kick_off_tests(tests, &reporter, &tasks); |
| 273 | kick_off_gms(gms, configs, gpuAPI, &reporter, &tasks); |
| 274 | kick_off_skps(skps, &reporter, &tasks); |
| 275 | kick_off_images(images, &reporter, &tasks); |
| 276 | tasks.wait(); |
| 277 | |
| 278 | DM::JsonWriter::DumpJson(); |
| 279 | |
| 280 | SkDebugf("\n"); |
| 281 | #ifdef SK_DEBUG |
| 282 | if (FLAGS_portableFonts && FLAGS_reportUsedChars) { |
| 283 | sk_tool_utils::report_used_chars(); |
| 284 | } |
| 285 | #endif |
| 286 | |
| 287 | SkTArray<SkString> failures; |
| 288 | reporter.getFailures(&failures); |
| 289 | report_failures(failures); |
| 290 | return failures.count() > 0; |
mtklein@google.com | d36522d | 2013-10-16 13:02:15 +0000 | [diff] [blame] | 291 | } |
jcgregorio | 3b27ade | 2014-11-13 08:06:40 -0800 | [diff] [blame] | 292 | |
| 293 | #if !defined(SK_BUILD_FOR_IOS) && !defined(SK_BUILD_FOR_NACL) |
| 294 | int main(int argc, char** argv) { |
| 295 | SkCommandLineFlags::Parse(argc, argv); |
| 296 | return dm_main(); |
| 297 | } |
| 298 | #endif |