Apply paper size selection for local printers on OSX.

printing/printing_context_mac.mm

 // Copyright (c) 2011 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 "printing/printing_context_mac.h"
 
 #import <ApplicationServices/ApplicationServices.h>
 #import <AppKit/AppKit.h>
 
 #import <iomanip>
 #import <numeric>
 
 #include "base/logging.h"
 #include "base/mac/scoped_cftyperef.h"
 #include "base/mac/scoped_nsautorelease_pool.h"
 #include "base/mac/scoped_nsexception_enabler.h"
 #include "base/strings/sys_string_conversions.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/values.h"
 #include "printing/print_settings_initializer_mac.h"
 #include "printing/units.h"
 
 namespace printing {
 
 namespace {
 
 // Return true if PPD name of paper is equal.
-bool IsPaperNameEqual(const PMPaper& paper1, const PMPaper& paper2) {
-  CFStringRef name1 = NULL;
+bool IsPaperNameEqual(CFStringRef name1, const PMPaper& paper2) {
   CFStringRef name2 = NULL;
-  return (PMPaperGetPPDPaperName(paper1, &name1) == noErr) &&
-         (PMPaperGetPPDPaperName(paper2, &name2) == noErr) &&
-         (CFStringCompare(name1, name2,
-                          kCFCompareCaseInsensitive) == kCFCompareEqualTo);
+  return (PMPaperGetPPDPaperName(paper2, &name2) == noErr) &&
+         (CFStringCompare(name1, name2, kCFCompareCaseInsensitive) ==
+          kCFCompareEqualTo);
+}
+
+PMPaper MatchPaper(CFArrayRef paper_list,
+                   CFStringRef name,
+                   double width,
+                   double height) {
+  double best_match = std::numeric_limits<double>::max();
+  PMPaper best_matching_paper = kPMNoData;
+  int num_papers = CFArrayGetCount(paper_list);
+  for (int i = 0; i < num_papers; ++i) {
+    PMPaper paper = (PMPaper)[(NSArray*)paper_list objectAtIndex : i];
+    double paper_width = 0.0;
+    double paper_height = 0.0;
+    PMPaperGetWidth(paper, &paper_width);
+    PMPaperGetHeight(paper, &paper_height);
+    double current_match =
+        std::max(fabs(width - paper_width), fabs(height - paper_height));
+    // Ignore paper sizes that are very different.
+    if (current_match > 2)
+      continue;
+    current_match += IsPaperNameEqual(name, paper) ? 0 : 1;
+    if (current_match < best_match) {
+      best_matching_paper = paper;
+      best_match = current_match;
+    }
+  }
+  return best_matching_paper;
 }
 
 }  // namespace
 
 // static
 PrintingContext* PrintingContext::Create(const std::string& app_locale) {
   return static_cast<PrintingContext*>(new PrintingContextMac(app_locale));
 }
 
 PrintingContextMac::PrintingContextMac(const std::string& app_locale)
@@ skipping 177 matching lines @@
   return status == noErr;
 }
 
 bool PrintingContextMac::UpdatePageFormatWithPaperInfo() {
   PMPrintSession print_session =
       static_cast<PMPrintSession>([print_info_.get() PMPrintSession]);
 
   PMPageFormat default_page_format =
       static_cast<PMPageFormat>([print_info_.get() PMPageFormat]);
 
-  PMPaper default_paper;
-  if (PMGetPageFormatPaper(default_page_format, &default_paper) != noErr)
-    return false;
-
-  double default_page_width = 0.0;
-  double default_page_height = 0.0;
-  if (PMPaperGetWidth(default_paper, &default_page_width) != noErr)
-    return false;
-
-  if (PMPaperGetHeight(default_paper, &default_page_height) != noErr)
-    return false;
-
   PMPrinter current_printer = NULL;
   if (PMSessionGetCurrentPrinter(print_session, &current_printer) != noErr)
     return false;
 
   if (current_printer == nil)

[Aleksey Shlyapnikov, 2014/06/18 17:45:01]

nil -> NULL?

[Vitaly Buka (NO REVIEWS), 2014/06/19 00:39:08]

should be nil

     return false;
 
   CFArrayRef paper_list = NULL;
   if (PMPrinterGetPaperList(current_printer, &paper_list) != noErr)
     return false;

[Aleksey Shlyapnikov, 2014/06/18 17:45:02]

Move down to the first use.

[Vitaly Buka (NO REVIEWS), 2014/06/19 00:39:09]

Done.

 
-  double best_match = std::numeric_limits<double>::max();
-  PMPaper best_matching_paper = kPMNoData;
-  int num_papers = CFArrayGetCount(paper_list);
-  for (int i = 0; i < num_papers; ++i) {
-    PMPaper paper = (PMPaper)[(NSArray*)paper_list objectAtIndex: i];
-    double paper_width = 0.0;
-    double paper_height = 0.0;
-    PMPaperGetWidth(paper, &paper_width);
-    PMPaperGetHeight(paper, &paper_height);
-    double current_match = std::max(fabs(default_page_width - paper_width),
-                                    fabs(default_page_height - paper_height));
-    // Ignore paper sizes that are very different.
-    if (current_match > 2)
-      continue;
-    current_match += IsPaperNameEqual(paper, default_paper) ? 0 : 1;
-    if (current_match < best_match) {
-      best_matching_paper = paper;
-      best_match = current_match;
+  const PrintSettings::RequestedMedia& media = settings_.requested_media();
+
+  double page_width = 0.0;
+  double page_height = 0.0;
+  CFStringRef paper_name = NULL;
+  PMPaperMargins margins = {0};
+
+  if (media.size_microns.IsEmpty()) {
+    PMPaper default_paper;
+    if (PMGetPageFormatPaper(default_page_format, &default_paper) != noErr ||
+        PMPaperGetPPDPaperName(default_paper, &paper_name) != noErr ||
+        PMPaperGetWidth(default_paper, &page_width) != noErr ||
+        PMPaperGetHeight(default_paper, &page_height) != noErr ||
+        PMPaperGetMargins(default_paper, &margins) != noErr) {
+      return false;
     }
+  } else {
+    const double kMutiplier = kPointsPerInch / (10.0f * kHundrethsMMPerInch);
+    page_width = media.size_microns.width() * kMutiplier;
+    page_height = media.size_microns.height() * kMutiplier;
+    paper_name = base::SysUTF8ToCFStringRef(media.vendor_id);
   }
 
+  PMPaper best_matching_paper =
+      MatchPaper(paper_list, paper_name, page_width, page_height);
+
   if (best_matching_paper == kPMNoData) {
     PMPaper paper = kPMNoData;
-    // Create a custom paper for the specified default page size.
-    PMPaperMargins default_margins;
-    if (PMPaperGetMargins(default_paper, &default_margins) != noErr)
-      return false;
-
-    const PMPaperMargins margins =
-        {default_margins.top, default_margins.left, default_margins.bottom,
-         default_margins.right};
     CFStringRef paper_id = CFSTR("Custom paper ID");
     CFStringRef paper_name = CFSTR("Custom paper");
-    if (PMPaperCreateCustom(current_printer, paper_id, paper_name,
-            default_page_width, default_page_height, &margins, &paper) !=
-            noErr) {
+    if (PMPaperCreateCustom(current_printer,
+                            paper_id,
+                            paper_name,
+                            page_width,
+                            page_height,
+                            &margins,
+                            &paper) != noErr) {

[Aleksey Shlyapnikov, 2014/06/18 17:45:01]

What for paper is created here? It's released on the next line.

[Vitaly Buka (NO REVIEWS), 2014/06/19 00:39:08]

Done.

       return false;
     }
-    [print_info_.get() updateFromPMPageFormat];
     PMRelease(paper);
   } else {
     PMPageFormat chosen_page_format = NULL;
     if (PMCreatePageFormat((PMPageFormat*) &chosen_page_format) != noErr)
       return false;
 
     // Create page format from that paper.
     if (PMCreatePageFormatWithPMPaper(&chosen_page_format,
             best_matching_paper) != noErr) {
       PMRelease(chosen_page_format);
       return false;
     }
     // Copy over the original format with the new page format.
     if (PMCopyPageFormat(chosen_page_format, default_page_format) != noErr) {
       PMRelease(chosen_page_format);
       return false;
     }
-    [print_info_.get() updateFromPMPageFormat];
     PMRelease(chosen_page_format);
   }
+  [print_info_.get() updateFromPMPageFormat];
   return true;
 }
 
 bool PrintingContextMac::SetCopiesInPrintSettings(int copies) {
   if (copies < 1)
     return false;
 
   PMPrintSettings pmPrintSettings =
       static_cast<PMPrintSettings>([print_info_.get() PMPrintSettings]);
   return PMSetCopies(pmPrintSettings, copies, false) == noErr;
@@ skipping 175 matching lines @@
 void PrintingContextMac::ReleaseContext() {
   print_info_.reset();
   context_ = NULL;
 }
 
 gfx::NativeDrawingContext PrintingContextMac::context() const {
   return context_;
 }
 
 }  // namespace printing