Add QPDFNameTreeObjectHelper

1 files changed, 82 insertions, 0 deletions

diff --git a/libqpdf/QPDFNameTreeObjectHelper.cc b/libqpdf/QPDFNameTreeObjectHelper.cc
new file mode 100644
index 00000000..4eb0fb15
--- /dev/null
+++ b/libqpdf/QPDFNameTreeObjectHelper.cc
@@ -0,0 +1,82 @@
+#include <qpdf/QPDFNameTreeObjectHelper.hh>
+
+QPDFNameTreeObjectHelper::Members::~Members()
+{
+}
+
+QPDFNameTreeObjectHelper::Members::Members()
+{
+}
+
+QPDFNameTreeObjectHelper::QPDFNameTreeObjectHelper(QPDFObjectHandle oh) :
+    QPDFObjectHelper(oh),
+    m(new Members())
+{
+    updateMap(oh);
+}
+
+QPDFNameTreeObjectHelper::~QPDFNameTreeObjectHelper()
+{
+}
+
+void
+QPDFNameTreeObjectHelper::updateMap(QPDFObjectHandle oh)
+{
+    if (this->m->seen.count(oh.getObjGen()))
+    {
+        return;
+    }
+    this->m->seen.insert(oh.getObjGen());
+    QPDFObjectHandle names = oh.getKey("/Names");
+    if (names.isArray())
+    {
+        size_t nitems = names.getArrayNItems();
+        size_t i = 0;
+        while (i < nitems - 1)
+        {
+            QPDFObjectHandle name = names.getArrayItem(i);
+            if (name.isString())
+            {
+                ++i;
+                QPDFObjectHandle obj = names.getArrayItem(i);
+                this->m->entries[name.getUTF8Value()] = obj;
+            }
+            ++i;
+        }
+    }
+    QPDFObjectHandle kids = oh.getKey("/Kids");
+    if (kids.isArray())
+    {
+        size_t nitems = kids.getArrayNItems();
+        for (size_t i = 0; i < nitems; ++i)
+        {
+            updateMap(kids.getArrayItem(i));
+        }
+    }
+}
+
+bool
+QPDFNameTreeObjectHelper::hasName(std::string const& name)
+{
+    return this->m->entries.count(name) != 0;
+}
+
+bool
+QPDFNameTreeObjectHelper::findObject(
+    std::string const& name, QPDFObjectHandle& oh)
+{
+    std::map<std::string, QPDFObjectHandle>::iterator i =
+        this->m->entries.find(name);
+    if (i == this->m->entries.end())
+    {
+        return false;
+    }
+    oh = (*i).second;
+    return true;
+}
+
+std::map<std::string, QPDFObjectHandle>
+QPDFNameTreeObjectHelper::getAsMap() const
+{
+    return this->m->entries;
+}