mirror of https://github.com/qpdf/qpdf.git
117 lines
3.3 KiB
C++
117 lines
3.3 KiB
C++
#include <qpdf/QPDFOutlineDocumentHelper.hh>
|
|
|
|
#include <qpdf/QTC.hh>
|
|
|
|
QPDFOutlineDocumentHelper::QPDFOutlineDocumentHelper(QPDF& qpdf) :
|
|
QPDFDocumentHelper(qpdf),
|
|
m(new Members())
|
|
{
|
|
QPDFObjectHandle root = qpdf.getRoot();
|
|
if (!root.hasKey("/Outlines")) {
|
|
return;
|
|
}
|
|
QPDFObjectHandle outlines = root.getKey("/Outlines");
|
|
if (!(outlines.isDictionary() && outlines.hasKey("/First"))) {
|
|
return;
|
|
}
|
|
QPDFObjectHandle cur = outlines.getKey("/First");
|
|
std::set<QPDFObjGen> seen;
|
|
while (!cur.isNull()) {
|
|
auto og = cur.getObjGen();
|
|
if (seen.count(og)) {
|
|
break;
|
|
}
|
|
seen.insert(og);
|
|
this->m->outlines.push_back(
|
|
QPDFOutlineObjectHelper::Accessor::create(cur, *this, 1));
|
|
cur = cur.getKey("/Next");
|
|
}
|
|
}
|
|
|
|
bool
|
|
QPDFOutlineDocumentHelper::hasOutlines()
|
|
{
|
|
return !this->m->outlines.empty();
|
|
}
|
|
|
|
std::vector<QPDFOutlineObjectHelper>
|
|
QPDFOutlineDocumentHelper::getTopLevelOutlines()
|
|
{
|
|
return this->m->outlines;
|
|
}
|
|
|
|
void
|
|
QPDFOutlineDocumentHelper::initializeByPage()
|
|
{
|
|
std::list<QPDFOutlineObjectHelper> queue;
|
|
queue.insert(
|
|
queue.end(), this->m->outlines.begin(), this->m->outlines.end());
|
|
|
|
while (!queue.empty()) {
|
|
QPDFOutlineObjectHelper oh = queue.front();
|
|
queue.pop_front();
|
|
this->m->by_page[oh.getDestPage().getObjGen()].push_back(oh);
|
|
std::vector<QPDFOutlineObjectHelper> kids = oh.getKids();
|
|
queue.insert(queue.end(), kids.begin(), kids.end());
|
|
}
|
|
}
|
|
|
|
std::vector<QPDFOutlineObjectHelper>
|
|
QPDFOutlineDocumentHelper::getOutlinesForPage(QPDFObjGen const& og)
|
|
{
|
|
if (this->m->by_page.empty()) {
|
|
initializeByPage();
|
|
}
|
|
std::vector<QPDFOutlineObjectHelper> result;
|
|
if (this->m->by_page.count(og)) {
|
|
result = this->m->by_page[og];
|
|
}
|
|
return result;
|
|
}
|
|
|
|
QPDFObjectHandle
|
|
QPDFOutlineDocumentHelper::resolveNamedDest(QPDFObjectHandle name)
|
|
{
|
|
QPDFObjectHandle result;
|
|
if (name.isName()) {
|
|
if (!this->m->dest_dict.isInitialized()) {
|
|
this->m->dest_dict = this->qpdf.getRoot().getKey("/Dests");
|
|
}
|
|
if (this->m->dest_dict.isDictionary()) {
|
|
QTC::TC("qpdf", "QPDFOutlineDocumentHelper name named dest");
|
|
result = this->m->dest_dict.getKey(name.getName());
|
|
}
|
|
} else if (name.isString()) {
|
|
if (0 == this->m->names_dest.get()) {
|
|
QPDFObjectHandle names = this->qpdf.getRoot().getKey("/Names");
|
|
if (names.isDictionary()) {
|
|
QPDFObjectHandle dests = names.getKey("/Dests");
|
|
if (dests.isDictionary()) {
|
|
this->m->names_dest =
|
|
std::make_shared<QPDFNameTreeObjectHelper>(
|
|
dests, this->qpdf);
|
|
}
|
|
}
|
|
}
|
|
if (this->m->names_dest.get()) {
|
|
if (this->m->names_dest->findObject(name.getUTF8Value(), result)) {
|
|
QTC::TC("qpdf", "QPDFOutlineDocumentHelper string named dest");
|
|
}
|
|
}
|
|
}
|
|
if (!result.isInitialized()) {
|
|
result = QPDFObjectHandle::newNull();
|
|
}
|
|
return result;
|
|
}
|
|
|
|
bool
|
|
QPDFOutlineDocumentHelper::checkSeen(QPDFObjGen const& og)
|
|
{
|
|
if (this->m->seen.count(og) > 0) {
|
|
return true;
|
|
}
|
|
this->m->seen.insert(og);
|
|
return false;
|
|
}
|