devselect.c

   1 /*
   2  * This file is part of the sigrok project.
   3  *
   4  * Copyright (C) 2011 Gareth McMullin <gareth@blacksphere.co.nz>
   5  *
   6  * This program is free software: you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation, either version 3 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program 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
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18  */
  19
  20 #include <sigrok.h>
  21 #include <gtk/gtk.h>
  22 #include "sigrok-gtk.h"
  23
  24 static void dev_selected(GtkComboBox *devbox, GObject *parent)
  25 {
  26         GtkTreeModel *devlist = gtk_combo_box_get_model(devbox);
  27         GtkTreeIter iter;
  28         const gchar *name;
  29         GtkCheckMenuItem *menuitem;
  30         struct sr_dev *dev;
  31
  32         if (!gtk_combo_box_get_active_iter(devbox, &iter)) {
  33                 g_object_set_data(parent, "dev", NULL);
  34                 return;
  35         }
  36         gtk_tree_model_get(devlist, &iter, 0, &name, 1, &dev,
  37                                 2, &menuitem, -1);
  38
  39         gtk_check_menu_item_set_active(menuitem, TRUE);
  40
  41         sr_session_dev_remove_all();
  42         if (sr_session_dev_add(dev) != SR_OK) {
  43                 g_critical("Failed to use device.");
  44                 sr_session_destroy();
  45                 dev = NULL;
  46         }
  47         g_object_set_data(parent, "dev", dev);
  48 }
  49
  50 static void dev_menuitem_toggled(GtkMenuItem *item, GtkComboBox *combo)
  51 {
  52         GtkTreeModel *model = gtk_combo_box_get_model(combo);
  53         GtkTreeIter iter;
  54
  55         if (!gtk_check_menu_item_get_active(GTK_CHECK_MENU_ITEM(item)))
  56                 return;
  57
  58         if (gtk_tree_model_get_iter_first(model, &iter)) do {
  59                 gchar *name;
  60                 gtk_tree_model_get(model, &iter, 0, &name, -1);
  61                 if (g_str_equal(name, gtk_menu_item_get_label(item))) {
  62                         gtk_combo_box_set_active_iter(combo, &iter);
  63                         return;
  64                 }
  65         } while (gtk_tree_model_iter_next(model, &iter));
  66 }
  67
  68 #define GET_DEV_INST(dev) \
  69         (dev)->driver->dev_info_get((dev)->driver_index, SR_DI_INST);
  70
  71 void dev_select_rescan(GtkAction *action, GtkWindow *parent)
  72 {
  73         GtkComboBox *devbox = g_object_get_data(G_OBJECT(parent), "devcombo");
  74         g_return_if_fail(devbox != NULL);
  75         GtkListStore *devlist = GTK_LIST_STORE(gtk_combo_box_get_model(devbox));
  76         GtkTreeIter iter;
  77         struct sr_dev *dev;
  78         struct sr_dev_inst *sdi;
  79         gchar *sdevname = NULL;
  80         GSList *devs, *l;
  81         GtkUIManager *ui = g_object_get_data(G_OBJECT(parent), "ui_manager");
  82         GtkWidget *menuitem = gtk_ui_manager_get_widget(ui,
  83                                         "/menubar/DevMenu/DevSelectMenu");
  84         GtkMenuShell *devmenu = GTK_MENU_SHELL(gtk_menu_item_get_submenu(GTK_MENU_ITEM(menuitem)));
  85         GSList *radiolist = NULL;
  86
  87         (void)action;
  88
  89         /* Make a copy of the selected device's short name for comparison.
  90          * We wish to select the same device after the refresh if possible.
  91          */
  92         if (gtk_combo_box_get_active_iter(devbox, &iter)) {
  93                 gtk_tree_model_get(GTK_TREE_MODEL(devlist), &iter, 1, &dev, -1);
  94                 /* FIXME: Use something other than dev->driver->name */
  95                 sdevname = g_strdup(dev->driver->name);
  96         }
  97
  98         /* Destroy the old menu items */
  99         if (gtk_tree_model_get_iter_first(GTK_TREE_MODEL(devlist), &iter)) do {
 100                 GtkMenuItem *item;
 101                 gtk_tree_model_get(GTK_TREE_MODEL(devlist), &iter, 2, &item, -1);
 102                 gtk_object_destroy(GTK_OBJECT(item));
 103         } while (gtk_tree_model_iter_next(GTK_TREE_MODEL(devlist), &iter));
 104
 105         gtk_list_store_clear(devlist);
 106
 107         /* Scan for new devices and update our list */
 108         /* TODO: Fix this in libsigrok first. */
 109         /*sr_dev_scan();*/
 110         devs = sr_dev_list();
 111         for (l = devs; l; l = l->next) {
 112                 dev = l->data;
 113                 sdi = GET_DEV_INST(dev);
 114                 gchar *name = sdi->model ? sdi->model : sdi->vendor;
 115                 if (!name)
 116                         name = "(unknown)";
 117
 118                 menuitem = gtk_radio_menu_item_new_with_label(radiolist, name);
 119                 gtk_widget_show(GTK_WIDGET(menuitem));
 120                 if (!radiolist)
 121                         radiolist = gtk_radio_menu_item_get_group(
 122                                         GTK_RADIO_MENU_ITEM(menuitem));
 123                 g_signal_connect(menuitem, "toggled",
 124                                 G_CALLBACK(dev_menuitem_toggled), devbox);
 125                 gtk_menu_shell_prepend(devmenu, menuitem);
 126
 127                 gtk_list_store_append(devlist, &iter);
 128                 gtk_list_store_set(devlist, &iter,
 129                                 0, name,
 130                                 1, dev,
 131                                 2, menuitem,
 132                                 -1);
 133
 134                 if (sdevname && g_str_equal(sdevname, dev->driver->name))
 135                         gtk_combo_box_set_active_iter(devbox, &iter);
 136         }
 137         if (sdevname)
 138                 g_free(sdevname);
 139
 140         /* Select a default if nothing selected */
 141         if (!gtk_combo_box_get_active_iter(devbox, &iter)) {
 142                 if (!gtk_tree_model_get_iter_first(GTK_TREE_MODEL(devlist), &iter))
 143                         return;
 144                 /* Skip demo if there's another available */
 145                 GtkTreeIter first = iter;
 146                 if (gtk_tree_model_iter_next(GTK_TREE_MODEL(devlist), &iter))
 147                         gtk_combo_box_set_active_iter(devbox, &iter);
 148                 else
 149                         gtk_combo_box_set_active_iter(devbox, &first);
 150         }
 151 }
 152
 153 GtkWidget *dev_select_combo_box_new(GtkWindow *parent)
 154 {
 155         GtkWidget *devbox = gtk_combo_box_new();
 156
 157         /* Populate device list */
 158         GtkListStore *devlist = gtk_list_store_new(3,
 159                         G_TYPE_STRING, G_TYPE_POINTER, GTK_TYPE_CHECK_MENU_ITEM);
 160         GtkCellRenderer *cel = gtk_cell_renderer_text_new();
 161         gtk_cell_layout_pack_start(GTK_CELL_LAYOUT(devbox), cel, TRUE);
 162         gtk_cell_layout_add_attribute(GTK_CELL_LAYOUT(devbox), cel, "text", 0);
 163         gtk_combo_box_set_model(GTK_COMBO_BOX(devbox), GTK_TREE_MODEL(devlist));
 164         g_signal_connect(devbox, "changed", G_CALLBACK(dev_selected), parent);
 165
 166         g_object_set_data(G_OBJECT(parent), "devcombo", devbox);
 167         dev_select_rescan(NULL, parent);
 168
 169         return devbox;
 170 }
 171