SkPDF: SkPDFStream takes a unique_ptr

src/pdf/SkPDFStream.cpp

 /*
  * Copyright 2010 Google Inc.
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 
 
 #include "SkData.h"
 #include "SkDeflate.h"
@@ skipping 14 matching lines @@
     this->INHERITED::emitObject(stream, objNumMap, substitutes);
     // duplicate (a cheap operation) preserves const on fCompressedData.
     std::unique_ptr<SkStreamAsset> dup(fCompressedData->duplicate());
     SkASSERT(dup);
     SkASSERT(dup->hasLength());
     stream->writeText(" stream\n");
     stream->writeStream(dup.get(), dup->getLength());
     stream->writeText("\nendstream");
 }
 
-void SkPDFStream::setData(SkStreamAsset* stream) {
+void SkPDFStream::setData(std::unique_ptr<SkStreamAsset> stream) {
     SkASSERT(!fCompressedData);  // Only call this function once.
     SkASSERT(stream);
     // Code assumes that the stream starts at the beginning.
 
     #ifdef SK_PDF_LESS_COMPRESSION
-    fCompressedData.reset(stream->duplicate());
+    fCompressedData = std::move(stream);
     SkASSERT(fCompressedData && fCompressedData->hasLength());
     this->insertInt("Length", fCompressedData->getLength());
     #else
 
     SkASSERT(stream->hasLength());
     SkDynamicMemoryWStream compressedData;
     SkDeflateWStream deflateWStream(&compressedData);
-    SkStreamCopy(&deflateWStream, stream);
+    SkStreamCopy(&deflateWStream, stream.get());
     deflateWStream.finalize();
     size_t compressedLength = compressedData.bytesWritten();
     size_t originalLength = stream->getLength();
 
     if (originalLength <= compressedLength + strlen("/Filter_/FlateDecode_")) {
-        fCompressedData.reset(stream->duplicate());
+        SkAssertResult(stream->rewind());
+        fCompressedData = std::move(stream);
         this->insertInt("Length", originalLength);
         return;
     }
     fCompressedData.reset(compressedData.detachAsStream());
     this->insertName("Filter", "FlateDecode");
     this->insertInt("Length", compressedLength);
     #endif
 }