Convert PDF to PWG Raster using utility process.

chrome/utility/chrome_content_utility_client.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/utility/chrome_content_utility_client.h"
 
 #include "base/base64.h"
 #include "base/bind.h"
 #include "base/command_line.h"
 #include "base/file_util.h"
 #include "base/files/file_path.h"
 #include "base/json/json_reader.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/path_service.h"
 #include "base/scoped_native_library.h"
 #include "base/time/time.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/common/chrome_utility_messages.h"
 #include "chrome/common/extensions/chrome_extensions_client.h"
 #include "chrome/common/extensions/extension_l10n_util.h"
 #include "chrome/common/extensions/update_manifest.h"
 #include "chrome/common/safe_browsing/zip_analyzer.h"
+#include "chrome/utility/cloud_print/bitmap_image.h"
+#include "chrome/utility/cloud_print/pwg_encoder.h"
 #include "chrome/utility/extensions/unpacker.h"
 #include "chrome/utility/profile_import_handler.h"
 #include "chrome/utility/web_resource_unpacker.h"
 #include "content/public/child/image_decoder_utils.h"
 #include "content/public/common/content_paths.h"
 #include "content/public/utility/utility_thread.h"
 #include "extensions/common/extension.h"
 #include "extensions/common/manifest.h"
 #include "media/base/media.h"
 #include "media/base/media_file_checker.h"
@@ skipping 286 matching lines @@
   IPC_BEGIN_MESSAGE_MAP(ChromeContentUtilityClient, message)
     IPC_MESSAGE_HANDLER(ChromeUtilityMsg_UnpackExtension, OnUnpackExtension)
     IPC_MESSAGE_HANDLER(ChromeUtilityMsg_UnpackWebResource,
                         OnUnpackWebResource)
     IPC_MESSAGE_HANDLER(ChromeUtilityMsg_ParseUpdateManifest,
                         OnParseUpdateManifest)
     IPC_MESSAGE_HANDLER(ChromeUtilityMsg_DecodeImage, OnDecodeImage)
     IPC_MESSAGE_HANDLER(ChromeUtilityMsg_DecodeImageBase64, OnDecodeImageBase64)
     IPC_MESSAGE_HANDLER(ChromeUtilityMsg_RenderPDFPagesToMetafile,
                         OnRenderPDFPagesToMetafile)
+    IPC_MESSAGE_HANDLER(ChromeUtilityMsg_RenderPDFPagesToPWGRaster,
+                        OnRenderPDFPagesToPWGRaster)
     IPC_MESSAGE_HANDLER(ChromeUtilityMsg_RobustJPEGDecodeImage,
                         OnRobustJPEGDecodeImage)
     IPC_MESSAGE_HANDLER(ChromeUtilityMsg_ParseJSON, OnParseJSON)
     IPC_MESSAGE_HANDLER(ChromeUtilityMsg_GetPrinterCapsAndDefaults,
                         OnGetPrinterCapsAndDefaults)
     IPC_MESSAGE_HANDLER(ChromeUtilityMsg_StartupPing, OnStartupPing)
     IPC_MESSAGE_HANDLER(ChromeUtilityMsg_AnalyzeZipFileForDownloadProtection,
                         OnAnalyzeZipFileForDownloadProtection)
 
 #if !defined(OS_ANDROID) && !defined(OS_IOS)
@@ skipping 182 matching lines @@
     Send(new ChromeUtilityHostMsg_RenderPDFPagesToMetafile_Succeeded(
         highest_rendered_page_number, scale_factor));
   }
 #endif  // defined(OS_WIN)
   if (!succeeded) {
     Send(new ChromeUtilityHostMsg_RenderPDFPagesToMetafile_Failed());
   }
   ReleaseProcessIfNeeded();
 }
 
+void ChromeContentUtilityClient::OnRenderPDFPagesToPWGRaster(
+    IPC::PlatformFileForTransit pdf_transit,
+    const printing::PdfRenderSettings& settings,
+    IPC::PlatformFileForTransit bitmap_transit) {
+  base::PlatformFile pdf =
+      IPC::PlatformFileForTransitToPlatformFile(pdf_transit);
+  base::PlatformFile bitmap =
+      IPC::PlatformFileForTransitToPlatformFile(bitmap_transit);
+  if (RenderPDFPagesToPWGRaster(pdf, settings, bitmap)) {
+    Send(new ChromeUtilityHostMsg_RenderPDFPagesToPWGRaster_Succeeded());
+  } else {
+    Send(new ChromeUtilityHostMsg_RenderPDFPagesToPWGRaster_Failed());
+  }
+  ReleaseProcessIfNeeded();
+}
+
 #if defined(OS_WIN)
 bool ChromeContentUtilityClient::RenderPDFToWinMetafile(
     base::PlatformFile pdf_file,
     const base::FilePath& metafile_path,
     const printing::PdfRenderSettings& settings,
     const std::vector<printing::PageRange>& page_ranges,
     int* highest_rendered_page_number,
     double* scale_factor) {
   *highest_rendered_page_number = -1;
   *scale_factor = 1.0;
@@ skipping 57 matching lines @@
         ret = true;
       }
       metafile.FinishPage();
     }
   }
   metafile.FinishDocument();
   return ret;
 }
 #endif  // defined(OS_WIN)
 
+bool ChromeContentUtilityClient::RenderPDFPagesToPWGRaster(
+    base::PlatformFile pdf_file,
+    const printing::PdfRenderSettings& settings,
+    base::PlatformFile bitmap_file) {
+  bool autoupdate = true;
+  if (!g_pdf_lib.Get().IsValid())
+    return false;
+
+  base::PlatformFileInfo info;
+  if (!base::GetPlatformFileInfo(pdf_file, &info) || info.size <= 0)
+    return false;
+
+  std::vector<char> data(info.size, 0);
+  data.resize(base::ReadPlatformFile(pdf_file, 0, &data[0], data.size()));

[Lei Zhang, 2013/11/21 20:51:04]

Can't ReadPlatformFile() fail for some reason and return -1?

[Vitaly Buka (NO REVIEWS), 2013/11/22 00:13:00]

Done.

+
+  int total_page_count = 0;
+  if (!g_pdf_lib.Get().GetPDFDocInfo(&data.front(), data.size(),
+                                     &total_page_count, NULL)) {
+    return false;
+  }
+
+  cloud_print::PwgEncoder encoder;
+  std::string pwg_header;
+  encoder.EncodeDocumentHeader(&pwg_header);
+  if (!base::WritePlatformFile(bitmap_file, 0, &pwg_header.front(),
+                               pwg_header.size())) {
+    return false;
+  }
+
+  cloud_print::BitmapImage image(settings.area().size(),
+                                 cloud_print::BitmapImage::BGRA);
+  for (int i = 0; i < total_page_count; ++i) {
+    if (!g_pdf_lib.Get().RenderPDFPageToBitmap(
+             &data.front(), data.size(), i, image.pixel_data(),
+             image.size().width(), image.size().height(), settings.dpi(),
+             settings.dpi(), autoupdate)) {
+      return false;
+    }
+    std::string pwg_page;
+    if (!encoder.EncodePage(image, settings.dpi(), total_page_count,
+                            &pwg_page) ||
+        !base::WritePlatformFile(bitmap_file, 0, &pwg_page.front(),

[Lei Zhang, 2013/11/21 20:51:04]

Doesn't this overwrite |bitmap_file| on every loop?

[Vitaly Buka (NO REVIEWS), 2013/11/22 00:13:00]

File is opened in APPEND mode, so offset is ignored.

but WritePlatformFileAtCurrentPos without APPEND is more appropriate here.

+                                 pwg_page.size())) {
+      return false;
+    }
+  }
+  return true;
+}
+
 void ChromeContentUtilityClient::OnRobustJPEGDecodeImage(
     const std::vector<unsigned char>& encoded_data) {
   // Our robust jpeg decoding is using IJG libjpeg.
   if (gfx::JPEGCodec::JpegLibraryVariant() == gfx::JPEGCodec::IJG_LIBJPEG) {
     scoped_ptr<SkBitmap> decoded_image(gfx::JPEGCodec::Decode(
         &encoded_data[0], encoded_data.size()));
     if (!decoded_image.get() || decoded_image->empty()) {
       Send(new ChromeUtilityHostMsg_DecodeImage_Failed());
     } else {
       Send(new ChromeUtilityHostMsg_DecodeImage_Succeeded(*decoded_image));
@@ skipping 135 matching lines @@
   picasa::PicasaAlbumsIndexer indexer(album_uids);
   indexer.ParseFolderINI(folders_inis);
 
   Send(new ChromeUtilityHostMsg_IndexPicasaAlbumsContents_Finished(
       indexer.albums_images()));
   ReleaseProcessIfNeeded();
 }
 #endif  // defined(OS_WIN) || defined(OS_MACOSX)
 
 }  // namespace chrome