Print Preview: Make preview generation event driven to eliminate synchronous messages.

chrome/browser/printing/print_preview_message_handler.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/browser/printing/print_preview_message_handler.h"
 
 #include <vector>
 
 #include "base/memory/ref_counted.h"
 #include "chrome/browser/browser_process.h"
@@ skipping 45 matching lines @@
       printing::PrintPreviewTabController::GetInstance();
   if (!tab_controller)
     return NULL;
   return tab_controller->GetPrintPreviewForTab(tab_contents());
 }
 
 void PrintPreviewMessageHandler::OnRequestPrintPreview() {
   PrintPreviewTabController::PrintPreview(tab_contents());
 }
 
-void PrintPreviewMessageHandler::OnDidGetPreviewPageCount(int page_count) {
+void PrintPreviewMessageHandler::OnDidGetPreviewPageCount(int document_cookie,
+                                                          int page_count) {
   if (page_count <= 0)
     return;
   TabContents* print_preview_tab = GetPrintPreviewTab();
   if (!print_preview_tab)
     return;
 
   PrintPreviewUI* print_preview_ui =
       static_cast<PrintPreviewUI*>(print_preview_tab->web_ui());
-  print_preview_ui->OnDidGetPreviewPageCount(page_count);
+  print_preview_ui->OnDidGetPreviewPageCount(document_cookie, page_count);
 }
 
-void PrintPreviewMessageHandler::OnDidPreviewPage(int page_number,
-                                                  bool* cancel) {
+void PrintPreviewMessageHandler::OnDidPreviewPage(int page_number) {
+  RenderViewHost* rvh = tab_contents()->render_view_host();
   TabContents* print_preview_tab = GetPrintPreviewTab();
-  if (!print_preview_tab) {
-    // Can't find print preview tab means we should cancel.
-    *cancel = true;
+  if (!(print_preview_tab && print_preview_tab->web_ui())) {
+    // Can't find print preview tab means we should abort.
+    rvh->Send(new PrintMsg_AbortPreview(rvh->routing_id()));
     return;
   }
 
   PrintPreviewUI* print_preview_ui =
       static_cast<PrintPreviewUI*>(print_preview_tab->web_ui());
   bool has_pending = print_preview_ui->HasPendingRequests();
-  if (!has_pending && page_number >= 0)
+  if (has_pending) {
+    // Cancel. Next print preview request will cancel the current one.
+    // Just do the required maintainance work here.
+    StopWorker(print_preview_ui->document_cookie());
+    print_preview_ui->OnPrintPreviewCancelled();
+    return;
+  }
+
+  // Continue
+  if (page_number >= 0)
     print_preview_ui->OnDidPreviewPage(page_number);
-  *cancel = has_pending;
+  rvh->Send(new PrintMsg_ContinuePreview(rvh->routing_id()));
 }
 
 void PrintPreviewMessageHandler::OnPagesReadyForPreview(
     const PrintHostMsg_DidPreviewDocument_Params& params) {
-  // Always need to stop the worker and send PrintMsg_PrintingDone.
   StopWorker(params.document_cookie);
 
-  RenderViewHost* rvh = tab_contents()->render_view_host();
-  rvh->Send(new PrintMsg_PrintingDone(rvh->routing_id(), true));
-
   // Get the print preview tab.
   TabContents* print_preview_tab = GetPrintPreviewTab();
   // User might have closed it already.
   if (!print_preview_tab)
     return;
 
   PrintPreviewUI* print_preview_ui =
       static_cast<PrintPreviewUI*>(print_preview_tab->web_ui());
 
   TabContentsWrapper* wrapper =
       TabContentsWrapper::GetCurrentWrapperForContents(print_preview_tab);
 
   if (params.reuse_existing_data) {
     // Need to match normal rendering where we are expected to send this.
-    print_preview_ui->OnDidGetPreviewPageCount(params.expected_pages_count);
+    print_preview_ui->OnDidGetPreviewPageCount(params.document_cookie,
+                                               params.expected_pages_count);
 
     print_preview_ui->OnPreviewDataIsAvailable(
         params.expected_pages_count,
         wrapper->print_view_manager()->RenderSourceName(),
         params.modifiable,
         params.preview_request_id);
     return;
   }
 
   base::SharedMemory* shared_buf =
@@ skipping 40 matching lines @@
           HasTabContents(wrapper)) {
     // Preview tab was hidden to serve the print request.
     delete wrapper;
   } else {
     PrintPreviewUI* print_preview_ui =
       static_cast<PrintPreviewUI*>(print_preview_tab->web_ui());
     print_preview_ui->OnPrintPreviewFailed();
   }
 }
 
-void PrintPreviewMessageHandler::OnPrintPreviewCancelled(int document_cookie) {
-  // Always need to stop the worker.
-  StopWorker(document_cookie);
-
-  TabContents* print_preview_tab = GetPrintPreviewTab();
-  if (!print_preview_tab)
-    return;
-
-  PrintPreviewUI* print_preview_ui =
-      static_cast<PrintPreviewUI*>(print_preview_tab->web_ui());
-  print_preview_ui->OnPrintPreviewCancelled();
-}
-
 bool PrintPreviewMessageHandler::OnMessageReceived(
     const IPC::Message& message) {
   bool handled = true;
   IPC_BEGIN_MESSAGE_MAP(PrintPreviewMessageHandler, message)
     IPC_MESSAGE_HANDLER(PrintHostMsg_RequestPrintPreview,
                         OnRequestPrintPreview)
     IPC_MESSAGE_HANDLER(PrintHostMsg_DidGetPreviewPageCount,
                         OnDidGetPreviewPageCount)
     IPC_MESSAGE_HANDLER(PrintHostMsg_DidPreviewPage,
                         OnDidPreviewPage)
     IPC_MESSAGE_HANDLER(PrintHostMsg_PagesReadyForPreview,
                         OnPagesReadyForPreview)
     IPC_MESSAGE_HANDLER(PrintHostMsg_PrintPreviewFailed,
                         OnPrintPreviewFailed)
-    IPC_MESSAGE_HANDLER(PrintHostMsg_PrintPreviewCancelled,
-                        OnPrintPreviewCancelled)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
   return handled;
 }
 
 void PrintPreviewMessageHandler::DidStartLoading() {
   if (tab_contents()->delegate() &&
       printing::PrintPreviewTabController::IsPrintPreviewTab(tab_contents())) {
     tab_contents()->SetContentRestrictions(CONTENT_RESTRICTION_PRINT);
   }
 }
 
 }  // namespace printing