OLD | NEW |
| (Empty) |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "chrome/renderer/printing/print_web_view_helper.h" | |
6 | |
7 #include "base/logging.h" | |
8 #include "base/memory/scoped_ptr.h" | |
9 #include "chrome/common/print_messages.h" | |
10 #include "content/public/renderer/render_thread.h" | |
11 #include "printing/metafile_skia_wrapper.h" | |
12 #include "printing/page_size_margins.h" | |
13 #include "printing/pdf_metafile_skia.h" | |
14 #include "skia/ext/platform_device.h" | |
15 #include "skia/ext/vector_canvas.h" | |
16 #include "third_party/WebKit/public/web/WebLocalFrame.h" | |
17 | |
18 #if !defined(OS_CHROMEOS) && !defined(OS_ANDROID) | |
19 #include "base/process/process_handle.h" | |
20 #else | |
21 #include "base/file_descriptor_posix.h" | |
22 #endif // !defined(OS_CHROMEOS) && !defined(OS_ANDROID) | |
23 | |
24 namespace printing { | |
25 | |
26 using blink::WebFrame; | |
27 | |
28 bool PrintWebViewHelper::RenderPreviewPage( | |
29 int page_number, | |
30 const PrintMsg_Print_Params& print_params) { | |
31 PrintMsg_PrintPage_Params page_params; | |
32 page_params.params = print_params; | |
33 page_params.page_number = page_number; | |
34 scoped_ptr<PdfMetafileSkia> draft_metafile; | |
35 PdfMetafileSkia* initial_render_metafile = print_preview_context_.metafile(); | |
36 if (print_preview_context_.IsModifiable() && is_print_ready_metafile_sent_) { | |
37 draft_metafile.reset(new PdfMetafileSkia); | |
38 initial_render_metafile = draft_metafile.get(); | |
39 } | |
40 | |
41 base::TimeTicks begin_time = base::TimeTicks::Now(); | |
42 PrintPageInternal(page_params, | |
43 print_preview_context_.prepared_frame(), | |
44 initial_render_metafile); | |
45 print_preview_context_.RenderedPreviewPage( | |
46 base::TimeTicks::Now() - begin_time); | |
47 if (draft_metafile.get()) { | |
48 draft_metafile->FinishDocument(); | |
49 } else if (print_preview_context_.IsModifiable() && | |
50 print_preview_context_.generate_draft_pages()) { | |
51 DCHECK(!draft_metafile.get()); | |
52 draft_metafile = | |
53 print_preview_context_.metafile()->GetMetafileForCurrentPage(); | |
54 } | |
55 return PreviewPageRendered(page_number, draft_metafile.get()); | |
56 } | |
57 | |
58 bool PrintWebViewHelper::PrintPagesNative(blink::WebFrame* frame, | |
59 int page_count) { | |
60 PdfMetafileSkia metafile; | |
61 if (!metafile.Init()) | |
62 return false; | |
63 | |
64 const PrintMsg_PrintPages_Params& params = *print_pages_params_; | |
65 std::vector<int> printed_pages; | |
66 | |
67 if (params.pages.empty()) { | |
68 for (int i = 0; i < page_count; ++i) { | |
69 printed_pages.push_back(i); | |
70 } | |
71 } else { | |
72 // TODO(vitalybuka): redesign to make more code cross platform. | |
73 for (size_t i = 0; i < params.pages.size(); ++i) { | |
74 if (params.pages[i] >= 0 && params.pages[i] < page_count) { | |
75 printed_pages.push_back(params.pages[i]); | |
76 } | |
77 } | |
78 } | |
79 | |
80 if (printed_pages.empty()) | |
81 return false; | |
82 | |
83 PrintMsg_PrintPage_Params page_params; | |
84 page_params.params = params.params; | |
85 for (size_t i = 0; i < printed_pages.size(); ++i) { | |
86 page_params.page_number = printed_pages[i]; | |
87 PrintPageInternal(page_params, frame, &metafile); | |
88 } | |
89 | |
90 // blink::printEnd() for PDF should be called before metafile is closed. | |
91 FinishFramePrinting(); | |
92 | |
93 metafile.FinishDocument(); | |
94 | |
95 // Get the size of the resulting metafile. | |
96 uint32 buf_size = metafile.GetDataSize(); | |
97 DCHECK_GT(buf_size, 0u); | |
98 | |
99 #if defined(OS_CHROMEOS) || defined(OS_ANDROID) | |
100 int sequence_number = -1; | |
101 base::FileDescriptor fd; | |
102 | |
103 // Ask the browser to open a file for us. | |
104 Send(new PrintHostMsg_AllocateTempFileForPrinting(routing_id(), | |
105 &fd, | |
106 &sequence_number)); | |
107 if (!metafile.SaveToFD(fd)) | |
108 return false; | |
109 | |
110 // Tell the browser we've finished writing the file. | |
111 Send(new PrintHostMsg_TempFileForPrintingWritten(routing_id(), | |
112 sequence_number)); | |
113 return true; | |
114 #else | |
115 PrintHostMsg_DidPrintPage_Params printed_page_params; | |
116 printed_page_params.data_size = 0; | |
117 printed_page_params.document_cookie = params.params.document_cookie; | |
118 | |
119 { | |
120 scoped_ptr<base::SharedMemory> shared_mem( | |
121 content::RenderThread::Get()->HostAllocateSharedMemoryBuffer( | |
122 buf_size).release()); | |
123 if (!shared_mem.get()) { | |
124 NOTREACHED() << "AllocateSharedMemoryBuffer failed"; | |
125 return false; | |
126 } | |
127 | |
128 if (!shared_mem->Map(buf_size)) { | |
129 NOTREACHED() << "Map failed"; | |
130 return false; | |
131 } | |
132 metafile.GetData(shared_mem->memory(), buf_size); | |
133 printed_page_params.data_size = buf_size; | |
134 shared_mem->GiveToProcess(base::GetCurrentProcessHandle(), | |
135 &(printed_page_params.metafile_data_handle)); | |
136 } | |
137 | |
138 for (size_t i = 0; i < printed_pages.size(); ++i) { | |
139 printed_page_params.page_number = printed_pages[i]; | |
140 Send(new PrintHostMsg_DidPrintPage(routing_id(), printed_page_params)); | |
141 // Send the rest of the pages with an invalid metafile handle. | |
142 printed_page_params.metafile_data_handle.fd = -1; | |
143 } | |
144 return true; | |
145 #endif // defined(OS_CHROMEOS) | |
146 } | |
147 | |
148 void PrintWebViewHelper::PrintPageInternal( | |
149 const PrintMsg_PrintPage_Params& params, | |
150 WebFrame* frame, | |
151 PdfMetafileSkia* metafile) { | |
152 PageSizeMargins page_layout_in_points; | |
153 double scale_factor = 1.0f; | |
154 ComputePageLayoutInPointsForCss(frame, params.page_number, params.params, | |
155 ignore_css_margins_, &scale_factor, | |
156 &page_layout_in_points); | |
157 gfx::Size page_size; | |
158 gfx::Rect content_area; | |
159 GetPageSizeAndContentAreaFromPageLayout(page_layout_in_points, &page_size, | |
160 &content_area); | |
161 gfx::Rect canvas_area = | |
162 params.params.display_header_footer ? gfx::Rect(page_size) : content_area; | |
163 | |
164 skia::VectorCanvas* canvas = | |
165 metafile->GetVectorCanvasForNewPage(page_size, canvas_area, scale_factor); | |
166 if (!canvas) | |
167 return; | |
168 | |
169 MetafileSkiaWrapper::SetMetafileOnCanvas(*canvas, metafile); | |
170 skia::SetIsDraftMode(*canvas, is_print_ready_metafile_sent_); | |
171 | |
172 #if defined(ENABLE_PRINT_PREVIEW) | |
173 if (params.params.display_header_footer) { | |
174 // |page_number| is 0-based, so 1 is added. | |
175 // TODO(vitalybuka) : why does it work only with 1.25? | |
176 PrintHeaderAndFooter(canvas, params.page_number + 1, | |
177 print_preview_context_.total_page_count(), *frame, | |
178 scale_factor / 1.25, page_layout_in_points, | |
179 params.params); | |
180 } | |
181 #endif // defined(ENABLE_PRINT_PREVIEW) | |
182 | |
183 RenderPageContent(frame, params.page_number, canvas_area, content_area, | |
184 scale_factor, canvas); | |
185 | |
186 // Done printing. Close the canvas to retrieve the compiled metafile. | |
187 if (!metafile->FinishPage()) | |
188 NOTREACHED() << "metafile failed"; | |
189 } | |
190 | |
191 } // namespace printing | |
OLD | NEW |