include/pdf/SkPDFCatalog.h

/*
 * Copyright (C) 2010 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef SkPDFCatalog_DEFINED
#define SkPDFCatalog_DEFINED

#include "SkPDFTypes.h"
#include "SkRefCnt.h"
#include "SkTDArray.h"

/** \class SkPDFCatalog

    The PDF catalog object manages object numbers and when emitted to the
    PDF stream, indexes all the objects in the file by offset.
*/
class SkPDFCatalog {
public:
    /** Create a PDF catalog.
     */
    SkPDFCatalog()
        : fNextObjNum(1),
          fStartedAssigningObjNums(false),
          fAssigningFirstPageObjNums(false) {
    }
    virtual ~SkPDFCatalog() {}

    /** Add the passed object to the catalog.
     *  @param obj        The object to add.
     *  @param onFirstPage  Is the object on the first page.
     */
    void addObject(SkPDFObject* obj, bool onFirstPage);

    /** Output the object number for the passed object.
     *  @param obj        The object of interest.
     *  @param stream     The writable output stream to send the output to.
     */
    void emitObjectNumber(SkWStream* stream, SkPDFObject* obj);

    /** Return the number of bytes that would be emitted for the passed
     *  object's object number.
     *  @param obj        The object of interest
     */
    size_t getObjectNumberSize(SkPDFObject* obj);

private:
    struct Rec {
        Rec(SkPDFObject* object, bool onFirstPage)
            : fObject(object),
              fFileOffset(0),
              fObjNumAssigned(false),
              fOnFirstPage(onFirstPage) {
        }
        SkPDFObject* fObject;
        off_t fFileOffset;
        bool fObjNumAssigned;
        bool fOnFirstPage;
    };

    // TODO(vandebo) Make this a hash if it's a performance problem.
    SkTDArray<struct Rec> fCatalog;

    uint32_t fNextObjNum;
    bool fStartedAssigningObjNums;
    bool fAssigningFirstPageObjNums;

    int findObjectIndex(SkPDFObject* obj) const;

    int assignObjNum(SkPDFObject* obj);
};

#endif