Added exit dialog question to camera extrinsic calibration widget

ros-industrial · Jun 17, 2024 · 35ba323 · 35ba323
1 parent df92161
commit 35ba323
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/gui/include/industrial_calibration/gui/extrinsic_hand_eye_calibration_widget.h b/gui/include/industrial_calibration/gui/extrinsic_hand_eye_calibration_widget.h
@@ -51,7 +51,9 @@ class ExtrinsicHandEyeCalibrationWidget : public QMainWindow
    */
   void saveResults(const std::string& file);
 
-private:
+protected:
+  void closeEvent(QCloseEvent* event) override;
+
   void onLoadConfig();
   void onLoadObservations();
   void onCalibrate();

diff --git a/gui/src/extrinsic_hand_eye_calibration_widget.cpp b/gui/src/extrinsic_hand_eye_calibration_widget.cpp
@@ -17,12 +17,13 @@
 #include <boost_plugin_loader/plugin_loader.hpp>
 #include <fstream>
 #include <opencv2/opencv.hpp>
+#include <QCloseEvent>
 #include <QDialog>
-#include <QScrollBar>
 #include <QFileDialog>
 #include <QFile>
 #include <QMessageBox>
 #include <QPixmap>
+#include <QScrollBar>
 #include <QVBoxLayout>
 
 static const unsigned RANDOM_SEED = 1;
@@ -130,6 +131,19 @@ ExtrinsicHandEyeCalibrationWidget::ExtrinsicHandEyeCalibrationWidget(QWidget* pa
 
 ExtrinsicHandEyeCalibrationWidget::~ExtrinsicHandEyeCalibrationWidget() { delete ui_; }
 
+void ExtrinsicHandEyeCalibrationWidget::closeEvent(QCloseEvent* event)
+{
+  QMessageBox::StandardButton ret = QMessageBox::question(this, "Exit", "Are you sure you want to exit?");
+  switch (ret)
+  {
+    case QMessageBox::StandardButton::No:
+      event->ignore();
+      break;
+    default:
+      event->accept();
+  }
+}
+
 void ExtrinsicHandEyeCalibrationWidget::onLoadConfig()
 {
   try