aboutsummaryrefslogtreecommitdiffstats
path: root/examples/pdf-bookmarks.cc
diff options
context:
space:
mode:
authorJay Berkenbilt <ejb@ql.org>2008-04-29 14:55:25 +0200
committerJay Berkenbilt <ejb@ql.org>2008-04-29 14:55:25 +0200
commit9a0b88bf7777c153dc46ace22db74ef24d51583a (patch)
treef567ac1cf2bf5071a611eb49323a935b6ac938ff /examples/pdf-bookmarks.cc
downloadqpdf-9a0b88bf7777c153dc46ace22db74ef24d51583a.tar.zst
update release date to actual daterelease-qpdf-2.0
git-svn-id: svn+q:///qpdf/trunk@599 71b93d88-0707-0410-a8cf-f5a4172ac649
Diffstat (limited to 'examples/pdf-bookmarks.cc')
-rw-r--r--examples/pdf-bookmarks.cc262
1 files changed, 262 insertions, 0 deletions
diff --git a/examples/pdf-bookmarks.cc b/examples/pdf-bookmarks.cc
new file mode 100644
index 00000000..98bff7fe
--- /dev/null
+++ b/examples/pdf-bookmarks.cc
@@ -0,0 +1,262 @@
+#include <iostream>
+#include <string.h>
+#include <qpdf/QPDF.hh>
+#include <qpdf/QUtil.hh>
+#include <qpdf/QTC.hh>
+
+static char const* whoami = 0;
+static enum { st_none, st_numbers, st_lines } style = st_none;
+static bool show_open = false;
+static bool show_targets = false;
+static std::map<int, int> page_map;
+
+void usage()
+{
+ std::cerr << "Usage: " << whoami << " [options] file.pdf [password]"
+ << std::endl
+ << "Options:" << std::endl
+ << " -numbers give bookmarks outline-style numbers"
+ << std::endl
+ << " -lines draw lines to show bookmark hierarchy"
+ << std::endl
+ << " -show-open indicate whether a bookmark is initially open"
+ << std::endl
+ << " -show-targets show target if possible"
+ << std::endl;
+ exit(2);
+}
+
+void print_lines(std::vector<int>& numbers)
+{
+ for (unsigned int i = 0; i < numbers.size() - 1; ++i)
+ {
+ if (numbers[i])
+ {
+ std::cout << "| ";
+ }
+ else
+ {
+ std::cout << " ";
+ }
+ }
+}
+
+void generate_page_map(QPDF& qpdf)
+{
+ std::vector<QPDFObjectHandle> pages = qpdf.getAllPages();
+ int n = 0;
+ for (std::vector<QPDFObjectHandle>::iterator iter = pages.begin();
+ iter != pages.end(); ++iter)
+ {
+ QPDFObjectHandle& oh = *iter;
+ page_map[oh.getObjectID()] = ++n;
+ }
+}
+
+void extract_bookmarks(QPDFObjectHandle outlines, std::vector<int>& numbers)
+{
+ if (outlines.hasKey("/Title"))
+ {
+ // No default so gcc will warn on missing tag
+ switch (style)
+ {
+ case st_none:
+ QTC::TC("examples", "pdf-bookmarks none");
+ break;
+
+ case st_numbers:
+ QTC::TC("examples", "pdf-bookmarks numbers");
+ for (std::vector<int>::iterator iter = numbers.begin();
+ iter != numbers.end(); ++iter)
+ {
+ std::cout << *iter << ".";
+ }
+ std::cout << " ";
+ break;
+
+ case st_lines:
+ QTC::TC("examples", "pdf-bookmarks lines");
+ print_lines(numbers);
+ std::cout << "|" << std::endl;
+ print_lines(numbers);
+ std::cout << "+-+ ";
+ break;
+ }
+
+ if (show_open)
+ {
+ if (outlines.hasKey("/Count"))
+ {
+ QTC::TC("examples", "pdf-bookmarks has count");
+ int count = outlines.getKey("/Count").getIntValue();
+ if (count > 0)
+ {
+ // hierarchy is open at this point
+ QTC::TC("examples", "pdf-bookmarks open");
+ std::cout << "(v) ";
+ }
+ else
+ {
+ QTC::TC("examples", "pdf-bookmarks closed");
+ std::cout << "(>) ";
+ }
+ }
+ else
+ {
+ QTC::TC("examples", "pdf-bookmarks no count");
+ std::cout << "( ) ";
+ }
+ }
+
+ if (show_targets)
+ {
+ QTC::TC("examples", "pdf-bookmarks targets");
+ std::string target = "unknown";
+ // Only explicit destinations supported for now
+ if (outlines.hasKey("/Dest"))
+ {
+ QTC::TC("examples", "pdf-bookmarks dest");
+ QPDFObjectHandle dest = outlines.getKey("/Dest");
+ if ((dest.isArray()) && (dest.getArrayNItems() > 0))
+ {
+ QPDFObjectHandle first = dest.getArrayItem(0);
+ int object_id = first.getObjectID();
+ if (page_map.count(object_id))
+ {
+ target = QUtil::int_to_string(page_map[object_id]);
+ }
+ }
+
+ std::cout << "[ -> " << target << " ] ";
+ }
+ }
+
+ std::cout << outlines.getKey("/Title").getUTF8Value() << std::endl;
+ }
+
+ if (outlines.hasKey("/First"))
+ {
+ numbers.push_back(0);
+ QPDFObjectHandle child = outlines.getKey("/First");
+ while (1)
+ {
+ ++(numbers.back());
+ bool has_next = child.hasKey("/Next");
+ if ((style == st_lines) && (! has_next))
+ {
+ numbers.back() = 0;
+ }
+ extract_bookmarks(child, numbers);
+ if (has_next)
+ {
+ child = child.getKey("/Next");
+ }
+ else
+ {
+ break;
+ }
+ }
+ numbers.pop_back();
+ }
+}
+
+int main(int argc, char* argv[])
+{
+ if ((whoami = strrchr(argv[0], '/')) == NULL)
+ {
+ whoami = argv[0];
+ }
+ else
+ {
+ ++whoami;
+ }
+ // For libtool's sake....
+ if (strncmp(whoami, "lt-", 3) == 0)
+ {
+ whoami += 3;
+ }
+
+ if ((argc == 2) && (strcmp(argv[1], "--version") == 0))
+ {
+ std::cout << whoami << " version 1.5" << std::endl;
+ exit(0);
+ }
+
+ int arg;
+ for (arg = 1; arg < argc; ++arg)
+ {
+ if (argv[arg][0] == '-')
+ {
+ if (strcmp(argv[arg], "-numbers") == 0)
+ {
+ style = st_numbers;
+ }
+ else if (strcmp(argv[arg], "-lines") == 0)
+ {
+ style = st_lines;
+ }
+ else if (strcmp(argv[arg], "-show-open") == 0)
+ {
+ show_open = true;
+ }
+ else if (strcmp(argv[arg], "-show-targets") == 0)
+ {
+ show_targets = true;
+ }
+ else
+ {
+ usage();
+ }
+ }
+ else
+ {
+ break;
+ }
+ }
+
+ if (arg >= argc)
+ {
+ usage();
+ }
+
+ char const* filename = argv[arg++];
+ char const* password = "";
+
+ if (arg < argc)
+ {
+ password = argv[arg++];
+ }
+ if (arg != argc)
+ {
+ usage();
+ }
+
+ try
+ {
+ QPDF qpdf;
+ qpdf.processFile(filename, password);
+
+ QPDFObjectHandle root = qpdf.getRoot();
+ if (root.hasKey("/Outlines"))
+ {
+ std::vector<int> numbers;
+ if (show_targets)
+ {
+ generate_page_map(qpdf);
+ }
+ extract_bookmarks(root.getKey("/Outlines"), numbers);
+ }
+ else
+ {
+ std::cout << filename << " has no bookmarks" << std::endl;
+ }
+ }
+ catch (std::exception &e)
+ {
+ std::cerr << whoami << " processing file " << filename << ": "
+ << e.what() << std::endl;
+ exit(2);
+ }
+
+ return 0;
+}