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