SkDocument_PDF.cpp revision f4219ddf31b2171a5576ab98d1fd2deab4d816fd
199ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com/* 2a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary * Copyright 2011 Google Inc. 399ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com * 499ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com * Use of this source code is governed by a BSD-style license that can be 599ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com * found in the LICENSE file. 699ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com */ 799ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com 899ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com#include "SkDocument.h" 9a1f1ee98a1f6d0770f6243270ca2f0e6c92efabahalcanary#include "SkPDFCanon.h" 105867736b08d3689356b49f505bcf748c2194a0bcreed#include "SkPDFDevice.h" 11a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary#include "SkPDFFont.h" 122f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary#include "SkPDFStream.h" 13a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary#include "SkPDFTypes.h" 14a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary#include "SkStream.h" 15a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary 16a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanarystatic void emit_pdf_header(SkWStream* stream) { 17a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary stream->writeText("%PDF-1.4\n%"); 18a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary // The PDF spec recommends including a comment with four bytes, all 19a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary // with their high bits set. This is "Skia" with the high bits set. 20a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary stream->write32(0xD3EBE9E1); 21a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary stream->writeText("\n"); 22a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary} 23a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary 24a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanarystatic void emit_pdf_footer(SkWStream* stream, 2537c46cad21632cfc1411b08d73af37a1fffe2944halcanary const SkPDFObjNumMap& objNumMap, 2637c46cad21632cfc1411b08d73af37a1fffe2944halcanary const SkPDFSubstituteMap& substitutes, 27a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary SkPDFObject* docCatalog, 28a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary int64_t objCount, 29a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary int32_t xRefFileOffset) { 30a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary SkPDFDict trailerDict; 31a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary // TODO(vandebo): Linearized format will take a Prev entry too. 32a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary // TODO(vandebo): PDF/A requires an ID entry. 33a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary trailerDict.insertInt("Size", int(objCount)); 34a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary trailerDict.insert("Root", new SkPDFObjRef(docCatalog))->unref(); 35a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary 36a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary stream->writeText("trailer\n"); 3737c46cad21632cfc1411b08d73af37a1fffe2944halcanary trailerDict.emitObject(stream, objNumMap, substitutes); 38a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary stream->writeText("\nstartxref\n"); 39a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary stream->writeBigDecAsText(xRefFileOffset); 40a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary stream->writeText("\n%%EOF"); 41a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary} 42a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary 432f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanarystatic void perform_font_subsetting( 442f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary const SkTDArray<const SkPDFDevice*>& pageDevices, 4537c46cad21632cfc1411b08d73af37a1fffe2944halcanary SkPDFSubstituteMap* substituteMap) { 4637c46cad21632cfc1411b08d73af37a1fffe2944halcanary SkASSERT(substituteMap); 47a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary 48a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary SkPDFGlyphSetMap usage; 492f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary for (int i = 0; i < pageDevices.count(); ++i) { 502f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary usage.merge(pageDevices[i]->getFontGlyphUsage()); 51a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary } 52a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary SkPDFGlyphSetMap::F2BIter iterator(usage); 53a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary const SkPDFGlyphSetMap::FontGlyphSetPair* entry = iterator.next(); 54a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary while (entry) { 55a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary SkAutoTUnref<SkPDFFont> subsetFont( 56a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary entry->fFont->getFontSubset(entry->fGlyphSet)); 57a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary if (subsetFont) { 5837c46cad21632cfc1411b08d73af37a1fffe2944halcanary substituteMap->setSubstitute(entry->fFont, subsetFont.get()); 59a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary } 60a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary entry = iterator.next(); 61a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary } 62a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary} 63a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary 642f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanarystatic SkPDFDict* create_pdf_page(const SkPDFDevice* pageDevice) { 652f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary SkAutoTUnref<SkPDFDict> page(SkNEW_ARGS(SkPDFDict, ("Page"))); 662b86155b42c2493ff0c558ce105a464769962274halcanary SkAutoTUnref<SkPDFDict> deviceResourceDict( 672f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary pageDevice->createResourceDict()); 682f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary page->insert("Resources", deviceResourceDict.get()); 692f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary 702f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary SkAutoTUnref<SkPDFArray> mediaBox(pageDevice->copyMediaBox()); 712f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary page->insert("MediaBox", mediaBox.get()); 722f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary 732f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary SkPDFArray* annots = pageDevice->getAnnotations(); 742f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary if (annots && annots->size() > 0) { 752f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary page->insert("Annots", annots); 762f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary } 772f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary 782f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary SkAutoTDelete<SkStreamAsset> content(pageDevice->content()); 792f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary SkAutoTUnref<SkPDFStream> contentStream( 802f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary SkNEW_ARGS(SkPDFStream, (content.get()))); 812f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary page->insert("Contents", new SkPDFObjRef(contentStream.get()))->unref(); 822f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary return page.detach(); 832f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary} 842f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary 852f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanarystatic void generate_page_tree(const SkTDArray<SkPDFDict*>& pages, 862f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary SkTDArray<SkPDFDict*>* pageTree, 872f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary SkPDFDict** rootNode) { 882f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary // PDF wants a tree describing all the pages in the document. We arbitrary 892f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary // choose 8 (kNodeSize) as the number of allowed children. The internal 902f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary // nodes have type "Pages" with an array of children, a parent pointer, and 912f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary // the number of leaves below the node as "Count." The leaves are passed 922f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary // into the method, have type "Page" and need a parent pointer. This method 932f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary // builds the tree bottom up, skipping internal nodes that would have only 942f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary // one child. 952f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary static const int kNodeSize = 8; 962f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary 972f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary // curNodes takes a reference to its items, which it passes to pageTree. 982f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary SkTDArray<SkPDFDict*> curNodes; 992f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary curNodes.setReserve(pages.count()); 1002f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary for (int i = 0; i < pages.count(); i++) { 1012f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary SkSafeRef(pages[i]); 1022f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary curNodes.push(pages[i]); 1032f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary } 1042f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary 1052f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary // nextRoundNodes passes its references to nodes on to curNodes. 1062f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary SkTDArray<SkPDFDict*> nextRoundNodes; 1072f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary nextRoundNodes.setReserve((pages.count() + kNodeSize - 1)/kNodeSize); 1082f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary 1092f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary int treeCapacity = kNodeSize; 1102f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary do { 1112f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary for (int i = 0; i < curNodes.count(); ) { 1122f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary if (i > 0 && i + 1 == curNodes.count()) { 1132f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary nextRoundNodes.push(curNodes[i]); 1142f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary break; 1152f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary } 1162f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary 1172f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary SkPDFDict* newNode = new SkPDFDict("Pages"); 1182f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary SkAutoTUnref<SkPDFObjRef> newNodeRef(new SkPDFObjRef(newNode)); 1192f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary 1202f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary SkAutoTUnref<SkPDFArray> kids(new SkPDFArray); 1212f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary kids->reserve(kNodeSize); 1222f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary 1232f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary int count = 0; 1242f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary for (; i < curNodes.count() && count < kNodeSize; i++, count++) { 125130444fdaf737c5931a934018ab50b0c91a3499ehalcanary curNodes[i]->insert("Parent", newNodeRef.get()); 1262f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary kids->append(new SkPDFObjRef(curNodes[i]))->unref(); 1272f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary 1282f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary // TODO(vandebo): put the objects in strict access order. 1292f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary // Probably doesn't matter because they are so small. 1302f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary if (curNodes[i] != pages[0]) { 1312f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary pageTree->push(curNodes[i]); // Transfer reference. 1322f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary } else { 1332f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary SkSafeUnref(curNodes[i]); 1342f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary } 1352f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary } 1362f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary 1372f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary // treeCapacity is the number of leaf nodes possible for the 1382f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary // current set of subtrees being generated. (i.e. 8, 64, 512, ...). 1392f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary // It is hard to count the number of leaf nodes in the current 1402f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary // subtree. However, by construction, we know that unless it's the 1412f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary // last subtree for the current depth, the leaf count will be 1422f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary // treeCapacity, otherwise it's what ever is left over after 1432f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary // consuming treeCapacity chunks. 1442f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary int pageCount = treeCapacity; 1452f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary if (i == curNodes.count()) { 1462f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary pageCount = ((pages.count() - 1) % treeCapacity) + 1; 1472f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary } 148130444fdaf737c5931a934018ab50b0c91a3499ehalcanary newNode->insert("Count", new SkPDFInt(pageCount))->unref(); 149130444fdaf737c5931a934018ab50b0c91a3499ehalcanary newNode->insert("Kids", kids.get()); 1502f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary nextRoundNodes.push(newNode); // Transfer reference. 1512f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary } 1522f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary 1532f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary curNodes = nextRoundNodes; 1542f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary nextRoundNodes.rewind(); 1552f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary treeCapacity *= kNodeSize; 1562f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary } while (curNodes.count() > 1); 1572f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary 1582f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary pageTree->push(curNodes[0]); // Transfer reference. 1592f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary if (rootNode) { 1602f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary *rootNode = curNodes[0]; 1612f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary } 1622f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary} 1632f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary 164a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanarystatic bool emit_pdf_document(const SkTDArray<const SkPDFDevice*>& pageDevices, 165a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary SkWStream* stream) { 166a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary if (pageDevices.isEmpty()) { 167a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary return false; 168a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary } 169a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary 1702f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary SkTDArray<SkPDFDict*> pages; 171a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary SkAutoTUnref<SkPDFDict> dests(SkNEW(SkPDFDict)); 172a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary 173a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary for (int i = 0; i < pageDevices.count(); i++) { 174a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary SkASSERT(pageDevices[i]); 175a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary SkASSERT(i == 0 || 176a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary pageDevices[i - 1]->getCanon() == pageDevices[i]->getCanon()); 1772f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary SkAutoTUnref<SkPDFDict> page(create_pdf_page(pageDevices[i])); 1782f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary pageDevices[i]->appendDestinations(dests, page.get()); 179a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary pages.push(page.detach()); 180a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary } 181a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary 182a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary SkTDArray<SkPDFDict*> pageTree; 183a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary SkAutoTUnref<SkPDFDict> docCatalog(SkNEW_ARGS(SkPDFDict, ("Catalog"))); 184a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary 185a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary SkPDFDict* pageTreeRoot; 1862f7ebcb424cd1d1acf07478157f86b0a3eafd712halcanary generate_page_tree(pages, &pageTree, &pageTreeRoot); 187a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary 188a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary docCatalog->insert("Pages", new SkPDFObjRef(pageTreeRoot))->unref(); 189a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary 190a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary /* TODO(vandebo): output intent 191a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary SkAutoTUnref<SkPDFDict> outputIntent = new SkPDFDict("OutputIntent"); 192a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary outputIntent->insert("S", new SkPDFName("GTS_PDFA1"))->unref(); 193a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary outputIntent->insert("OutputConditionIdentifier", 194a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary new SkPDFString("sRGB"))->unref(); 195a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary SkAutoTUnref<SkPDFArray> intentArray = new SkPDFArray; 196a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary intentArray->append(outputIntent.get()); 197a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary docCatalog->insert("OutputIntent", intentArray.get()); 198a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary */ 199a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary 200a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary if (dests->size() > 0) { 201a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary docCatalog->insert("Dests", SkNEW_ARGS(SkPDFObjRef, (dests.get()))) 202a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary ->unref(); 203a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary } 204a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary 205a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary // Build font subsetting info before proceeding. 20637c46cad21632cfc1411b08d73af37a1fffe2944halcanary SkPDFSubstituteMap substitutes; 20737c46cad21632cfc1411b08d73af37a1fffe2944halcanary perform_font_subsetting(pageDevices, &substitutes); 208a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary 20937c46cad21632cfc1411b08d73af37a1fffe2944halcanary SkPDFObjNumMap objNumMap; 21037c46cad21632cfc1411b08d73af37a1fffe2944halcanary if (objNumMap.addObject(docCatalog.get())) { 21137c46cad21632cfc1411b08d73af37a1fffe2944halcanary docCatalog->addResources(&objNumMap, substitutes); 212a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary } 213a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary size_t baseOffset = SkToOffT(stream->bytesWritten()); 214a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary emit_pdf_header(stream); 215a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary SkTDArray<int32_t> offsets; 21637c46cad21632cfc1411b08d73af37a1fffe2944halcanary for (int i = 0; i < objNumMap.objects().count(); ++i) { 21737c46cad21632cfc1411b08d73af37a1fffe2944halcanary SkPDFObject* object = objNumMap.objects()[i]; 218a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary offsets.push(SkToS32(stream->bytesWritten() - baseOffset)); 21937c46cad21632cfc1411b08d73af37a1fffe2944halcanary SkASSERT(object == substitutes.getSubstitute(object)); 22037c46cad21632cfc1411b08d73af37a1fffe2944halcanary SkASSERT(objNumMap.getObjectNumber(object) == i + 1); 221a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary stream->writeDecAsText(i + 1); 222a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary stream->writeText(" 0 obj\n"); // Generation number is always 0. 22337c46cad21632cfc1411b08d73af37a1fffe2944halcanary object->emitObject(stream, objNumMap, substitutes); 224a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary stream->writeText("\nendobj\n"); 225a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary } 226a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary int32_t xRefFileOffset = SkToS32(stream->bytesWritten() - baseOffset); 227a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary 22841f88f0251cf48eb06f3f8d143aac8562a230889halcanary // Include the zeroth object in the count. 229a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary int32_t objCount = SkToS32(offsets.count() + 1); 230a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary 231a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary stream->writeText("xref\n0 "); 23241f88f0251cf48eb06f3f8d143aac8562a230889halcanary stream->writeDecAsText(objCount); 233a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary stream->writeText("\n0000000000 65535 f \n"); 234a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary for (int i = 0; i < offsets.count(); i++) { 235a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary SkASSERT(offsets[i] > 0); 236a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary stream->writeBigDecAsText(offsets[i], 10); 237a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary stream->writeText(" 00000 n \n"); 238a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary } 23937c46cad21632cfc1411b08d73af37a1fffe2944halcanary emit_pdf_footer(stream, objNumMap, substitutes, docCatalog.get(), objCount, 240a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary xRefFileOffset); 241a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary 242a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary // The page tree has both child and parent pointers, so it creates a 243a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary // reference cycle. We must clear that cycle to properly reclaim memory. 244a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary for (int i = 0; i < pageTree.count(); i++) { 245a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary pageTree[i]->clear(); 246a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary } 247a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary pageTree.safeUnrefAll(); 248a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary pages.unrefAll(); 249a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary return true; 250a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary} 251a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary 252a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary#if 0 253a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary// TODO(halcanary): expose notEmbeddableCount in SkDocument 254a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanaryvoid GetCountOfFontTypes( 255a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary const SkTDArray<SkPDFDevice*>& pageDevices, 256a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary int counts[SkAdvancedTypefaceMetrics::kOther_Font + 1], 257a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary int* notSubsettableCount, 258a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary int* notEmbeddableCount) { 259a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary sk_bzero(counts, sizeof(int) * 260a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary (SkAdvancedTypefaceMetrics::kOther_Font + 1)); 261a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary SkTDArray<SkFontID> seenFonts; 262a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary int notSubsettable = 0; 263a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary int notEmbeddable = 0; 264a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary 265a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary for (int pageNumber = 0; pageNumber < pageDevices.count(); pageNumber++) { 266a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary const SkTDArray<SkPDFFont*>& fontResources = 267a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary pageDevices[pageNumber]->getFontResources(); 268a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary for (int font = 0; font < fontResources.count(); font++) { 269a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary SkFontID fontID = fontResources[font]->typeface()->uniqueID(); 270a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary if (seenFonts.find(fontID) == -1) { 271a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary counts[fontResources[font]->getType()]++; 272a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary seenFonts.push(fontID); 273a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary if (!fontResources[font]->canSubset()) { 274a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary notSubsettable++; 275a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary } 276a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary if (!fontResources[font]->canEmbed()) { 277a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary notEmbeddable++; 278a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary } 279a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary } 280a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary } 281a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary } 282a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary if (notSubsettableCount) { 283a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary *notSubsettableCount = notSubsettable; 284a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary 285a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary } 286a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary if (notEmbeddableCount) { 287a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary *notEmbeddableCount = notEmbeddable; 288a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary } 289a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary} 290a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary#endif 291a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary//////////////////////////////////////////////////////////////////////////////// 29299ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com 2937a0118465a2dd1783d7b19b026bedd7e13afadf2halcanarynamespace { 29499ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.comclass SkDocument_PDF : public SkDocument { 29599ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.compublic: 2968c92dc1dc281649f9e6b0ff534c25bc89dded3eahalcanary SkDocument_PDF(SkWStream* stream, 297792c80f5a7b66e75d42664ccb298f31962c6654chalcanary void (*doneProc)(SkWStream*, bool), 2988c294900f31d8d032b62182f011019276d27d5d0commit-bot@chromium.org SkScalar rasterDpi) 2998c92dc1dc281649f9e6b0ff534c25bc89dded3eahalcanary : SkDocument(stream, doneProc) 3008c92dc1dc281649f9e6b0ff534c25bc89dded3eahalcanary , fRasterDpi(rasterDpi) {} 3016319367bdcbf5e5050632ab97973f5035d0dd8faskia.committer@gmail.com 30299ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com virtual ~SkDocument_PDF() { 30399ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com // subclasses must call close() in their destructors 30499ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com this->close(); 30599ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com } 30699ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com 30799ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.comprotected: 308f4219ddf31b2171a5576ab98d1fd2deab4d816fdtfarina SkCanvas* onBeginPage(SkScalar width, SkScalar height, 309f4219ddf31b2171a5576ab98d1fd2deab4d816fdtfarina const SkRect& trimBox) override { 310a1f1ee98a1f6d0770f6243270ca2f0e6c92efabahalcanary SkASSERT(!fCanvas.get()); 31199ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com 312a1f1ee98a1f6d0770f6243270ca2f0e6c92efabahalcanary SkISize pageSize = SkISize::Make( 313a1f1ee98a1f6d0770f6243270ca2f0e6c92efabahalcanary SkScalarRoundToInt(width), SkScalarRoundToInt(height)); 3147a0118465a2dd1783d7b19b026bedd7e13afadf2halcanary SkAutoTUnref<SkPDFDevice> device( 3157a0118465a2dd1783d7b19b026bedd7e13afadf2halcanary SkPDFDevice::Create(pageSize, fRasterDpi, &fCanon)); 3167a0118465a2dd1783d7b19b026bedd7e13afadf2halcanary fCanvas.reset(SkNEW_ARGS(SkCanvas, (device.get()))); 3177a0118465a2dd1783d7b19b026bedd7e13afadf2halcanary fPageDevices.push(device.detach()); 318be519ad718371cc40c432572173e47f851476486halcanary fCanvas->clipRect(trimBox); 31993f816129356d6764a9402817cc36bffd8f991fbhalcanary fCanvas->translate(trimBox.x(), trimBox.y()); 320a1f1ee98a1f6d0770f6243270ca2f0e6c92efabahalcanary return fCanvas.get(); 32199ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com } 32299ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com 32336352bf5e38f45a70ee4f4fc132a38048d38206dmtklein void onEndPage() override { 324a1f1ee98a1f6d0770f6243270ca2f0e6c92efabahalcanary SkASSERT(fCanvas.get()); 32599ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com fCanvas->flush(); 326a1f1ee98a1f6d0770f6243270ca2f0e6c92efabahalcanary fCanvas.reset(NULL); 32799ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com } 32899ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com 32936352bf5e38f45a70ee4f4fc132a38048d38206dmtklein bool onClose(SkWStream* stream) override { 330a1f1ee98a1f6d0770f6243270ca2f0e6c92efabahalcanary SkASSERT(!fCanvas.get()); 33199ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com 332a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary bool success = emit_pdf_document(fPageDevices, stream); 3337a0118465a2dd1783d7b19b026bedd7e13afadf2halcanary fPageDevices.unrefAll(); 3342e3f9d8a9309686eeb4c76ccfde5800da87a68b3halcanary fCanon.reset(); 335b5a6651f9f69570d964382134d64360915db9a29commit-bot@chromium.org return success; 336b5a6651f9f69570d964382134d64360915db9a29commit-bot@chromium.org } 337b5a6651f9f69570d964382134d64360915db9a29commit-bot@chromium.org 33836352bf5e38f45a70ee4f4fc132a38048d38206dmtklein void onAbort() override { 3397a0118465a2dd1783d7b19b026bedd7e13afadf2halcanary fPageDevices.unrefAll(); 3402e3f9d8a9309686eeb4c76ccfde5800da87a68b3halcanary fCanon.reset(); 34199ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com } 34299ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com 34399ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.comprivate: 344a1f1ee98a1f6d0770f6243270ca2f0e6c92efabahalcanary SkPDFCanon fCanon; 3456d622703e578eddc64ab4e3340d0ab0033268799halcanary SkTDArray<const SkPDFDevice*> fPageDevices; 346a1f1ee98a1f6d0770f6243270ca2f0e6c92efabahalcanary SkAutoTUnref<SkCanvas> fCanvas; 347a1f1ee98a1f6d0770f6243270ca2f0e6c92efabahalcanary SkScalar fRasterDpi; 34899ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com}; 3497a0118465a2dd1783d7b19b026bedd7e13afadf2halcanary} // namespace 35099ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com/////////////////////////////////////////////////////////////////////////////// 35199ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com 3528c92dc1dc281649f9e6b0ff534c25bc89dded3eahalcanarySkDocument* SkDocument::CreatePDF(SkWStream* stream, SkScalar dpi) { 3538c92dc1dc281649f9e6b0ff534c25bc89dded3eahalcanary return stream ? SkNEW_ARGS(SkDocument_PDF, (stream, NULL, dpi)) : NULL; 35499ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com} 35599ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com 3568c92dc1dc281649f9e6b0ff534c25bc89dded3eahalcanarySkDocument* SkDocument::CreatePDF(const char path[], SkScalar dpi) { 35799ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com SkFILEWStream* stream = SkNEW_ARGS(SkFILEWStream, (path)); 35899ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com if (!stream->isValid()) { 35999ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com SkDELETE(stream); 36099ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com return NULL; 36199ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com } 362a43b41538ad6af0f2b7742b4ade8265d344b9d64halcanary auto delete_wstream = [](SkWStream* stream, bool) { SkDELETE(stream); }; 3638c92dc1dc281649f9e6b0ff534c25bc89dded3eahalcanary return SkNEW_ARGS(SkDocument_PDF, (stream, delete_wstream, dpi)); 36499ac02bb701c1e30b20f2174aac25ffbe487c0afreed@google.com} 365