Print Preview: Make print preview tab modal.

chrome/renderer/print_web_view_helper.h

 // 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.
 
 #ifndef CHROME_RENDERER_PRINT_WEB_VIEW_HELPER_H_
 #define CHROME_RENDERER_PRINT_WEB_VIEW_HELPER_H_
 #pragma once
 
 #include <vector>
 
 #include "base/memory/scoped_ptr.h"
 #include "base/shared_memory.h"
 #include "base/time.h"
 #include "content/public/renderer/render_view_observer.h"
 #include "content/public/renderer/render_view_observer_tracker.h"
 #include "printing/metafile.h"
 #include "printing/metafile_impl.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebCanvas.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebFrameClient.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebNode.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebViewClient.h"
 #include "ui/gfx/size.h"
 
+class MessageLoop;
 struct PrintMsg_Print_Params;
 struct PrintMsg_PrintPage_Params;
 struct PrintMsg_PrintPages_Params;
 
 namespace base {
 class DictionaryValue;
 }
 namespace printing {
 struct PageSizeMargins;
 }
@@ skipping 256 matching lines @@
   // Returns true if cancelling, false if continuing.
   bool CheckForCancel();
 
   // Notifies the browser a print preview page has been rendered.
   // |page_number| is 0-based.
   // For a valid |page_number| with modifiable content,
   // |metafile| is the rendered page. Otherwise |metafile| is NULL.
   // Returns true if print preview should continue, false on failure.
   bool PreviewPageRendered(int page_number, printing::Metafile* metafile);
 
+  // Stop the nested message loop when print preview has finished.
+  void OnPrintPreviewDone();
+
   WebKit::WebView* print_web_view_;
 
   scoped_ptr<PrintMsg_PrintPages_Params> print_pages_params_;
   bool is_preview_;
   bool is_print_ready_metafile_sent_;
 
   // Used for scripted initiated printing blocking.
   base::Time last_cancelled_script_print_;
   int user_cancelled_scripted_print_count_;
 
@@ skipping 57 matching lines @@
     // Helper functions
     int GetNextPageNumber();
     bool IsReadyToRender() const;
     bool IsModifiable() const;
     bool IsLastPageOfPrintReadyMetafile() const;
     bool IsFinalPageRendered() const;
 
     // Setters
     void set_generate_draft_pages(bool generate_draft_pages);
     void set_error(enum PrintPreviewErrorBuckets error);
+    void set_message_loop(MessageLoop* message_loop);
 
     // Getters
     WebKit::WebFrame* frame() const;
     const WebKit::WebNode& node() const;
+    MessageLoop* message_loop() const;
     int total_page_count() const;
     bool generate_draft_pages();
     printing::PreviewMetafile* metafile() const;
     const PrintMsg_Print_Params& print_params() const;
     const gfx::Size& GetPrintCanvasSize() const;
     int last_error() const;
 
    private:
     enum State {
       UNINITIALIZED,  // Not ready to render.
       INITIALIZED,    // Ready to render.
       RENDERING,      // Rendering.
       DONE            // Finished rendering.
     };
 
     // Reset some of the internal rendering context.
     void ClearContext();
 
     // Specifies what to render for print preview.
     WebKit::WebFrame* frame_;
     WebKit::WebNode node_;
 
     scoped_ptr<PrepareFrameAndViewForPrint> prep_frame_view_;
     scoped_ptr<printing::PreviewMetafile> metafile_;
     scoped_ptr<PrintMsg_Print_Params> print_params_;
 
+    MessageLoop* message_loop_;
+
     // Total page count in the renderer.
     int total_page_count_;
 
     // The current page to render.
     int current_page_index_;
 
     // List of page indices that need to be rendered.
     std::vector<int> pages_to_render_;
 
     // True, when draft pages needs to be generated.
     bool generate_draft_pages_;
 
     // Specifies the total number of pages in the print ready metafile.
     int print_ready_metafile_page_count_;
 
     base::TimeDelta document_render_time_;
     base::TimeTicks begin_time_;
 
     enum PrintPreviewErrorBuckets error_;
 
     State state_;
   };
 
   PrintPreviewContext print_preview_context_;
 
   DISALLOW_COPY_AND_ASSIGN(PrintWebViewHelper);
 };
 
 #endif  // CHROME_RENDERER_PRINT_WEB_VIEW_HELPER_H_