1 /*
2 * Copyright 2015 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 #ifndef RecordTestUtils_DEFINED
9 #define RecordTestUtils_DEFINED
10
11 #include "SkRecord.h"
12 #include "SkRecords.h"
13 #include "Test.h"
14
15 // If the command we're reading is a U, set ptr to it, otherwise set it to nullptr.
16 template <typename U>
17 struct ReadAs {
ReadAsReadAs18 ReadAs() : ptr(nullptr), type(SkRecords::Type(~0)) {}
19
20 const U* ptr;
21 SkRecords::Type type;
22
operatorReadAs23 void operator()(const U& r) { ptr = &r; type = U::kType; }
24
25 template <typename T>
operatorReadAs26 void operator()(const T&) { type = U::kType; }
27 };
28
29 // Assert that the ith command in record is of type T, and return it.
30 template <typename T>
assert_type(skiatest::Reporter * r,const SkRecord & record,int index)31 static const T* assert_type(skiatest::Reporter* r, const SkRecord& record, int index) {
32 ReadAs<T> reader;
33 record.visit(index, reader);
34 REPORTER_ASSERT(r, T::kType == reader.type);
35 REPORTER_ASSERT(r, SkToBool(reader.ptr));
36 return reader.ptr;
37 }
38
39 template <typename DrawT> struct MatchType {
operatorMatchType40 template <typename T> int operator()(const T&) { return 0; }
operatorMatchType41 int operator()(const DrawT&) { return 1; }
42 };
43
count_instances_of_type(const SkRecord & record)44 template <typename DrawT> int count_instances_of_type(const SkRecord& record) {
45 MatchType<DrawT> matcher;
46 int counter = 0;
47 for (int i = 0; i < record.count(); i++) {
48 counter += record.visit(i, matcher);
49 }
50 return counter;
51 }
52
find_first_instances_of_type(const SkRecord & record)53 template <typename DrawT> int find_first_instances_of_type(const SkRecord& record) {
54 MatchType<DrawT> matcher;
55 for (int i = 0; i < record.count(); i++) {
56 if (record.visit(i, matcher)) {
57 return i;
58 }
59 }
60 return -1;
61 }
62
63 #endif//RecordTestUtils_DEFINED
64