Split page handling APIs into a separate source file

1 files changed, 195 insertions, 0 deletions

diff --git a/libqpdf/QPDF_pages.cc b/libqpdf/QPDF_pages.cc
new file mode 100644
index 00000000..bd631c96
--- /dev/null
+++ b/libqpdf/QPDF_pages.cc
@@ -0,0 +1,195 @@
+#include <qpdf/QPDF.hh>
+
+#include <assert.h>
+
+#include <qpdf/QTC.hh>
+#include <qpdf/QUtil.hh>
+#include <qpdf/QPDFExc.hh>
+
+std::vector<QPDFObjectHandle> const&
+QPDF::getAllPages()
+{
+    if (this->all_pages.empty())
+    {
+	getAllPagesInternal(
+	    this->trailer.getKey("/Root").getKey("/Pages"), this->all_pages);
+    }
+    return this->all_pages;
+}
+
+void
+QPDF::getAllPagesInternal(QPDFObjectHandle cur_pages,
+			  std::vector<QPDFObjectHandle>& result)
+{
+    std::string type = cur_pages.getKey("/Type").getName();
+    if (type == "/Pages")
+    {
+	QPDFObjectHandle kids = cur_pages.getKey("/Kids");
+	int n = kids.getArrayNItems();
+	for (int i = 0; i < n; ++i)
+	{
+	    getAllPagesInternal(kids.getArrayItem(i), result);
+	}
+    }
+    else if (type == "/Page")
+    {
+	result.push_back(cur_pages);
+    }
+    else
+    {
+	throw QPDFExc(qpdf_e_damaged_pdf, this->file->getName(),
+		      this->last_object_description,
+		      this->file->getLastOffset(),
+		      ": invalid Type in page tree");
+    }
+}
+
+// FIXXX here down
+
+void
+QPDF::clearPagesCache()
+{
+    this->all_pages.clear();
+    this->pageobj_to_pages_pos.clear();
+}
+
+void
+QPDF::flattenPagesTree()
+{
+    clearPagesCache();
+
+    // FIXME: more specific method, we don't want to generate the extra stuff.
+    // We also need cheap fixup after addPage/removePage.
+
+    // no compressed objects to be produced here...
+    std::map<int, int> object_stream_data;
+    optimize(object_stream_data); // push down inheritance
+
+    std::vector<QPDFObjectHandle> kids = this->getAllPages();
+    QPDFObjectHandle pages = this->trailer.getKey("/Root").getKey("/Pages");
+
+    const int len = kids.size();
+    for (int pos = 0; pos < len; ++pos)
+    {
+        // populate pageobj_to_pages_pos
+        ObjGen og(kids[pos].getObjectID(), kids[pos].getGeneration());
+        if (! this->pageobj_to_pages_pos.insert(std::make_pair(og, pos)).second)
+        {
+            // insert failed: duplicate entry found
+            *out_stream << "WARNING: duplicate page reference found, "
+                        << "but currently not fully supported." << std::endl;
+        }
+
+        // fix parent links
+        kids[pos].replaceKey("/Parent", pages);
+    }
+
+    pages.replaceKey("/Kids", QPDFObjectHandle::newArray(kids));
+    // /Count has not changed
+    assert(pages.getKey("/Count").getIntValue() == len);
+}
+
+int
+QPDF::findPage(int objid, int generation)
+{
+    if (this->pageobj_to_pages_pos.empty())
+    {
+        flattenPagesTree();
+    }
+    std::map<ObjGen, int>::iterator it =
+        this->pageobj_to_pages_pos.find(ObjGen(objid, generation));
+    if (it != this->pageobj_to_pages_pos.end())
+    {
+        return (*it).second;
+    }
+    return -1; // throw?
+}
+
+int
+QPDF::findPage(QPDFObjectHandle const& pageoh)
+{
+    if (!pageoh.isInitialized())
+    {
+        return -1;
+        // TODO? throw
+    }
+    return findPage(pageoh.getObjectID(), pageoh.getGeneration());
+}
+
+void
+QPDF::addPage(QPDFObjectHandle newpage, bool first)
+{
+    if (this->pageobj_to_pages_pos.empty())
+    {
+        flattenPagesTree();
+    }
+
+    newpage.assertPageObject(); // FIXME: currently private
+
+    QPDFObjectHandle pages = this->trailer.getKey("/Root").getKey("/Pages");
+    QPDFObjectHandle kids = pages.getKey("/Kids");
+
+    newpage.replaceKey("/Parent", pages);
+    if (first)
+    {
+        kids.insertItem(0, newpage);
+    }
+    else
+    {
+        kids.appendItem(newpage);
+    }
+    pages.replaceKey("/Count",
+                     QPDFObjectHandle::newInteger(kids.getArrayNItems()));
+
+    // FIXME: this is overkill, but cache is now stale
+    clearPagesCache();
+}
+
+void
+QPDF::addPageAt(QPDFObjectHandle newpage, bool before,
+                QPDFObjectHandle const &refpage)
+{
+    int refpos = findPage(refpage); // also ensures flat /Pages
+    if (refpos == -1)
+    {
+        throw "Could not find refpage";
+    }
+
+    newpage.assertPageObject();
+
+    QPDFObjectHandle pages = this->trailer.getKey("/Root").getKey("/Pages");
+    QPDFObjectHandle kids = pages.getKey("/Kids");
+
+    if (! before)
+    {
+        ++refpos;
+    }
+
+    newpage.replaceKey("/Parent", pages);
+    kids.insertItem(refpos, newpage);
+    pages.replaceKey("/Count",
+                     QPDFObjectHandle::newInteger(kids.getArrayNItems()));
+
+    // FIXME: this is overkill, but cache is now stale
+    clearPagesCache();
+}
+
+void
+QPDF::removePage(QPDFObjectHandle const& pageoh)
+{
+    int pos = findPage(pageoh); // also ensures flat /Pages
+    if (pos == -1)
+    {
+        throw "Can't remove non-existing page";
+    }
+
+    QPDFObjectHandle pages = this->trailer.getKey("/Root").getKey("/Pages");
+    QPDFObjectHandle kids = pages.getKey("/Kids");
+
+    kids.eraseItem(pos);
+    pages.replaceKey("/Count",
+                     QPDFObjectHandle::newInteger(kids.getArrayNItems()));
+
+    // FIXME: this is overkill, but cache is now stale
+    clearPagesCache();
+}