| 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 "base/metrics/histogram.h" | |
| 10 #include "base/process/process_handle.h" | |
| 11 #include "base/win/scoped_gdi_object.h" | |
| 12 #include "base/win/scoped_hdc.h" | |
| 13 #include "base/win/scoped_select_object.h" | |
| 14 #include "chrome/common/print_messages.h" | |
| 15 #include "printing/metafile.h" | |
| 16 #include "printing/metafile_impl.h" | |
| 17 #include "printing/metafile_skia_wrapper.h" | |
| 18 #include "printing/page_size_margins.h" | |
| 19 #include "printing/units.h" | |
| 20 #include "skia/ext/platform_device.h" | |
| 21 #include "skia/ext/refptr.h" | |
| 22 #include "skia/ext/vector_canvas.h" | |
| 23 #include "third_party/WebKit/public/web/WebLocalFrame.h" | |
| 24 #include "ui/gfx/gdi_util.h" | |
| 25 #include "ui/gfx/point.h" | |
| 26 #include "ui/gfx/rect.h" | |
| 27 #include "ui/gfx/size.h" | |
| 28 | |
| 29 namespace printing { | |
| 30 | |
| 31 using blink::WebFrame; | |
| 32 | |
| 33 void PrintWebViewHelper::PrintPageInternal( | |
| 34 const PrintMsg_PrintPage_Params& params, | |
| 35 const gfx::Size& canvas_size, | |
| 36 WebFrame* frame) { | |
| 37 // Generate a memory-based metafile. It will use the current screen's DPI. | |
| 38 // Each metafile contains a single page. | |
| 39 scoped_ptr<NativeMetafile> metafile(new NativeMetafile); | |
| 40 metafile->Init(); | |
| 41 DCHECK(metafile->context()); | |
| 42 skia::InitializeDC(metafile->context()); | |
| 43 | |
| 44 int page_number = params.page_number; | |
| 45 | |
| 46 // Calculate the dpi adjustment. | |
| 47 // Browser will render context using desired_dpi, so we need to calculate | |
| 48 // adjustment factor to play content on the printer DC later during the | |
| 49 // actual printing. | |
| 50 double actual_shrink = static_cast<float>(params.params.desired_dpi / | |
| 51 params.params.dpi); | |
| 52 gfx::Size page_size_in_dpi; | |
| 53 gfx::Rect content_area_in_dpi; | |
| 54 | |
| 55 // Render page for printing. | |
| 56 RenderPage(params.params, page_number, frame, false, metafile.get(), | |
| 57 &actual_shrink, &page_size_in_dpi, &content_area_in_dpi); | |
| 58 | |
| 59 // Close the device context to retrieve the compiled metafile. | |
| 60 if (!metafile->FinishDocument()) | |
| 61 NOTREACHED(); | |
| 62 | |
| 63 if (!params.params.supports_alpha_blend && metafile->IsAlphaBlendUsed()) { | |
| 64 scoped_ptr<NativeMetafile> raster_metafile( | |
| 65 metafile->RasterizeAlphaBlend()); | |
| 66 if (raster_metafile.get()) | |
| 67 metafile.swap(raster_metafile); | |
| 68 } | |
| 69 | |
| 70 // Get the size of the compiled metafile. | |
| 71 uint32 buf_size = metafile->GetDataSize(); | |
| 72 DCHECK_GT(buf_size, 128u); | |
| 73 | |
| 74 PrintHostMsg_DidPrintPage_Params page_params; | |
| 75 page_params.data_size = buf_size; | |
| 76 page_params.metafile_data_handle = NULL; | |
| 77 page_params.page_number = page_number; | |
| 78 page_params.document_cookie = params.params.document_cookie; | |
| 79 page_params.actual_shrink = actual_shrink; | |
| 80 page_params.page_size = page_size_in_dpi; | |
| 81 page_params.content_area = content_area_in_dpi; | |
| 82 | |
| 83 if (!CopyMetafileDataToSharedMem(metafile.get(), | |
| 84 &(page_params.metafile_data_handle))) { | |
| 85 page_params.data_size = 0; | |
| 86 } | |
| 87 | |
| 88 Send(new PrintHostMsg_DidPrintPage(routing_id(), page_params)); | |
| 89 } | |
| 90 | |
| 91 bool PrintWebViewHelper::RenderPreviewPage( | |
| 92 int page_number, | |
| 93 const PrintMsg_Print_Params& print_params) { | |
| 94 // Calculate the dpi adjustment. | |
| 95 double actual_shrink = static_cast<float>(print_params.desired_dpi / | |
| 96 print_params.dpi); | |
| 97 scoped_ptr<Metafile> draft_metafile; | |
| 98 Metafile* initial_render_metafile = print_preview_context_.metafile(); | |
| 99 | |
| 100 if (print_preview_context_.IsModifiable() && is_print_ready_metafile_sent_) { | |
| 101 draft_metafile.reset(new PreviewMetafile); | |
| 102 initial_render_metafile = draft_metafile.get(); | |
| 103 } | |
| 104 | |
| 105 base::TimeTicks begin_time = base::TimeTicks::Now(); | |
| 106 RenderPage(print_params, page_number, print_preview_context_.prepared_frame(), | |
| 107 true, initial_render_metafile, &actual_shrink, NULL, NULL); | |
| 108 print_preview_context_.RenderedPreviewPage( | |
| 109 base::TimeTicks::Now() - begin_time); | |
| 110 | |
| 111 if (draft_metafile.get()) { | |
| 112 draft_metafile->FinishDocument(); | |
| 113 } else if (print_preview_context_.IsModifiable() && | |
| 114 print_preview_context_.generate_draft_pages()) { | |
| 115 DCHECK(!draft_metafile.get()); | |
| 116 draft_metafile.reset( | |
| 117 print_preview_context_.metafile()->GetMetafileForCurrentPage()); | |
| 118 } | |
| 119 return PreviewPageRendered(page_number, draft_metafile.get()); | |
| 120 } | |
| 121 | |
| 122 void PrintWebViewHelper::RenderPage( | |
| 123 const PrintMsg_Print_Params& params, int page_number, WebFrame* frame, | |
| 124 bool is_preview, Metafile* metafile, double* actual_shrink, | |
| 125 gfx::Size* page_size_in_dpi, gfx::Rect* content_area_in_dpi) { | |
| 126 PageSizeMargins page_layout_in_points; | |
| 127 double css_scale_factor = 1.0f; | |
| 128 ComputePageLayoutInPointsForCss(frame, page_number, params, | |
| 129 ignore_css_margins_, &css_scale_factor, | |
| 130 &page_layout_in_points); | |
| 131 gfx::Size page_size; | |
| 132 gfx::Rect content_area; | |
| 133 GetPageSizeAndContentAreaFromPageLayout(page_layout_in_points, &page_size, | |
| 134 &content_area); | |
| 135 int dpi = static_cast<int>(params.dpi); | |
| 136 // Calculate the actual page size and content area in dpi. | |
| 137 if (page_size_in_dpi) { | |
| 138 *page_size_in_dpi = gfx::Size( | |
| 139 static_cast<int>(ConvertUnitDouble(page_size.width(), kPointsPerInch, | |
| 140 dpi)), | |
| 141 static_cast<int>(ConvertUnitDouble(page_size.height(), kPointsPerInch, | |
| 142 dpi))); | |
| 143 } | |
| 144 | |
| 145 if (content_area_in_dpi) { | |
| 146 *content_area_in_dpi = gfx::Rect( | |
| 147 static_cast<int>(ConvertUnitDouble(content_area.x(), kPointsPerInch, | |
| 148 dpi)), | |
| 149 static_cast<int>(ConvertUnitDouble(content_area.y(), kPointsPerInch, | |
| 150 dpi)), | |
| 151 static_cast<int>(ConvertUnitDouble(content_area.width(), kPointsPerInch, | |
| 152 dpi)), | |
| 153 static_cast<int>(ConvertUnitDouble(content_area.height(), | |
| 154 kPointsPerInch, dpi))); | |
| 155 } | |
| 156 | |
| 157 if (!is_preview) { | |
| 158 // Since WebKit extends the page width depending on the magical scale factor | |
| 159 // we make sure the canvas covers the worst case scenario (x2.0 currently). | |
| 160 // PrintContext will then set the correct clipping region. | |
| 161 page_size = gfx::Size( | |
| 162 static_cast<int>(page_layout_in_points.content_width * | |
| 163 params.max_shrink), | |
| 164 static_cast<int>(page_layout_in_points.content_height * | |
| 165 params.max_shrink)); | |
| 166 } | |
| 167 | |
| 168 float webkit_page_shrink_factor = frame->getPrintPageShrink(page_number); | |
| 169 float scale_factor = css_scale_factor * webkit_page_shrink_factor; | |
| 170 | |
| 171 gfx::Rect canvas_area = | |
| 172 params.display_header_footer ? gfx::Rect(page_size) : content_area; | |
| 173 | |
| 174 SkBaseDevice* device = metafile->StartPageForVectorCanvas( | |
| 175 page_size, canvas_area, scale_factor); | |
| 176 DCHECK(device); | |
| 177 // The printPage method may take a reference to the canvas we pass down, so it | |
| 178 // can't be a stack object. | |
| 179 skia::RefPtr<skia::VectorCanvas> canvas = | |
| 180 skia::AdoptRef(new skia::VectorCanvas(device)); | |
| 181 | |
| 182 if (is_preview) { | |
| 183 MetafileSkiaWrapper::SetMetafileOnCanvas(*canvas, metafile); | |
| 184 skia::SetIsDraftMode(*canvas, is_print_ready_metafile_sent_); | |
| 185 skia::SetIsPreviewMetafile(*canvas, is_preview); | |
| 186 } | |
| 187 | |
| 188 if (params.display_header_footer) { | |
| 189 // |page_number| is 0-based, so 1 is added. | |
| 190 PrintHeaderAndFooter(canvas.get(), page_number + 1, | |
| 191 print_preview_context_.total_page_count(), scale_factor, | |
| 192 page_layout_in_points, *header_footer_info_, params); | |
| 193 } | |
| 194 | |
| 195 float webkit_scale_factor = RenderPageContent(frame, page_number, canvas_area, | |
| 196 content_area, scale_factor, | |
| 197 canvas.get()); | |
| 198 | |
| 199 if (*actual_shrink <= 0 || webkit_scale_factor <= 0) { | |
| 200 NOTREACHED() << "Printing page " << page_number << " failed."; | |
| 201 } else { | |
| 202 // While rendering certain plugins (PDF) to metafile, we might need to | |
| 203 // set custom scale factor. Update |actual_shrink| with custom scale | |
| 204 // if it is set on canvas. | |
| 205 // TODO(gene): We should revisit this solution for the next versions. | |
| 206 // Consider creating metafile of the right size (or resizable) | |
| 207 // https://code.google.com/p/chromium/issues/detail?id=126037 | |
| 208 if (!MetafileSkiaWrapper::GetCustomScaleOnCanvas( | |
| 209 *canvas, actual_shrink)) { | |
| 210 // Update the dpi adjustment with the "page |actual_shrink|" calculated in | |
| 211 // webkit. | |
| 212 *actual_shrink /= (webkit_scale_factor * css_scale_factor); | |
| 213 } | |
| 214 } | |
| 215 | |
| 216 bool result = metafile->FinishPage(); | |
| 217 DCHECK(result); | |
| 218 } | |
| 219 | |
| 220 bool PrintWebViewHelper::CopyMetafileDataToSharedMem( | |
| 221 Metafile* metafile, base::SharedMemoryHandle* shared_mem_handle) { | |
| 222 uint32 buf_size = metafile->GetDataSize(); | |
| 223 base::SharedMemory shared_buf; | |
| 224 if (buf_size >= kMetafileMaxSize) { | |
| 225 NOTREACHED() << "Buffer too large: " << buf_size; | |
| 226 return false; | |
| 227 } | |
| 228 | |
| 229 // Allocate a shared memory buffer to hold the generated metafile data. | |
| 230 if (!shared_buf.CreateAndMapAnonymous(buf_size)) { | |
| 231 NOTREACHED() << "Buffer allocation failed"; | |
| 232 return false; | |
| 233 } | |
| 234 | |
| 235 // Copy the bits into shared memory. | |
| 236 if (!metafile->GetData(shared_buf.memory(), buf_size)) { | |
| 237 NOTREACHED() << "GetData() failed"; | |
| 238 shared_buf.Unmap(); | |
| 239 return false; | |
| 240 } | |
| 241 shared_buf.GiveToProcess(base::GetCurrentProcessHandle(), shared_mem_handle); | |
| 242 shared_buf.Unmap(); | |
| 243 | |
| 244 Send(new PrintHostMsg_DuplicateSection(routing_id(), *shared_mem_handle, | |
| 245 shared_mem_handle)); | |
| 246 return true; | |
| 247 } | |
| 248 | |
| 249 } // namespace printing | |
| OLD | NEW |