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#import <AppKit/AppKit.h>
8
9#include "base/logging.h"
10#include "base/mac/scoped_nsautorelease_pool.h"
11#include "base/metrics/histogram.h"
12#include "chrome/common/print_messages.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/platform/WebCanvas.h"
18#include "third_party/WebKit/public/web/WebLocalFrame.h"
19
20namespace printing {
21
22using blink::WebFrame;
23
24void PrintWebViewHelper::PrintPageInternal(
25    const PrintMsg_PrintPage_Params& params,
26    WebFrame* frame) {
27  PdfMetafileSkia metafile;
28  if (!metafile.Init())
29    return;
30
31  int page_number = params.page_number;
32  gfx::Size page_size_in_dpi;
33  gfx::Rect content_area_in_dpi;
34  RenderPage(print_pages_params_->params, page_number, frame, false, &metafile,
35             &page_size_in_dpi, &content_area_in_dpi);
36  metafile.FinishDocument();
37
38  PrintHostMsg_DidPrintPage_Params page_params;
39  page_params.data_size = metafile.GetDataSize();
40  page_params.page_number = page_number;
41  page_params.document_cookie = params.params.document_cookie;
42  page_params.page_size = page_size_in_dpi;
43  page_params.content_area = content_area_in_dpi;
44
45  // Ask the browser to create the shared memory for us.
46  if (!CopyMetafileDataToSharedMem(&metafile,
47                                   &(page_params.metafile_data_handle))) {
48    page_params.data_size = 0;
49  }
50
51  Send(new PrintHostMsg_DidPrintPage(routing_id(), page_params));
52}
53
54bool PrintWebViewHelper::RenderPreviewPage(
55    int page_number,
56    const PrintMsg_Print_Params& print_params) {
57  PrintMsg_Print_Params printParams = print_params;
58  scoped_ptr<PdfMetafileSkia> draft_metafile;
59  PdfMetafileSkia* initial_render_metafile = print_preview_context_.metafile();
60
61  bool render_to_draft = print_preview_context_.IsModifiable() &&
62                         is_print_ready_metafile_sent_;
63
64  if (render_to_draft) {
65    draft_metafile.reset(new PdfMetafileSkia());
66    if (!draft_metafile->Init()) {
67      print_preview_context_.set_error(
68          PREVIEW_ERROR_MAC_DRAFT_METAFILE_INIT_FAILED);
69      LOG(ERROR) << "Draft PdfMetafileSkia Init failed";
70      return false;
71    }
72    initial_render_metafile = draft_metafile.get();
73  }
74
75  base::TimeTicks begin_time = base::TimeTicks::Now();
76  gfx::Size page_size;
77  RenderPage(printParams, page_number, print_preview_context_.prepared_frame(),
78             true, initial_render_metafile, &page_size, NULL);
79  print_preview_context_.RenderedPreviewPage(
80      base::TimeTicks::Now() - begin_time);
81
82  if (draft_metafile.get()) {
83    draft_metafile->FinishDocument();
84  } else {
85    if (print_preview_context_.IsModifiable() &&
86        print_preview_context_.generate_draft_pages()) {
87      DCHECK(!draft_metafile.get());
88      draft_metafile =
89          print_preview_context_.metafile()->GetMetafileForCurrentPage();
90    }
91  }
92  return PreviewPageRendered(page_number, draft_metafile.get());
93}
94
95void PrintWebViewHelper::RenderPage(const PrintMsg_Print_Params& params,
96                                    int page_number,
97                                    WebFrame* frame,
98                                    bool is_preview,
99                                    PdfMetafileSkia* metafile,
100                                    gfx::Size* page_size,
101                                    gfx::Rect* content_rect) {
102  double scale_factor = 1.0f;
103  double webkit_shrink_factor = frame->getPrintPageShrink(page_number);
104  PageSizeMargins page_layout_in_points;
105  gfx::Rect content_area;
106
107  ComputePageLayoutInPointsForCss(frame, page_number, params,
108                                  ignore_css_margins_, &scale_factor,
109                                  &page_layout_in_points);
110  GetPageSizeAndContentAreaFromPageLayout(page_layout_in_points, page_size,
111                                          &content_area);
112  if (content_rect)
113    *content_rect = content_area;
114
115  scale_factor *= webkit_shrink_factor;
116
117  gfx::Rect canvas_area =
118      params.display_header_footer ? gfx::Rect(*page_size) : content_area;
119
120  {
121    SkBaseDevice* device = metafile->StartPageForVectorCanvas(
122        *page_size, canvas_area, scale_factor);
123    if (!device)
124      return;
125
126    skia::RefPtr<skia::VectorCanvas> canvas =
127        skia::AdoptRef(new skia::VectorCanvas(device));
128    blink::WebCanvas* canvas_ptr = canvas.get();
129    MetafileSkiaWrapper::SetMetafileOnCanvas(*canvas, metafile);
130    skia::SetIsDraftMode(*canvas, is_print_ready_metafile_sent_);
131    skia::SetIsPreviewMetafile(*canvas, is_preview);
132
133    if (params.display_header_footer) {
134      PrintHeaderAndFooter(canvas_ptr,
135                           page_number + 1,
136                           print_preview_context_.total_page_count(),
137                           *frame,
138                           scale_factor,
139                           page_layout_in_points,
140                           params);
141    }
142    RenderPageContent(frame, page_number, canvas_area, content_area,
143                      scale_factor, canvas_ptr);
144  }
145
146  // Done printing. Close the device context to retrieve the compiled metafile.
147  metafile->FinishPage();
148}
149
150}  // namespace printing
151