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, NULL != reader.ptr);
28 return reader.ptr;
29 }
30
31 #endif//RecordTestUtils_DEFINED
32