OLD | NEW |
---|---|
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/renderer/print_web_view_helper.h" | 5 #include "chrome/renderer/print_web_view_helper.h" |
6 | 6 |
7 #import <AppKit/AppKit.h> | 7 #import <AppKit/AppKit.h> |
8 | 8 |
9 #include "base/logging.h" | 9 #include "base/logging.h" |
10 #include "base/memory/scoped_ptr.h" | |
11 #include "chrome/common/print_messages.h" | 10 #include "chrome/common/print_messages.h" |
12 #include "printing/native_metafile_factory.h" | 11 #include "printing/metafile.h" |
13 #include "printing/native_metafile.h" | 12 #include "printing/metafile_impl.h" |
14 #include "third_party/WebKit/Source/WebKit/chromium/public/WebFrame.h" | 13 #include "third_party/WebKit/Source/WebKit/chromium/public/WebFrame.h" |
15 | 14 |
16 using WebKit::WebFrame; | 15 using WebKit::WebFrame; |
17 | 16 |
18 void PrintWebViewHelper::PrintPageInternal( | 17 void PrintWebViewHelper::PrintPageInternal( |
19 const PrintMsg_PrintPage_Params& params, | 18 const PrintMsg_PrintPage_Params& params, |
20 const gfx::Size& canvas_size, | 19 const gfx::Size& canvas_size, |
21 WebFrame* frame) { | 20 WebFrame* frame) { |
22 scoped_ptr<printing::NativeMetafile> metafile( | 21 printing::NativeMetafile metafile; |
23 printing::NativeMetafileFactory::Create()); | 22 if(!metafile.Init()) |
Lei Zhang
2011/04/11 02:51:10
if( -> if (
ditto below.
vandebo (ex-Chrome)
2011/04/11 06:16:08
Done.
| |
24 if(!metafile.get()) | |
25 return; | 23 return; |
26 | 24 |
27 float scale_factor = frame->getPrintPageShrink(params.page_number); | 25 float scale_factor = frame->getPrintPageShrink(params.page_number); |
28 int page_number = params.page_number; | 26 int page_number = params.page_number; |
29 | 27 |
30 // Render page for printing. | 28 // Render page for printing. |
31 gfx::Point origin(0.0f, 0.0f); | 29 gfx::Point origin(0.0f, 0.0f); |
32 RenderPage(params.params.printable_size, origin, scale_factor, page_number, | 30 RenderPage(params.params.printable_size, origin, scale_factor, page_number, |
33 frame, metafile.get()); | 31 frame, &metafile); |
34 metafile->FinishDocument(); | 32 metafile.FinishDocument(); |
35 | 33 |
36 PrintHostMsg_DidPrintPage_Params page_params; | 34 PrintHostMsg_DidPrintPage_Params page_params; |
37 page_params.data_size = metafile->GetDataSize(); | 35 page_params.data_size = metafile.GetDataSize(); |
38 page_params.page_number = page_number; | 36 page_params.page_number = page_number; |
39 page_params.document_cookie = params.params.document_cookie; | 37 page_params.document_cookie = params.params.document_cookie; |
40 page_params.actual_shrink = scale_factor; | 38 page_params.actual_shrink = scale_factor; |
41 page_params.page_size = params.params.page_size; | 39 page_params.page_size = params.params.page_size; |
42 page_params.content_area = gfx::Rect(params.params.margin_left, | 40 page_params.content_area = gfx::Rect(params.params.margin_left, |
43 params.params.margin_top, | 41 params.params.margin_top, |
44 params.params.printable_size.width(), | 42 params.params.printable_size.width(), |
45 params.params.printable_size.height()); | 43 params.params.printable_size.height()); |
46 | 44 |
47 // Ask the browser to create the shared memory for us. | 45 // Ask the browser to create the shared memory for us. |
48 if (!CopyMetafileDataToSharedMem(metafile.get(), | 46 if (!CopyMetafileDataToSharedMem(&metafile, |
49 &(page_params.metafile_data_handle))) { | 47 &(page_params.metafile_data_handle))) { |
50 page_params.data_size = 0; | 48 page_params.data_size = 0; |
51 } | 49 } |
52 | 50 |
53 Send(new PrintHostMsg_DidPrintPage(routing_id(), page_params)); | 51 Send(new PrintHostMsg_DidPrintPage(routing_id(), page_params)); |
54 } | 52 } |
55 | 53 |
56 void PrintWebViewHelper::CreatePreviewDocument( | 54 void PrintWebViewHelper::CreatePreviewDocument( |
57 const PrintMsg_PrintPages_Params& params, WebKit::WebFrame* frame, | 55 const PrintMsg_PrintPages_Params& params, WebKit::WebFrame* frame, |
58 WebKit::WebNode* node) { | 56 WebKit::WebNode* node) { |
59 PrintMsg_Print_Params printParams = params.params; | 57 PrintMsg_Print_Params printParams = params.params; |
60 UpdatePrintableSizeInPrintParameters(frame, node, &printParams); | 58 UpdatePrintableSizeInPrintParameters(frame, node, &printParams); |
61 | 59 |
62 PrepareFrameAndViewForPrint prep_frame_view(printParams, | 60 PrepareFrameAndViewForPrint prep_frame_view(printParams, |
63 frame, node, frame->view()); | 61 frame, node, frame->view()); |
64 int page_count = prep_frame_view.GetExpectedPageCount(); | 62 int page_count = prep_frame_view.GetExpectedPageCount(); |
65 | 63 |
66 if (!page_count) | 64 if (!page_count) |
67 return; | 65 return; |
68 | 66 |
69 scoped_ptr<printing::NativeMetafile> metafile( | 67 printing::PreviewMetafile metafile; |
70 printing::NativeMetafileFactory::Create()); | 68 if(!metafile.Init()) |
71 if(!metafile.get()) | |
72 return; | 69 return; |
73 | 70 |
74 float scale_factor = frame->getPrintPageShrink(0); | 71 float scale_factor = frame->getPrintPageShrink(0); |
75 gfx::Point origin(printParams.margin_left, printParams.margin_top); | 72 gfx::Point origin(printParams.margin_left, printParams.margin_top); |
76 if (params.pages.empty()) { | 73 if (params.pages.empty()) { |
77 for (int i = 0; i < page_count; ++i) { | 74 for (int i = 0; i < page_count; ++i) { |
78 RenderPage(printParams.page_size, origin, scale_factor, i, frame, | 75 RenderPage(printParams.page_size, origin, scale_factor, i, frame, |
79 metafile.get()); | 76 &metafile); |
80 } | 77 } |
81 } else { | 78 } else { |
82 for (size_t i = 0; i < params.pages.size(); ++i) { | 79 for (size_t i = 0; i < params.pages.size(); ++i) { |
83 if (params.pages[i] >= page_count) | 80 if (params.pages[i] >= page_count) |
84 break; | 81 break; |
85 RenderPage(printParams.page_size, origin, scale_factor, | 82 RenderPage(printParams.page_size, origin, scale_factor, |
86 static_cast<int>(params.pages[i]), frame, metafile.get()); | 83 static_cast<int>(params.pages[i]), frame, &metafile); |
87 } | 84 } |
88 } | 85 } |
89 metafile->FinishDocument(); | 86 metafile.FinishDocument(); |
90 | 87 |
91 PrintHostMsg_DidPreviewDocument_Params preview_params; | 88 PrintHostMsg_DidPreviewDocument_Params preview_params; |
92 preview_params.data_size = metafile->GetDataSize(); | 89 preview_params.data_size = metafile.GetDataSize(); |
93 preview_params.document_cookie = params.params.document_cookie; | 90 preview_params.document_cookie = params.params.document_cookie; |
94 preview_params.expected_pages_count = page_count; | 91 preview_params.expected_pages_count = page_count; |
95 | 92 |
96 // Ask the browser to create the shared memory for us. | 93 // Ask the browser to create the shared memory for us. |
97 if (!CopyMetafileDataToSharedMem(metafile.get(), | 94 if (!CopyMetafileDataToSharedMem(&metafile, |
98 &(preview_params.metafile_data_handle))) { | 95 &(preview_params.metafile_data_handle))) { |
99 preview_params.data_size = 0; | 96 preview_params.data_size = 0; |
100 preview_params.expected_pages_count = 0; | 97 preview_params.expected_pages_count = 0; |
101 } | 98 } |
102 Send(new PrintHostMsg_PagesReadyForPreview(routing_id(), preview_params)); | 99 Send(new PrintHostMsg_PagesReadyForPreview(routing_id(), preview_params)); |
103 } | 100 } |
104 | 101 |
105 void PrintWebViewHelper::RenderPage( | 102 void PrintWebViewHelper::RenderPage( |
106 const gfx::Size& page_size, const gfx::Point& content_origin, | 103 const gfx::Size& page_size, const gfx::Point& content_origin, |
107 const float& scale_factor, int page_number, WebFrame* frame, | 104 const float& scale_factor, int page_number, WebFrame* frame, |
108 printing::NativeMetafile* metafile) { | 105 printing::Metafile* metafile) { |
109 bool success = metafile->StartPage(page_size, content_origin, scale_factor); | 106 bool success = metafile->StartPage(page_size, content_origin, scale_factor); |
110 DCHECK(success); | 107 DCHECK(success); |
111 | 108 |
112 // printPage can create autoreleased references to |context|. PDF contexts | 109 // printPage can create autoreleased references to |context|. PDF contexts |
113 // don't write all their data until they are destroyed, so we need to make | 110 // don't write all their data until they are destroyed, so we need to make |
114 // certain that there are no lingering references. | 111 // certain that there are no lingering references. |
115 NSAutoreleasePool* pool = [[NSAutoreleasePool alloc] init]; | 112 NSAutoreleasePool* pool = [[NSAutoreleasePool alloc] init]; |
116 frame->printPage(page_number, metafile->context()); | 113 frame->printPage(page_number, metafile->context()); |
117 [pool release]; | 114 [pool release]; |
118 | 115 |
119 // Done printing. Close the device context to retrieve the compiled metafile. | 116 // Done printing. Close the device context to retrieve the compiled metafile. |
120 metafile->FinishPage(); | 117 metafile->FinishPage(); |
121 } | 118 } |
OLD | NEW |