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