Qt: OpenIGTLink: add server socket handling

Author: ClaudioHoffmann

examples/solum_qt/solum/openigtlink.cpp

@@ -0,0 +1,19 @@
+#include "openigtlink.h"
+
+bool SolumIGTL::serve(uint16_t port)
+{
+    server_ = igtl::ServerSocket::New();
+    if (server_->CreateServer(port) < 0)
+        return false;
+    return true;
+}
+
+void SolumIGTL::close()
+{
+    server_->CloseSocket();
+}
+
+bool SolumIGTL::isServing() const
+{
+    return (server_ && server_->GetConnected());
+}

examples/solum_qt/solum/openigtlink.h

@@ -0,0 +1,18 @@
+#pragma once
+
+#include "igtlServerSocket.h"
+
+/// OpenIGTLink module
+class SolumIGTL : public QObject
+{
+    Q_OBJECT;
+
+public:
+    bool serve(uint16_t port); /// Returns `false` on failure.
+    void close();
+
+    bool isServing() const;
+
+private:
+    igtl::ServerSocket::Pointer server_;
+};

examples/solum_qt/solum/solum.pro

@@ -7,13 +7,15 @@ PRECOMPILED_HEADER = pch.h
 
 CONFIG(release, debug|release): CONFIG += ltcg
 
+include($$PWD/../../../openigtlink/openigtlink.pri)
+
 # ensure to unpack the appropriate libs from the zip file into this folder
 LIBPATH = $$PWD/../../../lib
 INCLUDEPATH += $$PWD/../../../include/solum
 LIBS += -L$$LIBPATH/ -lsolum
 
-SOURCES += main.cpp solumqt.cpp ble.cpp display.cpp 3d.cpp jobbutton.cpp
-HEADERS += solumqt.h ble.h display.h 3d.h jobbutton.h
+SOURCES += main.cpp solumqt.cpp ble.cpp display.cpp 3d.cpp jobbutton.cpp openigtlink.cpp
+HEADERS += solumqt.h ble.h display.h 3d.h jobbutton.h openigtlink.h
 FORMS += solumqt.ui
 
 RESOURCES += \

examples/solum_qt/solum/solumqt.cpp

@@ -23,6 +23,8 @@ Solum::Solum(QWidget *parent) : QMainWindow(parent), imaging_(false)
                      .arg(portValidator_->bottom())
                      .arg(portValidator_->top());
 
+    connectPortChanged(ui_.igtlport, ui_.igtlserve);
+
     // UI polish handlers
     connect(ui_.token, &QLineEdit::textChanged, [this](auto& text)
     {
@@ -62,6 +64,9 @@ Solum::Solum(QWidget *parent) : QMainWindow(parent), imaging_(false)
         settings_->endGroup();
     }
     settings_->endGroup();
+    auto igtlport = settings_->value("OpenIGTLink/port").toString();
+    if (!igtlport.isEmpty())
+        ui_.igtlport->setText(igtlport);
 
     ui_.certtable->setColumnCount(4);
     ui_.certtable->setHorizontalHeaderLabels({tr("Serial number"),
@@ -1077,3 +1082,28 @@ void Solum::connectPortChanged(QLineEdit* portEdit, QPushButton* connectButton)
         }
     });
 }
+
+void Solum::onIGTLServe()
+{
+    const auto port = ui_.igtlport->text().toInt();
+    if (!igtl_.isServing())
+    {
+        settings_->setValue("OpenIGTLink/port", port);
+        if (igtl_.serve(port))
+        {
+            ui_.igtlStatusServer->setText(tr("🟢 Server is running"));
+            ui_.igtlport->setEnabled(false);
+            ui_.igtlserve->setText(tr("Stop"));
+        }
+        else
+            ui_.igtlStatusServer->setText(
+                tr("❌ Socket could not be opened on port %1").arg(port));
+    }
+    else
+    {
+        igtl_.close();
+        ui_.igtlStatusServer->setText(tr("🔴 Server is stopped"));
+        ui_.igtlport->setEnabled(true);
+        ui_.igtlserve->setText(tr("Serve"));
+    }
+}

examples/solum_qt/solum/solumqt.h

@@ -1,7 +1,7 @@
 #pragma once
 
 #include "ble.h"
-#include <solum_def.h>
+#include "openigtlink.h"
 #include "ui_solumqt.h"
 #include <solum_def.h>
 
@@ -272,6 +272,7 @@ public slots:
     void tgcTop(int);
     void tgcMid(int);
     void tgcBottom(int);
+    void onIGTLServe();
 
 private:
     QString bleConnectedProbe_;     ///< serial of the probe connected via BLE
@@ -291,4 +292,7 @@ public slots:
     std::unique_ptr<QSettings> settings_;   ///< persistent settings
     QIntValidator* portValidator_;  ///< keeps port fields between 1 and 65535
     QString portError_;             ///< error message for port validation
+
+    std::optional<uint16_t> igtlport_; ///< OpenIGTLink port
+    SolumIGTL igtl_;                   ///< OpenIGTLink module
 };

examples/solum_qt/solum/solumqt.ui

@@ -1498,6 +1498,22 @@ After clicking the button, wait for the SSID to be published via BLE, then conne
     </hint>
    </hints>
   </connection>
+  <connection>
+   <sender>igtlserve</sender>
+   <signal>clicked()</signal>
+   <receiver>Solum</receiver>
+   <slot>onIGTLServe()</slot>
+   <hints>
+    <hint type="sourcelabel">
+     <x>20</x>
+     <y>20</y>
+    </hint>
+    <hint type="destinationlabel">
+     <x>20</x>
+     <y>20</y>
+    </hint>
+   </hints>
+  </connection>
  </connections>
  <slots>
   <slot>onTcpConnect()</slot>
@@ -1528,5 +1544,6 @@ After clicking the button, wait for the SSID to be published via BLE, then conne
   <slot>onOpacity(int)</slot>
   <slot>onAutoFocus(int)</slot>
   <slot>onFocus(int)</slot>
+  <slot>onIGTLServe()</slot>
  </slots>
 </ui>