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