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   if (gtk_print_settings_get_printer_lpi (settings) == 0.0)
 215     gtk_print_settings_set_printer_lpi (settings, 150.0);
 216
 217   cairo_surface_set_fallback_resolution (surface,
 218                                          2.0 * gtk_print_settings_get_printer_lpi (settings),
 219                                          2.0 * gtk_print_settings_get_printer_lpi (settings));
 220
 221   return surface;
 222 }
 223
 224 typedef struct {
 225   GtkPrintBackend *backend;
 226   GtkPrintJobCompleteFunc callback;
 227   GtkPrintJob *job;
 228   gpointer user_data;
 229   GDestroyNotify dnotify;
 230
 231   GIOChannel *in;
 232 } _PrintStreamData;
 233
 234 static void
 235 lpr_print_cb (GtkPrintBackendLpr *print_backend,
 236               GError             *error,
 237               gpointer            user_data)
 238 {
 239   _PrintStreamData *ps = (_PrintStreamData *) user_data;
 240
 241   if (ps->in != NULL)
 242     g_io_channel_unref (ps->in);
 243
 244   if (ps->callback)
 245     ps->callback (ps->job, ps->user_data, error);
 246
 247   if (ps->dnotify)
 248     ps->dnotify (ps->user_data);
 249
 250   gtk_print_job_set_status (ps->job,
 251                             error ? GTK_PRINT_STATUS_FINISHED_ABORTED
 252                                   : GTK_PRINT_STATUS_FINISHED);
 253
 254   if (ps->job)
 255     g_object_unref (ps->job);
 256
 257   g_free (ps);
 258 }
 259
 260 static gboolean
 261 lpr_write (GIOChannel   *source,
 262            GIOCondition  con,
 263            gpointer      user_data)
 264 {
 265   gchar buf[_LPR_MAX_CHUNK_SIZE];
 266   gsize bytes_read;
 267   GError *error;
 268   GIOStatus status;
 269   _PrintStreamData *ps = (_PrintStreamData *) user_data;
 270
 271   error = NULL;
 272
 273   status =
 274     g_io_channel_read_chars (source,
 275                              buf,
 276                              _LPR_MAX_CHUNK_SIZE,
 277                              &bytes_read,
 278                              &error);
 279
 280   if (status != G_IO_STATUS_ERROR)
 281     {
 282       gsize bytes_written;
 283
 284       g_io_channel_write_chars (ps->in,
 285                                 buf,
 286                                 bytes_read,
 287                                 &bytes_written,
 288                                 &error);
 289     }
 290
 291   if (error != NULL || status == G_IO_STATUS_EOF)
 292     {
 293       lpr_print_cb (GTK_PRINT_BACKEND_LPR (ps->backend),
 294                     error, user_data);
 295
 296
 297       if (error != NULL)
 298         {
 299           GTK_NOTE (PRINTING,
 300                     g_print ("LPR Backend: %s\n", error->message));
 301
 302           g_error_free (error);
 303         }
 304
 305       return FALSE;
 306     }
 307
 308   GTK_NOTE (PRINTING,
 309             g_print ("LPR Backend: Writting %i byte chunk to lpr pipe\n", bytes_read));
 310
 311
 312   return TRUE;
 313 }
 314
 315 #define LPR_COMMAND "lpr"
 316
 317 static void
 318 gtk_print_backend_lpr_print_stream (GtkPrintBackend        *print_backend,
 319                                     GtkPrintJob            *job,
 320                                     GIOChannel             *data_io,
 321                                     GtkPrintJobCompleteFunc callback,
 322                                     gpointer                user_data,
 323                                     GDestroyNotify          dnotify)
 324 {
 325   GError *print_error = NULL;
 326   GtkPrinter *printer;
 327   _PrintStreamData *ps;
 328   GtkPrintSettings *settings;
 329   gint argc;
 330   gint in_fd;
 331   gchar **argv = NULL;
 332   const char *cmd_line;
 333
 334   printer = gtk_print_job_get_printer (job);
 335   settings = gtk_print_job_get_settings (job);
 336
 337   cmd_line = gtk_print_settings_get (settings, "lpr-commandline");
 338   if (cmd_line == NULL)
 339     cmd_line = LPR_COMMAND;
 340
 341   ps = g_new0 (_PrintStreamData, 1);
 342   ps->callback = callback;
 343   ps->user_data = user_data;
 344   ps->dnotify = dnotify;
 345   ps->job = g_object_ref (job);
 346   ps->in = NULL;
 347
 348  /* spawn lpr with pipes and pipe ps file to lpr */
 349   if (!g_shell_parse_argv (cmd_line, &argc, &argv, &print_error))
 350     goto out;
 351
 352   if (!g_spawn_async_with_pipes (NULL,
 353                                  argv,
 354                                  NULL,
 355                                  G_SPAWN_SEARCH_PATH,
 356                                  NULL,
 357                                  NULL,
 358                                  NULL,
 359                                  &in_fd,
 360                                  NULL,
 361                                  NULL,
 362                                  &print_error))
 363       goto out;
 364
 365   ps->in = g_io_channel_unix_new (in_fd);
 366
 367   g_io_channel_set_encoding (ps->in, NULL, &print_error);
 368   if (print_error != NULL)
 369     {
 370       if (ps->in != NULL)
 371         g_io_channel_unref (ps->in);
 372
 373       goto out;
 374     }
 375
 376   g_io_channel_set_close_on_unref (ps->in, TRUE);
 377
 378   g_io_add_watch (data_io,
 379                   G_IO_IN | G_IO_PRI | G_IO_ERR | G_IO_HUP,
 380                   (GIOFunc) lpr_write,
 381                   ps);
 382
 383  out:
 384   if (argv != NULL)
 385     g_strfreev (argv);
 386
 387   if (print_error != NULL)
 388     {
 389       lpr_print_cb (GTK_PRINT_BACKEND_LPR (print_backend),
 390                     print_error, ps);
 391       g_error_free (print_error);
 392     }
 393 }
 394
 395 static void
 396 gtk_print_backend_lpr_init (GtkPrintBackendLpr *backend)
 397 {
 398   GtkPrinter *printer;
 399
 400   printer = gtk_printer_new (_("Print to LPR"),
 401                              GTK_PRINT_BACKEND (backend),
 402                              TRUE);
 403   gtk_printer_set_has_details (printer, TRUE);
 404   gtk_printer_set_icon_name (printer, "gtk-print");
 405   gtk_printer_set_is_active (printer, TRUE);
 406   gtk_printer_set_is_default (printer, TRUE);
 407
 408   gtk_print_backend_add_printer (GTK_PRINT_BACKEND (backend), printer);
 409   g_object_unref (printer);
 410   gtk_print_backend_set_list_done (GTK_PRINT_BACKEND (backend));
 411 }
 412
 413 static GtkPrinterOptionSet *
 414 lpr_printer_get_options (GtkPrinter           *printer,
 415                          GtkPrintSettings     *settings,
 416                          GtkPageSetup         *page_setup,
 417                          GtkPrintCapabilities  capabilities)
 418 {
 419   GtkPrinterOptionSet *set;
 420   GtkPrinterOption *option;
 421   const char *command;
 422   char *n_up[] = {"1", "2", "4", "6", "9", "16" };
 423
 424   set = gtk_printer_option_set_new ();
 425
 426   option = gtk_printer_option_new ("gtk-n-up", _("Pages Per Sheet"), GTK_PRINTER_OPTION_TYPE_PICKONE);
 427   gtk_printer_option_choices_from_array (option, G_N_ELEMENTS (n_up),
 428                                          n_up, n_up);
 429   gtk_printer_option_set (option, "1");
 430   gtk_printer_option_set_add (set, option);
 431   g_object_unref (option);
 432
 433   option = gtk_printer_option_new ("gtk-main-page-custom-input", _("Command Line"), GTK_PRINTER_OPTION_TYPE_STRING);
 434   option->group = g_strdup ("GtkPrintDialogExtension");
 435   if (settings != NULL &&
 436       (command = gtk_print_settings_get (settings, "lpr-commandline"))!= NULL)
 437     gtk_printer_option_set (option, command);
 438   else
 439     gtk_printer_option_set (option, LPR_COMMAND);
 440   gtk_printer_option_set_add (set, option);
 441
 442   return set;
 443 }
 444
 445 static void
 446 lpr_printer_get_settings_from_options (GtkPrinter          *printer,
 447                                        GtkPrinterOptionSet *options,
 448                                        GtkPrintSettings    *settings)
 449 {
 450   GtkPrinterOption *option;
 451
 452   option = gtk_printer_option_set_lookup (options, "gtk-main-page-custom-input");
 453   gtk_print_settings_set (settings, "lpr-commandline", option->value);
 454 }
 455
 456 static void
 457 lpr_printer_prepare_for_print (GtkPrinter       *printer,
 458                                GtkPrintJob      *print_job,
 459                                GtkPrintSettings *settings,
 460                                GtkPageSetup     *page_setup)
 461 {
 462   double scale;
 463
 464   print_job->print_pages = gtk_print_settings_get_print_pages (settings);
 465   print_job->page_ranges = NULL;
 466   print_job->num_page_ranges = 0;
 467
 468   if (print_job->print_pages == GTK_PRINT_PAGES_RANGES)
 469     print_job->page_ranges =
 470       gtk_print_settings_get_page_ranges (settings,
 471                                           &print_job->num_page_ranges);
 472
 473   print_job->collate = gtk_print_settings_get_collate (settings);
 474   print_job->reverse = gtk_print_settings_get_reverse (settings);
 475   print_job->num_copies = gtk_print_settings_get_n_copies (settings);
 476
 477   scale = gtk_print_settings_get_scale (settings);
 478   if (scale != 100.0)
 479     print_job->scale = scale/100.0;
 480
 481   print_job->page_set = gtk_print_settings_get_page_set (settings);
 482   print_job->rotate_to_orientation = TRUE;
 483 }