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