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