1 #ifndef RecordTestUtils_DEFINED
2 #define RecordTestUtils_DEFINED
3
4 #include "SkRecord.h"
5 #include "SkRecords.h"
6
7 // If the command we're reading is a U, set ptr to it, otherwise set it to NULL.
8 template <typename U>
9 struct ReadAs {
ReadAsReadAs10 ReadAs() : ptr(NULL), type(SkRecords::Type(~0)) {}
11
12 const U* ptr;
13 SkRecords::Type type;
14
operatorReadAs15 void operator()(const U& r) { ptr = &r; type = U::kType; }
16
17 template <typename T>
operatorReadAs18 void operator()(const T&) { type = U::kType; }
19 };
20
21 // Assert that the ith command in record is of type T, and return it.
22 template <typename T>
assert_type(skiatest::Reporter * r,const SkRecord & record,unsigned index)23 static const T* assert_type(skiatest::Reporter* r, const SkRecord& record, unsigned index) {
24 ReadAs<T> reader;
25 record.visit<void>(index, reader);
26 REPORTER_ASSERT(r, T::kType == reader.type);
27 REPORTER_ASSERT(r, SkToBool(reader.ptr));
28 return reader.ptr;
29 }
30
31 template <typename DrawT> struct MatchType {
operatorMatchType32 template <typename T> int operator()(const T&) { return 0; }
operatorMatchType33 int operator()(const DrawT&) { return 1; }
34 };
35
count_instances_of_type(const SkRecord & record)36 template <typename DrawT> int count_instances_of_type(const SkRecord& record) {
37 MatchType<DrawT> matcher;
38 int counter = 0;
39 for (unsigned i = 0; i < record.count(); i++) {
40 counter += record.visit<int>(i, matcher);
41 }
42 return counter;
43 }
44
find_first_instances_of_type(const SkRecord & record)45 template <typename DrawT> int find_first_instances_of_type(const SkRecord& record) {
46 MatchType<DrawT> matcher;
47 for (unsigned i = 0; i < record.count(); i++) {
48 if (record.visit<int>(i, matcher)) {
49 return i;
50 }
51 }
52 return -1;
53 }
54
55 #endif//RecordTestUtils_DEFINED
56