[Print Preview] Modified PrepareFrameAndViewPrint class to call the new printBegin function to supp…

chrome/renderer/print_web_view_helper.cc

 // Copyright (c) 2012 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 <string>
 
 #include "base/auto_reset.h"
 #include "base/command_line.h"
@@ skipping 284 matching lines @@
   // Swap the |width| and |height| values.
   page_params->page_size.SetSize(page_params->page_size.height(),
                                  page_params->page_size.width());
   page_params->content_size.SetSize(page_params->content_size.height(),
                                     page_params->content_size.width());
   page_params->printable_area.set_size(
       gfx::Size(page_params->printable_area.height(),
                 page_params->printable_area.width()));
 }
 
-void CalculatePrintCanvasSize(const PrintMsg_Print_Params& print_params,
-                              gfx::Size* result) {
+void ComputePrintParamsInDesiredDpi(const PrintMsg_Print_Params& print_params,
+                                    gfx::Size* content_size,
+                                    gfx::Rect* printable_area,
+                                    gfx::Size* paper_size) {
   int dpi = GetDPI(&print_params);
-  result->set_width(ConvertUnit(print_params.content_size.width(), dpi,
-                                print_params.desired_dpi));
+  content_size->set_width(ConvertUnit(print_params.content_size.width(), dpi,
+                                      print_params.desired_dpi));
 
-  result->set_height(ConvertUnit(print_params.content_size.height(), dpi,
-                                 print_params.desired_dpi));
+  content_size->set_height(ConvertUnit(print_params.content_size.height(), dpi,
+                                       print_params.desired_dpi));
+
+  printable_area->SetRect(ConvertUnit(print_params.printable_area.x(), dpi,
+                                      print_params.desired_dpi),
+                          ConvertUnit(print_params.printable_area.y(), dpi,
+                                      print_params.desired_dpi),
+                          ConvertUnit(print_params.printable_area.width(), dpi,
+                                      print_params.desired_dpi),
+                          ConvertUnit(print_params.printable_area.height(), dpi,
+                                      print_params.desired_dpi));
+  paper_size->SetSize(ConvertUnit(print_params.page_size.width(), dpi,
+                                  print_params.desired_dpi),
+                      ConvertUnit(print_params.page_size.height(), dpi,
+                                  print_params.desired_dpi));
 }
 
 bool PrintingNodeOrPdfFrame(const WebFrame* frame, const WebNode& node) {
   if (!node.isNull())
     return true;
   if (!frame->document().isPluginDocument())
     return false;
   WebPlugin* plugin = frame->document().to<WebPluginDocument>().plugin();
   return plugin && plugin->supportsPaginatedPrint();
 }
@@ skipping 270 matching lines @@
   }
 
 #if defined(USE_SKIA)
   device->setDrawingArea(SkPDFDevice::kContent_DrawingArea);
 #endif
 }
 
 PrepareFrameAndViewForPrint::PrepareFrameAndViewForPrint(
     const PrintMsg_Print_Params& print_params,
     WebFrame* frame,
-    const WebNode& node)
+    const WebNode& node,
+    WebKit::WebPrintScalingOption scaling_option)
         : frame_(frame),
           node_to_print_(node),
           web_view_(frame->view()),
           dpi_(static_cast<int>(print_params.dpi)),
           expected_pages_count_(0),
           use_browser_overlays_(true),
+          print_scaling_option_(scaling_option),
           finished_(false) {
   gfx::Size canvas_size;
-  CalculatePrintCanvasSize(print_params, &canvas_size);
+  gfx::Rect printable_area;
+  gfx::Size paper_size;
+  ComputePrintParamsInDesiredDpi(print_params, &canvas_size, &printable_area,
+                                 &paper_size);
 
   if (WebFrame* web_frame = web_view_->mainFrame())
     prev_scroll_offset_ = web_frame->scrollOffset();
   prev_view_size_ = web_view_->size();
 
-  StartPrinting(canvas_size);
+  StartPrinting(canvas_size, printable_area, paper_size);
 }
 
 PrepareFrameAndViewForPrint::~PrepareFrameAndViewForPrint() {
   FinishPrinting();
 }
 
 void PrepareFrameAndViewForPrint::UpdatePrintParams(
     const PrintMsg_Print_Params& print_params) {
   DCHECK(!finished_);
   gfx::Size canvas_size;
-  CalculatePrintCanvasSize(print_params, &canvas_size);
-  if (canvas_size == print_canvas_size_)
+  gfx::Rect printable_area;
+  gfx::Size paper_size;
+  ComputePrintParamsInDesiredDpi(print_params, &canvas_size, &printable_area,
+                                 &paper_size);
+  if (canvas_size == print_canvas_size_ &&
+      printable_area == printable_area_ &&
+      paper_size == paper_size_) {
     return;
+  }
 
   frame_->printEnd();
   dpi_ = static_cast<int>(print_params.dpi);
-  StartPrinting(canvas_size);
+  StartPrinting(canvas_size, printable_area, paper_size);
 }
 
 void PrepareFrameAndViewForPrint::StartPrinting(
-    const gfx::Size& print_canvas_size) {
+    const gfx::Size& print_canvas_size,
+    const gfx::Rect& printable_area,
+    const gfx::Size& paper_size) {
   print_canvas_size_ = print_canvas_size;
+  printable_area_ = printable_area;
+  paper_size_ = paper_size;
 
   // Layout page according to printer page size. Since WebKit shrinks the
   // size of the page automatically (from 125% to 200%) we trick it to
   // think the page is 125% larger so the size of the page is correct for
   // minimum (default) scaling.
   // This is important for sites that try to fill the page.
   gfx::Size print_layout_size(print_canvas_size_);
   print_layout_size.set_height(static_cast<int>(
       static_cast<double>(print_layout_size.height()) * 1.25));
 
   web_view_->resize(print_layout_size);
 
-  expected_pages_count_ = frame_->printBegin(print_canvas_size_, node_to_print_,
-                                             dpi_, &use_browser_overlays_);
+  expected_pages_count_ = frame_->printBegin(print_canvas_size_,
+                                             printable_area_,
+                                             paper_size_,
+                                             node_to_print_,
+                                             dpi_,
+                                             print_scaling_option_,
+                                             &use_browser_overlays_);
 }
 
 void PrepareFrameAndViewForPrint::FinishPrinting() {
   if (!finished_) {
     finished_ = true;
     frame_->printEnd();
     web_view_->resize(prev_view_size_);
     if (WebFrame* web_frame = web_view_->mainFrame())
       web_frame->setScrollOffset(prev_scroll_offset_);
   }
 }
 
 PrintWebViewHelper::PrintWebViewHelper(content::RenderView* render_view)
     : content::RenderViewObserver(render_view),
       content::RenderViewObserverTracker<PrintWebViewHelper>(render_view),
       print_web_view_(NULL),
       is_preview_enabled_(IsPrintPreviewEnabled()),
       is_print_ready_metafile_sent_(false),
       ignore_css_margins_(false),
       user_cancelled_scripted_print_count_(0),
       is_scripted_printing_blocked_(false),
       notify_browser_of_print_failure_(true),
-      print_for_preview_(false) {
+      print_for_preview_(false),
+      print_scaling_option_(WebKit::WebPrintFitToPrintableArea) {
 }
 
 PrintWebViewHelper::~PrintWebViewHelper() {}
 
 bool PrintWebViewHelper::IsScriptInitiatedPrintAllowed(
     WebKit::WebFrame* frame) {
   if (is_scripted_printing_blocked_)
     return false;
   return !IsScriptInitiatedPrintTooFrequent(frame);
 }
@@ skipping 199 matching lines @@
     if (notify_browser_of_print_failure_)
       LOG(ERROR) << "CreatePreviewDocument failed";
     DidFinishPrinting(FAIL_PREVIEW);
   }
 }
 
 bool PrintWebViewHelper::CreatePreviewDocument() {
   PrintMsg_Print_Params print_params = print_pages_params_->params;
   const std::vector<int>& pages = print_pages_params_->pages;
   if (!print_preview_context_.CreatePreviewDocument(&print_params, pages,
-                                                    ignore_css_margins_)) {
+                                                    ignore_css_margins_,
+                                                    print_scaling_option_)) {
     return false;
   }
 
   PageSizeMargins default_page_layout;
   ComputePageLayoutInPointsForCss(print_preview_context_.frame(), 0,
                                   print_params, ignore_css_margins_, NULL,
                                   &default_page_layout);
 
   if (!old_print_pages_params_.get() ||
       !PageLayoutIsEqual(*old_print_pages_params_, *print_pages_params_)) {
@@ skipping 254 matching lines @@
   // actual printing.
   print_web_view_->mainFrame()->loadRequest(WebURLRequest(url));
 
   return true;
 }
 
 #if defined(OS_MACOSX) || defined(OS_WIN)
 bool PrintWebViewHelper::PrintPages(WebFrame* frame, const WebNode& node) {
   const PrintMsg_PrintPages_Params& params = *print_pages_params_;
   const PrintMsg_Print_Params& print_params = params.params;
-  PrepareFrameAndViewForPrint prep_frame_view(print_params, frame, node);
+  PrepareFrameAndViewForPrint prep_frame_view(print_params, frame, node,
+                                              print_scaling_option_);
   UpdateFrameAndViewFromCssPageLayout(frame, node, &prep_frame_view,
                                       print_params, ignore_css_margins_);
 
   int page_count = prep_frame_view.GetExpectedPageCount();
   if (!page_count)
     return false;
   Send(new PrintHostMsg_DidGetPrintedPagesCount(routing_id(),
                                                 print_params.document_cookie,
                                                 page_count));
 
@@ skipping 43 matching lines @@
     const PrintMsg_Print_Params& params,
     bool ignore_css_margins) {
   if (PrintingNodeOrPdfFrame(frame, node))
     return;
   PrintMsg_Print_Params print_params = CalculatePrintParamsForCss(
       frame, 0, params, ignore_css_margins,
       ignore_css_margins && params.fit_to_paper_size, NULL);
   prepare->UpdatePrintParams(print_params);
 }
 
-bool PrintWebViewHelper::InitPrintSettings() {
+bool PrintWebViewHelper::InitPrintSettings(bool fit_to_paper_size) {
   PrintMsg_PrintPages_Params settings;
   Send(new PrintHostMsg_GetDefaultPrintSettings(routing_id(),
                                                 &settings.params));
   // Check if the printer returned any settings, if the settings is empty, we
   // can safely assume there are no printer drivers configured. So we safely
   // terminate.
   bool result = true;
   if (!PrintMsg_Print_Params_IsValid(settings.params))
     result = false;
 
   if (result &&
       (settings.params.dpi < kMinDpi || settings.params.document_cookie == 0)) {
     // Invalid print page settings.
     NOTREACHED();
     result = false;
   }
 
   // Reset to default values.
   ignore_css_margins_ = false;
-  settings.params.fit_to_paper_size = true;
   settings.pages.clear();
+  settings.params.fit_to_paper_size = fit_to_paper_size;
+
+  // Update the print scaling option.
+  print_scaling_option_ = static_cast<WebKit::WebPrintScalingOption>(

[vandebo (ex-Chrome), 2012/04/20 23:26:11]

This cast is pretty ugly.  A conditional or trinary op would be better.

[kmadhusu, 2012/04/23 17:39:10]

Done... Removed print_scaling_option_ from PrintWebViewHelper. Modified
PrintMsg_Print_Params fit_to_page => print_scaling_option.

+      fit_to_paper_size);
 
   print_pages_params_.reset(new PrintMsg_PrintPages_Params(settings));
   return result;
 }
 
 bool PrintWebViewHelper::InitPrintSettingsAndPrepareFrame(
     WebKit::WebFrame* frame, const WebKit::WebNode& node,
     scoped_ptr<PrepareFrameAndViewForPrint>* prepare) {
   DCHECK(frame);
-  if (!InitPrintSettings()) {
+  bool fit_to_paper_size = !(PrintingNodeOrPdfFrame(frame, node) &&
+                             frame->isPrintScalingDisabledForPlugin(node));
+  if (!InitPrintSettings(fit_to_paper_size)) {
     notify_browser_of_print_failure_ = false;
     render_view()->RunModalAlertDialog(
         frame,
         l10n_util::GetStringUTF16(IDS_PRINT_PREVIEW_INVALID_PRINTER_SETTINGS));
     return false;
   }
 
   DCHECK(!prepare->get());
   prepare->reset(new PrepareFrameAndViewForPrint(print_pages_params_->params,
-                                                 frame, node));
+                                                 frame, node,
+                                                 print_scaling_option_));
   UpdateFrameAndViewFromCssPageLayout(frame, node, prepare->get(),
                                       print_pages_params_->params,
                                       ignore_css_margins_);
   Send(new PrintHostMsg_DidGetDocumentCookie(
         routing_id(), print_pages_params_->params.document_cookie));
   return true;
 }
 
 bool PrintWebViewHelper::UpdatePrintSettings(
     WebKit::WebFrame* frame, const WebKit::WebNode& node,
@@ skipping 87 matching lines @@
                                   &(settings.params.is_first_request))) {
       NOTREACHED();
       print_preview_context_.set_error(PREVIEW_ERROR_BAD_SETTING);
       return false;
     }
 
     settings.params.print_to_pdf = IsPrintToPdfRequested(*job_settings);
     UpdateFrameMarginsCssInfo(*job_settings);
 
     // Fit to paper size.
-    settings.params.fit_to_paper_size = source_is_html &&
-                                        !IsPrintToPdfRequested(*job_settings);
+    settings.params.fit_to_paper_size =

[vandebo (ex-Chrome), 2012/04/20 23:26:11]

At this point, it might be clearer to use an if or two...

settings.params.fir_to_paper_size = true
print_scaling_option_ = WebKit::WebPrintFitToPrintableArea;
if (print_for_preview_ || settings.params.print_to_pdf) {
  print_scaling_option_ = false;
  settings.params.fit_to_paper_size = WebKit::WebPrintActualSize;
} else if(PrintingNodeOrPdfFrame(frame, node) &&
          frame->isPrintScalingDisabledForPlugin(node)) {
  settings.params.fit_to_paper_size = false
  print_scaling_option_ = WebKit::WebPrintScaleNone;
}

Or something like this.

[kmadhusu, 2012/04/23 17:39:10]

Done.

+        !(print_for_preview_ || settings.params.print_to_pdf ||
+          (PrintingNodeOrPdfFrame(frame, node) &&
+           frame->isPrintScalingDisabledForPlugin(node)));
+
+    // Update the print scaling option.
+    print_scaling_option_ = (print_for_preview_ ||
+                             settings.params.print_to_pdf) ?
+                            WebKit::WebPrintActualSize :
+                            static_cast<WebKit::WebPrintScalingOption>(
+                                settings.params.fit_to_paper_size);
 
     // Header/Footer: Set |header_footer_info_|.
     if (settings.params.display_header_footer) {
       header_footer_info_.reset(new DictionaryValue());
       header_footer_info_->SetString(printing::kSettingHeaderFooterDate,
                                      settings.params.date);
       header_footer_info_->SetString(printing::kSettingHeaderFooterURL,
                                      settings.params.url);
       header_footer_info_->SetString(printing::kSettingHeaderFooterTitle,
                                      settings.params.title);
@@ skipping 233 matching lines @@
 }
 
 void PrintWebViewHelper::PrintPreviewContext::OnPrintPreview() {
   DCHECK_EQ(INITIALIZED, state_);
   ClearContext();
 }
 
 bool PrintWebViewHelper::PrintPreviewContext::CreatePreviewDocument(
     PrintMsg_Print_Params* print_params,
     const std::vector<int>& pages,
-    bool ignore_css_margins) {
+    bool ignore_css_margins,
+    WebKit::WebPrintScalingOption print_scaling_option) {
   DCHECK_EQ(INITIALIZED, state_);
   state_ = RENDERING;
 
   metafile_.reset(new printing::PreviewMetafile);
   if (!metafile_->Init()) {
     set_error(PREVIEW_ERROR_METAFILE_INIT_FAILED);
     LOG(ERROR) << "PreviewMetafile Init failed";
     return false;
   }
 
   // Need to make sure old object gets destroyed first.
   prep_frame_view_.reset(new PrepareFrameAndViewForPrint(*print_params, frame(),
-                                                         node()));
+                                                         node(),
+                                                         print_scaling_option));
   UpdateFrameAndViewFromCssPageLayout(frame_, node_, prep_frame_view_.get(),
                                       *print_params, ignore_css_margins);
 
   print_params_.reset(new PrintMsg_Print_Params(*print_params));
 
   total_page_count_ = prep_frame_view_->GetExpectedPageCount();
   if (total_page_count_ == 0) {
     LOG(ERROR) << "CreatePreviewDocument got 0 page count";
     set_error(PREVIEW_ERROR_ZERO_PAGES);
     return false;
@@ skipping 156 matching lines @@
   DCHECK(IsRendering());
   return prep_frame_view_->GetPrintCanvasSize();
 }
 
 void PrintWebViewHelper::PrintPreviewContext::ClearContext() {
   prep_frame_view_.reset();
   metafile_.reset();
   pages_to_render_.clear();
   error_ = PREVIEW_ERROR_NONE;
 }