X-Git-Url: https://sigrok.org/gitweb/?p=pulseview.git;a=blobdiff_plain;f=pv%2Fdialogs%2Fconnect.h;h=e1a2c30d6c68422fbc37858c2be491702860e42d;hp=eeb108c105dd5a4c28edaf4e658ed10fa2e137f6;hb=e9213170687cf4afd14e20acdc690ff9381d94ae;hpb=5eb0fa13029207c45748ff572568542782e6d2a8

diff --git a/pv/dialogs/connect.h b/pv/dialogs/connect.h
index eeb108c1..e1a2c30d 100644
--- a/pv/dialogs/connect.h
+++ b/pv/dialogs/connect.h
@@ -21,6 +21,8 @@
 #ifndef PULSEVIEW_PV_CONNECT_H
 #define PULSEVIEW_PV_CONNECT_H
 
+#include <memory>
+
 #include <QComboBox>
 #include <QDialog>
 #include <QDialogButtonBox>
@@ -34,6 +36,13 @@ struct sr_config;
 struct sr_dev_inst;
 
 namespace pv {
+
+class DeviceManager;
+
+namespace device {
+class Device;
+}
+
 namespace dialogs {
 
 class Connect : public QDialog
@@ -41,9 +50,9 @@ class Connect : public QDialog
 	Q_OBJECT
 
 public:
-	Connect(QWidget *parent);
+	Connect(QWidget *parent, pv::DeviceManager &device_manager);
 
-	struct sr_dev_inst* get_selected_device() const;
+	std::shared_ptr<device::Device> get_selected_device() const;
 
 private:
 	void populate_drivers();
@@ -52,7 +61,7 @@ private:
 
 	void set_serial_connection();
 
-private slots:
+private Q_SLOTS:
 	void device_selected(int index);
 
 	void scan_pressed();
@@ -61,6 +70,8 @@ private:
 	static void free_drvopts(sr_config *src);
 
 private:
+	pv::DeviceManager &_device_manager;
+
 	QVBoxLayout _layout;
 
 	QWidget _form;
@@ -72,6 +83,8 @@ private:
 
 	QPushButton _scan_button;
 	QListWidget _device_list;
+	std::map<const sr_dev_inst*, std::shared_ptr<pv::device::Device> >
+		_device_map;
 
 	QDialogButtonBox _button_box;
 };