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/WebLocalFrame.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 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 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 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