| // Main binary for DM. |
| // For a high-level overview, please see dm/README. |
| |
| #include "CrashHandler.h" |
| #include "SkCommonFlags.h" |
| #include "SkForceLinking.h" |
| #include "SkGraphics.h" |
| #include "SkPicture.h" |
| #include "SkString.h" |
| #include "Test.h" |
| #include "gm.h" |
| |
| #include "DMCpuGMTask.h" |
| #include "DMGpuGMTask.h" |
| #include "DMGpuSupport.h" |
| #include "DMPDFTask.h" |
| #include "DMReporter.h" |
| #include "DMSKPTask.h" |
| #include "DMTask.h" |
| #include "DMTaskRunner.h" |
| #include "DMTestTask.h" |
| #include "DMWriteTask.h" |
| |
| #ifdef SK_BUILD_POPPLER |
| # include "SkPDFRasterizer.h" |
| # define RASTERIZE_PDF_PROC SkPopplerRasterizePDF |
| #else |
| # define RASTERIZE_PDF_PROC NULL |
| #endif |
| |
| #include <ctype.h> |
| |
| using skiagm::GM; |
| using skiagm::GMRegistry; |
| using skiatest::Test; |
| using skiatest::TestRegistry; |
| |
| static const char kGpuAPINameGL[] = "gl"; |
| static const char kGpuAPINameGLES[] = "gles"; |
| |
| DEFINE_int32(gpuThreads, 1, "Threads for GPU work."); |
| DEFINE_string2(expectations, r, "", |
| "If a directory, compare generated images against images under this path. " |
| "If a file, compare generated images against JSON expectations at this path." |
| ); |
| |
| DEFINE_string(skps, "", "Directory to read skps from."); |
| |
| DEFINE_bool(gms, true, "Run GMs?"); |
| DEFINE_bool(tests, true, "Run tests?"); |
| |
| __SK_FORCE_IMAGE_DECODER_LINKING; |
| |
| // "FooBar" -> "foobar". Obviously, ASCII only. |
| static SkString lowercase(SkString s) { |
| for (size_t i = 0; i < s.size(); i++) { |
| s[i] = tolower(s[i]); |
| } |
| return s; |
| } |
| |
| static const GrContextFactory::GLContextType native = GrContextFactory::kNative_GLContextType; |
| static const GrContextFactory::GLContextType nvpr = GrContextFactory::kNVPR_GLContextType; |
| static const GrContextFactory::GLContextType null = GrContextFactory::kNull_GLContextType; |
| static const GrContextFactory::GLContextType debug = GrContextFactory::kDebug_GLContextType; |
| static const GrContextFactory::GLContextType angle = |
| #if SK_ANGLE |
| GrContextFactory::kANGLE_GLContextType; |
| #else |
| native; |
| #endif |
| static const GrContextFactory::GLContextType mesa = |
| #if SK_MESA |
| GrContextFactory::kMESA_GLContextType; |
| #else |
| native; |
| #endif |
| |
| static void kick_off_gms(const SkTDArray<GMRegistry::Factory>& gms, |
| const SkTArray<SkString>& configs, |
| GrGLStandard gpuAPI, |
| const DM::Expectations& expectations, |
| DM::Reporter* reporter, |
| DM::TaskRunner* tasks) { |
| #define START(name, type, ...) \ |
| if (lowercase(configs[j]).equals(name)) { \ |
| tasks->add(SkNEW_ARGS(DM::type, (name, reporter, tasks, gms[i], ## __VA_ARGS__))); \ |
| } |
| for (int i = 0; i < gms.count(); i++) { |
| for (int j = 0; j < configs.count(); j++) { |
| |
| START("565", CpuGMTask, expectations, kRGB_565_SkColorType); |
| START("8888", CpuGMTask, expectations, kN32_SkColorType); |
| START("gpu", GpuGMTask, expectations, native, gpuAPI, 0); |
| START("msaa4", GpuGMTask, expectations, native, gpuAPI, 4); |
| START("msaa16", GpuGMTask, expectations, native, gpuAPI, 16); |
| START("nvprmsaa4", GpuGMTask, expectations, nvpr, gpuAPI, 4); |
| START("nvprmsaa16", GpuGMTask, expectations, nvpr, gpuAPI, 16); |
| START("gpunull", GpuGMTask, expectations, null, gpuAPI, 0); |
| START("gpudebug", GpuGMTask, expectations, debug, gpuAPI, 0); |
| START("angle", GpuGMTask, expectations, angle, gpuAPI, 0); |
| START("mesa", GpuGMTask, expectations, mesa, gpuAPI, 0); |
| START("pdf", PDFTask, RASTERIZE_PDF_PROC); |
| } |
| } |
| #undef START |
| } |
| |
| static void kick_off_tests(const SkTDArray<TestRegistry::Factory>& tests, |
| DM::Reporter* reporter, |
| DM::TaskRunner* tasks) { |
| for (int i = 0; i < tests.count(); i++) { |
| SkAutoTDelete<Test> test(tests[i](NULL)); |
| if (test->isGPUTest()) { |
| tasks->add(SkNEW_ARGS(DM::GpuTestTask, (reporter, tasks, tests[i]))); |
| } else { |
| tasks->add(SkNEW_ARGS(DM::CpuTestTask, (reporter, tasks, tests[i]))); |
| } |
| } |
| } |
| |
| static void find_skps(SkTArray<SkString>* skps) { |
| if (FLAGS_skps.isEmpty()) { |
| return; |
| } |
| |
| SkOSFile::Iter it(FLAGS_skps[0], ".skp"); |
| SkString filename; |
| while (it.next(&filename)) { |
| if (!SkCommandLineFlags::ShouldSkip(FLAGS_match, filename.c_str())) { |
| skps->push_back(SkOSPath::Join(FLAGS_skps[0], filename.c_str())); |
| } |
| } |
| } |
| |
| static void kick_off_skps(const SkTArray<SkString>& skps, |
| DM::Reporter* reporter, DM::TaskRunner* tasks) { |
| for (int i = 0; i < skps.count(); ++i) { |
| SkAutoTUnref<SkStream> stream(SkStream::NewFromFile(skps[i].c_str())); |
| if (stream.get() == NULL) { |
| SkDebugf("Could not read %s.\n", skps[i].c_str()); |
| exit(1); |
| } |
| SkAutoTUnref<SkPicture> pic(SkPicture::CreateFromStream(stream.get())); |
| if (pic.get() == NULL) { |
| SkDebugf("Could not read %s as an SkPicture.\n", skps[i].c_str()); |
| exit(1); |
| } |
| |
| SkString filename = SkOSPath::Basename(skps[i].c_str()); |
| tasks->add(SkNEW_ARGS(DM::SKPTask, (reporter, tasks, pic, filename))); |
| tasks->add(SkNEW_ARGS(DM::PDFTask, (reporter, tasks, pic, filename, |
| RASTERIZE_PDF_PROC))); |
| } |
| } |
| |
| static void report_failures(const SkTArray<SkString>& failures) { |
| if (failures.count() == 0) { |
| return; |
| } |
| |
| SkDebugf("Failures:\n"); |
| for (int i = 0; i < failures.count(); i++) { |
| SkDebugf(" %s\n", failures[i].c_str()); |
| } |
| SkDebugf("%d failures.\n", failures.count()); |
| } |
| |
| static GrGLStandard get_gl_standard() { |
| if (FLAGS_gpuAPI.contains(kGpuAPINameGL)) { |
| return kGL_GrGLStandard; |
| } |
| if (FLAGS_gpuAPI.contains(kGpuAPINameGLES)) { |
| return kGLES_GrGLStandard; |
| } |
| return kNone_GrGLStandard; |
| } |
| |
| template <typename T, typename Registry> |
| static void append_matching_factories(Registry* head, SkTDArray<typename Registry::Factory>* out) { |
| for (const Registry* reg = head; reg != NULL; reg = reg->next()) { |
| SkAutoTDelete<T> forName(reg->factory()(NULL)); |
| if (!SkCommandLineFlags::ShouldSkip(FLAGS_match, forName->getName())) { |
| *out->append() = reg->factory(); |
| } |
| } |
| } |
| |
| int dm_main(); |
| int dm_main() { |
| SetupCrashHandler(); |
| SkAutoGraphics ag; |
| |
| if (FLAGS_dryRun) { |
| FLAGS_verbose = true; |
| } |
| #if SK_ENABLE_INST_COUNT |
| gPrintInstCount = FLAGS_leaks; |
| #endif |
| |
| SkTArray<SkString> configs; |
| for (int i = 0; i < FLAGS_config.count(); i++) { |
| SkStrSplit(FLAGS_config[i], ", ", &configs); |
| } |
| |
| GrGLStandard gpuAPI = get_gl_standard(); |
| |
| SkTDArray<GMRegistry::Factory> gms; |
| SkAutoTDelete<DM::Expectations> expectations(SkNEW(DM::NoExpectations)); |
| if (FLAGS_gms) { |
| append_matching_factories<GM>(GMRegistry::Head(), &gms); |
| |
| if (FLAGS_expectations.count() > 0) { |
| const char* path = FLAGS_expectations[0]; |
| if (sk_isdir(path)) { |
| expectations.reset(SkNEW_ARGS(DM::WriteTask::Expectations, (path))); |
| } else { |
| expectations.reset(SkNEW_ARGS(DM::JsonExpectations, (path))); |
| } |
| } |
| } |
| |
| SkTDArray<TestRegistry::Factory> tests; |
| if (FLAGS_tests) { |
| append_matching_factories<Test>(TestRegistry::Head(), &tests); |
| } |
| |
| SkTArray<SkString> skps; |
| find_skps(&skps); |
| |
| SkDebugf("%d GMs x %d configs, %d tests, %d pictures\n", |
| gms.count(), configs.count(), tests.count(), skps.count()); |
| DM::Reporter reporter; |
| DM::TaskRunner tasks(FLAGS_threads, FLAGS_gpuThreads); |
| kick_off_gms(gms, configs, gpuAPI, *expectations, &reporter, &tasks); |
| kick_off_tests(tests, &reporter, &tasks); |
| kick_off_skps(skps, &reporter, &tasks); |
| tasks.wait(); |
| |
| SkDebugf("\n"); |
| |
| SkTArray<SkString> failures; |
| reporter.getFailures(&failures); |
| report_failures(failures); |
| return failures.count() > 0; |
| } |
| |
| #if !defined(SK_BUILD_FOR_IOS) && !defined(SK_BUILD_FOR_NACL) |
| int main(int argc, char** argv) { |
| SkCommandLineFlags::Parse(argc, argv); |
| return dm_main(); |
| } |
| #endif |