Home | History | Annotate | Download | only in printing 
      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 using WebKit::WebNode;
     29 
     30 bool PrintWebViewHelper::RenderPreviewPage(
     31     int page_number,
     32     const PrintMsg_Print_Params& print_params) {
     33   PrintMsg_PrintPage_Params page_params;
     34   page_params.params = print_params;
     35   page_params.page_number = page_number;
     36   scoped_ptr<Metafile> draft_metafile;
     37   Metafile* initial_render_metafile = print_preview_context_.metafile();
     38   if (print_preview_context_.IsModifiable() && is_print_ready_metafile_sent_) {
     39     draft_metafile.reset(new PreviewMetafile);
     40     initial_render_metafile = draft_metafile.get();
     41   }
     42 
     43   base::TimeTicks begin_time = base::TimeTicks::Now();
     44   PrintPageInternal(page_params,
     45                     print_preview_context_.GetPrintCanvasSize(),
     46                     print_preview_context_.prepared_frame(),
     47                     initial_render_metafile);
     48   print_preview_context_.RenderedPreviewPage(
     49       base::TimeTicks::Now() - begin_time);
     50   if (draft_metafile.get()) {
     51     draft_metafile->FinishDocument();
     52   } else if (print_preview_context_.IsModifiable() &&
     53              print_preview_context_.generate_draft_pages()) {
     54     DCHECK(!draft_metafile.get());
     55     draft_metafile.reset(
     56         print_preview_context_.metafile()->GetMetafileForCurrentPage());
     57   }
     58   return PreviewPageRendered(page_number, draft_metafile.get());
     59 }
     60 
     61 bool PrintWebViewHelper::PrintPagesNative(WebKit::WebFrame* frame,
     62                                           const WebKit::WebNode& node,
     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   // WebKit::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)
    151 }
    152 
    153 void 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   SkDevice* device = metafile->StartPageForVectorCanvas(page_size, canvas_area,
    171                                                         scale_factor);
    172   if (!device)
    173     return;
    174 
    175   // The printPage method take a reference to the canvas we pass down, so it
    176   // can't be a stack object.
    177   skia::RefPtr<skia::VectorCanvas> canvas =
    178       skia::AdoptRef(new skia::VectorCanvas(device));
    179   MetafileSkiaWrapper::SetMetafileOnCanvas(*canvas, metafile);
    180   skia::SetIsDraftMode(*canvas, is_print_ready_metafile_sent_);
    181 
    182   if (params.params.display_header_footer) {
    183     // |page_number| is 0-based, so 1 is added.
    184     // TODO(vitalybuka) : why does it work only with 1.25?
    185     PrintHeaderAndFooter(canvas.get(), params.page_number + 1,
    186                          print_preview_context_.total_page_count(),
    187                          scale_factor / 1.25,
    188                          page_layout_in_points, *header_footer_info_,
    189                          params.params);
    190   }
    191   RenderPageContent(frame, params.page_number, canvas_area, content_area,
    192                     scale_factor, canvas.get());
    193 
    194   // Done printing. Close the device context to retrieve the compiled metafile.
    195   if (!metafile->FinishPage())
    196     NOTREACHED() << "metafile failed";
    197 }
    198 
    199 }  // namespace printing
    200