epoger@google.com | ec3ed6a | 2011-07-28 14:26:00 +0000 | [diff] [blame] | 1 | |
reed@google.com | dde0956 | 2011-05-23 12:21:05 +0000 | [diff] [blame] | 2 | /* |
epoger@google.com | ec3ed6a | 2011-07-28 14:26:00 +0000 | [diff] [blame] | 3 | * Copyright 2011 Google Inc. |
| 4 | * |
| 5 | * Use of this source code is governed by a BSD-style license that can be |
| 6 | * found in the LICENSE file. |
reed@google.com | dde0956 | 2011-05-23 12:21:05 +0000 | [diff] [blame] | 7 | */ |
| 8 | |
| 9 | |
epoger@google.com | ec3ed6a | 2011-07-28 14:26:00 +0000 | [diff] [blame] | 10 | |
scroggo@google.com | dd39488 | 2012-07-24 20:47:55 +0000 | [diff] [blame] | 11 | #include "SkRandom.h" |
reed@google.com | dde0956 | 2011-05-23 12:21:05 +0000 | [diff] [blame] | 12 | #include "SkReader32.h" |
| 13 | #include "SkWriter32.h" |
| 14 | #include "Test.h" |
| 15 | |
reed@google.com | 51c62a6 | 2012-06-12 20:47:53 +0000 | [diff] [blame] | 16 | static void test_ptr(skiatest::Reporter* reporter) { |
| 17 | SkSWriter32<32> writer(32); |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame^] | 18 | |
reed@google.com | 51c62a6 | 2012-06-12 20:47:53 +0000 | [diff] [blame] | 19 | void* p0 = reporter; |
| 20 | void* p1 = &writer; |
| 21 | |
| 22 | // try writing ptrs where at least one of them may be at a non-multiple of |
| 23 | // 8 boundary, to confirm this works on 64bit machines. |
| 24 | |
| 25 | writer.writePtr(p0); |
| 26 | writer.write8(0x33); |
| 27 | writer.writePtr(p1); |
| 28 | writer.write8(0x66); |
| 29 | |
| 30 | size_t size = writer.size(); |
| 31 | REPORTER_ASSERT(reporter, 2 * sizeof(void*) + 2 * sizeof(int32_t)); |
| 32 | |
| 33 | char buffer[32]; |
| 34 | SkASSERT(sizeof(buffer) >= size); |
| 35 | writer.flatten(buffer); |
| 36 | |
| 37 | SkReader32 reader(buffer, size); |
| 38 | REPORTER_ASSERT(reporter, reader.readPtr() == p0); |
| 39 | REPORTER_ASSERT(reporter, reader.readInt() == 0x33); |
| 40 | REPORTER_ASSERT(reporter, reader.readPtr() == p1); |
| 41 | REPORTER_ASSERT(reporter, reader.readInt() == 0x66); |
| 42 | } |
| 43 | |
reed@google.com | dde0956 | 2011-05-23 12:21:05 +0000 | [diff] [blame] | 44 | static void test1(skiatest::Reporter* reporter, SkWriter32* writer) { |
| 45 | const uint32_t data[] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 }; |
| 46 | for (size_t i = 0; i < SK_ARRAY_COUNT(data); ++i) { |
| 47 | REPORTER_ASSERT(reporter, i*4 == writer->size()); |
| 48 | writer->write32(data[i]); |
| 49 | uint32_t* addr = writer->peek32(i * 4); |
| 50 | REPORTER_ASSERT(reporter, data[i] == *addr); |
| 51 | } |
| 52 | |
| 53 | char buffer[sizeof(data)]; |
| 54 | REPORTER_ASSERT(reporter, sizeof(buffer) == writer->size()); |
| 55 | writer->flatten(buffer); |
| 56 | REPORTER_ASSERT(reporter, !memcmp(data, buffer, sizeof(buffer))); |
| 57 | } |
| 58 | |
| 59 | static void test2(skiatest::Reporter* reporter, SkWriter32* writer) { |
| 60 | static const char gStr[] = "abcdefghimjklmnopqrstuvwxyz"; |
| 61 | size_t i; |
| 62 | |
| 63 | size_t len = 0; |
| 64 | for (i = 0; i <= 26; ++i) { |
| 65 | len += SkWriter32::WriteStringSize(gStr, i); |
| 66 | writer->writeString(gStr, i); |
| 67 | } |
| 68 | REPORTER_ASSERT(reporter, writer->size() == len); |
| 69 | |
| 70 | SkAutoMalloc storage(len); |
| 71 | writer->flatten(storage.get()); |
| 72 | |
| 73 | SkReader32 reader; |
| 74 | reader.setMemory(storage.get(), len); |
| 75 | for (i = 0; i <= 26; ++i) { |
| 76 | REPORTER_ASSERT(reporter, !reader.eof()); |
| 77 | const char* str = reader.readString(&len); |
| 78 | REPORTER_ASSERT(reporter, i == len); |
| 79 | REPORTER_ASSERT(reporter, strlen(str) == len); |
| 80 | REPORTER_ASSERT(reporter, !memcmp(str, gStr, len)); |
scroggo@google.com | e9617eb | 2012-07-23 13:44:10 +0000 | [diff] [blame] | 81 | // Ensure that the align4 of the string is padded with zeroes. |
| 82 | size_t alignedSize = SkAlign4(len + 1); |
| 83 | for (size_t j = len; j < alignedSize; j++) { |
| 84 | REPORTER_ASSERT(reporter, 0 == str[j]); |
| 85 | } |
reed@google.com | dde0956 | 2011-05-23 12:21:05 +0000 | [diff] [blame] | 86 | } |
| 87 | REPORTER_ASSERT(reporter, reader.eof()); |
| 88 | } |
| 89 | |
scroggo@google.com | dd39488 | 2012-07-24 20:47:55 +0000 | [diff] [blame] | 90 | static void testWritePad(skiatest::Reporter* reporter, SkWriter32* writer) { |
| 91 | // Create some random data to write. |
| 92 | const size_t dataSize = 10<<2; |
| 93 | SkASSERT(SkIsAlign4(dataSize)); |
| 94 | |
| 95 | SkAutoMalloc originalData(dataSize); |
| 96 | { |
| 97 | SkRandom rand(0); |
| 98 | uint32_t* ptr = static_cast<uint32_t*>(originalData.get()); |
| 99 | uint32_t* stop = ptr + (dataSize>>2); |
| 100 | while (ptr < stop) { |
| 101 | *ptr++ = rand.nextU(); |
| 102 | } |
| 103 | |
| 104 | // Write the random data to the writer at different lengths for |
| 105 | // different alignments. |
| 106 | for (size_t len = 0; len < dataSize; len++) { |
| 107 | writer->writePad(originalData.get(), len); |
| 108 | } |
| 109 | } |
| 110 | |
| 111 | uint32_t totalBytes = writer->size(); |
| 112 | |
| 113 | SkAutoMalloc readStorage(totalBytes); |
| 114 | writer->flatten(readStorage.get()); |
| 115 | |
| 116 | SkReader32 reader; |
| 117 | reader.setMemory(readStorage.get(), totalBytes); |
| 118 | |
| 119 | for (size_t len = 0; len < dataSize; len++) { |
| 120 | const char* readPtr = static_cast<const char*>(reader.skip(len)); |
| 121 | // Ensure that the data read is the same as what was written. |
| 122 | REPORTER_ASSERT(reporter, memcmp(readPtr, originalData.get(), len) == 0); |
| 123 | // Ensure that the rest is padded with zeroes. |
| 124 | const char* stop = readPtr + SkAlign4(len); |
| 125 | readPtr += len; |
| 126 | while (readPtr < stop) { |
| 127 | REPORTER_ASSERT(reporter, *readPtr++ == 0); |
| 128 | } |
| 129 | } |
| 130 | } |
| 131 | |
reed@google.com | dde0956 | 2011-05-23 12:21:05 +0000 | [diff] [blame] | 132 | static void Tests(skiatest::Reporter* reporter) { |
| 133 | // dynamic allocator |
| 134 | { |
| 135 | SkWriter32 writer(256 * 4); |
| 136 | REPORTER_ASSERT(reporter, NULL == writer.getSingleBlock()); |
| 137 | test1(reporter, &writer); |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame^] | 138 | |
reed@google.com | dde0956 | 2011-05-23 12:21:05 +0000 | [diff] [blame] | 139 | writer.reset(); |
| 140 | test2(reporter, &writer); |
scroggo@google.com | dd39488 | 2012-07-24 20:47:55 +0000 | [diff] [blame] | 141 | |
| 142 | writer.reset(); |
| 143 | testWritePad(reporter, &writer); |
reed@google.com | dde0956 | 2011-05-23 12:21:05 +0000 | [diff] [blame] | 144 | } |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame^] | 145 | |
reed@google.com | dde0956 | 2011-05-23 12:21:05 +0000 | [diff] [blame] | 146 | // single-block |
| 147 | { |
| 148 | SkWriter32 writer(0); |
| 149 | uint32_t storage[256]; |
| 150 | REPORTER_ASSERT(reporter, NULL == writer.getSingleBlock()); |
| 151 | writer.reset(storage, sizeof(storage)); |
| 152 | REPORTER_ASSERT(reporter, (void*)storage == writer.getSingleBlock()); |
| 153 | test1(reporter, &writer); |
| 154 | |
| 155 | writer.reset(storage, sizeof(storage)); |
| 156 | test2(reporter, &writer); |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame^] | 157 | |
scroggo@google.com | dd39488 | 2012-07-24 20:47:55 +0000 | [diff] [blame] | 158 | writer.reset(storage, sizeof(storage)); |
| 159 | testWritePad(reporter, &writer); |
reed@google.com | dde0956 | 2011-05-23 12:21:05 +0000 | [diff] [blame] | 160 | } |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame^] | 161 | |
reed@google.com | e49aca9 | 2012-04-24 21:12:39 +0000 | [diff] [blame] | 162 | // small storage |
| 163 | { |
reed@google.com | 51c62a6 | 2012-06-12 20:47:53 +0000 | [diff] [blame] | 164 | SkSWriter32<8 * sizeof(intptr_t)> writer(100); |
reed@google.com | e49aca9 | 2012-04-24 21:12:39 +0000 | [diff] [blame] | 165 | test1(reporter, &writer); |
| 166 | writer.reset(); // should just rewind our storage |
| 167 | test2(reporter, &writer); |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame^] | 168 | |
scroggo@google.com | dd39488 | 2012-07-24 20:47:55 +0000 | [diff] [blame] | 169 | writer.reset(); |
| 170 | testWritePad(reporter, &writer); |
reed@google.com | e49aca9 | 2012-04-24 21:12:39 +0000 | [diff] [blame] | 171 | } |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame^] | 172 | |
reed@google.com | e49aca9 | 2012-04-24 21:12:39 +0000 | [diff] [blame] | 173 | // large storage |
| 174 | { |
reed@google.com | 51c62a6 | 2012-06-12 20:47:53 +0000 | [diff] [blame] | 175 | SkSWriter32<1024 * sizeof(intptr_t)> writer(100); |
reed@google.com | e49aca9 | 2012-04-24 21:12:39 +0000 | [diff] [blame] | 176 | test1(reporter, &writer); |
| 177 | writer.reset(); // should just rewind our storage |
| 178 | test2(reporter, &writer); |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame^] | 179 | |
scroggo@google.com | dd39488 | 2012-07-24 20:47:55 +0000 | [diff] [blame] | 180 | writer.reset(); |
| 181 | testWritePad(reporter, &writer); |
reed@google.com | e49aca9 | 2012-04-24 21:12:39 +0000 | [diff] [blame] | 182 | } |
rmistry@google.com | d6176b0 | 2012-08-23 18:14:13 +0000 | [diff] [blame^] | 183 | |
reed@google.com | 51c62a6 | 2012-06-12 20:47:53 +0000 | [diff] [blame] | 184 | test_ptr(reporter); |
reed@google.com | dde0956 | 2011-05-23 12:21:05 +0000 | [diff] [blame] | 185 | } |
| 186 | |
| 187 | #include "TestClassDef.h" |
| 188 | DEFINE_TESTCLASS("Writer32", Writer32Class, Tests) |
| 189 | |