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