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) { |
halcanary | bf799cd | 2015-02-10 13:32:09 -0800 | [diff] [blame] | 78 | stream->writeDecAsText(catalog->getObjectNumber(object)); |
| 79 | stream->writeText(" 0 obj\n"); // Generation number is always 0. |
halcanary | 6a14434 | 2015-01-23 11:45:10 -0800 | [diff] [blame] | 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 | |
halcanary | 91d1d62 | 2015-02-17 14:43:06 -0800 | [diff] [blame^] | 156 | #ifndef SK_NO_FLATE |
| 157 | { |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 158 | char streamBytes2[] = "This is a longer string, so that compression " |
| 159 | "can do something with it. With shorter strings, " |
| 160 | "the short circuit logic cuts in and we end up " |
| 161 | "with an uncompressed string."; |
| 162 | SkAutoDataUnref streamData2(SkData::NewWithCopy(streamBytes2, |
| 163 | strlen(streamBytes2))); |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 164 | SkAutoTUnref<SkPDFStream> stream(new SkPDFStream(streamData2.get())); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 165 | |
| 166 | SkDynamicMemoryWStream compressedByteStream; |
| 167 | SkFlate::Deflate(streamData2.get(), &compressedByteStream); |
| 168 | SkAutoDataUnref compressedData(compressedByteStream.copyToData()); |
| 169 | |
| 170 | // Check first without compression. |
| 171 | SkDynamicMemoryWStream expectedResult1; |
| 172 | expectedResult1.writeText("<</Length 167\n>> stream\n"); |
| 173 | expectedResult1.writeText(streamBytes2); |
| 174 | expectedResult1.writeText("\nendstream"); |
| 175 | SkAutoDataUnref expectedResultData1(expectedResult1.copyToData()); |
| 176 | CheckObjectOutput(reporter, stream.get(), |
robertphillips@google.com | 59f46b8 | 2012-07-10 17:30:58 +0000 | [diff] [blame] | 177 | (const char*) expectedResultData1->data(), |
| 178 | expectedResultData1->size(), true, false); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 179 | |
| 180 | // Then again with compression. |
| 181 | SkDynamicMemoryWStream expectedResult2; |
| 182 | expectedResult2.writeText("<</Filter /FlateDecode\n/Length 116\n" |
| 183 | ">> stream\n"); |
robertphillips@google.com | 59f46b8 | 2012-07-10 17:30:58 +0000 | [diff] [blame] | 184 | expectedResult2.write(compressedData->data(), compressedData->size()); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 185 | expectedResult2.writeText("\nendstream"); |
| 186 | SkAutoDataUnref expectedResultData2(expectedResult2.copyToData()); |
| 187 | CheckObjectOutput(reporter, stream.get(), |
robertphillips@google.com | 59f46b8 | 2012-07-10 17:30:58 +0000 | [diff] [blame] | 188 | (const char*) expectedResultData2->data(), |
| 189 | expectedResultData2->size(), true, true); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 190 | } |
halcanary | 91d1d62 | 2015-02-17 14:43:06 -0800 | [diff] [blame^] | 191 | #endif // SK_NO_FLATE |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 192 | } |
| 193 | |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 194 | static void TestCatalog(skiatest::Reporter* reporter) { |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 195 | SkPDFCatalog catalog((SkPDFDocument::Flags)0); |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 196 | SkAutoTUnref<SkPDFInt> int1(new SkPDFInt(1)); |
| 197 | SkAutoTUnref<SkPDFInt> int2(new SkPDFInt(2)); |
| 198 | SkAutoTUnref<SkPDFInt> int3(new SkPDFInt(3)); |
| 199 | int1.get()->ref(); |
| 200 | SkAutoTUnref<SkPDFInt> int1Again(int1.get()); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 201 | |
| 202 | catalog.addObject(int1.get(), false); |
| 203 | catalog.addObject(int2.get(), false); |
| 204 | catalog.addObject(int3.get(), false); |
| 205 | |
halcanary | bf799cd | 2015-02-10 13:32:09 -0800 | [diff] [blame] | 206 | REPORTER_ASSERT(reporter, catalog.getObjectNumber(int1.get()) == 1); |
| 207 | REPORTER_ASSERT(reporter, catalog.getObjectNumber(int2.get()) == 2); |
| 208 | REPORTER_ASSERT(reporter, catalog.getObjectNumber(int3.get()) == 3); |
| 209 | REPORTER_ASSERT(reporter, catalog.getObjectNumber(int1Again.get()) == 1); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 210 | } |
| 211 | |
| 212 | static void TestObjectRef(skiatest::Reporter* reporter) { |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 213 | SkAutoTUnref<SkPDFInt> int1(new SkPDFInt(1)); |
| 214 | SkAutoTUnref<SkPDFInt> int2(new SkPDFInt(2)); |
| 215 | SkAutoTUnref<SkPDFObjRef> int2ref(new SkPDFObjRef(int2.get())); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 216 | |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 217 | SkPDFCatalog catalog((SkPDFDocument::Flags)0); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 218 | catalog.addObject(int1.get(), false); |
| 219 | catalog.addObject(int2.get(), false); |
halcanary | bf799cd | 2015-02-10 13:32:09 -0800 | [diff] [blame] | 220 | REPORTER_ASSERT(reporter, catalog.getObjectNumber(int1.get()) == 1); |
| 221 | REPORTER_ASSERT(reporter, catalog.getObjectNumber(int2.get()) == 2); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 222 | |
| 223 | char expectedResult[] = "2 0 R"; |
| 224 | SkDynamicMemoryWStream buffer; |
halcanary | 4fc48af | 2015-01-12 10:07:50 -0800 | [diff] [blame] | 225 | int2ref->emitObject(&buffer, &catalog); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 226 | REPORTER_ASSERT(reporter, buffer.getOffset() == strlen(expectedResult)); |
reed@google.com | 8a85d0c | 2011-06-24 19:12:12 +0000 | [diff] [blame] | 227 | REPORTER_ASSERT(reporter, stream_equals(buffer, 0, expectedResult, |
| 228 | buffer.getOffset())); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 229 | } |
| 230 | |
vandebo@chromium.org | 2ef12d4 | 2011-07-06 23:31:24 +0000 | [diff] [blame] | 231 | static void TestSubstitute(skiatest::Reporter* reporter) { |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 232 | SkAutoTUnref<SkPDFTestDict> proxy(new SkPDFTestDict()); |
| 233 | SkAutoTUnref<SkPDFTestDict> stub(new SkPDFTestDict()); |
vandebo@chromium.org | 2ef12d4 | 2011-07-06 23:31:24 +0000 | [diff] [blame] | 234 | |
halcanary | bf799cd | 2015-02-10 13:32:09 -0800 | [diff] [blame] | 235 | proxy->insert("Value", new SkPDFInt(33))->unref(); |
| 236 | stub->insert("Value", new SkPDFInt(44))->unref(); |
vandebo@chromium.org | 2ef12d4 | 2011-07-06 23:31:24 +0000 | [diff] [blame] | 237 | |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 238 | SkPDFCatalog catalog((SkPDFDocument::Flags)0); |
vandebo@chromium.org | 2ef12d4 | 2011-07-06 23:31:24 +0000 | [diff] [blame] | 239 | catalog.addObject(proxy.get(), false); |
| 240 | catalog.setSubstitute(proxy.get(), stub.get()); |
| 241 | |
halcanary | bf799cd | 2015-02-10 13:32:09 -0800 | [diff] [blame] | 242 | REPORTER_ASSERT(reporter, stub.get() == catalog.getSubstituteObject(proxy)); |
| 243 | REPORTER_ASSERT(reporter, proxy.get() != catalog.getSubstituteObject(stub)); |
vandebo@chromium.org | 2ef12d4 | 2011-07-06 23:31:24 +0000 | [diff] [blame] | 244 | } |
| 245 | |
edisonn@google.com | d9dfa18 | 2013-04-24 13:01:01 +0000 | [diff] [blame] | 246 | // 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] | 247 | static void setup_bitmap(SkBitmap* bitmap, int width, int height) { |
mike@reedtribe.org | deee496 | 2014-02-13 14:41:43 +0000 | [diff] [blame] | 248 | bitmap->allocN32Pixels(width, height); |
edisonn@google.com | d9dfa18 | 2013-04-24 13:01:01 +0000 | [diff] [blame] | 249 | bitmap->eraseColor(SK_ColorWHITE); |
| 250 | } |
| 251 | |
| 252 | static void TestImage(skiatest::Reporter* reporter, const SkBitmap& bitmap, |
| 253 | const char* expected, bool useDCTEncoder) { |
| 254 | SkISize pageSize = SkISize::Make(bitmap.width(), bitmap.height()); |
robertphillips@google.com | 57c5672 | 2013-08-26 11:10:13 +0000 | [diff] [blame] | 255 | SkAutoTUnref<SkPDFDevice> dev(new SkPDFDevice(pageSize, pageSize, SkMatrix::I())); |
edisonn@google.com | d9dfa18 | 2013-04-24 13:01:01 +0000 | [diff] [blame] | 256 | |
| 257 | if (useDCTEncoder) { |
commit-bot@chromium.org | 608ea65 | 2013-10-03 19:29:21 +0000 | [diff] [blame] | 258 | dev->setDCTEncoder(encode_to_dct_data); |
edisonn@google.com | d9dfa18 | 2013-04-24 13:01:01 +0000 | [diff] [blame] | 259 | } |
| 260 | |
| 261 | SkCanvas c(dev); |
| 262 | c.drawBitmap(bitmap, 0, 0, NULL); |
| 263 | |
| 264 | SkPDFDocument doc; |
| 265 | doc.appendPage(dev); |
| 266 | |
| 267 | SkDynamicMemoryWStream stream; |
| 268 | doc.emitPDF(&stream); |
| 269 | |
| 270 | REPORTER_ASSERT(reporter, stream_contains(stream, expected)); |
| 271 | } |
| 272 | |
| 273 | static void TestUncompressed(skiatest::Reporter* reporter) { |
| 274 | SkBitmap bitmap; |
edisonn@google.com | f1a358f | 2013-04-24 13:21:49 +0000 | [diff] [blame] | 275 | setup_bitmap(&bitmap, 1, 1); |
edisonn@google.com | d9dfa18 | 2013-04-24 13:01:01 +0000 | [diff] [blame] | 276 | TestImage(reporter, bitmap, |
| 277 | "/Subtype /Image\n" |
| 278 | "/Width 1\n" |
| 279 | "/Height 1\n" |
| 280 | "/ColorSpace /DeviceRGB\n" |
| 281 | "/BitsPerComponent 8\n" |
| 282 | "/Length 3\n" |
| 283 | ">> stream", |
| 284 | true); |
| 285 | } |
| 286 | |
| 287 | static void TestFlateDecode(skiatest::Reporter* reporter) { |
halcanary | 91d1d62 | 2015-02-17 14:43:06 -0800 | [diff] [blame^] | 288 | #ifndef SK_NO_FLATE |
skia.committer@gmail.com | 83f0d30 | 2013-04-25 07:01:04 +0000 | [diff] [blame] | 289 | SkBitmap bitmap; |
edisonn@google.com | f1a358f | 2013-04-24 13:21:49 +0000 | [diff] [blame] | 290 | setup_bitmap(&bitmap, 10, 10); |
edisonn@google.com | d9dfa18 | 2013-04-24 13:01:01 +0000 | [diff] [blame] | 291 | TestImage(reporter, bitmap, |
| 292 | "/Subtype /Image\n" |
| 293 | "/Width 10\n" |
| 294 | "/Height 10\n" |
| 295 | "/ColorSpace /DeviceRGB\n" |
| 296 | "/BitsPerComponent 8\n" |
| 297 | "/Filter /FlateDecode\n" |
| 298 | "/Length 13\n" |
| 299 | ">> stream", |
| 300 | false); |
halcanary | 91d1d62 | 2015-02-17 14:43:06 -0800 | [diff] [blame^] | 301 | #endif // SK_NO_FLATE |
edisonn@google.com | d9dfa18 | 2013-04-24 13:01:01 +0000 | [diff] [blame] | 302 | } |
| 303 | |
| 304 | static void TestDCTDecode(skiatest::Reporter* reporter) { |
| 305 | SkBitmap bitmap; |
edisonn@google.com | f1a358f | 2013-04-24 13:21:49 +0000 | [diff] [blame] | 306 | setup_bitmap(&bitmap, 32, 32); |
edisonn@google.com | d9dfa18 | 2013-04-24 13:01:01 +0000 | [diff] [blame] | 307 | TestImage(reporter, bitmap, |
| 308 | "/Subtype /Image\n" |
| 309 | "/Width 32\n" |
| 310 | "/Height 32\n" |
| 311 | "/ColorSpace /DeviceRGB\n" |
| 312 | "/BitsPerComponent 8\n" |
| 313 | "/Filter /DCTDecode\n" |
| 314 | "/ColorTransform 0\n" |
| 315 | "/Length 21\n" |
| 316 | ">> stream", |
| 317 | true); |
| 318 | } |
| 319 | |
| 320 | static void TestImages(skiatest::Reporter* reporter) { |
| 321 | TestUncompressed(reporter); |
| 322 | TestFlateDecode(reporter); |
| 323 | TestDCTDecode(reporter); |
| 324 | } |
| 325 | |
vandebo@chromium.org | 4e1cc6a | 2013-01-25 19:27:23 +0000 | [diff] [blame] | 326 | // This test used to assert without the fix submitted for |
| 327 | // http://code.google.com/p/skia/issues/detail?id=1083. |
| 328 | // SKP files might have invalid glyph ids. This test ensures they are ignored, |
| 329 | // and there is no assert on input data in Debug mode. |
sugoi@google.com | 54f0d1b | 2013-02-27 19:17:41 +0000 | [diff] [blame] | 330 | static void test_issue1083() { |
vandebo@chromium.org | 4e1cc6a | 2013-01-25 19:27:23 +0000 | [diff] [blame] | 331 | SkISize pageSize = SkISize::Make(100, 100); |
robertphillips@google.com | 35300c4 | 2013-03-21 17:38:49 +0000 | [diff] [blame] | 332 | SkAutoTUnref<SkPDFDevice> dev(new SkPDFDevice(pageSize, pageSize, SkMatrix::I())); |
vandebo@chromium.org | 4e1cc6a | 2013-01-25 19:27:23 +0000 | [diff] [blame] | 333 | |
| 334 | SkCanvas c(dev); |
| 335 | SkPaint paint; |
| 336 | paint.setTextEncoding(SkPaint::kGlyphID_TextEncoding); |
| 337 | |
| 338 | uint16_t glyphID = 65000; |
| 339 | c.drawText(&glyphID, 2, 0, 0, paint); |
| 340 | |
| 341 | SkPDFDocument doc; |
| 342 | doc.appendPage(dev); |
| 343 | |
| 344 | SkDynamicMemoryWStream stream; |
| 345 | doc.emitPDF(&stream); |
| 346 | } |
| 347 | |
tfarina@chromium.org | e4fafb1 | 2013-12-12 21:11:12 +0000 | [diff] [blame] | 348 | DEF_TEST(PDFPrimitives, reporter) { |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 349 | SkAutoTUnref<SkPDFInt> int42(new SkPDFInt(42)); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 350 | SimpleCheckObjectOutput(reporter, int42.get(), "42"); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 351 | |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 352 | SkAutoTUnref<SkPDFScalar> realHalf(new SkPDFScalar(SK_ScalarHalf)); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 353 | SimpleCheckObjectOutput(reporter, realHalf.get(), "0.5"); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 354 | |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 355 | SkAutoTUnref<SkPDFScalar> bigScalar(new SkPDFScalar(110999.75f)); |
vandebo@chromium.org | 6cc26da | 2011-05-18 17:08:05 +0000 | [diff] [blame] | 356 | #if !defined(SK_ALLOW_LARGE_PDF_SCALARS) |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 357 | SimpleCheckObjectOutput(reporter, bigScalar.get(), "111000"); |
vandebo@chromium.org | 094316b | 2011-03-04 03:15:13 +0000 | [diff] [blame] | 358 | #else |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 359 | SimpleCheckObjectOutput(reporter, bigScalar.get(), "110999.75"); |
vandebo@chromium.org | 094316b | 2011-03-04 03:15:13 +0000 | [diff] [blame] | 360 | |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 361 | SkAutoTUnref<SkPDFScalar> biggerScalar(new SkPDFScalar(50000000.1)); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 362 | SimpleCheckObjectOutput(reporter, biggerScalar.get(), "50000000"); |
vandebo@chromium.org | 094316b | 2011-03-04 03:15:13 +0000 | [diff] [blame] | 363 | |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 364 | SkAutoTUnref<SkPDFScalar> smallestScalar(new SkPDFScalar(1.0/65536)); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 365 | SimpleCheckObjectOutput(reporter, smallestScalar.get(), "0.00001526"); |
vandebo@chromium.org | 094316b | 2011-03-04 03:15:13 +0000 | [diff] [blame] | 366 | #endif |
| 367 | |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 368 | SkAutoTUnref<SkPDFString> stringSimple( |
| 369 | new SkPDFString("test ) string ( foo")); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 370 | SimpleCheckObjectOutput(reporter, stringSimple.get(), |
| 371 | "(test \\) string \\( foo)"); |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 372 | SkAutoTUnref<SkPDFString> stringComplex( |
| 373 | new SkPDFString("\ttest ) string ( foo")); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 374 | SimpleCheckObjectOutput(reporter, stringComplex.get(), |
| 375 | "<0974657374202920737472696E67202820666F6F>"); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 376 | |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 377 | SkAutoTUnref<SkPDFName> name(new SkPDFName("Test name\twith#tab")); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 378 | const char expectedResult[] = "/Test#20name#09with#23tab"; |
| 379 | CheckObjectOutput(reporter, name.get(), expectedResult, |
| 380 | strlen(expectedResult), false, false); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 381 | |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 382 | SkAutoTUnref<SkPDFName> escapedName(new SkPDFName("A#/%()<>[]{}B")); |
vandebo@chromium.org | 251a766 | 2012-09-21 17:50:50 +0000 | [diff] [blame] | 383 | const char escapedNameExpected[] = "/A#23#2F#25#28#29#3C#3E#5B#5D#7B#7DB"; |
| 384 | CheckObjectOutput(reporter, escapedName.get(), escapedNameExpected, |
| 385 | strlen(escapedNameExpected), false, false); |
| 386 | |
vandebo@chromium.org | c0376fe | 2012-03-05 18:44:33 +0000 | [diff] [blame] | 387 | // Test that we correctly handle characters with the high-bit set. |
bungeman@google.com | f8aa18c | 2012-03-19 21:04:52 +0000 | [diff] [blame] | 388 | const unsigned char highBitCString[] = {0xDE, 0xAD, 'b', 'e', 0xEF, 0}; |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 389 | SkAutoTUnref<SkPDFName> highBitName( |
| 390 | new SkPDFName((const char*)highBitCString)); |
vandebo@chromium.org | c0376fe | 2012-03-05 18:44:33 +0000 | [diff] [blame] | 391 | const char highBitExpectedResult[] = "/#DE#ADbe#EF"; |
| 392 | CheckObjectOutput(reporter, highBitName.get(), highBitExpectedResult, |
| 393 | strlen(highBitExpectedResult), false, false); |
| 394 | |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 395 | SkAutoTUnref<SkPDFArray> array(new SkPDFArray); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 396 | SimpleCheckObjectOutput(reporter, array.get(), "[]"); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 397 | array->append(int42.get()); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 398 | SimpleCheckObjectOutput(reporter, array.get(), "[42]"); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 399 | array->append(realHalf.get()); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 400 | SimpleCheckObjectOutput(reporter, array.get(), "[42 0.5]"); |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 401 | SkAutoTUnref<SkPDFInt> int0(new SkPDFInt(0)); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 402 | array->append(int0.get()); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 403 | SimpleCheckObjectOutput(reporter, array.get(), "[42 0.5 0]"); |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 404 | SkAutoTUnref<SkPDFInt> int1(new SkPDFInt(1)); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 405 | array->setAt(0, int1.get()); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 406 | SimpleCheckObjectOutput(reporter, array.get(), "[1 0.5 0]"); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 407 | |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 408 | SkAutoTUnref<SkPDFDict> dict(new SkPDFDict); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 409 | SimpleCheckObjectOutput(reporter, dict.get(), "<<>>"); |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 410 | SkAutoTUnref<SkPDFName> n1(new SkPDFName("n1")); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 411 | dict->insert(n1.get(), int42.get()); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 412 | SimpleCheckObjectOutput(reporter, dict.get(), "<</n1 42\n>>"); |
vandebo@chromium.org | d96d17b | 2013-01-04 19:31:24 +0000 | [diff] [blame] | 413 | SkAutoTUnref<SkPDFName> n2(new SkPDFName("n2")); |
| 414 | SkAutoTUnref<SkPDFName> n3(new SkPDFName("n3")); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 415 | dict->insert(n2.get(), realHalf.get()); |
| 416 | dict->insert(n3.get(), array.get()); |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 417 | SimpleCheckObjectOutput(reporter, dict.get(), |
| 418 | "<</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] | 419 | |
vandebo@chromium.org | 421d644 | 2011-07-20 17:39:01 +0000 | [diff] [blame] | 420 | TestPDFStream(reporter); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 421 | |
| 422 | TestCatalog(reporter); |
| 423 | |
| 424 | TestObjectRef(reporter); |
vandebo@chromium.org | 2ef12d4 | 2011-07-06 23:31:24 +0000 | [diff] [blame] | 425 | |
| 426 | TestSubstitute(reporter); |
vandebo@chromium.org | 4e1cc6a | 2013-01-25 19:27:23 +0000 | [diff] [blame] | 427 | |
sugoi@google.com | 54f0d1b | 2013-02-27 19:17:41 +0000 | [diff] [blame] | 428 | test_issue1083(); |
edisonn@google.com | d9dfa18 | 2013-04-24 13:01:01 +0000 | [diff] [blame] | 429 | |
skia.committer@gmail.com | 83f0d30 | 2013-04-25 07:01:04 +0000 | [diff] [blame] | 430 | TestImages(reporter); |
vandebo@chromium.org | 8459d4e | 2010-09-24 22:25:30 +0000 | [diff] [blame] | 431 | } |
senorblanco | b0e89dc | 2014-10-20 14:03:12 -0700 | [diff] [blame] | 432 | |
| 433 | namespace { |
| 434 | |
| 435 | class DummyImageFilter : public SkImageFilter { |
| 436 | public: |
| 437 | DummyImageFilter(bool visited = false) : SkImageFilter(0, NULL), fVisited(visited) {} |
mtklein | 72c9faa | 2015-01-09 10:06:39 -0800 | [diff] [blame] | 438 | ~DummyImageFilter() SK_OVERRIDE {} |
senorblanco | b0e89dc | 2014-10-20 14:03:12 -0700 | [diff] [blame] | 439 | virtual bool onFilterImage(Proxy*, const SkBitmap& src, const Context&, |
rmistry | 465206a | 2015-02-02 12:08:18 -0800 | [diff] [blame] | 440 | SkBitmap* result, SkIPoint* offset) const SK_OVERRIDE { |
senorblanco | b0e89dc | 2014-10-20 14:03:12 -0700 | [diff] [blame] | 441 | fVisited = true; |
| 442 | offset->fX = offset->fY = 0; |
| 443 | *result = src; |
| 444 | return true; |
| 445 | } |
robertphillips | f3f5bad | 2014-12-19 13:49:15 -0800 | [diff] [blame] | 446 | SK_TO_STRING_OVERRIDE() |
senorblanco | b0e89dc | 2014-10-20 14:03:12 -0700 | [diff] [blame] | 447 | SK_DECLARE_PUBLIC_FLATTENABLE_DESERIALIZATION_PROCS(DummyImageFilter) |
senorblanco | b0e89dc | 2014-10-20 14:03:12 -0700 | [diff] [blame] | 448 | bool visited() const { return fVisited; } |
| 449 | |
| 450 | private: |
| 451 | mutable bool fVisited; |
| 452 | }; |
| 453 | |
| 454 | SkFlattenable* DummyImageFilter::CreateProc(SkReadBuffer& buffer) { |
| 455 | SK_IMAGEFILTER_UNFLATTEN_COMMON(common, 0); |
| 456 | bool visited = buffer.readBool(); |
| 457 | return SkNEW_ARGS(DummyImageFilter, (visited)); |
| 458 | } |
| 459 | |
robertphillips | f3f5bad | 2014-12-19 13:49:15 -0800 | [diff] [blame] | 460 | #ifndef SK_IGNORE_TO_STRING |
| 461 | void DummyImageFilter::toString(SkString* str) const { |
| 462 | str->appendf("DummyImageFilter: ("); |
| 463 | str->append(")"); |
| 464 | } |
| 465 | #endif |
| 466 | |
senorblanco | b0e89dc | 2014-10-20 14:03:12 -0700 | [diff] [blame] | 467 | }; |
| 468 | |
| 469 | // Check that PDF rendering of image filters successfully falls back to |
| 470 | // CPU rasterization. |
| 471 | DEF_TEST(PDFImageFilter, reporter) { |
| 472 | SkISize pageSize = SkISize::Make(100, 100); |
| 473 | SkAutoTUnref<SkPDFDevice> device(new SkPDFDevice(pageSize, pageSize, SkMatrix::I())); |
| 474 | SkCanvas canvas(device.get()); |
| 475 | SkAutoTUnref<DummyImageFilter> filter(new DummyImageFilter()); |
| 476 | |
| 477 | // Filter just created; should be unvisited. |
| 478 | REPORTER_ASSERT(reporter, !filter->visited()); |
| 479 | SkPaint paint; |
| 480 | paint.setImageFilter(filter.get()); |
| 481 | canvas.drawRect(SkRect::MakeWH(100, 100), paint); |
| 482 | |
| 483 | // Filter was used in rendering; should be visited. |
| 484 | REPORTER_ASSERT(reporter, filter->visited()); |
| 485 | } |