Connect the right metafiles for print preview on Linux and Windows.

chrome/renderer/print_web_view_helper_win.cc

 // 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 "base/scoped_ptr.h"
 #include "chrome/common/print_messages.h"
-#include "printing/native_metafile_factory.h"
-#include "printing/native_metafile.h"
+#include "printing/metafile.h"
+#include "printing/metafile_impl.h"
 #include "printing/units.h"
 #include "skia/ext/vector_canvas.h"
 #include "skia/ext/vector_platform_device_emf_win.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebFrame.h"
 #include "ui/gfx/gdi_util.h"
 #include "ui/gfx/point.h"
 #include "ui/gfx/rect.h"
 #include "ui/gfx/size.h"
 
 using printing::ConvertUnitDouble;
@@ skipping 43 matching lines @@
 }
 
 }  // namespace
 
 void PrintWebViewHelper::PrintPageInternal(
     const PrintMsg_PrintPage_Params& params,
     const gfx::Size& canvas_size,
     WebFrame* frame) {
   // Generate a memory-based metafile. It will use the current screen's DPI.
   // Each metafile contains a single page.
-  scoped_ptr<printing::NativeMetafile> metafile(
-      printing::NativeMetafileFactory::Create());
+  scoped_ptr<printing::Metafile> metafile(new printing::NativeMetafile);
+  metafile->Init();
   DCHECK(metafile->context());
   skia::PlatformDevice::InitializeDC(metafile->context());
 
   int page_number = params.page_number;
 
   // Calculate the dpi adjustment.
   float scale_factor = static_cast<float>(params.params.desired_dpi /
                                           params.params.dpi);
 
   // Render page for printing.
@@ skipping 37 matching lines @@
     WebKit::WebNode* node) {
   int page_count = 0;
   PrintMsg_Print_Params print_params = params.params;
   UpdatePrintableSizeInPrintParameters(frame, node, &print_params);
   PrepareFrameAndViewForPrint prep_frame_view(print_params, frame, node,
                                               frame->view());
   page_count = prep_frame_view.GetExpectedPageCount();
   if (!page_count)
     return;
 
-  // NOTE: This is an enhanced-format metafile(EMF) which has an appearance of
-  // single page metafile. For print preview, we need a metafile with multiple
-  // pages.
-  // TODO(kmadhusu): Use a PDF metafile to support multiple pages. After "Skia
-  // PDF backend" work is completed for windows, make changes to replace this
-  // EMF with PDF metafile.
-  // http://code.google.com/p/chromium/issues/detail?id=62889
-  scoped_ptr<printing::NativeMetafile> metafile(
-      printing::NativeMetafileFactory::Create());
-  DCHECK(metafile->context());
-  skia::PlatformDevice::InitializeDC(metafile->context());
+  scoped_ptr<printing::Metafile> metafile(new printing::PreviewMetafile);
+  metafile->Init();
 
   // Calculate the dpi adjustment.
-  float shrink = static_cast<float>(params.params.desired_dpi /
-                                    params.params.dpi);
+  float shrink = static_cast<float>(print_params.desired_dpi /
+                                    print_params.dpi);
 
   if (params.pages.empty()) {
     for (int i = 0; i < page_count; ++i) {
       float scale_factor = shrink;
-      RenderPage(params.params, &scale_factor, i, frame, &metafile);
+      RenderPage(print_params, &scale_factor, i, frame, &metafile);
     }
   } else {
     for (size_t i = 0; i < params.pages.size(); ++i) {
       if (params.pages[i] >= page_count)
         break;
       float scale_factor = shrink;
-      RenderPage(params.params, &scale_factor,
+      RenderPage(print_params, &scale_factor,
           static_cast<int>(params.pages[i]), frame, &metafile);
     }
   }
 
-  // Close the device context to retrieve the compiled metafile.
   if (!metafile->FinishDocument())
     NOTREACHED();
 
   // Get the size of the compiled metafile.
   uint32 buf_size = metafile->GetDataSize();
   DCHECK_GT(buf_size, 128u);
 
   PrintHostMsg_DidPreviewDocument_Params preview_params;
   preview_params.document_cookie = params.params.document_cookie;
   preview_params.data_size = buf_size;
   preview_params.metafile_data_handle = NULL;
   preview_params.expected_pages_count = page_count;
 
   if (!CopyMetafileDataToSharedMem(metafile.get(),
                                    &(preview_params.metafile_data_handle))) {
     preview_params.data_size = 0;
     preview_params.expected_pages_count = 0;
   }
   if (!Send(new PrintHostMsg_DuplicateSection(
           routing_id(),
           preview_params.metafile_data_handle,
           &preview_params.metafile_data_handle))) {
     NOTREACHED() << "Send message failed.";
   }
   Send(new PrintHostMsg_PagesReadyForPreview(routing_id(), preview_params));
 }
 
 void PrintWebViewHelper::RenderPage(
     const PrintMsg_Print_Params& params, float* scale_factor, int page_number,
-    WebFrame* frame, scoped_ptr<printing::NativeMetafile>* metafile) {
-  DCHECK(metafile->get()->context());
-
+    WebFrame* frame, scoped_ptr<printing::Metafile>* metafile) {
   double content_width_in_points;
   double content_height_in_points;
   double margin_top_in_points;
   double margin_left_in_points;
   GetPageSizeAndMarginsInPoints(frame, page_number, params,
                                 &content_width_in_points,
                                 &content_height_in_points,
                                 &margin_top_in_points, NULL, NULL,
                                 &margin_left_in_points);
 
@@ skipping 16 matching lines @@
     NOTREACHED() << "Printing page " << page_number << " failed.";
   } else {
     // Update the dpi adjustment with the "page |scale_factor|" calculated in
     // webkit.
     *scale_factor /= webkit_scale_factor;
   }
 
   bool result = (*metafile)->FinishPage();
   DCHECK(result);
 
-  skia::VectorPlatformDeviceEmf* platform_device =
-      static_cast<skia::VectorPlatformDeviceEmf*>(device);
-  if (platform_device->alpha_blend_used() && !params.supports_alpha_blend) {
-    // Currently, we handle alpha blend transparency for a single page.
-    // Therefore, expecting a metafile with page count 1.
-    DCHECK((*metafile)->GetPageCount() == 1);
+  if (!params.supports_alpha_blend) {
+    // PreviewMetafile (PDF) supports alpha blend.

[Lei Zhang, 2011/04/11 02:51:10]

Err, PrintWebViewHelper::PrintPageInternal() also calls RenderPage(). In which
case we are not using a PreviewMetafile.

[vandebo (ex-Chrome), 2011/04/11 06:16:08]

Comment clarified.

+    skia::VectorPlatformDeviceEmf* platform_device =
+        static_cast<skia::VectorPlatformDeviceEmf*>(device);
+    if (platform_device->alpha_blend_used()) {
+      // Currently, we handle alpha blend transparency for a single page.
+      // Therefore, expecting a metafile with page count 1.
+      DCHECK((*metafile)->GetPageCount() == 1);
 
-    // Close the device context to retrieve the compiled metafile.
-    if (!(*metafile)->FinishDocument())
-      NOTREACHED();
+      // Close the device context to retrieve the compiled metafile.
+      if (!(*metafile)->FinishDocument())
+        NOTREACHED();
 
-    scoped_ptr<printing::NativeMetafile> metafile2(
-        printing::NativeMetafileFactory::Create());
-    // 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";
+      // 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);
+      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);
 
-    HDC hdc = metafile2->context();
-    DCHECK(hdc);
-    skia::PlatformDevice::InitializeDC(hdc);
+      scoped_ptr<printing::Metafile> metafile2(new printing::NativeMetafile);
+      metafile2->Init();
+      HDC hdc = metafile2->context();
+      DCHECK(hdc);
+      skia::PlatformDevice::InitializeDC(hdc);
 
-    RECT metafile_bounds = (*metafile)->GetPageBounds(1).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(),
-                    EnhMetaFileProc,
-                    &bitmap_dc,
-                    &metafile_bounds);
+      RECT metafile_bounds = (*metafile)->GetPageBounds(1).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(),
+                      EnhMetaFileProc,
+                      &bitmap_dc,
+                      &metafile_bounds);
 
-    SelectObject(bitmap_dc, old_bitmap);
-    metafile->reset(metafile2.release());
+      SelectObject(bitmap_dc, old_bitmap);
+      metafile->reset(metafile2.release());
+    }
   }
 }
 
 bool PrintWebViewHelper::CopyMetafileDataToSharedMem(
-    printing::NativeMetafile* metafile,
+    printing::Metafile* metafile,
     base::SharedMemoryHandle* shared_mem_handle) {
   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) {
     NOTREACHED() << "Buffer too large: " << buf_size;
     return false;
   }
 
   // Allocate a shared memory buffer to hold the generated metafile data.
   if (!shared_buf.CreateAndMapAnonymous(buf_size)) {
     NOTREACHED() << "Buffer allocation failed";
     return false;
   }
 
   // Copy the bits into shared memory.
   if (!metafile->GetData(shared_buf.memory(), buf_size)) {
     NOTREACHED() << "GetData() failed";
     shared_buf.Unmap();
     return false;
   }
   shared_buf.GiveToProcess(base::GetCurrentProcessHandle(), shared_mem_handle);
   shared_buf.Unmap();
   return true;
 }