src/third_party/skia/dm/DMUtil.cpp

   1 #include "DMUtil.h"
   2
   3 #include "SkColorPriv.h"
   4 #include "SkPicture.h"
   5 #include "SkPictureRecorder.h"
   6
   7 namespace DM {
   8
   9 SkString UnderJoin(const char* a, const char* b) {
  10     SkString s;
  11     s.appendf("%s_%s", a, b);
  12     return s;
  13 }
  14
  15 SkString FileToTaskName(SkString filename) {
  16     for (size_t i = 0; i < filename.size(); i++) {
  17         if ('_' == filename[i]) { filename[i] = '-'; }
  18         if ('.' == filename[i]) { filename[i] = '_'; }
  19     }
  20     return filename;
  21 }
  22
  23 SkPicture* RecordPicture(skiagm::GM* gm, uint32_t recordFlags, SkBBHFactory* factory) {
  24     const SkISize size = gm->getISize();
  25     SkPictureRecorder recorder;
  26     SkCanvas* canvas = recorder.beginRecording(size.width(), size.height(), factory, recordFlags);
  27     canvas->concat(gm->getInitialTransform());
  28     gm->draw(canvas);
  29     canvas->flush();
  30     return recorder.endRecording();
  31 }
  32
  33 void AllocatePixels(SkColorType ct, int width, int height, SkBitmap* bitmap) {
  34     bitmap->allocPixels(SkImageInfo::Make(width, height, ct, kPremul_SkAlphaType));
  35     bitmap->eraseColor(0x00000000);
  36 }
  37
  38 void AllocatePixels(const SkBitmap& reference, SkBitmap* bitmap) {
  39     AllocatePixels(reference.colorType(), reference.width(), reference.height(), bitmap);
  40 }
  41
  42 void DrawPicture(SkPicture* picture, SkBitmap* bitmap) {
  43     SkASSERT(picture != NULL);
  44     SkASSERT(bitmap != NULL);
  45     SkCanvas canvas(*bitmap);
  46     canvas.drawPicture(picture);
  47     canvas.flush();
  48 }
  49
  50 static void unpack_565(uint16_t pixel, unsigned* r, unsigned* g, unsigned* b) {
  51     *r = SkGetPackedR16(pixel);
  52     *g = SkGetPackedG16(pixel);
  53     *b = SkGetPackedB16(pixel);
  54 }
  55
  56 // Returns |a-b|.
  57 static unsigned abs_diff(unsigned a, unsigned b) {
  58     return a > b ? a - b : b - a;
  59 }
  60
  61 unsigned MaxComponentDifference(const SkBitmap& a, const SkBitmap& b) {
  62     if (a.info() != b.info()) {
  63         SkFAIL("Can't compare bitmaps of different shapes.");
  64     }
  65
  66     unsigned max = 0;
  67
  68     const SkAutoLockPixels lockA(a), lockB(b);
  69     if (a.info().colorType() == kRGB_565_SkColorType) {
  70         // 565 is special/annoying because its 3 components straddle 2 bytes.
  71         const uint16_t* aPixels = (const uint16_t*)a.getPixels();
  72         const uint16_t* bPixels = (const uint16_t*)b.getPixels();
  73         for (size_t i = 0; i < a.getSize() / 2; i++) {
  74             unsigned ar, ag, ab,
  75                      br, bg, bb;
  76             unpack_565(aPixels[i], &ar, &ag, &ab);
  77             unpack_565(bPixels[i], &br, &bg, &bb);
  78             max = SkTMax(max, abs_diff(ar, br));
  79             max = SkTMax(max, abs_diff(ag, bg));
  80             max = SkTMax(max, abs_diff(ab, bb));
  81         }
  82     } else {
  83         // Everything else we produce is byte aligned, so max component diff == max byte diff.
  84         const uint8_t* aBytes = (const uint8_t*)a.getPixels();
  85         const uint8_t* bBytes = (const uint8_t*)b.getPixels();
  86         for (size_t i = 0; i < a.getSize(); i++) {
  87             max = SkTMax(max, abs_diff(aBytes[i], bBytes[i]));
  88         }
  89     }
  90
  91     return max;
  92 }
  93
  94 bool BitmapsEqual(const SkBitmap& a, const SkBitmap& b) {
  95     if (a.info() != b.info()) {
  96         return false;
  97     }
  98     const SkAutoLockPixels lockA(a), lockB(b);
  99     return 0 == memcmp(a.getPixels(), b.getPixels(), a.getSize());
 100 }
 101
 102 }  // namespace DM