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 { 10 ReadAs() : ptr(NULL), type(SkRecords::Type(~0)) {} 11 12 const U* ptr; 13 SkRecords::Type type; 14 15 void operator()(const U& r) { ptr = &r; type = U::kType; } 16 17 template <typename T> 18 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> 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
