| // Copyright 2014 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "components/printing/renderer/print_web_view_helper.h" |
| |
| #include "base/logging.h" |
| #include "base/memory/scoped_ptr.h" |
| #include "base/process/process_handle.h" |
| #include "components/printing/common/print_messages.h" |
| #include "content/public/renderer/render_thread.h" |
| #include "printing/metafile_skia_wrapper.h" |
| #include "printing/page_size_margins.h" |
| #include "printing/pdf_metafile_skia.h" |
| #include "printing/units.h" |
| #include "third_party/WebKit/public/web/WebLocalFrame.h" |
| |
| namespace printing { |
| |
| using blink::WebFrame; |
| |
| bool PrintWebViewHelper::RenderPreviewPage( |
| int page_number, |
| const PrintMsg_Print_Params& print_params) { |
| PrintMsg_PrintPage_Params page_params; |
| page_params.params = print_params; |
| page_params.page_number = page_number; |
| scoped_ptr<PdfMetafileSkia> draft_metafile; |
| PdfMetafileSkia* initial_render_metafile = print_preview_context_.metafile(); |
| if (print_preview_context_.IsModifiable() && is_print_ready_metafile_sent_) { |
| draft_metafile.reset(new PdfMetafileSkia); |
| initial_render_metafile = draft_metafile.get(); |
| } |
| |
| base::TimeTicks begin_time = base::TimeTicks::Now(); |
| PrintPageInternal(page_params, |
| print_preview_context_.prepared_frame(), |
| initial_render_metafile, |
| NULL, |
| NULL); |
| print_preview_context_.RenderedPreviewPage( |
| base::TimeTicks::Now() - begin_time); |
| if (draft_metafile.get()) { |
| draft_metafile->FinishDocument(); |
| } else if (print_preview_context_.IsModifiable() && |
| print_preview_context_.generate_draft_pages()) { |
| DCHECK(!draft_metafile.get()); |
| draft_metafile = |
| print_preview_context_.metafile()->GetMetafileForCurrentPage(); |
| } |
| return PreviewPageRendered(page_number, draft_metafile.get()); |
| } |
| |
| bool PrintWebViewHelper::PrintPagesNative(blink::WebFrame* frame, |
| int page_count) { |
| PdfMetafileSkia metafile; |
| if (!metafile.Init()) |
| return false; |
| |
| const PrintMsg_PrintPages_Params& params = *print_pages_params_; |
| std::vector<int> printed_pages = GetPrintedPages(params, page_count); |
| if (printed_pages.empty()) |
| return false; |
| |
| std::vector<gfx::Size> page_size_in_dpi(printed_pages.size()); |
| std::vector<gfx::Rect> content_area_in_dpi(printed_pages.size()); |
| |
| PrintMsg_PrintPage_Params page_params; |
| page_params.params = params.params; |
| for (size_t i = 0; i < printed_pages.size(); ++i) { |
| page_params.page_number = printed_pages[i]; |
| PrintPageInternal(page_params, |
| frame, |
| &metafile, |
| &page_size_in_dpi[i], |
| &content_area_in_dpi[i]); |
| } |
| |
| // blink::printEnd() for PDF should be called before metafile is closed. |
| FinishFramePrinting(); |
| |
| metafile.FinishDocument(); |
| |
| PrintHostMsg_DidPrintPage_Params printed_page_params; |
| if (!CopyMetafileDataToSharedMem( |
| metafile, &printed_page_params.metafile_data_handle)) { |
| return false; |
| } |
| |
| printed_page_params.content_area = params.params.printable_area; |
| printed_page_params.data_size = metafile.GetDataSize(); |
| printed_page_params.document_cookie = params.params.document_cookie; |
| printed_page_params.page_size = params.params.page_size; |
| |
| for (size_t i = 0; i < printed_pages.size(); ++i) { |
| printed_page_params.page_number = printed_pages[i]; |
| printed_page_params.page_size = page_size_in_dpi[i]; |
| printed_page_params.content_area = content_area_in_dpi[i]; |
| Send(new PrintHostMsg_DidPrintPage(routing_id(), printed_page_params)); |
| // Send the rest of the pages with an invalid metafile handle. |
| printed_page_params.metafile_data_handle = base::SharedMemoryHandle(); |
| } |
| return true; |
| } |
| |
| void PrintWebViewHelper::PrintPageInternal( |
| const PrintMsg_PrintPage_Params& params, |
| WebFrame* frame, |
| PdfMetafileSkia* metafile, |
| gfx::Size* page_size_in_dpi, |
| gfx::Rect* content_area_in_dpi) { |
| PageSizeMargins page_layout_in_points; |
| double css_scale_factor = 1.0f; |
| ComputePageLayoutInPointsForCss(frame, params.page_number, params.params, |
| ignore_css_margins_, &css_scale_factor, |
| &page_layout_in_points); |
| gfx::Size page_size; |
| gfx::Rect content_area; |
| GetPageSizeAndContentAreaFromPageLayout(page_layout_in_points, &page_size, |
| &content_area); |
| int dpi = static_cast<int>(params.params.dpi); |
| // Calculate the actual page size and content area in dpi. |
| if (page_size_in_dpi) { |
| *page_size_in_dpi = |
| gfx::Size(static_cast<int>(ConvertUnitDouble(page_size.width(), |
| kPointsPerInch, dpi)), |
| static_cast<int>(ConvertUnitDouble(page_size.height(), |
| kPointsPerInch, dpi))); |
| } |
| |
| if (content_area_in_dpi) { |
| // Output PDF matches paper size and should be printer edge to edge. |
| *content_area_in_dpi = |
| gfx::Rect(0, 0, page_size_in_dpi->width(), page_size_in_dpi->height()); |
| } |
| |
| gfx::Rect canvas_area = |
| params.params.display_header_footer ? gfx::Rect(page_size) : content_area; |
| |
| float webkit_page_shrink_factor = |
| frame->getPrintPageShrink(params.page_number); |
| float scale_factor = css_scale_factor * webkit_page_shrink_factor; |
| |
| skia::PlatformCanvas* canvas = |
| metafile->GetVectorCanvasForNewPage(page_size, canvas_area, scale_factor); |
| if (!canvas) |
| return; |
| |
| MetafileSkiaWrapper::SetMetafileOnCanvas(*canvas, metafile); |
| |
| #if defined(ENABLE_PRINT_PREVIEW) |
| if (params.params.display_header_footer) { |
| // |page_number| is 0-based, so 1 is added. |
| PrintHeaderAndFooter(canvas, params.page_number + 1, |
| print_preview_context_.total_page_count(), *frame, |
| scale_factor, page_layout_in_points, params.params); |
| } |
| #endif // defined(ENABLE_PRINT_PREVIEW) |
| |
| float webkit_scale_factor = |
| RenderPageContent(frame, params.page_number, canvas_area, content_area, |
| scale_factor, canvas); |
| DCHECK_GT(webkit_scale_factor, 0.0f); |
| // Done printing. Close the canvas to retrieve the compiled metafile. |
| if (!metafile->FinishPage()) |
| NOTREACHED() << "metafile failed"; |
| } |
| |
| bool PrintWebViewHelper::CopyMetafileDataToSharedMem( |
| const PdfMetafileSkia& metafile, |
| base::SharedMemoryHandle* shared_mem_handle) { |
| uint32_t buf_size = metafile.GetDataSize(); |
| if (buf_size == 0) |
| return false; |
| |
| base::SharedMemory shared_buf; |
| // Allocate a shared memory buffer to hold the generated metafile data. |
| if (!shared_buf.CreateAndMapAnonymous(buf_size)) |
| return false; |
| |
| // Copy the bits into shared memory. |
| if (!metafile.GetData(shared_buf.memory(), buf_size)) |
| return false; |
| |
| if (!shared_buf.GiveToProcess(base::GetCurrentProcessHandle(), |
| shared_mem_handle)) { |
| return false; |
| } |
| |
| Send(new PrintHostMsg_DuplicateSection(routing_id(), *shared_mem_handle, |
| shared_mem_handle)); |
| return true; |
| } |
| |
| } // namespace printing |
