Fix problem where Monai tool sets a background image even if the tool is never used.

Author: akenmorris

Studio/Data/Session.cpp

@@ -1092,6 +1092,7 @@ bool Session::set_image_name(std::string image_name) {
   }
   params_.set("image_name", image_name);
   Q_EMIT image_slice_settings_changed();
+  Q_EMIT image_name_changed();
   return true;
 }
 

Studio/Data/Session.h

@@ -322,6 +322,7 @@ class Session : public QObject, public QEnableSharedFromThis<Session> {
   void annotations_changed();
   void save();
   void session_title_changed();
+  void image_name_changed();
 
  public:
   // constants

Studio/Interface/ShapeWorksStudioApp.cpp

@@ -1006,6 +1006,7 @@ void ShapeWorksStudioApp::new_session() {
   connect(session_.data(), &Session::save, this, &ShapeWorksStudioApp::on_action_save_project_triggered);
   connect(session_.data(), &Session::tool_state_changed, this, &ShapeWorksStudioApp::update_tool_mode);
   connect(session_.data(), &Session::session_title_changed, this, &ShapeWorksStudioApp::update_window_title);
+  connect(session_.data(), &Session::image_name_changed, this, &ShapeWorksStudioApp::handle_image_name_changed);
 
   connect(ui_->feature_auto_scale, &QCheckBox::toggled, this, &ShapeWorksStudioApp::update_feature_map_scale);
   connect(ui_->feature_auto_scale, &QCheckBox::toggled, session_.data(), &Session::set_feature_auto_scale);
@@ -1103,6 +1104,7 @@ void ShapeWorksStudioApp::update_tool_mode() {
     ui_->stacked_widget->setCurrentWidget(monai_tool_.data());
     ui_->controlsDock->setWindowTitle("MONAI");
     update_display();
+    monai_tool_->activate();
     ui_->action_monai_mode->setChecked(true);
     session_->set_display_mode(DisplayMode::Original);
   } else {  // DATA
@@ -2112,6 +2114,11 @@ void ShapeWorksStudioApp::image_combo_changed(int index) {
   session_->set_image_name(ui_->image_combo_->itemText(index).toStdString());
 }
 
+//---------------------------------------------------------------------------
+void ShapeWorksStudioApp::handle_image_name_changed() {
+  ui_->image_combo_->setCurrentText(QString::fromStdString(session_->get_image_name()));
+}
+
 //---------------------------------------------------------------------------
 bool ShapeWorksStudioApp::get_feature_uniform_scale() { return session_->get_feature_uniform_scale(); }
 

Studio/Interface/ShapeWorksStudioApp.h

@@ -148,6 +148,7 @@ class ShapeWorksStudioApp : public QMainWindow {
   void update_feature_map_scale();
 
   void image_combo_changed(int index);
+  void handle_image_name_changed();
 
   void show_splash_screen();
   void hide_splash_screen();

Studio/ShapeWorksMONAI/MonaiLabelTool.cpp

@@ -122,6 +122,10 @@ void MonaiLabelTool::enable_actions() {
   ui_->uploadSampleButton->setEnabled(false);
   ui_->runSegmentationButton->setEnabled(false);
   ui_->submitLabelButton->setEnabled(false);
+}
+
+//---------------------------------------------------------------------------
+void MonaiLabelTool::activate() {
   std::string feature_name = MonaiLabelUtils::getFeatureName(session_);
   if (!feature_name.empty()) {
     session_->set_image_name(feature_name);
@@ -134,10 +138,11 @@ void MonaiLabelTool::handleSampleNumberChanged() {
   if (monai_label_job_) {
     monai_label_job_->setCurrentSampleNumber(getCurrentSampleNumber());
   }
-  if (tool_is_running_)
+  if (tool_is_running_) {
     ui_->uploadSampleButton->setEnabled(true);
-  else
+  } else {
     ui_->uploadSampleButton->setEnabled(false);
+  }
   ui_->runSegmentationButton->setEnabled(false);
   ui_->submitLabelButton->setEnabled(false);
   Q_EMIT progress(-1);

Studio/ShapeWorksMONAI/MonaiLabelTool.h

@@ -48,6 +48,7 @@ class MonaiLabelTool : public QWidget {
   void resizeEvent(QResizeEvent* event) override;
   int getCurrentSampleNumber();
   void enable_actions();
+  void activate();
 
  public Q_SLOTS:
   void handle_error(QString msg);