PDFPrimitivesTest.cpp revision d1c53aae59ee44377be8bc0cc15e54d46aa530ce 
1
2/*
3 * Copyright 2010 The Android Open Source Project
4 *
5 * Use of this source code is governed by a BSD-style license that can be
6 * found in the LICENSE file.
7 */
8
9
10#include "Test.h"
11#include "SkCanvas.h"
12#include "SkData.h"
13#include "SkFlate.h"
14#include "SkPDFCatalog.h"
15#include "SkPDFDevice.h"
16#include "SkPDFStream.h"
17#include "SkPDFTypes.h"
18#include "SkScalar.h"
19#include "SkStream.h"
20#include "SkTypes.h"
21
22class SkPDFTestDict : public SkPDFDict {
23public:
24    void getResources(SkTDArray<SkPDFObject*>* resourceList) {
25        resourceList->setReserve(resourceList->count() + fResources.count());
26        for (int i = 0; i < fResources.count(); i++) {
27            resourceList->push(fResources[i]);
28            fResources[i]->ref();
29        }
30    }
31
32    void addResource(SkPDFObject* object) {
33        fResources.append(1, &object);
34    }
35
36private:
37    SkTDArray<SkPDFObject*> fResources;
38};
39
40static bool stream_equals(const SkDynamicMemoryWStream& stream, size_t offset,
41                          const void* buffer, size_t len) {
42    SkAutoDataUnref data(stream.copyToData());
43    if (offset + len > data->size()) {
44        return false;
45    }
46    return memcmp(data->bytes() + offset, buffer, len) == 0;
47}
48
49static void CheckObjectOutput(skiatest::Reporter* reporter, SkPDFObject* obj,
50                              const char* expectedData, size_t expectedSize,
51                              bool indirect, bool compression) {
52    SkPDFDocument::Flags docFlags = (SkPDFDocument::Flags) 0;
53    if (!compression) {
54        docFlags = SkTBitOr(docFlags, SkPDFDocument::kNoCompression_Flags);
55    }
56    SkPDFCatalog catalog(docFlags);
57    size_t directSize = obj->getOutputSize(&catalog, false);
58    REPORTER_ASSERT(reporter, directSize == expectedSize);
59
60    SkDynamicMemoryWStream buffer;
61    obj->emit(&buffer, &catalog, false);
62    REPORTER_ASSERT(reporter, directSize == buffer.getOffset());
63    REPORTER_ASSERT(reporter, stream_equals(buffer, 0, expectedData,
64                                            directSize));
65
66    if (indirect) {
67        // Indirect output.
68        static char header[] = "1 0 obj\n";
69        static size_t headerLen = strlen(header);
70        static char footer[] = "\nendobj\n";
71        static size_t footerLen = strlen(footer);
72
73        catalog.addObject(obj, false);
74
75        size_t indirectSize = obj->getOutputSize(&catalog, true);
76        REPORTER_ASSERT(reporter,
77                        indirectSize == directSize + headerLen + footerLen);
78
79        buffer.reset();
80        obj->emit(&buffer, &catalog, true);
81        REPORTER_ASSERT(reporter, indirectSize == buffer.getOffset());
82        REPORTER_ASSERT(reporter, stream_equals(buffer, 0, header, headerLen));
83        REPORTER_ASSERT(reporter, stream_equals(buffer, headerLen, expectedData,
84                                                directSize));
85        REPORTER_ASSERT(reporter, stream_equals(buffer, headerLen + directSize,
86                                                footer, footerLen));
87    }
88}
89
90static void SimpleCheckObjectOutput(skiatest::Reporter* reporter,
91                                    SkPDFObject* obj,
92                                    const char* expectedResult) {
93    CheckObjectOutput(reporter, obj, expectedResult,
94                      strlen(expectedResult), true, false);
95}
96
97static void TestPDFStream(skiatest::Reporter* reporter) {
98    char streamBytes[] = "Test\nFoo\tBar";
99    SkAutoTUnref<SkMemoryStream> streamData(new SkMemoryStream(
100        streamBytes, strlen(streamBytes), true));
101    SkAutoTUnref<SkPDFStream> stream(new SkPDFStream(streamData.get()));
102    SimpleCheckObjectOutput(
103        reporter, stream.get(),
104        "<</Length 12\n>> stream\nTest\nFoo\tBar\nendstream");
105    stream->insert("Attribute", new SkPDFInt(42))->unref();
106    SimpleCheckObjectOutput(reporter, stream.get(),
107                            "<</Length 12\n/Attribute 42\n>> stream\n"
108                                "Test\nFoo\tBar\nendstream");
109
110    if (SkFlate::HaveFlate()) {
111        char streamBytes2[] = "This is a longer string, so that compression "
112                              "can do something with it. With shorter strings, "
113                              "the short circuit logic cuts in and we end up "
114                              "with an uncompressed string.";
115        SkAutoDataUnref streamData2(SkData::NewWithCopy(streamBytes2,
116                                                        strlen(streamBytes2)));
117        SkAutoTUnref<SkPDFStream> stream(new SkPDFStream(streamData2.get()));
118
119        SkDynamicMemoryWStream compressedByteStream;
120        SkFlate::Deflate(streamData2.get(), &compressedByteStream);
121        SkAutoDataUnref compressedData(compressedByteStream.copyToData());
122
123        // Check first without compression.
124        SkDynamicMemoryWStream expectedResult1;
125        expectedResult1.writeText("<</Length 167\n>> stream\n");
126        expectedResult1.writeText(streamBytes2);
127        expectedResult1.writeText("\nendstream");
128        SkAutoDataUnref expectedResultData1(expectedResult1.copyToData());
129        CheckObjectOutput(reporter, stream.get(),
130                          (const char*) expectedResultData1->data(),
131                          expectedResultData1->size(), true, false);
132
133        // Then again with compression.
134        SkDynamicMemoryWStream expectedResult2;
135        expectedResult2.writeText("<</Filter /FlateDecode\n/Length 116\n"
136                                 ">> stream\n");
137        expectedResult2.write(compressedData->data(), compressedData->size());
138        expectedResult2.writeText("\nendstream");
139        SkAutoDataUnref expectedResultData2(expectedResult2.copyToData());
140        CheckObjectOutput(reporter, stream.get(),
141                          (const char*) expectedResultData2->data(),
142                          expectedResultData2->size(), true, true);
143    }
144}
145
146static void TestCatalog(skiatest::Reporter* reporter) {
147    SkPDFCatalog catalog((SkPDFDocument::Flags)0);
148    SkAutoTUnref<SkPDFInt> int1(new SkPDFInt(1));
149    SkAutoTUnref<SkPDFInt> int2(new SkPDFInt(2));
150    SkAutoTUnref<SkPDFInt> int3(new SkPDFInt(3));
151    int1.get()->ref();
152    SkAutoTUnref<SkPDFInt> int1Again(int1.get());
153
154    catalog.addObject(int1.get(), false);
155    catalog.addObject(int2.get(), false);
156    catalog.addObject(int3.get(), false);
157
158    REPORTER_ASSERT(reporter, catalog.getObjectNumberSize(int1.get()) == 3);
159    REPORTER_ASSERT(reporter, catalog.getObjectNumberSize(int2.get()) == 3);
160    REPORTER_ASSERT(reporter, catalog.getObjectNumberSize(int3.get()) == 3);
161
162    SkDynamicMemoryWStream buffer;
163    catalog.emitObjectNumber(&buffer, int1.get());
164    catalog.emitObjectNumber(&buffer, int2.get());
165    catalog.emitObjectNumber(&buffer, int3.get());
166    catalog.emitObjectNumber(&buffer, int1Again.get());
167    char expectedResult[] = "1 02 03 01 0";
168    REPORTER_ASSERT(reporter, stream_equals(buffer, 0, expectedResult,
169                                            strlen(expectedResult)));
170}
171
172static void TestObjectRef(skiatest::Reporter* reporter) {
173    SkAutoTUnref<SkPDFInt> int1(new SkPDFInt(1));
174    SkAutoTUnref<SkPDFInt> int2(new SkPDFInt(2));
175    SkAutoTUnref<SkPDFObjRef> int2ref(new SkPDFObjRef(int2.get()));
176
177    SkPDFCatalog catalog((SkPDFDocument::Flags)0);
178    catalog.addObject(int1.get(), false);
179    catalog.addObject(int2.get(), false);
180    REPORTER_ASSERT(reporter, catalog.getObjectNumberSize(int1.get()) == 3);
181    REPORTER_ASSERT(reporter, catalog.getObjectNumberSize(int2.get()) == 3);
182
183    char expectedResult[] = "2 0 R";
184    SkDynamicMemoryWStream buffer;
185    int2ref->emitObject(&buffer, &catalog, false);
186    REPORTER_ASSERT(reporter, buffer.getOffset() == strlen(expectedResult));
187    REPORTER_ASSERT(reporter, stream_equals(buffer, 0, expectedResult,
188                                            buffer.getOffset()));
189}
190
191static void TestSubstitute(skiatest::Reporter* reporter) {
192    SkAutoTUnref<SkPDFTestDict> proxy(new SkPDFTestDict());
193    SkAutoTUnref<SkPDFTestDict> stub(new SkPDFTestDict());
194    SkAutoTUnref<SkPDFInt> int33(new SkPDFInt(33));
195    SkAutoTUnref<SkPDFDict> stubResource(new SkPDFDict());
196    SkAutoTUnref<SkPDFInt> int44(new SkPDFInt(44));
197
198    stub->insert("Value", int33.get());
199    stubResource->insert("InnerValue", int44.get());
200    stub->addResource(stubResource.get());
201
202    SkPDFCatalog catalog((SkPDFDocument::Flags)0);
203    catalog.addObject(proxy.get(), false);
204    catalog.setSubstitute(proxy.get(), stub.get());
205
206    SkDynamicMemoryWStream buffer;
207    proxy->emit(&buffer, &catalog, false);
208    catalog.emitSubstituteResources(&buffer, false);
209
210    char objectResult[] = "2 0 obj\n<</Value 33\n>>\nendobj\n";
211    REPORTER_ASSERT(
212        reporter,
213        catalog.setFileOffset(proxy.get(), 0) == strlen(objectResult));
214
215    char expectedResult[] =
216        "<</Value 33\n>>1 0 obj\n<</InnerValue 44\n>>\nendobj\n";
217    REPORTER_ASSERT(reporter, buffer.getOffset() == strlen(expectedResult));
218    REPORTER_ASSERT(reporter, stream_equals(buffer, 0, expectedResult,
219                                            buffer.getOffset()));
220}
221
222// This test used to assert without the fix submitted for
223// http://code.google.com/p/skia/issues/detail?id=1083.
224// SKP files might have invalid glyph ids. This test ensures they are ignored,
225// and there is no assert on input data in Debug mode.
226static void test_issue1083(skiatest::Reporter* reporter) {
227    SkISize pageSize = SkISize::Make(100, 100);
228    SkPDFDevice* dev = new SkPDFDevice(pageSize, pageSize, SkMatrix::I());
229
230    SkCanvas c(dev);
231    SkPaint paint;
232    paint.setTextEncoding(SkPaint::kGlyphID_TextEncoding);
233
234    uint16_t glyphID = 65000;
235    c.drawText(&glyphID, 2, 0, 0, paint);
236
237    SkPDFDocument doc;
238    doc.appendPage(dev);
239
240    SkDynamicMemoryWStream stream;
241    doc.emitPDF(&stream);
242}
243
244static void TestPDFPrimitives(skiatest::Reporter* reporter) {
245    SkAutoTUnref<SkPDFInt> int42(new SkPDFInt(42));
246    SimpleCheckObjectOutput(reporter, int42.get(), "42");
247
248    SkAutoTUnref<SkPDFScalar> realHalf(new SkPDFScalar(SK_ScalarHalf));
249    SimpleCheckObjectOutput(reporter, realHalf.get(), "0.5");
250
251#if defined(SK_SCALAR_IS_FLOAT)
252    SkAutoTUnref<SkPDFScalar> bigScalar(new SkPDFScalar(110999.75f));
253#if !defined(SK_ALLOW_LARGE_PDF_SCALARS)
254    SimpleCheckObjectOutput(reporter, bigScalar.get(), "111000");
255#else
256    SimpleCheckObjectOutput(reporter, bigScalar.get(), "110999.75");
257
258    SkAutoTUnref<SkPDFScalar> biggerScalar(new SkPDFScalar(50000000.1));
259    SimpleCheckObjectOutput(reporter, biggerScalar.get(), "50000000");
260
261    SkAutoTUnref<SkPDFScalar> smallestScalar(new SkPDFScalar(1.0/65536));
262    SimpleCheckObjectOutput(reporter, smallestScalar.get(), "0.00001526");
263#endif
264#endif
265
266    SkAutoTUnref<SkPDFString> stringSimple(
267        new SkPDFString("test ) string ( foo"));
268    SimpleCheckObjectOutput(reporter, stringSimple.get(),
269                            "(test \\) string \\( foo)");
270    SkAutoTUnref<SkPDFString> stringComplex(
271        new SkPDFString("\ttest ) string ( foo"));
272    SimpleCheckObjectOutput(reporter, stringComplex.get(),
273                            "<0974657374202920737472696E67202820666F6F>");
274
275    SkAutoTUnref<SkPDFName> name(new SkPDFName("Test name\twith#tab"));
276    const char expectedResult[] = "/Test#20name#09with#23tab";
277    CheckObjectOutput(reporter, name.get(), expectedResult,
278                      strlen(expectedResult), false, false);
279
280    SkAutoTUnref<SkPDFName> escapedName(new SkPDFName("A#/%()<>[]{}B"));
281    const char escapedNameExpected[] = "/A#23#2F#25#28#29#3C#3E#5B#5D#7B#7DB";
282    CheckObjectOutput(reporter, escapedName.get(), escapedNameExpected,
283                      strlen(escapedNameExpected), false, false);
284
285    // Test that we correctly handle characters with the high-bit set.
286    const unsigned char highBitCString[] = {0xDE, 0xAD, 'b', 'e', 0xEF, 0};
287    SkAutoTUnref<SkPDFName> highBitName(
288        new SkPDFName((const char*)highBitCString));
289    const char highBitExpectedResult[] = "/#DE#ADbe#EF";
290    CheckObjectOutput(reporter, highBitName.get(), highBitExpectedResult,
291                      strlen(highBitExpectedResult), false, false);
292
293    SkAutoTUnref<SkPDFArray> array(new SkPDFArray);
294    SimpleCheckObjectOutput(reporter, array.get(), "[]");
295    array->append(int42.get());
296    SimpleCheckObjectOutput(reporter, array.get(), "[42]");
297    array->append(realHalf.get());
298    SimpleCheckObjectOutput(reporter, array.get(), "[42 0.5]");
299    SkAutoTUnref<SkPDFInt> int0(new SkPDFInt(0));
300    array->append(int0.get());
301    SimpleCheckObjectOutput(reporter, array.get(), "[42 0.5 0]");
302    SkAutoTUnref<SkPDFInt> int1(new SkPDFInt(1));
303    array->setAt(0, int1.get());
304    SimpleCheckObjectOutput(reporter, array.get(), "[1 0.5 0]");
305
306    SkAutoTUnref<SkPDFDict> dict(new SkPDFDict);
307    SimpleCheckObjectOutput(reporter, dict.get(), "<<>>");
308    SkAutoTUnref<SkPDFName> n1(new SkPDFName("n1"));
309    dict->insert(n1.get(), int42.get());
310    SimpleCheckObjectOutput(reporter, dict.get(), "<</n1 42\n>>");
311    SkAutoTUnref<SkPDFName> n2(new SkPDFName("n2"));
312    SkAutoTUnref<SkPDFName> n3(new SkPDFName("n3"));
313    dict->insert(n2.get(), realHalf.get());
314    dict->insert(n3.get(), array.get());
315    SimpleCheckObjectOutput(reporter, dict.get(),
316                            "<</n1 42\n/n2 0.5\n/n3 [1 0.5 0]\n>>");
317
318    TestPDFStream(reporter);
319
320    TestCatalog(reporter);
321
322    TestObjectRef(reporter);
323
324    TestSubstitute(reporter);
325
326    test_issue1083(reporter);
327}
328
329#include "TestClassDef.h"
330DEFINE_TESTCLASS("PDFPrimitives", PDFPrimitivesTestClass, TestPDFPrimitives)
331