SkData to SkStreamAsset to avoid unneeded copying

dm/DMPDFTask.cpp

 /*
  * Copyright 2014 Google Inc.
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 
 #include "DMPDFTask.h"
 #include "DMPDFRasterizeTask.h"
 #include "DMUtil.h"
@@ skipping 32 matching lines @@
 namespace {
 
 class SinglePagePDF {
 public:
     SinglePagePDF(SkScalar width, SkScalar height)
         : fDocument(SkDocument::CreatePDF(&fWriteStream))
         , fCanvas(fDocument->beginPage(width, height)) {}
 
     SkCanvas* canvas() { return fCanvas; }
 
-    SkData* end() {
+    SkStreamAsset* end() {
         fCanvas->flush();
         fDocument->endPage();
         fDocument->close();
-        return fWriteStream.copyToData();
+        return fWriteStream.detachAsStream();
     }
 
 private:
     SkDynamicMemoryWStream fWriteStream;
     SkAutoTUnref<SkDocument> fDocument;
     SkCanvas* fCanvas;
 };
 
 }  // namespace
 
 void PDFTask::draw() {
-    SkAutoTUnref<SkData> pdfData;
+    SkAutoTDelete<SkStreamAsset> pdfData;
     bool rasterize = true;
     if (fGM.get()) {
         rasterize = 0 == (fGM->getFlags() & skiagm::GM::kSkipPDFRasterization_Flag);
         SinglePagePDF pdf(fGM->width(), fGM->height());
         //TODO(mtklein): GM doesn't do this.  Why not?
         //pdf.canvas()->concat(fGM->getInitialTransform());
         fGM->draw(pdf.canvas());
         pdfData.reset(pdf.end());
     } else {
         SinglePagePDF pdf(SkIntToScalar(fPicture->width()), SkIntToScalar(fPicture->height()));
         fPicture->draw(pdf.canvas());
         pdfData.reset(pdf.end());
     }
 
     SkASSERT(pdfData.get());
     if (rasterize) {
-        this->spawnChild(SkNEW_ARGS(PDFRasterizeTask, (*this, pdfData.get(), fRasterize)));
+        this->spawnChild(SkNEW_ARGS(PDFRasterizeTask,
+                                    (*this, pdfData->duplicate(), fRasterize)));
     }
-    this->spawnChild(SkNEW_ARGS(WriteTask, (*this, pdfData.get(), ".pdf")));
+    this->spawnChild(SkNEW_ARGS(WriteTask,
+                                (*this, pdfData->duplicate(), ".pdf")));
 }
 
 bool PDFTask::shouldSkip() const {
     if (!FLAGS_pdf) {
         return true;
     }
     if (fGM.get() && 0 != (fGM->getFlags() & skiagm::GM::kSkipPDF_Flag)) {
         return true;
     }
     return false;
 }
 
 }  // namespace DM