| #include "SkCanvas.h" |
| #include "SkColorPriv.h" |
| #include "SkGraphics.h" |
| #include "SkImageEncoder.h" |
| #include "SkNWayCanvas.h" |
| #include "SkPicture.h" |
| #include "SkString.h" |
| #include "SkTime.h" |
| |
| #include "SkBenchmark.h" |
| |
| #ifdef ANDROID |
| static void log_error(const char msg[]) { SkDebugf("%s", msg); } |
| static void log_progress(const char msg[]) { SkDebugf("%s", msg); } |
| #else |
| static void log_error(const char msg[]) { fprintf(stderr, "%s", msg); } |
| static void log_progress(const char msg[]) { printf("%s", msg); } |
| #endif |
| |
| static void log_error(const SkString& str) { log_error(str.c_str()); } |
| static void log_progress(const SkString& str) { log_progress(str.c_str()); } |
| |
| /////////////////////////////////////////////////////////////////////////////// |
| |
| static void erase(SkBitmap& bm) { |
| if (bm.config() == SkBitmap::kA8_Config) { |
| bm.eraseColor(0); |
| } else { |
| bm.eraseColor(SK_ColorWHITE); |
| } |
| } |
| |
| static bool equal(const SkBitmap& bm1, const SkBitmap& bm2) { |
| if (bm1.width() != bm2.width() || |
| bm1.height() != bm2.height() || |
| bm1.config() != bm2.config()) { |
| return false; |
| } |
| |
| size_t pixelBytes = bm1.width() * bm1.bytesPerPixel(); |
| for (int y = 0; y < bm1.height(); y++) { |
| if (memcmp(bm1.getAddr(0, y), bm2.getAddr(0, y), pixelBytes)) { |
| return false; |
| } |
| } |
| |
| return true; |
| } |
| |
| class Iter { |
| public: |
| Iter(void* param) { |
| fBench = BenchRegistry::Head(); |
| fParam = param; |
| } |
| |
| SkBenchmark* next() { |
| if (fBench) { |
| BenchRegistry::Factory f = fBench->factory(); |
| fBench = fBench->next(); |
| return f(fParam); |
| } |
| return NULL; |
| } |
| |
| private: |
| const BenchRegistry* fBench; |
| void* fParam; |
| }; |
| |
| static void make_filename(const char name[], SkString* path) { |
| path->set(name); |
| for (int i = 0; name[i]; i++) { |
| switch (name[i]) { |
| case '/': |
| case '\\': |
| case ' ': |
| case ':': |
| path->writable_str()[i] = '-'; |
| break; |
| default: |
| break; |
| } |
| } |
| } |
| |
| static void saveFile(const char name[], const char config[], const char dir[], |
| const SkBitmap& bm) { |
| SkBitmap copy; |
| if (!bm.copyTo(©, SkBitmap::kARGB_8888_Config)) { |
| return; |
| } |
| |
| if (bm.config() == SkBitmap::kA8_Config) { |
| // turn alpha into gray-scale |
| size_t size = copy.getSize() >> 2; |
| SkPMColor* p = copy.getAddr32(0, 0); |
| for (size_t i = 0; i < size; i++) { |
| int c = (*p >> SK_A32_SHIFT) & 0xFF; |
| c = 255 - c; |
| c |= (c << 24) | (c << 16) | (c << 8); |
| *p++ = c | (SK_A32_MASK << SK_A32_SHIFT); |
| } |
| } |
| |
| SkString str; |
| make_filename(name, &str); |
| str.appendf("_%s.png", config); |
| str.prepend(dir); |
| ::remove(str.c_str()); |
| SkImageEncoder::EncodeFile(str.c_str(), copy, SkImageEncoder::kPNG_Type, |
| 100); |
| } |
| |
| static void performClip(SkCanvas* canvas, int w, int h) { |
| SkRect r; |
| |
| r.set(SkIntToScalar(10), SkIntToScalar(10), |
| SkIntToScalar(w*2/3), SkIntToScalar(h*2/3)); |
| canvas->clipRect(r, SkRegion::kIntersect_Op); |
| |
| r.set(SkIntToScalar(w/3), SkIntToScalar(h/3), |
| SkIntToScalar(w-10), SkIntToScalar(h-10)); |
| canvas->clipRect(r, SkRegion::kXOR_Op); |
| } |
| |
| static void performRotate(SkCanvas* canvas, int w, int h) { |
| const SkScalar x = SkIntToScalar(w) / 2; |
| const SkScalar y = SkIntToScalar(h) / 2; |
| |
| canvas->translate(x, y); |
| canvas->rotate(SkIntToScalar(35)); |
| canvas->translate(-x, -y); |
| } |
| |
| static void performScale(SkCanvas* canvas, int w, int h) { |
| const SkScalar x = SkIntToScalar(w) / 2; |
| const SkScalar y = SkIntToScalar(h) / 2; |
| |
| canvas->translate(x, y); |
| // just enough so we can't take the sprite case |
| canvas->scale(SK_Scalar1 * 99/100, SK_Scalar1 * 99/100); |
| canvas->translate(-x, -y); |
| } |
| |
| static void compare_pict_to_bitmap(SkPicture* pict, const SkBitmap& bm) { |
| SkBitmap bm2; |
| |
| bm2.setConfig(bm.config(), bm.width(), bm.height()); |
| bm2.allocPixels(); |
| erase(bm2); |
| |
| SkCanvas canvas(bm2); |
| canvas.drawPicture(*pict); |
| |
| if (!equal(bm, bm2)) { |
| SkDebugf("----- compare_pict_to_bitmap failed\n"); |
| } |
| } |
| |
| static bool parse_bool_arg(char * const* argv, char* const* stop, bool* var) { |
| if (argv < stop) { |
| *var = atoi(*argv) != 0; |
| return true; |
| } |
| return false; |
| } |
| |
| static const struct { |
| SkBitmap::Config fConfig; |
| const char* fName; |
| } gConfigs[] = { |
| { SkBitmap::kARGB_8888_Config, "8888" }, |
| { SkBitmap::kRGB_565_Config, "565", }, |
| #if 0 |
| { SkBitmap::kARGB_4444_Config, "4444", }, |
| { SkBitmap::kA8_Config, "A8", } |
| #endif |
| }; |
| |
| static int findConfig(const char config[]) { |
| for (size_t i = 0; i < SK_ARRAY_COUNT(gConfigs); i++) { |
| if (!strcmp(config, gConfigs[i].fName)) { |
| return i; |
| } |
| } |
| return -1; |
| } |
| |
| int main (int argc, char * const argv[]) { |
| SkAutoGraphics ag; |
| |
| SkTDict<const char*> defineDict(1024); |
| int repeatDraw = 1; |
| int forceAlpha = 0xFF; |
| bool forceAA = true; |
| bool forceFilter = false; |
| SkTriState::State forceDither = SkTriState::kDefault; |
| bool doScale = false; |
| bool doRotate = false; |
| bool doClip = false; |
| bool doPict = false; |
| const char* matchStr = NULL; |
| |
| SkString outDir; |
| SkBitmap::Config outConfig = SkBitmap::kNo_Config; |
| const char* configName = ""; |
| int configCount = SK_ARRAY_COUNT(gConfigs); |
| |
| char* const* stop = argv + argc; |
| for (++argv; argv < stop; ++argv) { |
| if (strcmp(*argv, "-o") == 0) { |
| argv++; |
| if (argv < stop && **argv) { |
| outDir.set(*argv); |
| if (outDir.c_str()[outDir.size() - 1] != '/') { |
| outDir.append("/"); |
| } |
| } |
| } else if (strcmp(*argv, "-pict") == 0) { |
| doPict = true; |
| } else if (strcmp(*argv, "-repeat") == 0) { |
| argv++; |
| if (argv < stop) { |
| repeatDraw = atoi(*argv); |
| if (repeatDraw < 1) { |
| repeatDraw = 1; |
| } |
| } else { |
| log_error("missing arg for -repeat\n"); |
| return -1; |
| } |
| } else if (!strcmp(*argv, "-rotate")) { |
| doRotate = true; |
| } else if (!strcmp(*argv, "-scale")) { |
| doScale = true; |
| } else if (!strcmp(*argv, "-clip")) { |
| doClip = true; |
| } else if (strcmp(*argv, "-forceAA") == 0) { |
| if (!parse_bool_arg(++argv, stop, &forceAA)) { |
| log_error("missing arg for -forceAA\n"); |
| return -1; |
| } |
| } else if (strcmp(*argv, "-forceFilter") == 0) { |
| if (!parse_bool_arg(++argv, stop, &forceFilter)) { |
| log_error("missing arg for -forceFilter\n"); |
| return -1; |
| } |
| } else if (strcmp(*argv, "-forceDither") == 0) { |
| bool tmp; |
| if (!parse_bool_arg(++argv, stop, &tmp)) { |
| log_error("missing arg for -forceDither\n"); |
| return -1; |
| } |
| forceDither = tmp ? SkTriState::kTrue : SkTriState::kFalse; |
| } else if (strcmp(*argv, "-forceBlend") == 0) { |
| bool wantAlpha = false; |
| if (!parse_bool_arg(++argv, stop, &wantAlpha)) { |
| log_error("missing arg for -forceBlend\n"); |
| return -1; |
| } |
| forceAlpha = wantAlpha ? 0x80 : 0xFF; |
| } else if (strcmp(*argv, "-match") == 0) { |
| argv++; |
| if (argv < stop) { |
| matchStr = *argv; |
| } else { |
| log_error("missing arg for -match\n"); |
| return -1; |
| } |
| } else if (strcmp(*argv, "-config") == 0) { |
| argv++; |
| if (argv < stop) { |
| int index = findConfig(*argv); |
| if (index >= 0) { |
| outConfig = gConfigs[index].fConfig; |
| configName = gConfigs[index].fName; |
| configCount = 1; |
| } else { |
| SkString str; |
| str.printf("unrecognized config %s\n", *argv); |
| log_error(str); |
| return -1; |
| } |
| } else { |
| log_error("missing arg for -config\n"); |
| return -1; |
| } |
| } else if (strlen(*argv) > 2 && strncmp(*argv, "-D", 2) == 0) { |
| argv++; |
| if (argv < stop) { |
| defineDict.set(argv[-1] + 2, *argv); |
| } else { |
| log_error("incomplete '-Dfoo bar' definition\n"); |
| return -1; |
| } |
| } else { |
| SkString str; |
| str.printf("unrecognized arg %s\n", *argv); |
| log_error(str); |
| return -1; |
| } |
| } |
| |
| Iter iter(&defineDict); |
| SkBenchmark* bench; |
| while ((bench = iter.next()) != NULL) { |
| SkIPoint dim = bench->getSize(); |
| if (dim.fX <= 0 || dim.fY <= 0) { |
| continue; |
| } |
| |
| bench->setForceAlpha(forceAlpha); |
| bench->setForceAA(forceAA); |
| bench->setForceFilter(forceFilter); |
| bench->setDither(forceDither); |
| |
| // only run benchmarks if their name contains matchStr |
| if (matchStr && strstr(bench->getName(), matchStr) == NULL) { |
| continue; |
| } |
| |
| { |
| SkString str; |
| str.printf("running bench [%d %d] %16s", dim.fX, dim.fY, |
| bench->getName()); |
| log_progress(str); |
| } |
| |
| for (int configIndex = 0; configIndex < configCount; configIndex++) { |
| if (configCount > 1) { |
| outConfig = gConfigs[configIndex].fConfig; |
| configName = gConfigs[configIndex].fName; |
| } |
| |
| SkBitmap bm; |
| bm.setConfig(outConfig, dim.fX, dim.fY); |
| bm.allocPixels(); |
| erase(bm); |
| |
| SkCanvas canvas(bm); |
| |
| if (doClip) { |
| performClip(&canvas, dim.fX, dim.fY); |
| } |
| if (doScale) { |
| performScale(&canvas, dim.fX, dim.fY); |
| } |
| if (doRotate) { |
| performRotate(&canvas, dim.fX, dim.fY); |
| } |
| |
| SkMSec now = SkTime::GetMSecs(); |
| for (int i = 0; i < repeatDraw; i++) { |
| SkCanvas* c = &canvas; |
| |
| SkNWayCanvas nway; |
| SkPicture* pict = NULL; |
| if (doPict) { |
| pict = new SkPicture; |
| nway.addCanvas(pict->beginRecording(bm.width(), bm.height())); |
| nway.addCanvas(&canvas); |
| c = &nway; |
| } |
| |
| SkAutoCanvasRestore acr(c, true); |
| bench->draw(c); |
| |
| if (pict) { |
| compare_pict_to_bitmap(pict, bm); |
| pict->unref(); |
| } |
| } |
| if (repeatDraw > 1) { |
| double duration = SkTime::GetMSecs() - now; |
| SkString str; |
| str.printf(" %4s: msecs = %7.2f, fps = %7.2f", configName, |
| duration / repeatDraw, repeatDraw * 1000.0 / duration); |
| log_progress(str); |
| } |
| if (outDir.size() > 0) { |
| saveFile(bench->getName(), configName, outDir.c_str(), bm); |
| } |
| } |
| log_progress("\n"); |
| } |
| |
| return 0; |
| } |
