Fix bar displaying undecoded area
[pulseview.git] / 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, see <http://www.gnu.org/licenses/>.
18  */
19
20 #include <cassert>
21
22 #include <libsigrokcxx/libsigrokcxx.hpp>
23
24 #include <QGroupBox>
25 #include <QLabel>
26 #include <QRadioButton>
27
28 #include "connect.hpp"
29
30 #include <pv/devicemanager.hpp>
31 #include <pv/devices/hardwaredevice.hpp>
32
33 using std::list;
34 using std::map;
35 using std::shared_ptr;
36 using std::string;
37
38 using Glib::ustring;
39 using Glib::Variant;
40 using Glib::VariantBase;
41
42 using sigrok::ConfigKey;
43 using sigrok::Driver;
44
45 using pv::devices::HardwareDevice;
46
47 namespace pv {
48 namespace dialogs {
49
50 Connect::Connect(QWidget *parent, pv::DeviceManager &device_manager) :
51         QDialog(parent),
52         device_manager_(device_manager),
53         layout_(this),
54         form_(this),
55         form_layout_(&form_),
56         drivers_(&form_),
57         serial_devices_(&form_),
58         scan_button_(tr("&Scan for devices using driver above"), this),
59         device_list_(this),
60         button_box_(QDialogButtonBox::Ok | QDialogButtonBox::Cancel,
61                 Qt::Horizontal, this)
62 {
63         setWindowTitle(tr("Connect to Device"));
64
65         connect(&button_box_, SIGNAL(accepted()), this, SLOT(accept()));
66         connect(&button_box_, SIGNAL(rejected()), this, SLOT(reject()));
67
68         populate_drivers();
69         connect(&drivers_, SIGNAL(activated(int)), this, SLOT(driver_selected(int)));
70
71         form_.setLayout(&form_layout_);
72
73         QVBoxLayout *vbox_drv = new QVBoxLayout;
74         vbox_drv->addWidget(&drivers_);
75         QGroupBox *groupbox_drv = new QGroupBox(tr("Step 1: Choose the driver"));
76         groupbox_drv->setLayout(vbox_drv);
77         form_layout_.addRow(groupbox_drv);
78
79         QRadioButton *radiobtn_usb = new QRadioButton(tr("&USB"), this);
80         QRadioButton *radiobtn_serial = new QRadioButton(tr("Serial &Port"), this);
81         QRadioButton *radiobtn_tcp = new QRadioButton(tr("&TCP/IP"), this);
82
83         radiobtn_usb->setChecked(true);
84
85         serial_devices_.setEditable(true);
86         serial_devices_.setEnabled(false);
87
88         tcp_config_ = new QWidget();
89         QHBoxLayout *tcp_config_layout = new QHBoxLayout(tcp_config_);
90         tcp_host_ = new QLineEdit;
91         tcp_host_->setText("192.168.1.100");
92         tcp_config_layout->addWidget(tcp_host_);
93         tcp_config_layout->addWidget(new QLabel(":"));
94         tcp_port_ = new QSpinBox;
95         tcp_port_->setRange(1, 65535);
96         tcp_port_->setValue(5555);
97         tcp_config_layout->addWidget(tcp_port_);
98         tcp_use_vxi_ = new QCheckBox();
99         tcp_use_vxi_->setText(tr("Use VXI"));
100         tcp_config_layout->addSpacing(30);
101         tcp_config_layout->addWidget(tcp_use_vxi_);
102         tcp_config_layout->setContentsMargins(0, 0, 0, 0);
103         tcp_config_->setEnabled(false);
104
105         QVBoxLayout *vbox_if = new QVBoxLayout;
106         vbox_if->addWidget(radiobtn_usb);
107         vbox_if->addWidget(radiobtn_serial);
108         vbox_if->addWidget(&serial_devices_);
109         vbox_if->addWidget(radiobtn_tcp);
110         vbox_if->addWidget(tcp_config_);
111
112         QGroupBox *groupbox_if = new QGroupBox(tr("Step 2: Choose the interface"));
113         groupbox_if->setLayout(vbox_if);
114         form_layout_.addRow(groupbox_if);
115
116         QVBoxLayout *vbox_scan = new QVBoxLayout;
117         vbox_scan->addWidget(&scan_button_);
118         QGroupBox *groupbox_scan = new QGroupBox(tr("Step 3: Scan for devices"));
119         groupbox_scan->setLayout(vbox_scan);
120         form_layout_.addRow(groupbox_scan);
121
122         QVBoxLayout *vbox_select = new QVBoxLayout;
123         vbox_select->addWidget(&device_list_);
124         QGroupBox *groupbox_select = new QGroupBox(tr("Step 4: Select the device"));
125         groupbox_select->setLayout(vbox_select);
126         form_layout_.addRow(groupbox_select);
127
128         unset_connection();
129
130         connect(radiobtn_serial, SIGNAL(toggled(bool)), this, SLOT(serial_toggled(bool)));
131         connect(radiobtn_tcp, SIGNAL(toggled(bool)), this, SLOT(tcp_toggled(bool)));
132         connect(&scan_button_, SIGNAL(pressed()), this, SLOT(scan_pressed()));
133
134         setLayout(&layout_);
135
136         layout_.addWidget(&form_);
137         layout_.addWidget(&button_box_);
138 }
139
140 shared_ptr<HardwareDevice> Connect::get_selected_device() const
141 {
142         const QListWidgetItem *const item = device_list_.currentItem();
143         if (!item)
144                 return shared_ptr<HardwareDevice>();
145
146         return item->data(Qt::UserRole).value<shared_ptr<HardwareDevice>>();
147 }
148
149 void Connect::populate_drivers()
150 {
151         for (auto entry : device_manager_.context()->drivers()) {
152                 auto name = entry.first;
153                 auto driver = entry.second;
154                 /**
155                  * We currently only support devices that can deliver
156                  * samples at a fixed samplerate i.e. oscilloscopes and
157                  * logic analysers.
158                  * @todo Add support for non-monotonic devices i.e. DMMs
159                  * and sensors.
160                  */
161                 const auto keys = driver->config_keys();
162
163                 bool supported_device = keys.count(ConfigKey::LOGIC_ANALYZER) |
164                         keys.count(ConfigKey::OSCILLOSCOPE);
165
166                 if (supported_device)
167                         drivers_.addItem(QString("%1 (%2)").arg(
168                                 driver->long_name().c_str(), name.c_str()),
169                                 qVariantFromValue(driver));
170         }
171 }
172
173 void Connect::populate_serials(shared_ptr<Driver> driver)
174 {
175         serial_devices_.clear();
176         for (auto serial : device_manager_.context()->serials(driver))
177                 serial_devices_.addItem(QString("%1 (%2)").arg(
178                         serial.first.c_str(), serial.second.c_str()),
179                         QString::fromStdString(serial.first));
180 }
181
182 void Connect::unset_connection()
183 {
184         device_list_.clear();
185         button_box_.button(QDialogButtonBox::Ok)->setDisabled(true);
186 }
187
188 void Connect::serial_toggled(bool checked)
189 {
190         serial_devices_.setEnabled(checked);
191 }
192
193 void Connect::tcp_toggled(bool checked)
194 {
195         tcp_config_->setEnabled(checked);
196 }
197
198 void Connect::scan_pressed()
199 {
200         device_list_.clear();
201
202         const int index = drivers_.currentIndex();
203         if (index == -1)
204                 return;
205
206         shared_ptr<Driver> driver =
207                 drivers_.itemData(index).value<shared_ptr<Driver>>();
208
209         assert(driver);
210
211         map<const ConfigKey *, VariantBase> drvopts;
212
213         if (serial_devices_.isEnabled()) {
214                 QString serial;
215                 const int index = serial_devices_.currentIndex();
216                 if (index >= 0 && index < serial_devices_.count() &&
217                     serial_devices_.currentText() == serial_devices_.itemText(index))
218                         serial = serial_devices_.itemData(index).value<QString>();
219                 else
220                         serial = serial_devices_.currentText();
221                 drvopts[ConfigKey::CONN] = Variant<ustring>::create(
222                         serial.toUtf8().constData());
223         }
224
225         if (tcp_config_->isEnabled()) {
226                 QString host = tcp_host_->text();
227                 QString port = tcp_port_->text();
228                 if (!host.isEmpty()) {
229                         QString conn;
230                         if (tcp_use_vxi_->isChecked())
231                                 conn = QString("vxi/%1/%2").arg(host, port);
232                         else
233                                 conn = QString("tcp-raw/%1/%2").arg(host, port);
234
235                         drvopts[ConfigKey::CONN] = Variant<ustring>::create(
236                                 conn.toUtf8().constData());
237                 }
238         }
239
240         const list< shared_ptr<HardwareDevice> > devices =
241                 device_manager_.driver_scan(driver, drvopts);
242
243         for (shared_ptr<HardwareDevice> device : devices) {
244                 assert(device);
245
246                 QString text = QString::fromStdString(
247                         device->display_name(device_manager_));
248                 text += QString(" with %1 channels").arg(
249                         device->device()->channels().size());
250
251                 QListWidgetItem *const item = new QListWidgetItem(text,
252                         &device_list_);
253                 item->setData(Qt::UserRole, qVariantFromValue(device));
254                 device_list_.addItem(item);
255         }
256
257         device_list_.setCurrentRow(0);
258         button_box_.button(QDialogButtonBox::Ok)->setDisabled(device_list_.count() == 0);
259 }
260
261 void Connect::driver_selected(int index)
262 {
263         shared_ptr<Driver> driver =
264                 drivers_.itemData(index).value<shared_ptr<Driver>>();
265
266         unset_connection();
267
268         populate_serials(driver);
269 }
270
271 } // namespace dialogs
272 } // namespace pv