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