src/chrome/renderer/printing/print_web_view_helper_linux.cc

   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
  25 namespace printing {
  26
  27 using WebKit::WebFrame;
  28
  29 bool 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
  60 bool 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
 151 void 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