X-Git-Url: https://sigrok.org/gitweb/?a=blobdiff_plain;f=pv%2Fdialogs%2Fconnect.h;h=a3f05ee1a6a224e1936843d11b2b30b508a50103;hb=512bfc565937c0c1b07ec0e6486831526a6e1eee;hp=862184b87326566415b2a142f59baef523dc6b6b;hpb=c9de8b3297469aad87554ad6fcc6909223424f54;p=pulseview.git

diff --git a/pv/dialogs/connect.h b/pv/dialogs/connect.h
index 862184b8..a3f05ee1 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>
@@ -31,8 +33,16 @@
 #include <QVBoxLayout>
 
 struct sr_config;
+struct sr_dev_inst;
 
 namespace pv {
+
+class DeviceManager;
+
+namespace device {
+class Device;
+}
+
 namespace dialogs {
 
 class Connect : public QDialog
@@ -40,7 +50,9 @@ class Connect : public QDialog
 	Q_OBJECT
 
 public:
-	Connect(QWidget *parent);
+	Connect(QWidget *parent, pv::DeviceManager &device_manager);
+
+	std::shared_ptr<device::Device> get_selected_device() const;
 
 private:
 	void populate_drivers();
@@ -58,6 +70,8 @@ private:
 	static void free_drvopts(sr_config *src);
 
 private:
+	pv::DeviceManager &_device_manager;
+
 	QVBoxLayout _layout;
 
 	QWidget _form;
@@ -69,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;
 };