QmitkMedSAMToolGUI.h

Go to the documentation of this file.

/*============================================================================
 
The Medical Imaging Interaction Toolkit (MITK)
 
Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.
 
Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.
 
============================================================================*/
 
#ifndef QmitkMedSAMToolGUI_h
#define QmitkMedSAMToolGUI_h
 
#include "QmitkSegmentAnythingToolGUI.h"
#include <MitkSegmentationUIExports.h>
#include "ui_QmitkMedSAMGUIControls.h"
#include <mitkIPreferences.h>
#include "QmitknnUNetGPU.h"
 
class MITKSEGMENTATIONUI_EXPORT QmitkMedSAMToolGUI : public QmitkSegWithPreviewToolGUIBase
{
  Q_OBJECT
 
public:
  mitkClassMacro(QmitkMedSAMToolGUI, QmitkSegWithPreviewToolGUIBase);
  itkFactorylessNewMacro(Self);
  itkCloneMacro(Self);
 
  void EnableAll(bool);
 
  void WriteStatusMessage(const QString &);
 
  void WriteErrorMessage(const QString &);
 
  void ShowProgressBar(bool);
 
  bool ActivateSAMDaemon();
 
  void OnPreferenceChangedEvent(const mitk::IPreferences::ChangeEvent &);
 
  void ShowErrorMessage(const std::string &, QMessageBox::Icon = QMessageBox::Critical);
 
  bool ValidatePrefences();
 
protected:
  QmitkMedSAMToolGUI();
  ~QmitkMedSAMToolGUI();
 
  void InitializeUI(QBoxLayout *mainLayout) override;
  
  void StatusMessageListener(const std::string&);
 
protected slots:
  void OnActivateBtnClicked();
 
  void OnPreviewBtnClicked();
 
  void OnResetPicksClicked();
 
private:
  mitk::IPreferences *m_Preferences;
  QmitkGPULoader m_GpuLoader;
  Ui_QmitkMedSAMGUIControls m_Controls;
  bool m_FirstPreviewComputation = true;
  const std::string WARNING_SAM_NOT_FOUND =
    "MedSAM is not detected in the selected python environment. Please reinstall MedSAM.";
};
#endif