[Experimental] Supporting OOPIF printing

components/printing/renderer/print_web_view_helper_pdf_win.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "components/printing/renderer/print_web_view_helper.h"
 
 #include <stddef.h>
 
 #include "base/logging.h"
 #include "base/process/process_handle.h"
 #include "components/printing/common/print_messages.h"
+#include "content/public/renderer/render_thread.h"
 #include "printing/features/features.h"
 #include "printing/metafile_skia_wrapper.h"
+#include "printing/pdf_metafile_skia.h"
+#include "third_party/WebKit/public/web/WebLocalFrame.h"
 
 namespace printing {
 
 #if BUILDFLAG(ENABLE_BASIC_PRINTING)
 bool PrintWebViewHelper::PrintPagesNative(blink::WebLocalFrame* frame,
                                           int page_count) {
   const PrintMsg_PrintPages_Params& params = *print_pages_params_;
   std::vector<int> printed_pages = GetPrintedPages(params, page_count);
   if (printed_pages.empty())
     return false;
 
   std::vector<gfx::Size> page_size_in_dpi(printed_pages.size());
   std::vector<gfx::Rect> content_area_in_dpi(printed_pages.size());
   std::vector<gfx::Rect> printable_area_in_dpi(printed_pages.size());
 
   PdfMetafileSkia metafile(PDF_SKIA_DOCUMENT_TYPE);
   CHECK(metafile.Init());
 
   PrintMsg_PrintPage_Params page_params;
   page_params.params = params.params;
   for (size_t i = 0; i < printed_pages.size(); ++i) {
     page_params.page_number = printed_pages[i];
     PrintPageInternal(page_params, frame, &metafile, &page_size_in_dpi[i],
                       &content_area_in_dpi[i], &printable_area_in_dpi[i]);
   }
 
   // blink::printEnd() for PDF should be called before metafile is closed.
   FinishFramePrinting();
 
-  metafile.FinishDocument();
+  if (true) {
+    // End the document.
+    metafile.FinishAllPages();
+    std::string file_name = base::StringPrintf(
+        "C:\\src\\test_only\\chrome_test_dump\\org_%lld.skp", (size_t)frame);
+    SkFILEWStream file(file_name.c_str());
+    metafile.GetSkp(&file);
+    file.fsync();
 
-  PrintHostMsg_DidPrintPage_Params printed_page_params;
-  if (!CopyMetafileDataToSharedMem(
-          metafile, &printed_page_params.metafile_data_handle)) {
-    return false;
+    // Test new utility logic.
+    size_t buf_size;
+    std::unique_ptr<char[]> buf = metafile.GetPageDataBuffer(&buf_size);
+    if (buf_size == 0) {
+      LOG(ERROR) << "!!! Page data is empty";
+      print_preview_context_.set_error(PREVIEW_ERROR_METAFILE_COPY_FAILED);
+      return false;
+    }
+
+    std::unique_ptr<base::SharedMemory> shared_buf(
+        content::RenderThread::Get()->HostAllocateSharedMemoryBuffer(buf_size));
+    if (!shared_buf || !shared_buf->Map(buf_size)) {
+      print_preview_context_.set_error(PREVIEW_ERROR_METAFILE_COPY_FAILED);
+      return false;
+    }
+
+    memcpy(shared_buf->memory(), buf.get(), buf_size);
+
+    PrintHostMsg_DidPrintPage_Params printed_page_params;
+    if (!CopyMetafileDataToSharedMem(metafile,
+                                     &printed_page_params.metafile_data_handle,
+                                     &printed_page_params.data_size)) {
+      print_preview_context_.set_error(PREVIEW_ERROR_METAFILE_COPY_FAILED);
+      return false;
+    }
+
+    printed_page_params.content_area = params.params.printable_area;
+    printed_page_params.document_cookie = params.params.document_cookie;
+    printed_page_params.page_size = params.params.page_size;
+    printed_page_params.is_subframe = params.params.is_subframe;
+
+    for (size_t i = 0; i < printed_pages.size(); ++i) {
+      printed_page_params.page_number = printed_pages[i];
+      printed_page_params.page_size = page_size_in_dpi[i];
+      printed_page_params.content_area = content_area_in_dpi[i];
+      Send(new PrintHostMsg_DidPrintPage(routing_id(), printed_page_params));
+      // Send the rest of the pages with an invalid metafile handle.
+      // TODO(erikchen): Fix semantics. See https://crbug.com/640840
+      if (printed_page_params.metafile_data_handle.IsValid())
+        printed_page_params.metafile_data_handle = base::SharedMemoryHandle();
+    }
+    return true;
+  } else {
+    metafile.FinishAllPages();
+    SkFILEWStream file("C:\\src\\test_only\\chrome_test_dump\\org.skp");
+    metafile.GetSkp(&file);
+    file.fsync();
+    // End the document.
+    metafile.FinishDocument();
+
+    PrintHostMsg_DidPrintPage_Params printed_page_params;
+    if (!CopyMetafileDataToSharedMem(metafile,
+                                     &printed_page_params.metafile_data_handle,
+                                     &printed_page_params.data_size)) {
+      return false;
+    }
+
+    printed_page_params.content_area = params.params.printable_area;
+    printed_page_params.data_size = metafile.GetDataSize();
+    printed_page_params.document_cookie = params.params.document_cookie;
+    printed_page_params.page_size = params.params.page_size;
+
+    for (size_t i = 0; i < printed_pages.size(); ++i) {
+      printed_page_params.page_number = printed_pages[i];
+      printed_page_params.page_size = page_size_in_dpi[i];
+      printed_page_params.content_area = content_area_in_dpi[i];
+      Send(new PrintHostMsg_DidPrintPage(routing_id(), printed_page_params));
+      // Send the rest of the pages with an invalid metafile handle.
+      // TODO(erikchen): Fix semantics. See https://crbug.com/640840
+      if (printed_page_params.metafile_data_handle.IsValid())
+        printed_page_params.metafile_data_handle = base::SharedMemoryHandle();
+    }
+    return true;
   }
-
-  printed_page_params.content_area = params.params.printable_area;
-  printed_page_params.data_size = metafile.GetDataSize();
-  printed_page_params.document_cookie = params.params.document_cookie;
-  printed_page_params.page_size = params.params.page_size;
-
-  for (size_t i = 0; i < printed_pages.size(); ++i) {
-    printed_page_params.page_number = printed_pages[i];
-    printed_page_params.page_size = page_size_in_dpi[i];
-    printed_page_params.content_area = content_area_in_dpi[i];
-    printed_page_params.physical_offsets =
-        gfx::Point(printable_area_in_dpi[i].x(), printable_area_in_dpi[i].y());
-    Send(new PrintHostMsg_DidPrintPage(routing_id(), printed_page_params));
-    // Send the rest of the pages with an invalid metafile handle.
-    // TODO(erikchen): Fix semantics. See https://crbug.com/640840
-    if (printed_page_params.metafile_data_handle.IsValid())
-      printed_page_params.metafile_data_handle = base::SharedMemoryHandle();
-  }
-  return true;
 }
 #endif  // BUILDFLAG(ENABLE_BASIC_PRINTING)
 
 }  // namespace printing