vandebo@chromium.org | f66025d | 2010-10-01 23:26:55 +0000 | [diff] [blame] | 1 | /* |
epoger@google.com | ec3ed6a | 2011-07-28 14:26:00 +0000 | [diff] [blame] | 2 | * Copyright 2010 The Android Open Source Project |
vandebo@chromium.org | f66025d | 2010-10-01 23:26:55 +0000 | [diff] [blame] | 3 | * |
epoger@google.com | ec3ed6a | 2011-07-28 14:26:00 +0000 | [diff] [blame] | 4 | * Use of this source code is governed by a BSD-style license that can be |
| 5 | * found in the LICENSE file. |
vandebo@chromium.org | f66025d | 2010-10-01 23:26:55 +0000 | [diff] [blame] | 6 | */ |
| 7 | |
edisonn@google.com | d9dfa18 | 2013-04-24 13:01:01 +0000 | [diff] [blame] | 8 | #include "SkBitmap.h" |
vandebo@chromium.org | 4e1cc6a | 2013-01-25 19:27:23 +0000 | [diff] [blame] | 9 | #include "SkCanvas.h" |
reed@google.com | 8a85d0c | 2011-06-24 19:12:12 +0000 | [diff] [blame] | 10 | #include "SkData.h" |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 11 | #include "SkFlate.h" |
edisonn@google.com | d9dfa18 | 2013-04-24 13:01:01 +0000 | [diff] [blame] | 12 | #include "SkImageEncoder.h" |
| 13 | #include "SkMatrix.h" |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 14 | #include "SkPDFCatalog.h" |
vandebo@chromium.org | 4e1cc6a | 2013-01-25 19:27:23 +0000 | [diff] [blame] | 15 | #include "SkPDFDevice.h" |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 16 | #include "SkPDFStream.h" |
| 17 | #include "SkPDFTypes.h" |
senorblanco | b0e89dc | 2014-10-20 14:03:12 -0700 | [diff] [blame] | 18 | #include "SkReadBuffer.h" |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 19 | #include "SkScalar.h" |
| 20 | #include "SkStream.h" |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 21 | #include "SkTypes.h" |
tfarina@chromium.org | 8f6884a | 2014-01-24 20:56:26 +0000 | [diff] [blame] | 22 | #include "Test.h" |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 23 | |
vandebo@chromium.org | 2ef12d4 | 2011-07-06 23:31:24 +0000 | [diff] [blame] | 24 | class SkPDFTestDict : public SkPDFDict { |
| 25 | public: |
edisonn@google.com | 6addb19 | 2013-04-02 15:33:08 +0000 | [diff] [blame] | 26 | virtual void getResources(const SkTSet<SkPDFObject*>& knownResourceObjects, |
| 27 | SkTSet<SkPDFObject*>* newResourceObjects) { |
vandebo@chromium.org | 2ef12d4 | 2011-07-06 23:31:24 +0000 | [diff] [blame] | 28 | for (int i = 0; i < fResources.count(); i++) { |
edisonn@google.com | 6addb19 | 2013-04-02 15:33:08 +0000 | [diff] [blame] | 29 | newResourceObjects->add(fResources[i]); |
vandebo@chromium.org | 2ef12d4 | 2011-07-06 23:31:24 +0000 | [diff] [blame] | 30 | fResources[i]->ref(); |
| 31 | } |
| 32 | } |
| 33 | |
| 34 | void addResource(SkPDFObject* object) { |
| 35 | fResources.append(1, &object); |
| 36 | } |
| 37 | |
| 38 | private: |
| 39 | SkTDArray<SkPDFObject*> fResources; |
| 40 | }; |
| 41 | |
commit-bot@chromium.org | 608ea65 | 2013-10-03 19:29:21 +0000 | [diff] [blame] | 42 | #define DUMMY_TEXT "DCT compessed stream." |
| 43 | |
| 44 | static SkData* encode_to_dct_data(size_t* pixelRefOffset, const SkBitmap& bitmap) { |
| 45 | *pixelRefOffset = 0; |
| 46 | return SkData::NewWithProc(DUMMY_TEXT, sizeof(DUMMY_TEXT) - 1, NULL, NULL); |
edisonn@google.com | d9dfa18 | 2013-04-24 13:01:01 +0000 | [diff] [blame] | 47 | } |
| 48 | |
reed@google.com | 8a85d0c | 2011-06-24 19:12:12 +0000 | [diff] [blame] | 49 | static bool stream_equals(const SkDynamicMemoryWStream& stream, size_t offset, |
| 50 | const void* buffer, size_t len) { |
| 51 | SkAutoDataUnref data(stream.copyToData()); |
robertphillips@google.com | 59f46b8 | 2012-07-10 17:30:58 +0000 | [diff] [blame] | 52 | if (offset + len > data->size()) { |
reed@google.com | 8a85d0c | 2011-06-24 19:12:12 +0000 | [diff] [blame] | 53 | return false; |
| 54 | } |
robertphillips@google.com | 59f46b8 | 2012-07-10 17:30:58 +0000 | [diff] [blame] | 55 | return memcmp(data->bytes() + offset, buffer, len) == 0; |
reed@google.com | 8a85d0c | 2011-06-24 19:12:12 +0000 | [diff] [blame] | 56 | } |
| 57 | |
edisonn@google.com | d9dfa18 | 2013-04-24 13:01:01 +0000 | [diff] [blame] | 58 | static bool stream_contains(const SkDynamicMemoryWStream& stream, |
| 59 | const char* buffer) { |
| 60 | SkAutoDataUnref data(stream.copyToData()); |
bsalomon | fbaace0 | 2014-12-12 16:41:46 -0800 | [diff] [blame] | 61 | size_t len = strlen(buffer); // our buffer does not have EOSs. |
edisonn@google.com | d9dfa18 | 2013-04-24 13:01:01 +0000 | [diff] [blame] | 62 | |
bsalomon | fbaace0 | 2014-12-12 16:41:46 -0800 | [diff] [blame] | 63 | for (size_t offset = 0 ; offset < data->size() - len; offset++) { |
edisonn@google.com | d9dfa18 | 2013-04-24 13:01:01 +0000 | [diff] [blame] | 64 | if (memcmp(data->bytes() + offset, buffer, len) == 0) { |
| 65 | return true; |
| 66 | } |
| 67 | } |
| 68 | |
| 69 | return false; |
| 70 | } |
| 71 | |
halcanary | 6a14434 | 2015-01-23 11:45:10 -0800 | [diff] [blame] | 72 | static void emit_object(SkPDFObject* object, |
| 73 | SkWStream* stream, |
| 74 | SkPDFCatalog* catalog, |
| 75 | bool indirect) { |
| 76 | SkPDFObject* realObject = catalog->getSubstituteObject(object); |
| 77 | if (indirect) { |
| 78 | catalog->emitObjectNumber(stream, realObject); |
| 79 | stream->writeText(" obj\n"); |
| 80 | realObject->emitObject(stream, catalog); |
| 81 | stream->writeText("\nendobj\n"); |
| 82 | } else { |
| 83 | realObject->emitObject(stream, catalog); |
| 84 | } |
| 85 | } |
| 86 | |
| 87 | static size_t get_output_size(SkPDFObject* object, |
| 88 | SkPDFCatalog* catalog, |
| 89 | bool indirect) { |
| 90 | SkDynamicMemoryWStream buffer; |
| 91 | emit_object(object, &buffer, catalog, indirect); |
| 92 | return buffer.getOffset(); |
| 93 | } |
| 94 | |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 95 | static void CheckObjectOutput(skiatest::Reporter* reporter, SkPDFObject* obj, |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 96 | const char* expectedData, size_t expectedSize, |
| 97 | bool indirect, bool compression) { |
| 98 | SkPDFDocument::Flags docFlags = (SkPDFDocument::Flags) 0; |
| 99 | if (!compression) { |
edisonn@google.com | 8c02061 | 2013-03-12 19:53:16 +0000 | [diff] [blame] | 100 | docFlags = SkTBitOr(docFlags, SkPDFDocument::kFavorSpeedOverSize_Flags); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 101 | } |
| 102 | SkPDFCatalog catalog(docFlags); |
halcanary | 6a14434 | 2015-01-23 11:45:10 -0800 | [diff] [blame] | 103 | size_t directSize = get_output_size(obj, &catalog, false); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 104 | REPORTER_ASSERT(reporter, directSize == expectedSize); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 105 | |
| 106 | SkDynamicMemoryWStream buffer; |
halcanary | 6a14434 | 2015-01-23 11:45:10 -0800 | [diff] [blame] | 107 | emit_object(obj, &buffer, &catalog, false); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 108 | REPORTER_ASSERT(reporter, directSize == buffer.getOffset()); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 109 | REPORTER_ASSERT(reporter, stream_equals(buffer, 0, expectedData, |
reed@google.com | 8a85d0c | 2011-06-24 19:12:12 +0000 | [diff] [blame] | 110 | directSize)); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 111 | |
| 112 | if (indirect) { |
| 113 | // Indirect output. |
| 114 | static char header[] = "1 0 obj\n"; |
| 115 | static size_t headerLen = strlen(header); |
| 116 | static char footer[] = "\nendobj\n"; |
| 117 | static size_t footerLen = strlen(footer); |
| 118 | |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 119 | catalog.addObject(obj, false); |
| 120 | |
halcanary | 6a14434 | 2015-01-23 11:45:10 -0800 | [diff] [blame] | 121 | size_t indirectSize = get_output_size(obj, &catalog, true); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 122 | REPORTER_ASSERT(reporter, |
| 123 | indirectSize == directSize + headerLen + footerLen); |
| 124 | |
| 125 | buffer.reset(); |
halcanary | 6a14434 | 2015-01-23 11:45:10 -0800 | [diff] [blame] | 126 | emit_object(obj, &buffer, &catalog, true); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 127 | REPORTER_ASSERT(reporter, indirectSize == buffer.getOffset()); |
reed@google.com | 8a85d0c | 2011-06-24 19:12:12 +0000 | [diff] [blame] | 128 | REPORTER_ASSERT(reporter, stream_equals(buffer, 0, header, headerLen)); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 129 | REPORTER_ASSERT(reporter, stream_equals(buffer, headerLen, expectedData, |
reed@google.com | 8a85d0c | 2011-06-24 19:12:12 +0000 | [diff] [blame] | 130 | directSize)); |
| 131 | REPORTER_ASSERT(reporter, stream_equals(buffer, headerLen + directSize, |
| 132 | footer, footerLen)); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 133 | } |
| 134 | } |
| 135 | |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 136 | static void SimpleCheckObjectOutput(skiatest::Reporter* reporter, |
| 137 | SkPDFObject* obj, |
robertphillips@google.com | 4debcac | 2012-05-14 16:33:36 +0000 | [diff] [blame] | 138 | const char* expectedResult) { |
| 139 | CheckObjectOutput(reporter, obj, expectedResult, |
| 140 | strlen(expectedResult), true, false); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 141 | } |
| 142 | |
| 143 | static void TestPDFStream(skiatest::Reporter* reporter) { |
| 144 | char streamBytes[] = "Test\nFoo\tBar"; |
scroggo | a1193e4 | 2015-01-21 12:09:53 -0800 | [diff] [blame] | 145 | SkAutoTDelete<SkMemoryStream> streamData(new SkMemoryStream( |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 146 | streamBytes, strlen(streamBytes), true)); |
| 147 | SkAutoTUnref<SkPDFStream> stream(new SkPDFStream(streamData.get())); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 148 | SimpleCheckObjectOutput( |
| 149 | reporter, stream.get(), |
| 150 | "<</Length 12\n>> stream\nTest\nFoo\tBar\nendstream"); |
| 151 | stream->insert("Attribute", new SkPDFInt(42))->unref(); |
| 152 | SimpleCheckObjectOutput(reporter, stream.get(), |
| 153 | "<</Length 12\n/Attribute 42\n>> stream\n" |
| 154 | "Test\nFoo\tBar\nendstream"); |
| 155 | |
| 156 | if (SkFlate::HaveFlate()) { |
| 157 | char streamBytes2[] = "This is a longer string, so that compression " |
| 158 | "can do something with it. With shorter strings, " |
| 159 | "the short circuit logic cuts in and we end up " |
| 160 | "with an uncompressed string."; |
| 161 | SkAutoDataUnref streamData2(SkData::NewWithCopy(streamBytes2, |
| 162 | strlen(streamBytes2))); |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 163 | SkAutoTUnref<SkPDFStream> stream(new SkPDFStream(streamData2.get())); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 164 | |
| 165 | SkDynamicMemoryWStream compressedByteStream; |
| 166 | SkFlate::Deflate(streamData2.get(), &compressedByteStream); |
| 167 | SkAutoDataUnref compressedData(compressedByteStream.copyToData()); |
| 168 | |
| 169 | // Check first without compression. |
| 170 | SkDynamicMemoryWStream expectedResult1; |
| 171 | expectedResult1.writeText("<</Length 167\n>> stream\n"); |
| 172 | expectedResult1.writeText(streamBytes2); |
| 173 | expectedResult1.writeText("\nendstream"); |
| 174 | SkAutoDataUnref expectedResultData1(expectedResult1.copyToData()); |
| 175 | CheckObjectOutput(reporter, stream.get(), |
robertphillips@google.com | 59f46b8 | 2012-07-10 17:30:58 +0000 | [diff] [blame] | 176 | (const char*) expectedResultData1->data(), |
| 177 | expectedResultData1->size(), true, false); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 178 | |
| 179 | // Then again with compression. |
| 180 | SkDynamicMemoryWStream expectedResult2; |
| 181 | expectedResult2.writeText("<</Filter /FlateDecode\n/Length 116\n" |
| 182 | ">> stream\n"); |
robertphillips@google.com | 59f46b8 | 2012-07-10 17:30:58 +0000 | [diff] [blame] | 183 | expectedResult2.write(compressedData->data(), compressedData->size()); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 184 | expectedResult2.writeText("\nendstream"); |
| 185 | SkAutoDataUnref expectedResultData2(expectedResult2.copyToData()); |
| 186 | CheckObjectOutput(reporter, stream.get(), |
robertphillips@google.com | 59f46b8 | 2012-07-10 17:30:58 +0000 | [diff] [blame] | 187 | (const char*) expectedResultData2->data(), |
| 188 | expectedResultData2->size(), true, true); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 189 | } |
| 190 | } |
| 191 | |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 192 | static void TestCatalog(skiatest::Reporter* reporter) { |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 193 | SkPDFCatalog catalog((SkPDFDocument::Flags)0); |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 194 | SkAutoTUnref<SkPDFInt> int1(new SkPDFInt(1)); |
| 195 | SkAutoTUnref<SkPDFInt> int2(new SkPDFInt(2)); |
| 196 | SkAutoTUnref<SkPDFInt> int3(new SkPDFInt(3)); |
| 197 | int1.get()->ref(); |
| 198 | SkAutoTUnref<SkPDFInt> int1Again(int1.get()); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 199 | |
| 200 | catalog.addObject(int1.get(), false); |
| 201 | catalog.addObject(int2.get(), false); |
| 202 | catalog.addObject(int3.get(), false); |
| 203 | |
| 204 | REPORTER_ASSERT(reporter, catalog.getObjectNumberSize(int1.get()) == 3); |
| 205 | REPORTER_ASSERT(reporter, catalog.getObjectNumberSize(int2.get()) == 3); |
| 206 | REPORTER_ASSERT(reporter, catalog.getObjectNumberSize(int3.get()) == 3); |
| 207 | |
| 208 | SkDynamicMemoryWStream buffer; |
| 209 | catalog.emitObjectNumber(&buffer, int1.get()); |
| 210 | catalog.emitObjectNumber(&buffer, int2.get()); |
| 211 | catalog.emitObjectNumber(&buffer, int3.get()); |
| 212 | catalog.emitObjectNumber(&buffer, int1Again.get()); |
| 213 | char expectedResult[] = "1 02 03 01 0"; |
reed@google.com | 8a85d0c | 2011-06-24 19:12:12 +0000 | [diff] [blame] | 214 | REPORTER_ASSERT(reporter, stream_equals(buffer, 0, expectedResult, |
| 215 | strlen(expectedResult))); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 216 | } |
| 217 | |
| 218 | static void TestObjectRef(skiatest::Reporter* reporter) { |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 219 | SkAutoTUnref<SkPDFInt> int1(new SkPDFInt(1)); |
| 220 | SkAutoTUnref<SkPDFInt> int2(new SkPDFInt(2)); |
| 221 | SkAutoTUnref<SkPDFObjRef> int2ref(new SkPDFObjRef(int2.get())); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 222 | |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 223 | SkPDFCatalog catalog((SkPDFDocument::Flags)0); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 224 | catalog.addObject(int1.get(), false); |
| 225 | catalog.addObject(int2.get(), false); |
| 226 | REPORTER_ASSERT(reporter, catalog.getObjectNumberSize(int1.get()) == 3); |
| 227 | REPORTER_ASSERT(reporter, catalog.getObjectNumberSize(int2.get()) == 3); |
| 228 | |
| 229 | char expectedResult[] = "2 0 R"; |
| 230 | SkDynamicMemoryWStream buffer; |
halcanary | 4fc48af | 2015-01-12 10:07:50 -0800 | [diff] [blame] | 231 | int2ref->emitObject(&buffer, &catalog); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 232 | REPORTER_ASSERT(reporter, buffer.getOffset() == strlen(expectedResult)); |
reed@google.com | 8a85d0c | 2011-06-24 19:12:12 +0000 | [diff] [blame] | 233 | REPORTER_ASSERT(reporter, stream_equals(buffer, 0, expectedResult, |
| 234 | buffer.getOffset())); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 235 | } |
| 236 | |
vandebo@chromium.org | 2ef12d4 | 2011-07-06 23:31:24 +0000 | [diff] [blame] | 237 | static void TestSubstitute(skiatest::Reporter* reporter) { |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 238 | SkAutoTUnref<SkPDFTestDict> proxy(new SkPDFTestDict()); |
| 239 | SkAutoTUnref<SkPDFTestDict> stub(new SkPDFTestDict()); |
| 240 | SkAutoTUnref<SkPDFInt> int33(new SkPDFInt(33)); |
| 241 | SkAutoTUnref<SkPDFDict> stubResource(new SkPDFDict()); |
| 242 | SkAutoTUnref<SkPDFInt> int44(new SkPDFInt(44)); |
vandebo@chromium.org | 2ef12d4 | 2011-07-06 23:31:24 +0000 | [diff] [blame] | 243 | |
| 244 | stub->insert("Value", int33.get()); |
| 245 | stubResource->insert("InnerValue", int44.get()); |
| 246 | stub->addResource(stubResource.get()); |
| 247 | |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 248 | SkPDFCatalog catalog((SkPDFDocument::Flags)0); |
vandebo@chromium.org | 2ef12d4 | 2011-07-06 23:31:24 +0000 | [diff] [blame] | 249 | catalog.addObject(proxy.get(), false); |
| 250 | catalog.setSubstitute(proxy.get(), stub.get()); |
| 251 | |
| 252 | SkDynamicMemoryWStream buffer; |
halcanary | 6a14434 | 2015-01-23 11:45:10 -0800 | [diff] [blame] | 253 | emit_object(proxy, &buffer, &catalog, false); |
halcanary | f361b71 | 2015-01-13 07:12:57 -0800 | [diff] [blame] | 254 | SkTSet<SkPDFObject*>* substituteResources = |
| 255 | catalog.getSubstituteList(false); |
| 256 | for (int i = 0; i < substituteResources->count(); ++i) { |
halcanary | 6a14434 | 2015-01-23 11:45:10 -0800 | [diff] [blame] | 257 | emit_object((*substituteResources)[i], &buffer, &catalog, true); |
halcanary | f361b71 | 2015-01-13 07:12:57 -0800 | [diff] [blame] | 258 | } |
vandebo@chromium.org | 2ef12d4 | 2011-07-06 23:31:24 +0000 | [diff] [blame] | 259 | |
vandebo@chromium.org | d3a094c | 2011-07-25 22:22:25 +0000 | [diff] [blame] | 260 | char objectResult[] = "2 0 obj\n<</Value 33\n>>\nendobj\n"; |
halcanary | f361b71 | 2015-01-13 07:12:57 -0800 | [diff] [blame] | 261 | catalog.setFileOffset(proxy.get(), 0); |
| 262 | |
halcanary | 6a14434 | 2015-01-23 11:45:10 -0800 | [diff] [blame] | 263 | size_t outputSize = get_output_size( |
| 264 | catalog.getSubstituteObject(proxy.get()), &catalog, true); |
halcanary | f361b71 | 2015-01-13 07:12:57 -0800 | [diff] [blame] | 265 | REPORTER_ASSERT(reporter, outputSize == strlen(objectResult)); |
vandebo@chromium.org | d3a094c | 2011-07-25 22:22:25 +0000 | [diff] [blame] | 266 | |
vandebo@chromium.org | 2ef12d4 | 2011-07-06 23:31:24 +0000 | [diff] [blame] | 267 | char expectedResult[] = |
| 268 | "<</Value 33\n>>1 0 obj\n<</InnerValue 44\n>>\nendobj\n"; |
| 269 | REPORTER_ASSERT(reporter, buffer.getOffset() == strlen(expectedResult)); |
| 270 | REPORTER_ASSERT(reporter, stream_equals(buffer, 0, expectedResult, |
| 271 | buffer.getOffset())); |
| 272 | } |
| 273 | |
edisonn@google.com | d9dfa18 | 2013-04-24 13:01:01 +0000 | [diff] [blame] | 274 | // Create a bitmap that would be very eficiently compressed in a ZIP. |
edisonn@google.com | f1a358f | 2013-04-24 13:21:49 +0000 | [diff] [blame] | 275 | static void setup_bitmap(SkBitmap* bitmap, int width, int height) { |
mike@reedtribe.org | deee496 | 2014-02-13 14:41:43 +0000 | [diff] [blame] | 276 | bitmap->allocN32Pixels(width, height); |
edisonn@google.com | d9dfa18 | 2013-04-24 13:01:01 +0000 | [diff] [blame] | 277 | bitmap->eraseColor(SK_ColorWHITE); |
| 278 | } |
| 279 | |
| 280 | static void TestImage(skiatest::Reporter* reporter, const SkBitmap& bitmap, |
| 281 | const char* expected, bool useDCTEncoder) { |
| 282 | SkISize pageSize = SkISize::Make(bitmap.width(), bitmap.height()); |
robertphillips@google.com | 57c5672 | 2013-08-26 11:10:13 +0000 | [diff] [blame] | 283 | SkAutoTUnref<SkPDFDevice> dev(new SkPDFDevice(pageSize, pageSize, SkMatrix::I())); |
edisonn@google.com | d9dfa18 | 2013-04-24 13:01:01 +0000 | [diff] [blame] | 284 | |
| 285 | if (useDCTEncoder) { |
commit-bot@chromium.org | 608ea65 | 2013-10-03 19:29:21 +0000 | [diff] [blame] | 286 | dev->setDCTEncoder(encode_to_dct_data); |
edisonn@google.com | d9dfa18 | 2013-04-24 13:01:01 +0000 | [diff] [blame] | 287 | } |
| 288 | |
| 289 | SkCanvas c(dev); |
| 290 | c.drawBitmap(bitmap, 0, 0, NULL); |
| 291 | |
| 292 | SkPDFDocument doc; |
| 293 | doc.appendPage(dev); |
| 294 | |
| 295 | SkDynamicMemoryWStream stream; |
| 296 | doc.emitPDF(&stream); |
| 297 | |
| 298 | REPORTER_ASSERT(reporter, stream_contains(stream, expected)); |
| 299 | } |
| 300 | |
| 301 | static void TestUncompressed(skiatest::Reporter* reporter) { |
| 302 | SkBitmap bitmap; |
edisonn@google.com | f1a358f | 2013-04-24 13:21:49 +0000 | [diff] [blame] | 303 | setup_bitmap(&bitmap, 1, 1); |
edisonn@google.com | d9dfa18 | 2013-04-24 13:01:01 +0000 | [diff] [blame] | 304 | TestImage(reporter, bitmap, |
| 305 | "/Subtype /Image\n" |
| 306 | "/Width 1\n" |
| 307 | "/Height 1\n" |
| 308 | "/ColorSpace /DeviceRGB\n" |
| 309 | "/BitsPerComponent 8\n" |
| 310 | "/Length 3\n" |
| 311 | ">> stream", |
| 312 | true); |
| 313 | } |
| 314 | |
| 315 | static void TestFlateDecode(skiatest::Reporter* reporter) { |
| 316 | if (!SkFlate::HaveFlate()) { |
| 317 | return; |
| 318 | } |
skia.committer@gmail.com | 83f0d30 | 2013-04-25 07:01:04 +0000 | [diff] [blame] | 319 | SkBitmap bitmap; |
edisonn@google.com | f1a358f | 2013-04-24 13:21:49 +0000 | [diff] [blame] | 320 | setup_bitmap(&bitmap, 10, 10); |
edisonn@google.com | d9dfa18 | 2013-04-24 13:01:01 +0000 | [diff] [blame] | 321 | TestImage(reporter, bitmap, |
| 322 | "/Subtype /Image\n" |
| 323 | "/Width 10\n" |
| 324 | "/Height 10\n" |
| 325 | "/ColorSpace /DeviceRGB\n" |
| 326 | "/BitsPerComponent 8\n" |
| 327 | "/Filter /FlateDecode\n" |
| 328 | "/Length 13\n" |
| 329 | ">> stream", |
| 330 | false); |
| 331 | } |
| 332 | |
| 333 | static void TestDCTDecode(skiatest::Reporter* reporter) { |
| 334 | SkBitmap bitmap; |
edisonn@google.com | f1a358f | 2013-04-24 13:21:49 +0000 | [diff] [blame] | 335 | setup_bitmap(&bitmap, 32, 32); |
edisonn@google.com | d9dfa18 | 2013-04-24 13:01:01 +0000 | [diff] [blame] | 336 | TestImage(reporter, bitmap, |
| 337 | "/Subtype /Image\n" |
| 338 | "/Width 32\n" |
| 339 | "/Height 32\n" |
| 340 | "/ColorSpace /DeviceRGB\n" |
| 341 | "/BitsPerComponent 8\n" |
| 342 | "/Filter /DCTDecode\n" |
| 343 | "/ColorTransform 0\n" |
| 344 | "/Length 21\n" |
| 345 | ">> stream", |
| 346 | true); |
| 347 | } |
| 348 | |
| 349 | static void TestImages(skiatest::Reporter* reporter) { |
| 350 | TestUncompressed(reporter); |
| 351 | TestFlateDecode(reporter); |
| 352 | TestDCTDecode(reporter); |
| 353 | } |
| 354 | |
vandebo@chromium.org | 4e1cc6a | 2013-01-25 19:27:23 +0000 | [diff] [blame] | 355 | // This test used to assert without the fix submitted for |
| 356 | // http://code.google.com/p/skia/issues/detail?id=1083. |
| 357 | // SKP files might have invalid glyph ids. This test ensures they are ignored, |
| 358 | // and there is no assert on input data in Debug mode. |
sugoi@google.com | 54f0d1b | 2013-02-27 19:17:41 +0000 | [diff] [blame] | 359 | static void test_issue1083() { |
vandebo@chromium.org | 4e1cc6a | 2013-01-25 19:27:23 +0000 | [diff] [blame] | 360 | SkISize pageSize = SkISize::Make(100, 100); |
robertphillips@google.com | 35300c4 | 2013-03-21 17:38:49 +0000 | [diff] [blame] | 361 | SkAutoTUnref<SkPDFDevice> dev(new SkPDFDevice(pageSize, pageSize, SkMatrix::I())); |
vandebo@chromium.org | 4e1cc6a | 2013-01-25 19:27:23 +0000 | [diff] [blame] | 362 | |
| 363 | SkCanvas c(dev); |
| 364 | SkPaint paint; |
| 365 | paint.setTextEncoding(SkPaint::kGlyphID_TextEncoding); |
| 366 | |
| 367 | uint16_t glyphID = 65000; |
| 368 | c.drawText(&glyphID, 2, 0, 0, paint); |
| 369 | |
| 370 | SkPDFDocument doc; |
| 371 | doc.appendPage(dev); |
| 372 | |
| 373 | SkDynamicMemoryWStream stream; |
| 374 | doc.emitPDF(&stream); |
| 375 | } |
| 376 | |
tfarina@chromium.org | e4fafb1 | 2013-12-12 21:11:12 +0000 | [diff] [blame] | 377 | DEF_TEST(PDFPrimitives, reporter) { |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 378 | SkAutoTUnref<SkPDFInt> int42(new SkPDFInt(42)); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 379 | SimpleCheckObjectOutput(reporter, int42.get(), "42"); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 380 | |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 381 | SkAutoTUnref<SkPDFScalar> realHalf(new SkPDFScalar(SK_ScalarHalf)); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 382 | SimpleCheckObjectOutput(reporter, realHalf.get(), "0.5"); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 383 | |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 384 | SkAutoTUnref<SkPDFScalar> bigScalar(new SkPDFScalar(110999.75f)); |
vandebo@chromium.org | 6cc26da | 2011-05-18 17:08:05 +0000 | [diff] [blame] | 385 | #if !defined(SK_ALLOW_LARGE_PDF_SCALARS) |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 386 | SimpleCheckObjectOutput(reporter, bigScalar.get(), "111000"); |
vandebo@chromium.org | 094316b | 2011-03-04 03:15:13 +0000 | [diff] [blame] | 387 | #else |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 388 | SimpleCheckObjectOutput(reporter, bigScalar.get(), "110999.75"); |
vandebo@chromium.org | 094316b | 2011-03-04 03:15:13 +0000 | [diff] [blame] | 389 | |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 390 | SkAutoTUnref<SkPDFScalar> biggerScalar(new SkPDFScalar(50000000.1)); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 391 | SimpleCheckObjectOutput(reporter, biggerScalar.get(), "50000000"); |
vandebo@chromium.org | 094316b | 2011-03-04 03:15:13 +0000 | [diff] [blame] | 392 | |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 393 | SkAutoTUnref<SkPDFScalar> smallestScalar(new SkPDFScalar(1.0/65536)); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 394 | SimpleCheckObjectOutput(reporter, smallestScalar.get(), "0.00001526"); |
vandebo@chromium.org | 094316b | 2011-03-04 03:15:13 +0000 | [diff] [blame] | 395 | #endif |
| 396 | |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 397 | SkAutoTUnref<SkPDFString> stringSimple( |
| 398 | new SkPDFString("test ) string ( foo")); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 399 | SimpleCheckObjectOutput(reporter, stringSimple.get(), |
| 400 | "(test \\) string \\( foo)"); |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 401 | SkAutoTUnref<SkPDFString> stringComplex( |
| 402 | new SkPDFString("\ttest ) string ( foo")); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 403 | SimpleCheckObjectOutput(reporter, stringComplex.get(), |
| 404 | "<0974657374202920737472696E67202820666F6F>"); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 405 | |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 406 | SkAutoTUnref<SkPDFName> name(new SkPDFName("Test name\twith#tab")); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 407 | const char expectedResult[] = "/Test#20name#09with#23tab"; |
| 408 | CheckObjectOutput(reporter, name.get(), expectedResult, |
| 409 | strlen(expectedResult), false, false); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 410 | |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 411 | SkAutoTUnref<SkPDFName> escapedName(new SkPDFName("A#/%()<>[]{}B")); |
vandebo@chromium.org | 251a766 | 2012-09-21 17:50:50 +0000 | [diff] [blame] | 412 | const char escapedNameExpected[] = "/A#23#2F#25#28#29#3C#3E#5B#5D#7B#7DB"; |
| 413 | CheckObjectOutput(reporter, escapedName.get(), escapedNameExpected, |
| 414 | strlen(escapedNameExpected), false, false); |
| 415 | |
vandebo@chromium.org | c0376fe | 2012-03-05 18:44:33 +0000 | [diff] [blame] | 416 | // Test that we correctly handle characters with the high-bit set. |
bungeman@google.com | f8aa18c | 2012-03-19 21:04:52 +0000 | [diff] [blame] | 417 | const unsigned char highBitCString[] = {0xDE, 0xAD, 'b', 'e', 0xEF, 0}; |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 418 | SkAutoTUnref<SkPDFName> highBitName( |
| 419 | new SkPDFName((const char*)highBitCString)); |
vandebo@chromium.org | c0376fe | 2012-03-05 18:44:33 +0000 | [diff] [blame] | 420 | const char highBitExpectedResult[] = "/#DE#ADbe#EF"; |
| 421 | CheckObjectOutput(reporter, highBitName.get(), highBitExpectedResult, |
| 422 | strlen(highBitExpectedResult), false, false); |
| 423 | |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 424 | SkAutoTUnref<SkPDFArray> array(new SkPDFArray); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 425 | SimpleCheckObjectOutput(reporter, array.get(), "[]"); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 426 | array->append(int42.get()); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 427 | SimpleCheckObjectOutput(reporter, array.get(), "[42]"); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 428 | array->append(realHalf.get()); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 429 | SimpleCheckObjectOutput(reporter, array.get(), "[42 0.5]"); |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 430 | SkAutoTUnref<SkPDFInt> int0(new SkPDFInt(0)); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 431 | array->append(int0.get()); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 432 | SimpleCheckObjectOutput(reporter, array.get(), "[42 0.5 0]"); |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 433 | SkAutoTUnref<SkPDFInt> int1(new SkPDFInt(1)); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 434 | array->setAt(0, int1.get()); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 435 | SimpleCheckObjectOutput(reporter, array.get(), "[1 0.5 0]"); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 436 | |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 437 | SkAutoTUnref<SkPDFDict> dict(new SkPDFDict); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 438 | SimpleCheckObjectOutput(reporter, dict.get(), "<<>>"); |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 439 | SkAutoTUnref<SkPDFName> n1(new SkPDFName("n1")); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 440 | dict->insert(n1.get(), int42.get()); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 441 | SimpleCheckObjectOutput(reporter, dict.get(), "<</n1 42\n>>"); |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 442 | SkAutoTUnref<SkPDFName> n2(new SkPDFName("n2")); |
| 443 | SkAutoTUnref<SkPDFName> n3(new SkPDFName("n3")); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 444 | dict->insert(n2.get(), realHalf.get()); |
| 445 | dict->insert(n3.get(), array.get()); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 446 | SimpleCheckObjectOutput(reporter, dict.get(), |
| 447 | "<</n1 42\n/n2 0.5\n/n3 [1 0.5 0]\n>>"); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 448 | |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 449 | TestPDFStream(reporter); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 450 | |
| 451 | TestCatalog(reporter); |
| 452 | |
| 453 | TestObjectRef(reporter); |
vandebo@chromium.org | 2ef12d4 | 2011-07-06 23:31:24 +0000 | [diff] [blame] | 454 | |
| 455 | TestSubstitute(reporter); |
vandebo@chromium.org | 4e1cc6a | 2013-01-25 19:27:23 +0000 | [diff] [blame] | 456 | |
sugoi@google.com | 54f0d1b | 2013-02-27 19:17:41 +0000 | [diff] [blame] | 457 | test_issue1083(); |
edisonn@google.com | d9dfa18 | 2013-04-24 13:01:01 +0000 | [diff] [blame] | 458 | |
skia.committer@gmail.com | 83f0d30 | 2013-04-25 07:01:04 +0000 | [diff] [blame] | 459 | TestImages(reporter); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 460 | } |
senorblanco | b0e89dc | 2014-10-20 14:03:12 -0700 | [diff] [blame] | 461 | |
| 462 | namespace { |
| 463 | |
| 464 | class DummyImageFilter : public SkImageFilter { |
| 465 | public: |
| 466 | DummyImageFilter(bool visited = false) : SkImageFilter(0, NULL), fVisited(visited) {} |
mtklein | 72c9faa | 2015-01-09 10:06:39 -0800 | [diff] [blame] | 467 | ~DummyImageFilter() SK_OVERRIDE {} |
senorblanco | b0e89dc | 2014-10-20 14:03:12 -0700 | [diff] [blame] | 468 | virtual bool onFilterImage(Proxy*, const SkBitmap& src, const Context&, |
| 469 | SkBitmap* result, SkIPoint* offset) const { |
| 470 | fVisited = true; |
| 471 | offset->fX = offset->fY = 0; |
| 472 | *result = src; |
| 473 | return true; |
| 474 | } |
robertphillips | f3f5bad | 2014-12-19 13:49:15 -0800 | [diff] [blame] | 475 | SK_TO_STRING_OVERRIDE() |
senorblanco | b0e89dc | 2014-10-20 14:03:12 -0700 | [diff] [blame] | 476 | SK_DECLARE_PUBLIC_FLATTENABLE_DESERIALIZATION_PROCS(DummyImageFilter) |
senorblanco | b0e89dc | 2014-10-20 14:03:12 -0700 | [diff] [blame] | 477 | bool visited() const { return fVisited; } |
| 478 | |
| 479 | private: |
| 480 | mutable bool fVisited; |
| 481 | }; |
| 482 | |
| 483 | SkFlattenable* DummyImageFilter::CreateProc(SkReadBuffer& buffer) { |
| 484 | SK_IMAGEFILTER_UNFLATTEN_COMMON(common, 0); |
| 485 | bool visited = buffer.readBool(); |
| 486 | return SkNEW_ARGS(DummyImageFilter, (visited)); |
| 487 | } |
| 488 | |
robertphillips | f3f5bad | 2014-12-19 13:49:15 -0800 | [diff] [blame] | 489 | #ifndef SK_IGNORE_TO_STRING |
| 490 | void DummyImageFilter::toString(SkString* str) const { |
| 491 | str->appendf("DummyImageFilter: ("); |
| 492 | str->append(")"); |
| 493 | } |
| 494 | #endif |
| 495 | |
senorblanco | b0e89dc | 2014-10-20 14:03:12 -0700 | [diff] [blame] | 496 | }; |
| 497 | |
| 498 | // Check that PDF rendering of image filters successfully falls back to |
| 499 | // CPU rasterization. |
| 500 | DEF_TEST(PDFImageFilter, reporter) { |
| 501 | SkISize pageSize = SkISize::Make(100, 100); |
| 502 | SkAutoTUnref<SkPDFDevice> device(new SkPDFDevice(pageSize, pageSize, SkMatrix::I())); |
| 503 | SkCanvas canvas(device.get()); |
| 504 | SkAutoTUnref<DummyImageFilter> filter(new DummyImageFilter()); |
| 505 | |
| 506 | // Filter just created; should be unvisited. |
| 507 | REPORTER_ASSERT(reporter, !filter->visited()); |
| 508 | SkPaint paint; |
| 509 | paint.setImageFilter(filter.get()); |
| 510 | canvas.drawRect(SkRect::MakeWH(100, 100), paint); |
| 511 | |
| 512 | // Filter was used in rendering; should be visited. |
| 513 | REPORTER_ASSERT(reporter, filter->visited()); |
| 514 | } |