gtk/src/main.c

   1 /* -*- Mode: C; indent-tabs-mode: t; c-basic-offset: 4; tab-width: 4 -*- */
   2 /*
   3  * main.c
   4  * Copyright (C) John Stebbins 2008 <stebbins@stebbins>
   5  *
   6  * main.c is free software.
   7  *
   8  * You may redistribute it and/or modify it under the terms of the
   9  * GNU General Public License, as published by the Free Software
  10  * Foundation; either version 2 of the License, or (at your option)
  11  * any later version.
  12  *
  13  * main.c is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  16  * See the GNU General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * along with main.c.  If not, write to:
  20  *      The Free Software Foundation, Inc.,
  21  *      51 Franklin Street, Fifth Floor
  22  *      Boston, MA  02110-1301, USA.
  23  */
  24
  25 #include <sys/types.h>
  26 #include <sys/stat.h>
  27 #include <unistd.h>
  28 #include <string.h>
  29 #include <stdio.h>
  30 #include <stdlib.h>
  31 #include <fcntl.h>
  32
  33 #include <config.h>
  34
  35 #include <gtk/gtk.h>
  36 #include <glib/gstdio.h>
  37 #include <gio/gio.h>
  38 #include "renderer_button.h"
  39 #include "hb-backend.h"
  40 #include "ghb-dvd.h"
  41
  42
  43 /*
  44  * Standard gettext macros.
  45  */
  46 #ifdef ENABLE_NLS
  47 #  include <libintl.h>
  48 #  undef _
  49 #  define _(String) dgettext (PACKAGE, String)
  50 #  ifdef gettext_noop
  51 #    define N_(String) gettext_noop (String)
  52 #  else
  53 #    define N_(String) (String)
  54 #  endif
  55 #else
  56 #  define textdomain(String) (String)
  57 #  define gettext(String) (String)
  58 #  define dgettext(Domain,Message) (Message)
  59 #  define dcgettext(Domain,Message,Type) (Message)
  60 #  define bindtextdomain(Domain,Directory) (Domain)
  61 #  define _(String) (String)
  62 #  define N_(String) (String)
  63 #endif
  64
  65
  66
  67 #include "callbacks.h"
  68 #include "settings.h"
  69
  70 #define BUILDER_NAME "ghb"
  71
  72 static gchar*
  73 find_builder_file(gchar *builder_name)
  74 {
  75         const gchar *dir;
  76         const gchar * const *dirs;
  77         gchar *file;
  78         gint ii;
  79
  80         file = g_strdup_printf("./%s.ui", builder_name);
  81         if (g_file_test(file, G_FILE_TEST_IS_REGULAR))
  82         {
  83                 return file;
  84         }
  85         g_free(file);
  86         // The following resolves to /$HOME/.local/share currently
  87         dir = g_get_user_data_dir();
  88         file = g_strdup_printf("%s/%s/builder/%s.ui", dir, builder_name, builder_name);
  89         if (g_file_test(file, G_FILE_TEST_IS_REGULAR))
  90         {
  91                 return file;
  92         }
  93         g_free(file);
  94         dirs = g_get_system_data_dirs();
  95         if (dirs != NULL)
  96         {
  97                 for (ii = 0; dirs[ii] != NULL; ii++)
  98                 {
  99                         file = g_strdup_printf("%s/%s/builder/%s.ui", dirs[ii], builder_name, builder_name);
 100                         if (g_file_test(file, G_FILE_TEST_IS_REGULAR))
 101                         {
 102                                 return file;
 103                         }
 104                         g_free(file);
 105                 }
 106         }
 107         return NULL;
 108 }
 109
 110 GtkBuilder*
 111 create_builder_or_die(const gchar * name,
 112                     const gchar * path)
 113 {
 114         guint res;
 115     const gchar *markup =
 116         N_("<b><big>Unable to create %s.</big></b>\n"
 117         "\n"
 118         "Could not open Gtk Builder file (%s). Please check your "
 119         "installation.\n");
 120         g_debug("create_builder_or_die ()\n");
 121         GtkBuilder *xml = gtk_builder_new();
 122         if (xml != NULL)
 123                 res = gtk_builder_add_from_file(xml, path, NULL);
 124     if (!xml || !res)
 125         {
 126         GtkWidget *dialog = gtk_message_dialog_new_with_markup(NULL,
 127             GTK_DIALOG_MODAL,
 128             GTK_MESSAGE_ERROR,
 129             GTK_BUTTONS_CLOSE,
 130             _(markup),
 131             name, path);
 132         gtk_dialog_run(GTK_DIALOG(dialog));
 133         gtk_widget_destroy(dialog);
 134         exit(EXIT_FAILURE);
 135     }
 136     return xml;
 137 }
 138
 139 static GCallback
 140 self_symbol_lookup(const gchar * symbol_name)
 141 {
 142     static GModule *module = NULL;
 143     gpointer symbol = NULL;
 144
 145     if (!module)
 146         module = g_module_open(NULL, 0);
 147
 148     g_module_symbol(module, symbol_name, &symbol);
 149     return (GCallback) symbol;
 150 }
 151
 152
 153 static void
 154 MyConnect(
 155         GtkBuilder *builder,
 156         GObject *object,
 157         const gchar *signal_name,
 158         const gchar *handler_name,
 159         GObject *connect_object,
 160         GConnectFlags flags,
 161         gpointer user_data)
 162 {
 163         GCallback callback;
 164
 165     g_return_if_fail(object != NULL);
 166     g_return_if_fail(handler_name != NULL);
 167     g_return_if_fail(signal_name != NULL);
 168
 169         g_debug("handler_name %s\n", handler_name);
 170         g_debug("signal_name %s\n", signal_name);
 171     callback = self_symbol_lookup(handler_name);
 172     if (!callback)
 173         {
 174         g_message("Signal handler (%s) not found", handler_name);
 175         return;
 176     }
 177     if (connect_object)
 178         {
 179         g_signal_connect_object(object, signal_name, callback, connect_object, flags);
 180     }
 181     else
 182         {
 183         if (flags & G_CONNECT_AFTER)
 184                 {
 185             g_signal_connect_after( object, signal_name, callback, user_data);
 186                 }
 187         else
 188                 {
 189             g_signal_connect(object, signal_name, callback, user_data);
 190                 }
 191     }
 192 }
 193
 194 #if 0
 195 // If you should ever need to change the font for the running application..
 196 // Ugly but effective.
 197 void
 198 change_font(GtkWidget *widget, gpointer data)
 199 {
 200     PangoFontDescription *font_desc;
 201     gchar *font = (gchar*)data;
 202     const gchar *name;
 203
 204     font_desc = pango_font_description_from_string(font);
 205     if (font_desc == NULL) exit(1);
 206     gtk_widget_modify_font(widget, font_desc);
 207     name = gtk_widget_get_name(widget);
 208     g_debug("changing font for widget %s\n", name);
 209     if (GTK_IS_CONTAINER(widget))
 210     {
 211         gtk_container_foreach((GtkContainer*)widget, change_font, data);
 212     }
 213 }
 214     //gtk_container_foreach((GtkContainer*)window, change_font, "sans 20");
 215 #endif
 216
 217 extern void chapter_list_selection_changed_cb(void);
 218 extern void chapter_edited_cb(void);
 219
 220 // Create and bind the tree model to the tree view for the chapter list
 221 // Also, connect up the signal that lets us know the selection has changed
 222 static void
 223 bind_chapter_tree_model (signal_user_data_t *ud)
 224 {
 225         GtkCellRenderer *cell;
 226         GtkTreeViewColumn *column;
 227         GtkListStore *treestore;
 228         GtkTreeView  *treeview;
 229         GtkTreeSelection *selection;
 230
 231         g_debug("bind_chapter_tree_model ()\n");
 232         treeview = GTK_TREE_VIEW(GHB_WIDGET (ud->builder, "chapters_list"));
 233         selection = gtk_tree_view_get_selection (treeview);
 234         treestore = gtk_list_store_new(3, G_TYPE_INT, G_TYPE_STRING, G_TYPE_BOOLEAN);
 235         gtk_tree_view_set_model(treeview, GTK_TREE_MODEL(treestore));
 236
 237         cell = gtk_cell_renderer_text_new();
 238         column = gtk_tree_view_column_new_with_attributes(
 239                                                                         _("Chapter No."), cell, "text", 0, NULL);
 240     gtk_tree_view_append_column(treeview, GTK_TREE_VIEW_COLUMN(column));
 241
 242         cell = gtk_cell_renderer_text_new();
 243         column = gtk_tree_view_column_new_with_attributes(
 244                                         _("Chapter Title"), cell, "text", 1, "editable", 2, NULL);
 245     gtk_tree_view_append_column(treeview, GTK_TREE_VIEW_COLUMN(column));
 246         g_signal_connect(cell, "edited", chapter_edited_cb, ud);
 247
 248         g_signal_connect(selection, "changed", chapter_list_selection_changed_cb, ud);
 249         g_debug("Done\n");
 250 }
 251
 252 extern void queue_list_selection_changed_cb(void);
 253 extern void queue_remove_clicked_cb(void);
 254 extern void queue_list_size_allocate_cb(void);
 255
 256 // Create and bind the tree model to the tree view for the queue list
 257 // Also, connect up the signal that lets us know the selection has changed
 258 static void
 259 bind_queue_tree_model (signal_user_data_t *ud)
 260 {
 261         GtkCellRenderer *cell, *textcell;
 262         GtkTreeViewColumn *column;
 263         GtkTreeStore *treestore;
 264         GtkTreeView  *treeview;
 265         GtkTreeSelection *selection;
 266
 267         g_debug("bind_queue_tree_model ()\n");
 268         treeview = GTK_TREE_VIEW(GHB_WIDGET (ud->builder, "queue_list"));
 269         selection = gtk_tree_view_get_selection (treeview);
 270         treestore = gtk_tree_store_new(3, G_TYPE_STRING, G_TYPE_STRING, G_TYPE_STRING);
 271         gtk_tree_view_set_model(treeview, GTK_TREE_MODEL(treestore));
 272
 273         column = gtk_tree_view_column_new();
 274         gtk_tree_view_column_set_title (column, _("Job Information"));
 275         cell = gtk_cell_renderer_pixbuf_new();
 276         g_object_set(cell, "yalign", 0.0, NULL);
 277         gtk_tree_view_column_pack_start (column, cell, FALSE);
 278         gtk_tree_view_column_add_attribute (column, cell, "icon-name", 0);
 279         textcell = gtk_cell_renderer_text_new();
 280         g_object_set(textcell, "wrap-mode", PANGO_WRAP_CHAR, NULL);
 281         g_object_set(textcell, "wrap-width", 500, NULL);
 282         gtk_tree_view_column_pack_start (column, textcell, TRUE);
 283         gtk_tree_view_column_add_attribute (column, textcell, "markup", 1);
 284     gtk_tree_view_append_column(treeview, GTK_TREE_VIEW_COLUMN(column));
 285         gtk_tree_view_column_set_expand (column, TRUE);
 286         gtk_tree_view_column_set_max_width (column, 550);
 287
 288         cell = custom_cell_renderer_button_new();
 289         g_object_set(cell, "yalign", 0.0, NULL);
 290         column = gtk_tree_view_column_new_with_attributes(
 291                                                                         _(""), cell, "icon-name", 2, NULL);
 292         gtk_tree_view_column_set_min_width (column, 24);
 293     gtk_tree_view_append_column(treeview, GTK_TREE_VIEW_COLUMN(column));
 294
 295         g_signal_connect(selection, "changed", queue_list_selection_changed_cb, ud);
 296         g_signal_connect(cell, "clicked", queue_remove_clicked_cb, ud);
 297         g_signal_connect(treeview, "size-allocate", queue_list_size_allocate_cb, textcell);
 298 }
 299
 300 extern void audio_list_selection_changed_cb(void);
 301
 302 // Create and bind the tree model to the tree view for the audio track list
 303 // Also, connect up the signal that lets us know the selection has changed
 304 static void
 305 bind_audio_tree_model (signal_user_data_t *ud)
 306 {
 307         GtkCellRenderer *cell;
 308         GtkTreeViewColumn *column;
 309         GtkListStore *treestore;
 310         GtkTreeView  *treeview;
 311         GtkTreeSelection *selection;
 312         GtkWidget *widget;
 313
 314         g_debug("bind_audio_tree_model ()\n");
 315         treeview = GTK_TREE_VIEW(GHB_WIDGET (ud->builder, "audio_list"));
 316         selection = gtk_tree_view_get_selection (treeview);
 317         // 11 columns in model.  6 are visible, the other 5 are for storing
 318         // values that I need
 319         treestore = gtk_list_store_new(11, G_TYPE_STRING, G_TYPE_STRING,
 320                                                                    G_TYPE_STRING, G_TYPE_STRING,
 321                                                                    G_TYPE_STRING, G_TYPE_STRING,
 322                                                                    G_TYPE_STRING, G_TYPE_STRING,
 323                                                                    G_TYPE_STRING, G_TYPE_STRING,
 324                                                                    G_TYPE_STRING);
 325         gtk_tree_view_set_model(treeview, GTK_TREE_MODEL(treestore));
 326
 327         cell = gtk_cell_renderer_text_new();
 328         column = gtk_tree_view_column_new_with_attributes(
 329                                                                         _("Track"), cell, "text", 0, NULL);
 330     gtk_tree_view_append_column(treeview, GTK_TREE_VIEW_COLUMN(column));
 331
 332         cell = gtk_cell_renderer_text_new();
 333         column = gtk_tree_view_column_new_with_attributes(
 334                                                                         _("Codec"), cell, "text", 1, NULL);
 335     gtk_tree_view_append_column(treeview, GTK_TREE_VIEW_COLUMN(column));
 336
 337         cell = gtk_cell_renderer_text_new();
 338         column = gtk_tree_view_column_new_with_attributes(
 339                                                                         _("Bitrate"), cell, "text", 2, NULL);
 340     gtk_tree_view_append_column(treeview, GTK_TREE_VIEW_COLUMN(column));
 341
 342         cell = gtk_cell_renderer_text_new();
 343         column = gtk_tree_view_column_new_with_attributes(
 344                                                                         _("Sample Rate"), cell, "text", 3, NULL);
 345     gtk_tree_view_append_column(treeview, GTK_TREE_VIEW_COLUMN(column));
 346
 347         cell = gtk_cell_renderer_text_new();
 348         column = gtk_tree_view_column_new_with_attributes(
 349                                                                         _("Mix"), cell, "text", 4, NULL);
 350     gtk_tree_view_append_column(treeview, GTK_TREE_VIEW_COLUMN(column));
 351
 352         cell = gtk_cell_renderer_text_new();
 353         column = gtk_tree_view_column_new_with_attributes(
 354                                                                         _("DRC"), cell, "text", 5, NULL);
 355     gtk_tree_view_append_column(treeview, GTK_TREE_VIEW_COLUMN(column));
 356
 357         g_signal_connect(selection, "changed", audio_list_selection_changed_cb, ud);
 358         // Need to disable remove and update buttons since there are initially
 359         // no selections
 360         widget = GHB_WIDGET (ud->builder, "audio_remove");
 361         gtk_widget_set_sensitive(widget, FALSE);
 362         g_debug("Done\n");
 363 }
 364
 365 extern void presets_list_selection_changed_cb(void);
 366
 367 // Create and bind the tree model to the tree view for the preset list
 368 // Also, connect up the signal that lets us know the selection has changed
 369 static void
 370 bind_presets_tree_model (signal_user_data_t *ud)
 371 {
 372         GtkCellRenderer *cell;
 373         GtkTreeViewColumn *column;
 374         GtkListStore *treestore;
 375         GtkTreeView  *treeview;
 376         GtkTreeSelection *selection;
 377         GtkWidget *widget;
 378
 379         g_debug("bind_presets_tree_model ()\n");
 380         treeview = GTK_TREE_VIEW(GHB_WIDGET (ud->builder, "presets_list"));
 381         selection = gtk_tree_view_get_selection (treeview);
 382         treestore = gtk_list_store_new(5, G_TYPE_STRING, G_TYPE_INT, G_TYPE_INT,
 383                                                                    G_TYPE_STRING, G_TYPE_STRING);
 384         gtk_tree_view_set_model(treeview, GTK_TREE_MODEL(treestore));
 385
 386         cell = gtk_cell_renderer_text_new();
 387         column = gtk_tree_view_column_new_with_attributes(_("Preset Name"), cell,
 388                                         "text", 0, "weight", 1, "style", 2, "foreground", 3, NULL);
 389     gtk_tree_view_append_column(treeview, GTK_TREE_VIEW_COLUMN(column));
 390         gtk_tree_view_set_tooltip_column (treeview, 4);
 391         g_signal_connect(selection, "changed", presets_list_selection_changed_cb, ud);
 392         widget = GHB_WIDGET (ud->builder, "presets_remove");
 393         gtk_widget_set_sensitive(widget, FALSE);
 394         g_debug("Done\n");
 395 }
 396
 397 static void
 398 IoRedirect(signal_user_data_t *ud)
 399 {
 400         GIOChannel *channel;
 401         gint pfd[2];
 402
 403         // I'm opening a pipe and attaching the writer end to stderr
 404         // The reader end will be polled by main event loop and I'll get
 405         // a callback when there is data available.
 406         if (pipe( pfd ) < 0)
 407         {
 408                 g_warning("Failed to redirect IO. Logging impaired\n");
 409                 return;
 410         }
 411         // Open activity log.
 412         // TODO: Put this in the same directory as the encode destination
 413         ud->activity_log = g_io_channel_new_file ("Activity.log", "w", NULL);
 414         // Set encoding to raw.
 415         g_io_channel_set_encoding (ud->activity_log, NULL, NULL);
 416         stderr->_fileno = pfd[1];
 417         stdin->_fileno = pfd[0];
 418         channel = g_io_channel_unix_new (pfd[0]);
 419         // I was getting an this error:
 420         // "Invalid byte sequence in conversion input"
 421         // Set disable encoding on the channel.
 422         g_io_channel_set_encoding (channel, NULL, NULL);
 423         g_io_add_watch (channel, G_IO_IN, ghb_log_cb, (gpointer)ud );
 424 }
 425
 426 typedef struct
 427 {
 428         gchar *filename;
 429         gchar *iconname;
 430 } icon_map_t;
 431
 432 gboolean ghb_autostart;
 433 static gchar *dvd_device = NULL;
 434 static gchar *arg_preset = NULL;
 435
 436 static GOptionEntry entries[] =
 437 {
 438         { "autostart", 'a', 0, G_OPTION_ARG_NONE, &ghb_autostart, "Automatically scan and start encoding", NULL },
 439         { "device", 'd', 0, G_OPTION_ARG_FILENAME, &dvd_device, "The device or file to encode", NULL },
 440         { "preset", 'p', 0, G_OPTION_ARG_STRING, &arg_preset, "The preset values to use for encoding", NULL },
 441         { NULL }
 442 };
 443
 444 #if defined(__linux__)
 445 void drive_changed_cb(GVolumeMonitor *gvm, GDrive *gd, signal_user_data_t *ud);
 446 //void drive_disconnected_cb(GnomeVFSVolumeMonitor *gvm, GnomeVFSDrive *gd, signal_user_data_t *ud);
 447
 448 void
 449 watch_volumes(signal_user_data_t *ud)
 450 {
 451         GVolumeMonitor *gvm;
 452         gvm = g_volume_monitor_get ();
 453
 454         g_signal_connect(gvm, "drive-changed", (GCallback)drive_changed_cb, ud);
 455         //g_signal_connect(gvm, "drive-connected", (GCallback)drive_connected_cb, ud);
 456 }
 457 #else
 458 void
 459 watch_volumes(signal_user_data_t *ud)
 460 {
 461 }
 462 #endif
 463
 464 // Hack to avoid a segfault in libavcodec
 465 extern int mm_flags;
 466 int mm_support();
 467
 468 int
 469 main (int argc, char *argv[])
 470 {
 471         GtkWidget *window;
 472         signal_user_data_t *ud;
 473         const gchar *preset;
 474         gchar *builder_file;
 475         GError *error = NULL;
 476         GOptionContext *context;
 477
 478         mm_flags = mm_support();
 479 #ifdef ENABLE_NLS
 480         bindtextdomain (GETTEXT_PACKAGE, PACKAGE_LOCALE_DIR);
 481         bind_textdomain_codeset (GETTEXT_PACKAGE, "UTF-8");
 482         textdomain (GETTEXT_PACKAGE);
 483 #endif
 484
 485         context = g_option_context_new ("- Rip and encode DVD or MPEG file");
 486         g_option_context_add_main_entries (context, entries, GETTEXT_PACKAGE);
 487         g_option_context_add_group (context, gtk_get_option_group (TRUE));
 488         g_option_context_parse (context, &argc, &argv, &error);
 489
 490         gtk_set_locale ();
 491         gtk_init (&argc, &argv);
 492
 493 #if defined(__linux__)
 494         ghb_hal_init();
 495 #endif
 496
 497         // Add ./icons so that we can find the icons if we haven't installed
 498         GtkIconTheme *theme;
 499         theme = gtk_icon_theme_get_default ();
 500         gtk_icon_theme_append_search_path (theme, "./icons");
 501
 502         ud = g_malloc(sizeof(signal_user_data_t));
 503         ud->state = GHB_STATE_START;
 504         ud->debug = FALSE;
 505         g_log_set_handler (NULL, G_LOG_LEVEL_DEBUG, debug_log_handler, ud);
 506         ud->settings = ghb_settings_new();
 507         ud->audio_settings = NULL;
 508         ud->chapter_list = NULL;
 509         ud->queue = NULL;
 510         ud->current_dvd_device = NULL;
 511         // Redirect stderr to the activity window
 512         IoRedirect(ud);
 513         // Enable events that alert us to media change events
 514         watch_volumes (ud);
 515         builder_file = find_builder_file(BUILDER_NAME);
 516         if (builder_file != NULL)
 517         {
 518                 ud->builder = create_builder_or_die (BUILDER_NAME, builder_file);
 519                 g_free(builder_file);
 520         }
 521         ghb_file_menu_add_dvd(ud);
 522         ghb_backend_init(ud->builder, 1, 0);
 523
 524         g_debug("ud %p\n", ud);
 525         g_debug("ud->builder %p\n", ud->builder);
 526         bind_audio_tree_model(ud);
 527         bind_presets_tree_model(ud);
 528         bind_queue_tree_model(ud);
 529         bind_chapter_tree_model(ud);
 530         // Connect up the signals to their callbacks
 531         // I wrote my own connector so that I could pass user data
 532         // to the callbacks.  Builder's standard autoconnect doesn't all this.
 533         gtk_builder_connect_signals_full (ud->builder, MyConnect, ud);
 534
 535         ghb_prefs_load(ud);
 536         if (ghb_settings_get_bool(ud->settings, "hbfd"))
 537         {
 538                 ghb_hbfd(ud, TRUE);
 539         }
 540         const gchar *source = ghb_settings_get_string(ud->settings, "default_source");
 541         ghb_dvd_set_current(source, ud);
 542         // Start the show.
 543         window = GHB_WIDGET (ud->builder, "hb_window");
 544         gtk_widget_show (window);
 545
 546         // Load the presets files
 547         ghb_presets_load(ud);
 548         // Populate the presets tree view
 549         ghb_presets_list_update(ud);
 550         // Get the first preset name
 551         if (arg_preset != NULL)
 552         {
 553                 preset = arg_preset;
 554         }
 555         else
 556         {
 557                 preset = ghb_settings_get_string (ud->settings, "default_preset");
 558                 if (preset == NULL)
 559                         preset = ghb_presets_get_name(0);
 560         }
 561         ghb_select_preset(ud->builder, preset);
 562         ghb_prefs_to_ui(ud);
 563         // Grey out widgets that are dependent on a disabled feature
 564         ghb_check_all_depencencies (ud);
 565
 566         if (dvd_device != NULL)
 567         {
 568                 // Source overridden from command line option
 569                 ghb_settings_set_string(ud->settings, "source", dvd_device);
 570         }
 571         if (ghb_autostart)
 572         {
 573                 ghb_hbfd(ud, TRUE);
 574         }
 575         // Start timer for monitoring libhb status, 500ms
 576         g_timeout_add (500, ghb_timer_cb, (gpointer)ud);
 577         // Everything should be go-to-go.  Lets rock!
 578         gtk_main ();
 579         g_free(ud);
 580         return 0;
 581 }
 582