src/third_party/skia/tests/RecordTestUtils.h

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