Printing on Windows via PDF

chrome/browser/printing/print_view_manager_base.cc

 // Copyright 2013 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/browser/printing/print_view_manager_base.h"
 
 #include <map>
 
 #include "base/bind.h"
+#include "base/memory/ref_counted.h"
+#include "base/memory/ref_counted_memory.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/prefs/pref_service.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/timer/timer.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/chrome_notification_types.h"
+#include "chrome/browser/printing/pdf_to_emf_converter.h"
 #include "chrome/browser/printing/print_job.h"
 #include "chrome/browser/printing/print_job_manager.h"
 #include "chrome/browser/printing/printer_query.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/simple_message_box.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/common/print_messages.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/notification_details.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/notification_source.h"
 #include "content/public/browser/render_view_host.h"
 #include "content/public/browser/web_contents.h"
 #include "grit/generated_resources.h"
 #include "printing/metafile_impl.h"
+#include "printing/pdf_render_settings.h"
 #include "printing/printed_document.h"
 #include "ui/base/l10n/l10n_util.h"
 
 #if defined(OS_WIN)
 #include "base/command_line.h"
 #include "chrome/common/chrome_switches.h"
 #endif
 
 #if defined(ENABLE_FULL_PRINTING)
 #include "chrome/browser/printing/print_error_dialog.h"
 #endif
 
 using base::TimeDelta;
 using content::BrowserThread;
 
-#if defined(OS_WIN)
+#if !defined(WIN_PDF_METAFILE_FOR_PRINTING)
 // Limits memory usage by raster to 64 MiB.
 const int kMaxRasterSizeInPixels = 16*1024*1024;
 #endif
 
 namespace printing {
 
 PrintViewManagerBase::PrintViewManagerBase(content::WebContents* web_contents)
     : content::WebContentsObserver(web_contents),
       number_pages_(0),
       printing_succeeded_(false),
       inside_inner_message_loop_(false),
       cookie_(0),
       queue_(g_browser_process->print_job_manager()->queue()) {
   DCHECK(queue_);
-#if defined(OS_POSIX) && !defined(OS_MACOSX)
+#if (defined(OS_POSIX) && !defined(OS_MACOSX)) || \
+    defined(WIN_PDF_METAFILE_FOR_PRINTING)
   expecting_first_page_ = true;
 #endif
   Profile* profile =
       Profile::FromBrowserContext(web_contents->GetBrowserContext());
   printing_enabled_.Init(
       prefs::kPrintingEnabled,
       profile->GetPrefs(),
       base::Bind(&PrintViewManagerBase::UpdateScriptedPrintingBlocked,
                  base::Unretained(this)));
 }
@@ skipping 45 matching lines @@
   DCHECK_GT(cookie, 0);
   DCHECK_GT(number_pages, 0);
   number_pages_ = number_pages;
   OpportunisticallyCreatePrintJob(cookie);
 }
 
 void PrintViewManagerBase::OnDidGetDocumentCookie(int cookie) {
   cookie_ = cookie;
 }
 
+#if defined(WIN_PDF_METAFILE_FOR_PRINTING)
+void PrintViewManagerBase::OnPdfToEmfConverted(
+    const PrintHostMsg_DidPrintPage_Params& params,
+    double scale_factor,
+    const std::vector<base::FilePath>& emf_files) {
+  PrintedDocument* document = print_job_->document();
+  if (!document)
+    return;
+
+  for (size_t i = 0; i < emf_files.size(); ++i) {
+    scoped_ptr<printing::Emf> metafile(new printing::Emf);
+    if (!metafile->InitFromFile(emf_files[i])) {
+      NOTREACHED() << "Invalid metafile";
+      web_contents()->Stop();
+      return;
+    }
+    // Update the rendered document. It will send notifications to the listener.
+    document->SetPage(i,
+                      metafile.release(),
+                      scale_factor,
+                      params.page_size,
+                      params.content_area);
+  }
+
+  ShouldQuitFromInnerMessageLoop();
+}
+#endif  // WIN_PDF_METAFILE_FOR_PRINTING
+
 void PrintViewManagerBase::OnDidPrintPage(
-    const PrintHostMsg_DidPrintPage_Params& params) {
+  const PrintHostMsg_DidPrintPage_Params& params) {
   if (!OpportunisticallyCreatePrintJob(params.document_cookie))
     return;
 
   PrintedDocument* document = print_job_->document();
   if (!document || params.document_cookie != document->cookie()) {
     // Out of sync. It may happen since we are completely asynchronous. Old
     // spurious messages can be received if one of the processes is overloaded.
     return;
   }
 
-#if defined(OS_WIN) || defined(OS_MACOSX)
+#if !defined(WIN_PDF_METAFILE_FOR_PRINTING) || defined(OS_MACOSX)
   const bool metafile_must_be_valid = true;
-#elif defined(OS_POSIX)
+#elif defined(OS_POSIX) || defined(WIN_PDF_METAFILE_FOR_PRINTING)
   const bool metafile_must_be_valid = expecting_first_page_;
   expecting_first_page_ = false;
 #endif
 
   base::SharedMemory shared_buf(params.metafile_data_handle, true);
   if (metafile_must_be_valid) {
     if (!shared_buf.Map(params.data_size)) {
       NOTREACHED() << "couldn't map";
       web_contents()->Stop();
       return;
     }
   }
 
+#if defined(OS_WIN) && !defined(WIN_PDF_METAFILE_FOR_PRINTING)
   scoped_ptr<NativeMetafile> metafile(new NativeMetafile);
   if (metafile_must_be_valid) {
     if (!metafile->InitFromData(shared_buf.memory(), params.data_size)) {
       NOTREACHED() << "Invalid metafile header";
       web_contents()->Stop();
       return;
     }
   }
 
-#if defined(OS_WIN)
   bool big_emf = (params.data_size && params.data_size >= kMetafileMaxSize);
-  int raster_size = std::min(params.page_size.GetArea(),
-                             kMaxRasterSizeInPixels);
+  int raster_size =
+      std::min(params.page_size.GetArea(), kMaxRasterSizeInPixels);
   if (big_emf) {
     scoped_ptr<NativeMetafile> raster_metafile(
         metafile->RasterizeMetafile(raster_size));
     if (raster_metafile.get()) {
       metafile.swap(raster_metafile);
     } else if (big_emf) {
       // Don't fall back to emf here.
       NOTREACHED() << "size:" << params.data_size;
       TerminatePrintJob(true);
       web_contents()->Stop();
       return;
     }
   }
-#endif
+#endif  // OS_WIN && !WIN_PDF_METAFILE_FOR_PRINTING
 
+#if !defined(WIN_PDF_METAFILE_FOR_PRINTING)
   // Update the rendered document. It will send notifications to the listener.
   document->SetPage(params.page_number,
-                    metafile.release(),
-                    params.actual_shrink,
-                    params.page_size,
-                    params.content_area);
+    metafile.release(),
+    params.actual_shrink,
+    params.page_size,
+    params.content_area);
 
   ShouldQuitFromInnerMessageLoop();
+#else
+  if (metafile_must_be_valid) {
+    scoped_refptr<base::RefCountedBytes> bytes = new base::RefCountedBytes(
+        reinterpret_cast<const unsigned char*>(shared_buf.memory()),
+        params.data_size);
+
+    if (!pdf_to_emf_converter_)
+      pdf_to_emf_converter_ = PdfToEmfConverter::CreateDefault();
+
+    const int printer_dpi = 600;
+    pdf_to_emf_converter_->Start(
+        bytes,
+        printing::PdfRenderSettings(params.content_area, printer_dpi, false),
+        base::Bind(&PrintViewManagerBase::OnPdfToEmfConverted,
+                   base::Unretained(this),
+                   params));
+  }
+#endif  // !WIN_PDF_METAFILE_FOR_PRINTING
 }
 
 void PrintViewManagerBase::OnPrintingFailed(int cookie) {
   if (cookie != cookie_) {
     NOTREACHED();
     return;
   }
 
 #if defined(ENABLE_FULL_PRINTING)
   chrome::ShowPrintErrorDialog();
@@ skipping 186 matching lines @@
   if (print_job_.get() &&
       print_job_->document() &&
       !print_job_->document()->IsComplete()) {
     DCHECK(!result);
     // That failed.
     TerminatePrintJob(true);
   } else {
     // DO NOT wait for the job to finish.
     ReleasePrintJob();
   }
-#if defined(OS_POSIX) && !defined(OS_MACOSX)
+#if (defined(OS_POSIX) && !defined(OS_MACOSX)) || \
+    defined(WIN_PDF_METAFILE_FOR_PRINTING)
   expecting_first_page_ = true;
 #endif
 }
 
 void PrintViewManagerBase::PrintingDone(bool success) {
   if (!print_job_.get())
     return;
   Send(new PrintMsg_PrintingDone(routing_id(), success));
 }
 
@@ skipping 123 matching lines @@
   scoped_refptr<printing::PrinterQuery> printer_query;
   printer_query = queue_->PopPrinterQuery(cookie);
   if (!printer_query)
     return;
   BrowserThread::PostTask(
       BrowserThread::IO, FROM_HERE,
       base::Bind(&PrinterQuery::StopWorker, printer_query.get()));
 }
 
 }  // namespace printing