diff options
| -rw-r--r-- | printing/printing_context_win.cc | 169 |
1 files changed, 120 insertions, 49 deletions
diff --git a/printing/printing_context_win.cc b/printing/printing_context_win.cc index 884ae37..74de8ba 100644 --- a/printing/printing_context_win.cc +++ b/printing/printing_context_win.cc @@ -19,6 +19,63 @@ using base::Time; +namespace { + +// Retrieves the printer's PRINTER_INFO_* structure. +// Output |level| can be 9 (user-default), 8 (admin-default), or 2 +// (printer-default). +// |devmode| is a pointer points to the start of DEVMODE structure in +// |buffer|. +bool GetPrinterInfo(HANDLE printer, + const std::wstring &device_name, + int* level, + scoped_array<uint8>* buffer, + DEVMODE** dev_mode) { + DCHECK(buffer); + + // A PRINTER_INFO_9 structure specifying the per-user default printer + // settings. + printing::PrintingContextWin::GetPrinterHelper(printer, 9, buffer); + if (buffer->get()) { + PRINTER_INFO_9* info_9 = reinterpret_cast<PRINTER_INFO_9*>(buffer->get()); + if (info_9->pDevMode != NULL) { + *level = 9; + *dev_mode = info_9->pDevMode; + return true; + } + buffer->reset(); + } + + // A PRINTER_INFO_8 structure specifying the global default printer settings. + printing::PrintingContextWin::GetPrinterHelper(printer, 8, buffer); + if (buffer->get()) { + PRINTER_INFO_8* info_8 = reinterpret_cast<PRINTER_INFO_8*>(buffer->get()); + if (info_8->pDevMode != NULL) { + *level = 8; + *dev_mode = info_8->pDevMode; + return true; + } + buffer->reset(); + } + + // A PRINTER_INFO_2 structure specifying the driver's default printer + // settings. + printing::PrintingContextWin::GetPrinterHelper(printer, 2, buffer); + if (buffer->get()) { + PRINTER_INFO_2* info_2 = reinterpret_cast<PRINTER_INFO_2*>(buffer->get()); + if (info_2->pDevMode != NULL) { + *level = 2; + *dev_mode = info_2->pDevMode; + return true; + } + buffer->reset(); + } + + return false; +} + +} // anonymous namespace + namespace printing { class PrintingContextWin::CallbackHandler : public IPrintDialogCallback, @@ -244,20 +301,70 @@ PrintingContext::Result PrintingContextWin::UseDefaultSettings() { } PrintingContext::Result PrintingContextWin::UpdatePrintSettings( - const DictionaryValue& job_settings, const PageRanges& ranges) { + const DictionaryValue& job_settings, + const PageRanges& ranges) { DCHECK(!in_print_job_); - bool landscape; - if (!job_settings.GetBoolean(kSettingLandscape, &landscape)) + bool is_landscape, is_collate, is_color; + string16 printer_name; + int copies, duplex_mode; + if (!job_settings.GetBoolean(kSettingLandscape, &is_landscape) || + !job_settings.GetString(kSettingPrinterName, &printer_name) || + !job_settings.GetInteger(kSettingCopies, &copies) || + !job_settings.GetBoolean(kSettingCollate, &is_collate) || + !job_settings.GetInteger(kSettingDuplexMode, &duplex_mode) || + !job_settings.GetBoolean(kSettingColor, &is_color)) { return OnError(); + } - settings_.SetOrientation(landscape); + // Underlying |settings_| do not have these attributes, so we need to + // operate on printer directly, which involves reloading settings. + // Therefore, reset the settings anyway. + ResetSettings(); - // TODO(kmadhusu): Update other print settings such as number of copies, - // collate, duplex printing, job title, etc., + HANDLE printer; + if (!OpenPrinter(const_cast<wchar_t*>(printer_name.c_str()), + &printer, NULL)) { + return OnError(); + } - settings_.ranges = ranges; + scoped_array<uint8> buffer; + int level; + DEVMODE* dev_mode = NULL; + if (!GetPrinterInfo(printer, printer_name, &level, &buffer, &dev_mode) || + dev_mode == NULL) { + ClosePrinter(printer); + return OnError(); + } + dev_mode->dmColor = is_color ? DMCOLOR_COLOR : DMCOLOR_MONOCHROME; + dev_mode->dmCopies = std::max(copies, 1); + if (dev_mode->dmCopies > 1) // do not change collate unless multiple copies + dev_mode->dmCollate = is_collate ? DMCOLLATE_TRUE : DMCOLLATE_FALSE; + switch (duplex_mode) { + case LONG_EDGE: + dev_mode->dmDuplex = DMDUP_VERTICAL; + break; + case SHORT_EDGE: + dev_mode->dmDuplex = DMDUP_HORIZONTAL; + break; + default: // simplex + dev_mode->dmDuplex = DMDUP_SIMPLEX; + break; + } + dev_mode->dmOrientation = is_landscape ? DMORIENT_LANDSCAPE : + DMORIENT_PORTRAIT; + + // Set printer then refresh printer settings. + if (!SetPrinter(printer, level, buffer.get(), 0) || + !AllocateContext(printer_name, dev_mode, &context_)) { + ClosePrinter(printer); + return OnError(); + } + PrintSettingsInitializerWin::InitPrintSettings(context_, *dev_mode, + ranges, printer_name, + false, &settings_); + ClosePrinter(printer); return OK; } @@ -457,51 +564,15 @@ bool PrintingContextWin::GetPrinterSettings(HANDLE printer, const std::wstring& device_name) { DCHECK(!in_print_job_); scoped_array<uint8> buffer; + int level = 0; + DEVMODE* dev_mode = NULL; - // A PRINTER_INFO_9 structure specifying the per-user default printer - // settings. - GetPrinterHelper(printer, 9, &buffer); - if (buffer.get()) { - PRINTER_INFO_9* info_9 = reinterpret_cast<PRINTER_INFO_9*>(buffer.get()); - if (info_9->pDevMode != NULL) { - if (!AllocateContext(device_name, info_9->pDevMode, &context_)) { - ResetSettings(); - return false; - } - return InitializeSettings(*info_9->pDevMode, device_name, NULL, 0, false); - } - buffer.reset(); - } - - // A PRINTER_INFO_8 structure specifying the global default printer settings. - GetPrinterHelper(printer, 8, &buffer); - if (buffer.get()) { - PRINTER_INFO_8* info_8 = reinterpret_cast<PRINTER_INFO_8*>(buffer.get()); - if (info_8->pDevMode != NULL) { - if (!AllocateContext(device_name, info_8->pDevMode, &context_)) { - ResetSettings(); - return false; - } - return InitializeSettings(*info_8->pDevMode, device_name, NULL, 0, false); - } - buffer.reset(); + if (GetPrinterInfo(printer, device_name, &level, &buffer, &dev_mode) && + AllocateContext(device_name, dev_mode, &context_)) { + return InitializeSettings(*dev_mode, device_name, NULL, 0, false); } - // A PRINTER_INFO_2 structure specifying the driver's default printer - // settings. - GetPrinterHelper(printer, 2, &buffer); - if (buffer.get()) { - PRINTER_INFO_2* info_2 = reinterpret_cast<PRINTER_INFO_2*>(buffer.get()); - if (info_2->pDevMode != NULL) { - if (!AllocateContext(device_name, info_2->pDevMode, &context_)) { - ResetSettings(); - return false; - } - return InitializeSettings(*info_2->pDevMode, device_name, NULL, 0, false); - } - buffer.reset(); - } - // Failed to retrieve the printer settings. + buffer.reset(); ResetSettings(); return false; } |