mirror of
https://github.com/qpdf/qpdf.git
synced 2025-01-23 07:08:30 +00:00
Split page handling APIs into a separate source file
This commit is contained in:
Jay Berkenbilt
parent
df493c352f
commit
e01ae1968b
189
libqpdf/QPDF.cc
189
libqpdf/QPDF.cc
@ -4,7 +4,6 @@
|
||||
#include <map>
|
||||
#include <string.h>
|
||||
#include <memory.h>
|
||||
#include <assert.h>
|
||||
|
||||
#include <qpdf/QTC.hh>
|
||||
#include <qpdf/QUtil.hh>
|
||||
@ -2165,191 +2164,3 @@ QPDF::decodeStreams()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
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();
|
||||
}
|
||||
|
||||
195
libqpdf/QPDF_pages.cc
Normal file
195
libqpdf/QPDF_pages.cc
Normal file
@ -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();
|
||||
}
|
||||
@ -43,6 +43,7 @@ SRCS_libqpdf = \
|
||||
libqpdf/QPDF_encryption.cc \
|
||||
libqpdf/QPDF_linearization.cc \
|
||||
libqpdf/QPDF_optimization.cc \
|
||||
libqpdf/QPDF_pages.cc \
|
||||
libqpdf/QTC.cc \
|
||||
libqpdf/QUtil.cc \
|
||||
libqpdf/RC4.cc \
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user