1 /*
2 * Copyright 2014 Google Inc.
3 *
4 * Use of this source code is governed by a BSD-style license that can be
5 * found in the LICENSE file.
6 */
7
8 #include "Test.h"
9
10 #include "SkPictureRecorder.h"
11 #include "SkRecord.h"
12 #include "SkRecorder.h"
13 #include "SkRecords.h"
14 #include "SkShader.h"
15 #include "SkSurface.h"
16
17 #define COUNT(T) + 1
18 static const int kRecordTypes = SK_RECORD_TYPES(COUNT);
19 #undef COUNT
20
21 // Tallies the types of commands it sees into a histogram.
22 class Tally {
23 public:
Tally()24 Tally() { sk_bzero(&fHistogram, sizeof(fHistogram)); }
25
26 template <typename T>
operator ()(const T &)27 void operator()(const T&) { ++fHistogram[T::kType]; }
28
29 template <typename T>
count() const30 int count() const { return fHistogram[T::kType]; }
31
apply(const SkRecord & record)32 void apply(const SkRecord& record) {
33 for (unsigned i = 0; i < record.count(); i++) {
34 record.visit<void>(i, *this);
35 }
36 }
37
38 private:
39 int fHistogram[kRecordTypes];
40 };
41
DEF_TEST(Recorder,r)42 DEF_TEST(Recorder, r) {
43 SkRecord record;
44 SkRecorder recorder(&record, 1920, 1080);
45
46 recorder.drawRect(SkRect::MakeWH(10, 10), SkPaint());
47
48 Tally tally;
49 tally.apply(record);
50 REPORTER_ASSERT(r, 1 == tally.count<SkRecords::DrawRect>());
51 }
52
53 // All of Skia will work fine without support for comment groups, but
54 // Chrome's inspector can break. This serves as a simple regression test.
DEF_TEST(Recorder_CommentGroups,r)55 DEF_TEST(Recorder_CommentGroups, r) {
56 SkRecord record;
57 SkRecorder recorder(&record, 1920, 1080);
58
59 recorder.beginCommentGroup("test");
60 recorder.addComment("foo", "bar");
61 recorder.addComment("baz", "quux");
62 recorder.endCommentGroup();
63
64 Tally tally;
65 tally.apply(record);
66
67 REPORTER_ASSERT(r, 1 == tally.count<SkRecords::BeginCommentGroup>());
68 REPORTER_ASSERT(r, 2 == tally.count<SkRecords::AddComment>());
69 REPORTER_ASSERT(r, 1 == tally.count<SkRecords::EndCommentGroup>());
70 }
71
72 // Regression test for leaking refs held by optional arguments.
DEF_TEST(Recorder_RefLeaking,r)73 DEF_TEST(Recorder_RefLeaking, r) {
74 // We use SaveLayer to test:
75 // - its SkRect argument is optional and SkRect is POD. Just testing that that works.
76 // - its SkPaint argument is optional and SkPaint is not POD. The bug was here.
77
78 SkRect bounds = SkRect::MakeWH(320, 240);
79 SkPaint paint;
80 paint.setShader(SkShader::CreateEmptyShader())->unref();
81
82 REPORTER_ASSERT(r, paint.getShader()->unique());
83 {
84 SkRecord record;
85 SkRecorder recorder(&record, 1920, 1080);
86 recorder.saveLayer(&bounds, &paint);
87 REPORTER_ASSERT(r, !paint.getShader()->unique());
88 }
89 REPORTER_ASSERT(r, paint.getShader()->unique());
90 }
91
DEF_TEST(Recorder_drawImage_takeReference,reporter)92 DEF_TEST(Recorder_drawImage_takeReference, reporter) {
93
94 SkAutoTUnref<SkImage> image;
95 {
96 SkAutoTUnref<SkSurface> surface(SkSurface::NewRasterN32Premul(100, 100));
97 surface->getCanvas()->clear(SK_ColorGREEN);
98 image.reset(surface->newImageSnapshot());
99 }
100 {
101 SkRecord record;
102 SkRecorder recorder(&record, 100, 100);
103
104 // DrawImage is supposed to take a reference
105 recorder.drawImage(image.get(), 0, 0);
106 REPORTER_ASSERT(reporter, !image->unique());
107
108 Tally tally;
109 tally.apply(record);
110
111 REPORTER_ASSERT(reporter, 1 == tally.count<SkRecords::DrawImage>());
112 }
113 REPORTER_ASSERT(reporter, image->unique());
114
115 {
116 SkRecord record;
117 SkRecorder recorder(&record, 100, 100);
118
119 // DrawImageRect is supposed to take a reference
120 recorder.drawImageRect(image.get(), 0, SkRect::MakeWH(100, 100));
121 REPORTER_ASSERT(reporter, !image->unique());
122
123 Tally tally;
124 tally.apply(record);
125
126 REPORTER_ASSERT(reporter, 1 == tally.count<SkRecords::DrawImageRect>());
127 }
128 REPORTER_ASSERT(reporter, image->unique());
129 }
130