Pileus Git - ~andy/gtk/blob - modules/printbackends/lpr/gtkprintbackendlpr.c

   1 /* GTK - The GIMP Toolkit
   2  * gtkprintbackendlpr.c: LPR implementation of GtkPrintBackend
   3  * for printing to lpr
   4  * Copyright (C) 2006, 2007 Red Hat, Inc.
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the
  18  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  19  * Boston, MA 02111-1307, USA.
  20  */
  21
  22 #include "config.h"
  23 #include <unistd.h>
  24 #include <sys/types.h>
  25 #include <sys/stat.h>
  26 #include <fcntl.h>
  27 #include <stdlib.h>
  28 #include <string.h>
  29
  30 #include <errno.h>
  31 #include <cairo.h>
  32 #include <cairo-ps.h>
  33
  34 #include <glib/gi18n-lib.h>
  35
  36 #include <gtk/gtk.h>
  37 #include "gtkprinter-private.h"
  38
  39 #include "gtkprintbackendlpr.h"
  40
  41 typedef struct _GtkPrintBackendLprClass GtkPrintBackendLprClass;
  42
  43 #define GTK_PRINT_BACKEND_LPR_CLASS(klass)     (G_TYPE_CHECK_CLASS_CAST ((klass), GTK_TYPE_PRINT_BACKEND_LPR, GtkPrintBackendLprClass))
  44 #define GTK_IS_PRINT_BACKEND_LPR_CLASS(klass)  (G_TYPE_CHECK_CLASS_TYPE ((klass), GTK_TYPE_PRINT_BACKEND_LPR))
  45 #define GTK_PRINT_BACKEND_LPR_GET_CLASS(obj)   (G_TYPE_INSTANCE_GET_CLASS ((obj), GTK_TYPE_PRINT_BACKEND_LPR, GtkPrintBackendLprClass))
  46
  47 #define _LPR_MAX_CHUNK_SIZE 8192
  48
  49 static GType print_backend_lpr_type = 0;
  50
  51 struct _GtkPrintBackendLprClass
  52 {
  53   GtkPrintBackendClass parent_class;
  54 };
  55
  56 struct _GtkPrintBackendLpr
  57 {
  58   GtkPrintBackend parent_instance;
  59 };
  60
  61 static GObjectClass *backend_parent_class;
  62
  63 static void                 gtk_print_backend_lpr_class_init      (GtkPrintBackendLprClass *class);
  64 static void                 gtk_print_backend_lpr_init            (GtkPrintBackendLpr      *impl);
  65 static void                 lpr_printer_get_settings_from_options (GtkPrinter              *printer,
  66                                                                    GtkPrinterOptionSet     *options,
  67                                                                    GtkPrintSettings        *settings);
  68 static GtkPrinterOptionSet *lpr_printer_get_options               (GtkPrinter              *printer,
  69                                                                    GtkPrintSettings        *settings,
  70                                                                    GtkPageSetup            *page_setup,
  71                                                                    GtkPrintCapabilities     capabilities);
  72 static void                 lpr_printer_prepare_for_print         (GtkPrinter              *printer,
  73                                                                    GtkPrintJob             *print_job,
  74                                                                    GtkPrintSettings        *settings,
  75                                                                    GtkPageSetup            *page_setup);
  76 static cairo_surface_t *    lpr_printer_create_cairo_surface      (GtkPrinter              *printer,
  77                                                                    GtkPrintSettings        *settings,
  78                                                                    gdouble                  width,
  79                                                                    gdouble                  height,
  80                                                                    GIOChannel              *cache_io);
  81 static void                 gtk_print_backend_lpr_print_stream    (GtkPrintBackend         *print_backend,
  82                                                                    GtkPrintJob             *job,
  83                                                                    GIOChannel              *data_io,
  84                                                                    GtkPrintJobCompleteFunc  callback,
  85                                                                    gpointer                 user_data,
  86                                                                    GDestroyNotify           dnotify);
  87
  88 static void
  89 gtk_print_backend_lpr_register_type (GTypeModule *module)
  90 {
  91   static const GTypeInfo print_backend_lpr_info =
  92   {
  93     sizeof (GtkPrintBackendLprClass),
  94     NULL,               /* base_init */
  95     NULL,               /* base_finalize */
  96     (GClassInitFunc) gtk_print_backend_lpr_class_init,
  97     NULL,               /* class_finalize */
  98     NULL,               /* class_data */
  99     sizeof (GtkPrintBackendLpr),
 100     0,          /* n_preallocs */
 101     (GInstanceInitFunc) gtk_print_backend_lpr_init,
 102   };
 103
 104   print_backend_lpr_type = g_type_module_register_type (module,
 105                                                         GTK_TYPE_PRINT_BACKEND,
 106                                                         "GtkPrintBackendLpr",
 107                                                         &print_backend_lpr_info, 0);
 108 }
 109
 110 G_MODULE_EXPORT void
 111 pb_module_init (GTypeModule *module)
 112 {
 113   gtk_print_backend_lpr_register_type (module);
 114 }
 115
 116 G_MODULE_EXPORT void
 117 pb_module_exit (void)
 118 {
 119
 120 }
 121
 122 G_MODULE_EXPORT GtkPrintBackend *
 123 pb_module_create (void)
 124 {
 125   return gtk_print_backend_lpr_new ();
 126 }
 127
 128 /*
 129  * GtkPrintBackendLpr
 130  */
 131 GType
 132 gtk_print_backend_lpr_get_type (void)
 133 {
 134   return print_backend_lpr_type;
 135 }
 136
 137 /**
 138  * gtk_print_backend_lpr_new:
 139  *
 140  * Creates a new #GtkPrintBackendLpr object. #GtkPrintBackendLpr
 141  * implements the #GtkPrintBackend interface with direct access to
 142  * the filesystem using Unix/Linux API calls
 143  *
 144  * Return value: the new #GtkPrintBackendLpr object
 145  **/
 146 GtkPrintBackend *
 147 gtk_print_backend_lpr_new (void)
 148 {
 149   return g_object_new (GTK_TYPE_PRINT_BACKEND_LPR, NULL);
 150 }
 151
 152 static void
 153 gtk_print_backend_lpr_class_init (GtkPrintBackendLprClass *class)
 154 {
 155   GtkPrintBackendClass *backend_class = GTK_PRINT_BACKEND_CLASS (class);
 156
 157   backend_parent_class = g_type_class_peek_parent (class);
 158
 159   backend_class->print_stream = gtk_print_backend_lpr_print_stream;
 160   backend_class->printer_create_cairo_surface = lpr_printer_create_cairo_surface;
 161   backend_class->printer_get_options = lpr_printer_get_options;
 162   backend_class->printer_get_settings_from_options = lpr_printer_get_settings_from_options;
 163   backend_class->printer_prepare_for_print = lpr_printer_prepare_for_print;
 164 }
 165
 166 static cairo_status_t
 167 _cairo_write (void                *closure,
 168               const unsigned char *data,
 169               unsigned int         length)
 170 {
 171   GIOChannel *io = (GIOChannel *)closure;
 172   gsize written;
 173   GError *error;
 174
 175   error = NULL;
 176
 177   GTK_NOTE (PRINTING,
 178             g_print ("LPR Backend: Writting %i byte chunk to temp file\n", length));
 179
 180   while (length > 0)
 181     {
 182       g_io_channel_write_chars (io, data, length, &written, &error);
 183
 184       if (error != NULL)
 185         {
 186           GTK_NOTE (PRINTING,
 187                      g_print ("LPR Backend: Error writting to temp file, %s\n", error->message));
 188
 189           g_error_free (error);
 190           return CAIRO_STATUS_WRITE_ERROR;
 191         }
 192
 193       GTK_NOTE (PRINTING,
 194                 g_print ("LPR Backend: Wrote %i bytes to temp file\n", written));
 195
 196       data += written;
 197       length -= written;
 198     }
 199
 200   return CAIRO_STATUS_SUCCESS;
 201 }
 202
 203 static cairo_surface_t *
 204 lpr_printer_create_cairo_surface (GtkPrinter       *printer,
 205                                   GtkPrintSettings *settings,
 206                                   gdouble           width,
 207                                   gdouble           height,
 208                                   GIOChannel       *cache_io)
 209 {
 210   cairo_surface_t *surface;
 211
 212   surface = cairo_ps_surface_create_for_stream (_cairo_write, cache_io, width, height);
 213
 214   /* TODO: DPI from settings object? */
 215   cairo_surface_set_fallback_resolution (surface, 300, 300);
 216
 217   return surface;
 218 }
 219
 220 typedef struct {
 221   GtkPrintBackend *backend;
 222   GtkPrintJobCompleteFunc callback;
 223   GtkPrintJob *job;
 224   gpointer user_data;
 225   GDestroyNotify dnotify;
 226
 227   GIOChannel *in;
 228 } _PrintStreamData;
 229
 230 static void
 231 lpr_print_cb (GtkPrintBackendLpr *print_backend,
 232               GError             *error,
 233               gpointer            user_data)
 234 {
 235   _PrintStreamData *ps = (_PrintStreamData *) user_data;
 236
 237   if (ps->in != NULL)
 238     g_io_channel_unref (ps->in);
 239
 240   if (ps->callback)
 241     ps->callback (ps->job, ps->user_data, error);
 242
 243   if (ps->dnotify)
 244     ps->dnotify (ps->user_data);
 245
 246   gtk_print_job_set_status (ps->job,
 247                             error ? GTK_PRINT_STATUS_FINISHED_ABORTED
 248                                   : GTK_PRINT_STATUS_FINISHED);
 249
 250   if (ps->job)
 251     g_object_unref (ps->job);
 252
 253   g_free (ps);
 254 }
 255
 256 static gboolean
 257 lpr_write (GIOChannel   *source,
 258            GIOCondition  con,
 259            gpointer      user_data)
 260 {
 261   gchar buf[_LPR_MAX_CHUNK_SIZE];
 262   gsize bytes_read;
 263   GError *error;
 264   GIOStatus status;
 265   _PrintStreamData *ps = (_PrintStreamData *) user_data;
 266
 267   error = NULL;
 268
 269   status =
 270     g_io_channel_read_chars (source,
 271                              buf,
 272                              _LPR_MAX_CHUNK_SIZE,
 273                              &bytes_read,
 274                              &error);
 275
 276   if (status != G_IO_STATUS_ERROR)
 277     {
 278       gsize bytes_written;
 279
 280       g_io_channel_write_chars (ps->in,
 281                                 buf,
 282                                 bytes_read,
 283                                 &bytes_written,
 284                                 &error);
 285     }
 286
 287   if (error != NULL || status == G_IO_STATUS_EOF)
 288     {
 289       lpr_print_cb (GTK_PRINT_BACKEND_LPR (ps->backend),
 290                     error, user_data);
 291
 292       if (error)
 293         g_error_free (error);
 294
 295       if (error != NULL)
 296         {
 297           GTK_NOTE (PRINTING,
 298                     g_print ("LPR Backend: %s\n", error->message));
 299
 300           g_error_free (error);
 301         }
 302
 303       return FALSE;
 304     }
 305
 306   GTK_NOTE (PRINTING,
 307             g_print ("LPR Backend: Writting %i byte chunk to lpr pipe\n", bytes_read));
 308
 309
 310   return TRUE;
 311 }
 312
 313 #define LPR_COMMAND "lpr"
 314
 315 static void
 316 gtk_print_backend_lpr_print_stream (GtkPrintBackend        *print_backend,
 317                                     GtkPrintJob            *job,
 318                                     GIOChannel             *data_io,
 319                                     GtkPrintJobCompleteFunc callback,
 320                                     gpointer                user_data,
 321                                     GDestroyNotify          dnotify)
 322 {
 323   GError *print_error = NULL;
 324   GtkPrinter *printer;
 325   _PrintStreamData *ps;
 326   GtkPrintSettings *settings;
 327   gint argc;
 328   gint in_fd;
 329   gchar **argv = NULL;
 330   const char *cmd_line;
 331
 332   printer = gtk_print_job_get_printer (job);
 333   settings = gtk_print_job_get_settings (job);
 334
 335   cmd_line = gtk_print_settings_get (settings, "lpr-commandline");
 336   if (cmd_line == NULL)
 337     cmd_line = LPR_COMMAND;
 338
 339   ps = g_new0 (_PrintStreamData, 1);
 340   ps->callback = callback;
 341   ps->user_data = user_data;
 342   ps->dnotify = dnotify;
 343   ps->job = g_object_ref (job);
 344   ps->in = NULL;
 345
 346  /* spawn lpr with pipes and pipe ps file to lpr */
 347   if (!g_shell_parse_argv (cmd_line, &argc, &argv, &print_error))
 348     goto out;
 349
 350   if (!g_spawn_async_with_pipes (NULL,
 351                                  argv,
 352                                  NULL,
 353                                  G_SPAWN_SEARCH_PATH,
 354                                  NULL,
 355                                  NULL,
 356                                  NULL,
 357                                  &in_fd,
 358                                  NULL,
 359                                  NULL,
 360                                  &print_error))
 361       goto out;
 362
 363   ps->in = g_io_channel_unix_new (in_fd);
 364
 365   g_io_channel_set_encoding (ps->in, NULL, &print_error);
 366   if (print_error != NULL)
 367     {
 368       if (ps->in != NULL)
 369         g_io_channel_unref (ps->in);
 370
 371       goto out;
 372     }
 373
 374   g_io_channel_set_close_on_unref (ps->in, TRUE);
 375
 376   g_io_add_watch (data_io,
 377                   G_IO_IN | G_IO_PRI | G_IO_ERR | G_IO_HUP,
 378                   (GIOFunc) lpr_write,
 379                   ps);
 380
 381  out:
 382   if (argv != NULL)
 383     g_strfreev (argv);
 384
 385   if (print_error != NULL)
 386     {
 387       lpr_print_cb (GTK_PRINT_BACKEND_LPR (print_backend),
 388                     print_error, ps);
 389       g_error_free (print_error);
 390     }
 391 }
 392
 393 static void
 394 gtk_print_backend_lpr_init (GtkPrintBackendLpr *backend)
 395 {
 396   GtkPrinter *printer;
 397
 398   printer = gtk_printer_new (_("Print to LPR"),
 399                              GTK_PRINT_BACKEND (backend),
 400                              TRUE);
 401   gtk_printer_set_has_details (printer, TRUE);
 402   gtk_printer_set_icon_name (printer, "gtk-print");
 403   gtk_printer_set_is_active (printer, TRUE);
 404   gtk_printer_set_is_default (printer, TRUE);
 405
 406   gtk_print_backend_add_printer (GTK_PRINT_BACKEND (backend), printer);
 407   g_object_unref (printer);
 408   gtk_print_backend_set_list_done (GTK_PRINT_BACKEND (backend));
 409 }
 410
 411 static GtkPrinterOptionSet *
 412 lpr_printer_get_options (GtkPrinter           *printer,
 413                          GtkPrintSettings     *settings,
 414                          GtkPageSetup         *page_setup,
 415                          GtkPrintCapabilities  capabilities)
 416 {
 417   GtkPrinterOptionSet *set;
 418   GtkPrinterOption *option;
 419   const char *command;
 420   char *n_up[] = {"1", "2", "4", "6", "9", "16" };
 421
 422   set = gtk_printer_option_set_new ();
 423
 424   option = gtk_printer_option_new ("gtk-n-up", _("Pages Per Sheet"), GTK_PRINTER_OPTION_TYPE_PICKONE);
 425   gtk_printer_option_choices_from_array (option, G_N_ELEMENTS (n_up),
 426                                          n_up, n_up);
 427   gtk_printer_option_set (option, "1");
 428   gtk_printer_option_set_add (set, option);
 429   g_object_unref (option);
 430
 431   option = gtk_printer_option_new ("gtk-main-page-custom-input", _("Command Line"), GTK_PRINTER_OPTION_TYPE_STRING);
 432   option->group = g_strdup ("GtkPrintDialogExtension");
 433   if (settings != NULL &&
 434       (command = gtk_print_settings_get (settings, "lpr-commandline"))!= NULL)
 435     gtk_printer_option_set (option, command);
 436   else
 437     gtk_printer_option_set (option, LPR_COMMAND);
 438   gtk_printer_option_set_add (set, option);
 439
 440   return set;
 441 }
 442
 443 static void
 444 lpr_printer_get_settings_from_options (GtkPrinter          *printer,
 445                                        GtkPrinterOptionSet *options,
 446                                        GtkPrintSettings    *settings)
 447 {
 448   GtkPrinterOption *option;
 449
 450   option = gtk_printer_option_set_lookup (options, "gtk-main-page-custom-input");
 451   gtk_print_settings_set (settings, "lpr-commandline", option->value);
 452 }
 453
 454 static void
 455 lpr_printer_prepare_for_print (GtkPrinter       *printer,
 456                                GtkPrintJob      *print_job,
 457                                GtkPrintSettings *settings,
 458                                GtkPageSetup     *page_setup)
 459 {
 460   double scale;
 461
 462   print_job->print_pages = gtk_print_settings_get_print_pages (settings);
 463   print_job->page_ranges = NULL;
 464   print_job->num_page_ranges = 0;
 465
 466   if (print_job->print_pages == GTK_PRINT_PAGES_RANGES)
 467     print_job->page_ranges =
 468       gtk_print_settings_get_page_ranges (settings,
 469                                           &print_job->num_page_ranges);
 470
 471   print_job->collate = gtk_print_settings_get_collate (settings);
 472   print_job->reverse = gtk_print_settings_get_reverse (settings);
 473   print_job->num_copies = gtk_print_settings_get_n_copies (settings);
 474
 475   scale = gtk_print_settings_get_scale (settings);
 476   if (scale != 100.0)
 477     print_job->scale = scale/100.0;
 478
 479   print_job->page_set = gtk_print_settings_get_page_set (settings);
 480   print_job->rotate_to_orientation = TRUE;
 481 }