Added exit dialog question to camera intrinsic calibration widget

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

Hide file tree

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

diff --git a/gui/src/camera_intrinsic_calibration_widget.cpp b/gui/src/camera_intrinsic_calibration_widget.cpp
@@ -15,12 +15,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;
@@ -177,6 +178,19 @@ CameraIntrinsicCalibrationWidget::CameraIntrinsicCalibrationWidget(QWidget* pare
 
 CameraIntrinsicCalibrationWidget::~CameraIntrinsicCalibrationWidget() { delete ui_; }
 
+void CameraIntrinsicCalibrationWidget::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 CameraIntrinsicCalibrationWidget::onLoadConfig()
 {
   try