Refactor win printing workflow.

chrome/renderer/print_web_view_helper_win.cc

-// Copyright (c) 2009 The Chromium Authors. All rights reserved.
+// Copyright (c) 2011 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 "chrome/renderer/print_web_view_helper.h"
 
 #include "base/logging.h"
+#include "base/process_util.h"
 #include "chrome/common/render_messages.h"
 #include "chrome/common/render_messages_params.h"
 #include "chrome/renderer/render_view.h"
 #include "gfx/gdi_util.h"
 #include "gfx/size.h"
 #include "grit/generated_resources.h"
 #include "printing/native_metafile.h"
 #include "printing/units.h"
 #include "skia/ext/vector_canvas.h"
 #include "skia/ext/vector_platform_device.h"
@@ skipping 41 matching lines @@
     // Play this command to the metafile DC.
     PlayEnhMetaFileRecord(dc, handle_table, record, num_objects);
   }
   return 1;  // Continue enumeration
 }
 
 void PrintWebViewHelper::PrintPage(const ViewMsg_PrintPage_Params& params,
                                    const gfx::Size& canvas_size,
                                    WebFrame* frame) {
   // Generate a memory-based metafile. It will use the current screen's DPI.
-  printing::NativeMetafile metafile;
-
-  metafile.CreateDc(NULL, NULL);
-  HDC hdc = metafile.hdc();
-  DCHECK(hdc);
-  skia::PlatformDevice::InitializeDC(hdc);
+  // Each metafile contains a single page.
+  scoped_ptr<printing::NativeMetafile> metafile(new printing::NativeMetafile());
+  metafile->CreateDc(NULL, NULL);
+  DCHECK(metafile->hdc());
+  skia::PlatformDevice::InitializeDC(metafile->hdc());
 
   int page_number = params.page_number;
 
   double content_width_in_points;
   double content_height_in_points;
   GetPageSizeAndMarginsInPoints(frame, page_number, params.params,
       &content_width_in_points, &content_height_in_points, NULL, NULL, NULL,
       NULL);
 
   // Calculate the dpi adjustment.
   float scale_factor = static_cast<float>(params.params.desired_dpi /
                                           params.params.dpi);
 
   // Since WebKit extends the page width depending on the magical |scale_factor|
   // we make sure the canvas covers the worst case scenario (x2.0 currently).
   // PrintContext will then set the correct clipping region.
-  int width = static_cast<int>(content_width_in_points *
-                               params.params.max_shrink);
-  int height = static_cast<int>(content_height_in_points *
-                                params.params.max_shrink);
+  gfx::Size page_size(
+      static_cast<int>(content_width_in_points * params.params.max_shrink),
+      static_cast<int>(content_height_in_points * params.params.max_shrink));
+
+  // Render page for printing.
+  RenderPage(page_size, &scale_factor, page_number, frame, &metafile,
+      params.params.supports_alpha_blend);
+
+  // Close the device context to retrieve the compiled metafile.
+  if (!metafile->CloseDc())
+    NOTREACHED();
+
+  // Get the size of the compiled metafile.
+  uint32 buf_size = metafile->GetDataSize();
+  DCHECK_GT(buf_size, 128u);
+
+  ViewHostMsg_DidPrintPage_Params page_params;
+  page_params.data_size = 0;
+  page_params.metafile_data_handle = NULL;
+  page_params.page_number = page_number;
+  page_params.document_cookie = params.params.document_cookie;
+  page_params.actual_shrink = scale_factor;
+  page_params.page_size = params.params.page_size;
+  page_params.content_area = gfx::Rect(params.params.margin_left,
+      params.params.margin_top, params.params.printable_size.width(),
+      params.params.printable_size.height());
+  page_params.has_visible_overlays = frame->isPageBoxVisible(page_number);
+
+  if (CopyMetafileDataToSharedMem(metafile.get(),
+          &(page_params.metafile_data_handle))) {
+    page_params.data_size = buf_size;
+  }
+  (*metafile).CloseEmf();

[Lei Zhang, 2011/01/25 03:02:54]

Isn't this the same as metafile->CloseEmf()?

[kmadhusu, 2011/01/25 17:39:18]

Fixed.

+  if (Send(new ViewHostMsg_DuplicateSection(
+          routing_id(),
+          page_params.metafile_data_handle,
+          &page_params.metafile_data_handle))) {
+    Send(new ViewHostMsg_DidPrintPage(routing_id(), page_params));
+  }
+}
+
+void PrintWebViewHelper::RenderPage(
+    const gfx::Size& page_size, float* scale_factor, int page_number,
+    WebFrame* frame, scoped_ptr<printing::NativeMetafile>* metafile,
+    bool supports_alpha_blend) {
+  HDC hdc = (*metafile)->hdc();
+  DCHECK(hdc);
+
+  int width = page_size.width();
+  int height = page_size.height();
 #if 0
   // TODO(maruel): This code is kept for testing until the 100% GDI drawing
   // code is stable. maruels use this code's output as a reference when the
   // GDI drawing code fails.
 
   // Mix of Skia and GDI based.
   skia::PlatformCanvas canvas(width, height, true);
   canvas.drawARGB(255, 255, 255, 255, SkXfermode::kSrc_Mode);
   float webkit_scale_factor = frame->printPage(page_number, &canvas);
-  if (scale_factor <= 0 || webkit_scale_factor <= 0) {
+  if (*scale_factor <= 0 || webkit_scale_factor <= 0) {
     NOTREACHED() << "Printing page " << page_number << " failed.";
   } else {
-    // Update the dpi adjustment with the "page |scale_factor|" calculated
-    // in webkit.
-    scale_factor /= webkit_scale_factor;
+    // Update the dpi adjustment with the "page |scale_factor|" calculated in
+    // webkit.
+    *scale_factor /= webkit_scale_factor;
   }
 
   // Create a BMP v4 header that we can serialize.
   BITMAPV4HEADER bitmap_header;
   gfx::CreateBitmapV4Header(width, height, &bitmap_header);
   const SkBitmap& src_bmp = canvas.getDevice()->accessBitmap(true);
   SkAutoLockPixels src_lock(src_bmp);
   int retval = StretchDIBits(hdc,
                              0,
                              0,
                              width, height,
                              0, 0,
                              width, height,
                              src_bmp.getPixels(),
                              reinterpret_cast<BITMAPINFO*>(&bitmap_header),
                              DIB_RGB_COLORS,
                              SRCCOPY);
   DCHECK(retval != GDI_ERROR);
 #else
   // 100% GDI based.
   skia::VectorCanvas canvas(hdc, width, height);
   float webkit_scale_factor = frame->printPage(page_number, &canvas);
-  if (scale_factor <= 0 || webkit_scale_factor <= 0) {
+  if (*scale_factor <= 0 || webkit_scale_factor <= 0) {
     NOTREACHED() << "Printing page " << page_number << " failed.";
   } else {
-    // Update the dpi adjustment with the "page scale_factor" calculated
-    // in webkit.
-    scale_factor /= webkit_scale_factor;
+    // Update the dpi adjustment with the "page |scale_factor|" calculated in
+    // webkit.
+    *scale_factor /= webkit_scale_factor;
   }
 #endif
 
-  // Done printing. Close the device context to retrieve the compiled metafile.
-  if (!metafile.CloseDc()) {
-    NOTREACHED() << "metafile failed";
-  }
-  printing::NativeMetafile* mf = &metafile;
-  printing::NativeMetafile metafile2;
+  skia::VectorPlatformDevice* platform_device =
+      static_cast<skia::VectorPlatformDevice*>(canvas.getDevice());
+  if (platform_device->alpha_blend_used() && !supports_alpha_blend) {
+    // Close the device context to retrieve the compiled metafile.
+    if (!(*metafile)->CloseDc())
+      NOTREACHED();
 
-  skia::VectorPlatformDevice* platform_device =
-    static_cast<skia::VectorPlatformDevice*>(canvas.getDevice());
-  if (platform_device->alpha_blend_used() &&
-      !params.params.supports_alpha_blend) {
+    scoped_ptr<printing::NativeMetafile> metafile2(
+        new printing::NativeMetafile());
     // Page used alpha blend, but printer doesn't support it.  Rewrite the
     // metafile and flatten out the transparency.
     HDC bitmap_dc = CreateCompatibleDC(GetDC(NULL));
     if (!bitmap_dc) {
       NOTREACHED() << "Bitmap DC creation failed";
     }
     SetGraphicsMode(bitmap_dc, GM_ADVANCED);
     void* bits = NULL;
     BITMAPINFO hdr;
     gfx::CreateBitmapHeader(width, height, &hdr.bmiHeader);
     HBITMAP hbitmap = CreateDIBSection(
         bitmap_dc, &hdr, DIB_RGB_COLORS, &bits, NULL, 0);
     if (!hbitmap) {
       NOTREACHED() << "Raster bitmap creation for printing failed";
     }
 
     HGDIOBJ old_bitmap = SelectObject(bitmap_dc, hbitmap);
     RECT rect = {0, 0, width, height };
     HBRUSH whiteBrush = static_cast<HBRUSH>(GetStockObject(WHITE_BRUSH));
     FillRect(bitmap_dc, &rect, whiteBrush);
 
-    metafile2.CreateDc(NULL, NULL);
-    HDC hdc = metafile2.hdc();
+    metafile2->CreateDc(NULL, NULL);
+    HDC hdc = metafile2->hdc();
     DCHECK(hdc);
     skia::PlatformDevice::InitializeDC(hdc);
 
-    RECT metafile_bounds = metafile.GetBounds().ToRECT();
+    RECT metafile_bounds = (*metafile)->GetBounds().ToRECT();
     // Process the old metafile, placing all non-AlphaBlend calls into the
     // new metafile, and copying the results of all the AlphaBlend calls
     // from the bitmap DC.
     EnumEnhMetaFile(hdc,
-                    metafile.emf(),
+                    (*metafile)->emf(),
                     EnhMetaFileProc,
                     &bitmap_dc,
                     &metafile_bounds);
 
     SelectObject(bitmap_dc, old_bitmap);
+    metafile->reset(metafile2.release());
+  }
+}
 
-    if (!metafile2.CloseDc()) {
-      NOTREACHED() << "metafile failed";
-    }
-    mf = &metafile2;
-  }
-
-  // Get the size of the compiled metafile.
-  uint32 buf_size = mf->GetDataSize();
-  DCHECK_GT(buf_size, 128u);
-  ViewHostMsg_DidPrintPage_Params page_params;
-  page_params.data_size = 0;
-  page_params.metafile_data_handle = NULL;
-  page_params.page_number = page_number;
-  page_params.document_cookie = params.params.document_cookie;
-  page_params.actual_shrink = scale_factor;
-  page_params.page_size = params.params.page_size;
-  page_params.content_area = gfx::Rect(params.params.margin_left,
-      params.params.margin_top, params.params.printable_size.width(),
-      params.params.printable_size.height());
-  page_params.has_visible_overlays = frame->isPageBoxVisible(page_number);
+bool PrintWebViewHelper::CopyMetafileDataToSharedMem(
+    printing::NativeMetafile* metafile,
+    base::SharedMemoryHandle* shared_mem_handle) {
+  bool ret_val = false;
+  uint32 buf_size = metafile->GetDataSize();
   base::SharedMemory shared_buf;
 
   // http://msdn2.microsoft.com/en-us/library/ms535522.aspx
   // Windows 2000/XP: When a page in a spooled file exceeds approximately 350
   // MB, it can fail to print and not send an error message.
-  if (buf_size < 350*1024*1024) {
-    // Allocate a shared memory buffer to hold the generated metafile data.
-    if (shared_buf.CreateAndMapAnonymous(buf_size)) {
-      // Copy the bits into shared memory.
-      if (mf->GetData(shared_buf.memory(), buf_size)) {
-        page_params.metafile_data_handle = shared_buf.handle();
-        page_params.data_size = buf_size;
-      } else {
-        NOTREACHED() << "GetData() failed";
-      }
-      shared_buf.Unmap();
-    } else {
-      NOTREACHED() << "Buffer allocation failed";
-    }
+  if (buf_size > 350*1024*1024) {

[Lei Zhang, 2011/01/25 03:02:54]

Change > to >= to preserve the original code.

[kmadhusu, 2011/01/25 17:39:18]

Done.

+    NOTREACHED() << "Buffer too large: " << buf_size;
+    return ret_val;
+  }
+
+  // Allocate a shared memory buffer to hold the generated metafile data.
+  if (!shared_buf.CreateAndMapAnonymous(buf_size)) {
+    NOTREACHED() << "Buffer allocation failed";
+    return ret_val;
+  }
+
+  // Copy the bits into shared memory.
+  if (metafile->GetData(shared_buf.memory(), buf_size)) {

[Lei Zhang, 2011/01/25 03:02:54]

If you change this to:

if (!metafile->GetData())

then you can get rid of |ret_val| altogether.

[kmadhusu, 2011/01/25 17:39:18]

Done.

+    shared_buf.GiveToProcess(base::GetCurrentProcessHandle(),
+                             shared_mem_handle);
+    ret_val = true;
   } else {
-    NOTREACHED() << "Buffer too large: " << buf_size;
+    NOTREACHED() << "GetData() failed";
   }
-  mf->CloseEmf();
-  if (Send(new ViewHostMsg_DuplicateSection(
-      routing_id(),
-      page_params.metafile_data_handle,
-      &page_params.metafile_data_handle))) {
-    if (!is_preview_) {

[Lei Zhang, 2011/01/25 03:02:54]

Do you need to add a placeholder here to say what we are going to do in the
preview case?

[kmadhusu, 2011/01/25 17:39:18]

No. For preview workflow, we are not going to call this function.

[Lei Zhang, 2011/01/25 20:01:05]

If you're never going to call this function for print preview, then why have
this check?

-      Send(new ViewHostMsg_DidPrintPage(routing_id(), page_params));
-    }
-  }
+  shared_buf.Unmap();
+  return ret_val;
 }