michael@0: 
michael@0: /*
michael@0:  * Copyright 2010 The Android Open Source Project
michael@0:  *
michael@0:  * Use of this source code is governed by a BSD-style license that can be
michael@0:  * found in the LICENSE file.
michael@0:  */
michael@0: 
michael@0: 
michael@0: #include "SkPDFCatalog.h"
michael@0: #include "SkPDFTypes.h"
michael@0: #include "SkStream.h"
michael@0: #include "SkTypes.h"
michael@0: 
michael@0: SkPDFCatalog::SkPDFCatalog(SkPDFDocument::Flags flags)
michael@0:     : fFirstPageCount(0),
michael@0:       fNextObjNum(1),
michael@0:       fNextFirstPageObjNum(0),
michael@0:       fDocumentFlags(flags) {
michael@0: }
michael@0: 
michael@0: SkPDFCatalog::~SkPDFCatalog() {
michael@0:     fSubstituteResourcesRemaining.safeUnrefAll();
michael@0:     fSubstituteResourcesFirstPage.safeUnrefAll();
michael@0: }
michael@0: 
michael@0: SkPDFObject* SkPDFCatalog::addObject(SkPDFObject* obj, bool onFirstPage) {
michael@0:     if (findObjectIndex(obj) != -1) {  // object already added
michael@0:         return obj;
michael@0:     }
michael@0:     SkASSERT(fNextFirstPageObjNum == 0);
michael@0:     if (onFirstPage) {
michael@0:         fFirstPageCount++;
michael@0:     }
michael@0: 
michael@0:     struct Rec newEntry(obj, onFirstPage);
michael@0:     fCatalog.append(1, &newEntry);
michael@0:     return obj;
michael@0: }
michael@0: 
michael@0: size_t SkPDFCatalog::setFileOffset(SkPDFObject* obj, off_t offset) {
michael@0:     int objIndex = assignObjNum(obj) - 1;
michael@0:     SkASSERT(fCatalog[objIndex].fObjNumAssigned);
michael@0:     SkASSERT(fCatalog[objIndex].fFileOffset == 0);
michael@0:     fCatalog[objIndex].fFileOffset = offset;
michael@0: 
michael@0:     return getSubstituteObject(obj)->getOutputSize(this, true);
michael@0: }
michael@0: 
michael@0: void SkPDFCatalog::emitObjectNumber(SkWStream* stream, SkPDFObject* obj) {
michael@0:     stream->writeDecAsText(assignObjNum(obj));
michael@0:     stream->writeText(" 0");  // Generation number is always 0.
michael@0: }
michael@0: 
michael@0: size_t SkPDFCatalog::getObjectNumberSize(SkPDFObject* obj) {
michael@0:     SkDynamicMemoryWStream buffer;
michael@0:     emitObjectNumber(&buffer, obj);
michael@0:     return buffer.getOffset();
michael@0: }
michael@0: 
michael@0: int SkPDFCatalog::findObjectIndex(SkPDFObject* obj) const {
michael@0:     for (int i = 0; i < fCatalog.count(); i++) {
michael@0:         if (fCatalog[i].fObject == obj) {
michael@0:             return i;
michael@0:         }
michael@0:     }
michael@0:     // If it's not in the main array, check if it's a substitute object.
michael@0:     for (int i = 0; i < fSubstituteMap.count(); ++i) {
michael@0:         if (fSubstituteMap[i].fSubstitute == obj) {
michael@0:             return findObjectIndex(fSubstituteMap[i].fOriginal);
michael@0:         }
michael@0:     }
michael@0:     return -1;
michael@0: }
michael@0: 
michael@0: int SkPDFCatalog::assignObjNum(SkPDFObject* obj) {
michael@0:     int pos = findObjectIndex(obj);
michael@0:     // If this assert fails, it means you probably forgot to add an object
michael@0:     // to the resource list.
michael@0:     SkASSERT(pos >= 0);
michael@0:     uint32_t currentIndex = pos;
michael@0:     if (fCatalog[currentIndex].fObjNumAssigned) {
michael@0:         return currentIndex + 1;
michael@0:     }
michael@0: 
michael@0:     // First assignment.
michael@0:     if (fNextFirstPageObjNum == 0) {
michael@0:         fNextFirstPageObjNum = fCatalog.count() - fFirstPageCount + 1;
michael@0:     }
michael@0: 
michael@0:     uint32_t objNum;
michael@0:     if (fCatalog[currentIndex].fOnFirstPage) {
michael@0:         objNum = fNextFirstPageObjNum;
michael@0:         fNextFirstPageObjNum++;
michael@0:     } else {
michael@0:         objNum = fNextObjNum;
michael@0:         fNextObjNum++;
michael@0:     }
michael@0: 
michael@0:     // When we assign an object an object number, we put it in that array
michael@0:     // offset (minus 1 because object number 0 is reserved).
michael@0:     SkASSERT(!fCatalog[objNum - 1].fObjNumAssigned);
michael@0:     if (objNum - 1 != currentIndex) {
michael@0:         SkTSwap(fCatalog[objNum - 1], fCatalog[currentIndex]);
michael@0:     }
michael@0:     fCatalog[objNum - 1].fObjNumAssigned = true;
michael@0:     return objNum;
michael@0: }
michael@0: 
michael@0: int32_t SkPDFCatalog::emitXrefTable(SkWStream* stream, bool firstPage) {
michael@0:     int first = -1;
michael@0:     int last = fCatalog.count() - 1;
michael@0:     // TODO(vandebo): Support linearized format.
michael@0:     // int last = fCatalog.count() - fFirstPageCount - 1;
michael@0:     // if (firstPage) {
michael@0:     //     first = fCatalog.count() - fFirstPageCount;
michael@0:     //     last = fCatalog.count() - 1;
michael@0:     // }
michael@0: 
michael@0:     stream->writeText("xref\n");
michael@0:     stream->writeDecAsText(first + 1);
michael@0:     stream->writeText(" ");
michael@0:     stream->writeDecAsText(last - first + 1);
michael@0:     stream->writeText("\n");
michael@0: 
michael@0:     if (first == -1) {
michael@0:         stream->writeText("0000000000 65535 f \n");
michael@0:         first++;
michael@0:     }
michael@0:     for (int i = first; i <= last; i++) {
michael@0:         // For 32 bits platforms, the maximum offset has to fit within off_t
michael@0:         // which is a 32 bits signed integer on these platforms.
michael@0:         SkDEBUGCODE(static const off_t kMaxOff = SK_MaxS32;)
michael@0:         SkASSERT(fCatalog[i].fFileOffset > 0);
michael@0:         SkASSERT(fCatalog[i].fFileOffset < kMaxOff);
michael@0:         stream->writeBigDecAsText(fCatalog[i].fFileOffset, 10);
michael@0:         stream->writeText(" 00000 n \n");
michael@0:     }
michael@0: 
michael@0:     return fCatalog.count() + 1;
michael@0: }
michael@0: 
michael@0: void SkPDFCatalog::setSubstitute(SkPDFObject* original,
michael@0:                                  SkPDFObject* substitute) {
michael@0: #if defined(SK_DEBUG)
michael@0:     // Sanity check: is the original already in substitute list?
michael@0:     for (int i = 0; i < fSubstituteMap.count(); ++i) {
michael@0:         if (original == fSubstituteMap[i].fSubstitute ||
michael@0:             original == fSubstituteMap[i].fOriginal) {
michael@0:             SkASSERT(false);
michael@0:             return;
michael@0:         }
michael@0:     }
michael@0: #endif
michael@0:     // Check if the original is on first page.
michael@0:     bool onFirstPage = false;
michael@0:     for (int i = 0; i < fCatalog.count(); ++i) {
michael@0:         if (fCatalog[i].fObject == original) {
michael@0:             onFirstPage = fCatalog[i].fOnFirstPage;
michael@0:             break;
michael@0:         }
michael@0: #if defined(SK_DEBUG)
michael@0:         if (i == fCatalog.count() - 1) {
michael@0:             SkASSERT(false);  // original not in catalog
michael@0:             return;
michael@0:         }
michael@0: #endif
michael@0:     }
michael@0: 
michael@0:     SubstituteMapping newMapping(original, substitute);
michael@0:     fSubstituteMap.append(1, &newMapping);
michael@0: 
michael@0:     // Add resource objects of substitute object to catalog.
michael@0:     SkTSet<SkPDFObject*>* targetSet = getSubstituteList(onFirstPage);
michael@0:     SkTSet<SkPDFObject*> newResourceObjects;
michael@0:     newMapping.fSubstitute->getResources(*targetSet, &newResourceObjects);
michael@0:     for (int i = 0; i < newResourceObjects.count(); ++i) {
michael@0:         addObject(newResourceObjects[i], onFirstPage);
michael@0:     }
michael@0:     // mergeInto returns the number of duplicates.
michael@0:     // If there are duplicates, there is a bug and we mess ref counting.
michael@0:     SkDEBUGCODE(int duplicates =) targetSet->mergeInto(newResourceObjects);
michael@0:     SkASSERT(duplicates == 0);
michael@0: }
michael@0: 
michael@0: SkPDFObject* SkPDFCatalog::getSubstituteObject(SkPDFObject* object) {
michael@0:     for (int i = 0; i < fSubstituteMap.count(); ++i) {
michael@0:         if (object == fSubstituteMap[i].fOriginal) {
michael@0:             return fSubstituteMap[i].fSubstitute;
michael@0:         }
michael@0:     }
michael@0:     return object;
michael@0: }
michael@0: 
michael@0: off_t SkPDFCatalog::setSubstituteResourcesOffsets(off_t fileOffset,
michael@0:                                                   bool firstPage) {
michael@0:     SkTSet<SkPDFObject*>* targetSet = getSubstituteList(firstPage);
michael@0:     off_t offsetSum = fileOffset;
michael@0:     for (int i = 0; i < targetSet->count(); ++i) {
michael@0:         offsetSum += setFileOffset((*targetSet)[i], offsetSum);
michael@0:     }
michael@0:     return offsetSum - fileOffset;
michael@0: }
michael@0: 
michael@0: void SkPDFCatalog::emitSubstituteResources(SkWStream *stream, bool firstPage) {
michael@0:     SkTSet<SkPDFObject*>* targetSet = getSubstituteList(firstPage);
michael@0:     for (int i = 0; i < targetSet->count(); ++i) {
michael@0:         (*targetSet)[i]->emit(stream, this, true);
michael@0:     }
michael@0: }
michael@0: 
michael@0: SkTSet<SkPDFObject*>* SkPDFCatalog::getSubstituteList(bool firstPage) {
michael@0:     return firstPage ? &fSubstituteResourcesFirstPage :
michael@0:                        &fSubstituteResourcesRemaining;
michael@0: }