Simplify reference management in SkPDF

src/pdf/SkPDFCatalog.cpp

 
 /*
  * Copyright 2010 The Android Open Source Project
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 
 
 #include "SkPDFCatalog.h"
@@ skipping 15 matching lines @@
 
 SkPDFObject* SkPDFCatalog::addObject(SkPDFObject* obj, bool onFirstPage) {
     if (findObjectIndex(obj) != -1) {  // object already added
         return obj;
     }
     SkASSERT(fNextFirstPageObjNum == 0);
     if (onFirstPage) {
         fFirstPageCount++;
     }
 
-    struct Rec newEntry(obj, onFirstPage);
+    Rec newEntry(obj, onFirstPage);
     fCatalog.append(1, &newEntry);
     return obj;
 }
 
 void SkPDFCatalog::setFileOffset(SkPDFObject* obj, off_t offset) {
     int objIndex = assignObjNum(obj) - 1;
     SkASSERT(fCatalog[objIndex].fObjNumAssigned);
     SkASSERT(fCatalog[objIndex].fFileOffset == 0);
     fCatalog[objIndex].fFileOffset = offset;
 }
 
-void SkPDFCatalog::emitObjectNumber(SkWStream* stream, SkPDFObject* obj) {
-    stream->writeDecAsText(assignObjNum(obj));
-    stream->writeText(" 0");  // Generation number is always 0.
+int32_t SkPDFCatalog::getObjectNumber(SkPDFObject* obj) {
+    return (int32_t)assignObjNum(obj);
 }
 
-size_t SkPDFCatalog::getObjectNumberSize(SkPDFObject* obj) {
-    SkDynamicMemoryWStream buffer;
-    emitObjectNumber(&buffer, obj);
-    return buffer.getOffset();
-}
-
-int SkPDFCatalog::findObjectIndex(SkPDFObject* obj) const {
+int SkPDFCatalog::findObjectIndex(SkPDFObject* obj) {
     for (int i = 0; i < fCatalog.count(); i++) {
         if (fCatalog[i].fObject == obj) {
             return i;
         }
     }
     // If it's not in the main array, check if it's a substitute object.
     for (int i = 0; i < fSubstituteMap.count(); ++i) {
         if (fSubstituteMap[i].fSubstitute == obj) {
             return findObjectIndex(fSubstituteMap[i].fOriginal);
         }
     }
-    return -1;
+    Rec newEntry(obj, false);
+    fCatalog.append(1, &newEntry);
+    return fCatalog.count() - 1;
 }
 
 int SkPDFCatalog::assignObjNum(SkPDFObject* obj) {
     int pos = findObjectIndex(obj);
     // If this assert fails, it means you probably forgot to add an object
     // to the resource list.
     SkASSERT(pos >= 0);
     uint32_t currentIndex = pos;
     if (fCatalog[currentIndex].fObjNumAssigned) {
         return currentIndex + 1;
@@ skipping 61 matching lines @@
 #if defined(SK_DEBUG)
     // Sanity check: is the original already in substitute list?
     for (int i = 0; i < fSubstituteMap.count(); ++i) {
         if (original == fSubstituteMap[i].fSubstitute ||
             original == fSubstituteMap[i].fOriginal) {
             SkASSERT(false);
             return;
         }
     }
 #endif
-    // Check if the original is on first page.
-    bool onFirstPage = false;
-    for (int i = 0; i < fCatalog.count(); ++i) {
-        if (fCatalog[i].fObject == original) {
-            onFirstPage = fCatalog[i].fOnFirstPage;
-            break;
-        }
-#if defined(SK_DEBUG)
-        if (i == fCatalog.count() - 1) {
-            SkASSERT(false);  // original not in catalog
-            return;
-        }
-#endif
-    }
-
     SubstituteMapping newMapping(original, substitute);
     fSubstituteMap.append(1, &newMapping);
-
-    // Add resource objects of substitute object to catalog.
-    SkTSet<SkPDFObject*>* targetSet = getSubstituteList(onFirstPage);
-    SkTSet<SkPDFObject*> newResourceObjects;
-    newMapping.fSubstitute->getResources(*targetSet, &newResourceObjects);
-    for (int i = 0; i < newResourceObjects.count(); ++i) {
-        addObject(newResourceObjects[i], onFirstPage);
-    }
-    // mergeInto returns the number of duplicates.
-    // If there are duplicates, there is a bug and we mess ref counting.
-    SkDEBUGCODE(int duplicates =) targetSet->mergeInto(newResourceObjects);
-    SkASSERT(duplicates == 0);
 }
 
 SkPDFObject* SkPDFCatalog::getSubstituteObject(SkPDFObject* object) {
     for (int i = 0; i < fSubstituteMap.count(); ++i) {
         if (object == fSubstituteMap[i].fOriginal) {
             return fSubstituteMap[i].fSubstitute;
         }
     }
     return object;
 }
 
-SkTSet<SkPDFObject*>* SkPDFCatalog::getSubstituteList(bool firstPage) {
-    return firstPage ? &fSubstituteResourcesFirstPage :
-                       &fSubstituteResourcesRemaining;
-}