commit-bot@chromium.org | 0a98d87 | 2014-05-19 15:15:24 +0000 | [diff] [blame] | 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 |