From 8ed3e8c79b5cbccfeccee865e555b68025ee2c1f Mon Sep 17 00:00:00 2001
From: Jay Berkenbilt <ejb@ql.org>
Date: Mon, 25 Jan 2021 08:05:43 -0500
Subject: NNTree: rework iterators to be more memory efficient

Keep a std::pair internal to the iterators so that operator* can
return a reference and operator-> can work, and each can work without
copying pairs of objects around.
---
 libqpdf/QPDFNameTreeObjectHelper.cc | 33 ++++++++++++++++++++++++++++++---
 1 file changed, 30 insertions(+), 3 deletions(-)

(limited to 'libqpdf/QPDFNameTreeObjectHelper.cc')

diff --git a/libqpdf/QPDFNameTreeObjectHelper.cc b/libqpdf/QPDFNameTreeObjectHelper.cc
index 7abc761a..1dc7f205 100644
--- a/libqpdf/QPDFNameTreeObjectHelper.cc
+++ b/libqpdf/QPDFNameTreeObjectHelper.cc
@@ -79,6 +79,7 @@ QPDFNameTreeObjectHelper::iterator&
 QPDFNameTreeObjectHelper::iterator::operator++()
 {
     ++(*impl);
+    updateIValue();
     return *this;
 }
 
@@ -86,14 +87,38 @@ QPDFNameTreeObjectHelper::iterator&
 QPDFNameTreeObjectHelper::iterator::operator--()
 {
     --(*impl);
+    updateIValue();
     return *this;
 }
 
+void
+QPDFNameTreeObjectHelper::iterator::updateIValue()
+{
+    if (impl->valid())
+    {
+        auto p = *impl;
+        this->ivalue.first = p->first.getUTF8Value();
+        this->ivalue.second = p->second;
+    }
+    else
+    {
+        this->ivalue.first = "";
+        this->ivalue.second = QPDFObjectHandle();
+    }
+}
+
 QPDFNameTreeObjectHelper::iterator::reference
 QPDFNameTreeObjectHelper::iterator::operator*()
 {
-    auto p = **impl;
-    return std::make_pair(p.first.getUTF8Value(), p.second);
+    updateIValue();
+    return this->ivalue;
+}
+
+QPDFNameTreeObjectHelper::iterator::pointer
+QPDFNameTreeObjectHelper::iterator::operator->()
+{
+    updateIValue();
+    return &this->ivalue;
 }
 
 bool
@@ -107,12 +132,14 @@ QPDFNameTreeObjectHelper::iterator::insertAfter(
     std::string const& key, QPDFObjectHandle value)
 {
     impl->insertAfter(QPDFObjectHandle::newUnicodeString(key), value);
+    updateIValue();
 }
 
 void
 QPDFNameTreeObjectHelper::iterator::remove()
 {
     impl->remove();
+    updateIValue();
 }
 
 QPDFNameTreeObjectHelper::iterator
@@ -175,7 +202,7 @@ QPDFNameTreeObjectHelper::findObject(
     {
         return false;
     }
-    oh = (*i).second;
+    oh = i->second;
     return true;
 }
 
-- 
cgit v1.2.3-70-g09d2