pv/dialogs/connect.cpp

   1 /*
   2  * This file is part of the PulseView project.
   3  *
   4  * Copyright (C) 2012-2013 Joel Holdsworth <joel@airwebreathe.org.uk>
   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 2 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, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
  19  */
  20
  21 #include <boost/foreach.hpp>
  22
  23 #include "connect.h"
  24
  25 #include "pv/devicemanager.h"
  26
  27 extern "C" {
  28 /* __STDC_FORMAT_MACROS is required for PRIu64 and friends (in C++). */
  29 #define __STDC_FORMAT_MACROS
  30 #include <glib.h>
  31 #include <libsigrok/libsigrok.h>
  32 }
  33
  34 using namespace std;
  35
  36 extern sr_context *sr_ctx;
  37
  38 namespace pv {
  39 namespace dialogs {
  40
  41 Connect::Connect(QWidget *parent, pv::DeviceManager &device_manager) :
  42         QDialog(parent),
  43         _device_manager(device_manager),
  44         _layout(this),
  45         _form(this),
  46         _form_layout(&_form),
  47         _drivers(&_form),
  48         _serial_device(&_form),
  49         _scan_button(tr("Scan for Devices"), this),
  50         _device_list(this),
  51         _button_box(QDialogButtonBox::Ok | QDialogButtonBox::Cancel,
  52                 Qt::Horizontal, this)
  53 {
  54         setWindowTitle(tr("Connect to Device"));
  55
  56         connect(&_button_box, SIGNAL(accepted()), this, SLOT(accept()));
  57         connect(&_button_box, SIGNAL(rejected()), this, SLOT(reject()));
  58
  59         populate_drivers();
  60         connect(&_drivers, SIGNAL(activated(int)),
  61                 this, SLOT(device_selected(int)));
  62
  63         _form.setLayout(&_form_layout);
  64         _form_layout.addRow(tr("Driver"), &_drivers);
  65
  66         _form_layout.addRow(tr("Serial Port"), &_serial_device);
  67
  68         unset_connection();
  69
  70         connect(&_scan_button, SIGNAL(pressed()),
  71                 this, SLOT(scan_pressed()));
  72
  73         setLayout(&_layout);
  74         _layout.addWidget(&_form);
  75         _layout.addWidget(&_scan_button);
  76         _layout.addWidget(&_device_list);
  77         _layout.addWidget(&_button_box);
  78 }
  79
  80 struct sr_dev_inst* Connect::get_selected_device() const
  81 {
  82         const QListWidgetItem *const item = _device_list.currentItem();
  83         if (!item)
  84                 return NULL;
  85
  86         return (sr_dev_inst*)item->data(Qt::UserRole).value<void*>();
  87 }
  88
  89 void Connect::populate_drivers()
  90 {
  91         gsize num_opts = 0;
  92         const int32_t *hwopts;
  93         struct sr_dev_driver **drivers = sr_driver_list();
  94         GVariant *gvar_opts;
  95
  96         for (int i = 0; drivers[i]; ++i) {
  97                 /**
  98                  * We currently only support devices that can deliver
  99                  * samples at a fixed samplerate i.e. oscilloscopes and
 100                  * logic analysers.
 101                  * @todo Add support for non-monotonic devices i.e. DMMs
 102                  * and sensors.
 103                  */
 104                 bool supported_device = false;
 105                 if ((sr_config_list(drivers[i], SR_CONF_DEVICE_OPTIONS,
 106                                 &gvar_opts, NULL) == SR_OK)) {
 107                         hwopts = (const int32_t *)g_variant_get_fixed_array(gvar_opts,
 108                                         &num_opts, sizeof(int32_t));
 109                         for (unsigned int j = 0; j < num_opts; j++)
 110                                 if (hwopts[j] == SR_CONF_SAMPLERATE) {
 111                                         supported_device = true;
 112                                         break;
 113                                 }
 114                 }
 115
 116                 if (supported_device)
 117                         _drivers.addItem(QString("%1 (%2)").arg(
 118                                 drivers[i]->longname).arg(drivers[i]->name),
 119                                 qVariantFromValue((void*)drivers[i]));
 120         }
 121 }
 122
 123 void Connect::unset_connection()
 124 {
 125         _device_list.clear();
 126         _serial_device.hide();
 127         _form_layout.labelForField(&_serial_device)->hide();
 128         _button_box.button(QDialogButtonBox::Ok)->setDisabled(true);
 129 }
 130
 131 void Connect::set_serial_connection()
 132 {
 133         _serial_device.show();
 134         _form_layout.labelForField(&_serial_device)->show();
 135 }
 136
 137 void Connect::scan_pressed()
 138 {
 139         _device_list.clear();
 140
 141         const int index = _drivers.currentIndex();
 142         if (index == -1)
 143                 return;
 144
 145         sr_dev_driver *const driver = (sr_dev_driver*)_drivers.itemData(
 146                 index).value<void*>();
 147
 148         GSList *drvopts = NULL;
 149
 150         if (_serial_device.isVisible()) {
 151                 sr_config *const src = (sr_config*)g_try_malloc(sizeof(sr_config));
 152                 src->key = SR_CONF_CONN;
 153                 const QByteArray byteArray = _serial_device.text().toUtf8();
 154                 src->data = g_variant_new_string((const gchar*)byteArray.constData());
 155                 drvopts = g_slist_append(drvopts, src);
 156         }
 157
 158         const list<sr_dev_inst*> devices = _device_manager.driver_scan(
 159                 driver, drvopts);
 160
 161         g_slist_free_full(drvopts, (GDestroyNotify)free_drvopts);
 162
 163         BOOST_FOREACH(sr_dev_inst *const sdi, devices) {
 164
 165                 QString text;
 166                 if (sdi->vendor && sdi->vendor[0])
 167                         text += QString("%1 ").arg(sdi->vendor);
 168                 if (sdi->model && sdi->model[0])
 169                         text += QString("%1 ").arg(sdi->model);
 170                 if (sdi->version && sdi->version[0])
 171                         text += QString("%1 ").arg(sdi->version);
 172                 if (sdi->probes) {
 173                         text += QString("with %1 probes").arg(
 174                                 g_slist_length(sdi->probes));
 175                 }
 176
 177                 QListWidgetItem *const item = new QListWidgetItem(text,
 178                         &_device_list);
 179                 item->setData(Qt::UserRole, qVariantFromValue((void*)sdi));
 180                 _device_list.addItem(item);
 181         }
 182
 183         _device_list.setCurrentRow(0);
 184         _button_box.button(QDialogButtonBox::Ok)->setDisabled(false);
 185 }
 186
 187 void Connect::device_selected(int index)
 188 {
 189         gsize num_opts = 0;
 190         const int32_t *hwopts;
 191         GVariant *gvar_list;
 192         sr_dev_driver *const driver = (sr_dev_driver*)_drivers.itemData(
 193                 index).value<void*>();
 194
 195         unset_connection();
 196
 197         if ((sr_config_list(driver, SR_CONF_SCAN_OPTIONS,
 198                                 &gvar_list, NULL) == SR_OK)) {
 199                 hwopts = (const int32_t *)g_variant_get_fixed_array(gvar_list,
 200                                 &num_opts, sizeof(int32_t));
 201
 202                 for (unsigned int i = 0; i < num_opts; i++) {
 203                         switch(hwopts[i]) {
 204                         case SR_CONF_SERIALCOMM:
 205                                 set_serial_connection();
 206                                 break;
 207
 208                         default:
 209                                 continue;
 210                         }
 211
 212                         break;
 213                 }
 214                 g_variant_unref(gvar_list);
 215         }
 216 }
 217
 218 void Connect::free_drvopts(struct sr_config *src)
 219 {
 220         g_variant_unref(src->data);
 221         g_free(src);
 222 }
 223
 224 } // namespace dialogs
 225 } // namespace pv