print_web_view_helper_linux.cc revision 4e180b6a0b4720a9b8e9e959a882386f690f08ff 
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/WebFrame.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
25namespace printing {
26
27using WebKit::WebFrame;
28
29bool 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
60bool PrintWebViewHelper::PrintPagesNative(WebKit::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  // WebKit::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
151void 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