Refactor `QPDFObjectHandle::copyStream`: move stream copying logic to `Stream`.

m-holger
1 parent f797de7b
Showing 4 changed files with 30 additions and 34 deletions
libqpdf/QPDFObjectHandle.cc
libqpdf/QPDF_Stream.cc
libqpdf/qpdf/QPDFObjectHandle_private.hh
libqpdf/qpdf/QPDF_private.hh
@@ -1935,24 +1935,6 @@ QPDFObjectHandle::makeDirect(QPDFObjGen::set&amp; visited, bool stop_at_streams)
     visited.erase(cur_og);
 }
-QPDFObjectHandle
-QPDFObjectHandle::copyStream()
-{
-    assertStream();
-    QPDFObjectHandle result = newStream(getOwningQPDF());
-    QPDFObjectHandle dict = result.getDict();
-    QPDFObjectHandle old_dict = getDict();
-    for (auto& iter: QPDFDictItems(old_dict)) {
-        if (iter.second.isIndirect()) {
-            dict.replaceKey(iter.first, iter.second);
-        } else {
-            dict.replaceKey(iter.first, iter.second.shallowCopy());
-        }
-    }
-    QPDF::Doc::StreamCopier::copyStreamData(getOwningQPDF(), result, *this);
-    return result;
-}
-
 void
 QPDFObjectHandle::makeDirect(bool allow_streams)
 {
@@ -44,6 +44,12 @@ class QPDF::Doc::Streams
         return qpdf->pipeStreamData(
             og, offset, length, dict, is_root_metadata, pipeline, suppress_warnings, will_retry);
     }
+
+    static void
+    copyStreamData(QPDF* qpdf, QPDFObjectHandle const& dest, QPDFObjectHandle const& src)
+    {
+        qpdf->copyStreamData(dest, src);
+    }
 };
 namespace
@@ -206,6 +212,15 @@ Stream::Stream(
     setDictDescription();
 }
+Stream
+Stream::copy() const
+{
+    Stream result = qpdf()->newStream();
+    result.stream()->stream_dict = getDict().copy();
+    QPDF::Doc::Streams::copyStreamData(qpdf(), result, *this);
+    return result;
+}
+
 void
 Stream::registerStreamFilter(
     std::string const& filter_name, std::function<std::shared_ptr<QPDFStreamFilter>()> factory)
@@ -850,3 +865,9 @@ QPDFObjectHandle::getStreamJSON(
 {
     return as_stream(error).getStreamJSON(json_version, json_data, decode_level, p, data_filename);
 }
+
+QPDFObjectHandle
+QPDFObjectHandle::copyStream()
+{
+    return as_stream(error).copy();
+}
@@ -469,6 +469,8 @@ namespace qpdf
             qpdf_offset_t offset,
             size_t length);
+        Stream copy() const;
+
         Dictionary
         getDict() const
         {
@@ -13,10 +13,14 @@
 using namespace qpdf;
-namespace qpdf::is
+namespace qpdf
 {
-    class OffsetBuffer;
-} // namespace qpdf::is
+    class Stream;
+    namespace is
+    {
+        class OffsetBuffer;
+    } // namespace is
+} // namespace qpdf
 class BitStream;
 class BitWriter;
@@ -700,19 +704,6 @@ class QPDF::Doc
         QPDF::Members* m;
     }; // class QPDF::Doc::Pages
-    // StreamCopier class is restricted to QPDFObjectHandle so it can copy stream data.
-    class StreamCopier
-    {
-        friend class QPDFObjectHandle;
-
-      private:
-        static void
-        copyStreamData(QPDF* qpdf, QPDFObjectHandle const& dest, QPDFObjectHandle const& src)
-        {
-            qpdf->copyStreamData(dest, src);
-        }
-    };
-
     Doc() = delete;
     Doc(Doc const&) = delete;
     Doc(Doc&&) = delete;