QmitkSegmentAnythingToolGUI.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 QmitkSegmentAnythingToolGUI_h
#define QmitkSegmentAnythingToolGUI_h
 
#include "QmitkSegWithPreviewToolGUIBase.h"
#include <MitkSegmentationUIExports.h>
#include "ui_QmitkSegmentAnythingGUIControls.h"
#include "QmitknnUNetGPU.h"
#include "QmitkSetupVirtualEnvUtil.h"
#include <QMessageBox>
#include <QStandardPaths>
#include <mitkIPreferences.h>
 
class MITKSEGMENTATIONUI_EXPORT QmitkSegmentAnythingToolGUI : public QmitkSegWithPreviewToolGUIBase
{
  Q_OBJECT
 
public:
  mitkClassMacro(QmitkSegmentAnythingToolGUI, QmitkSegWithPreviewToolGUIBase);
  itkFactorylessNewMacro(Self);
  itkCloneMacro(Self);
 
  static bool IsSAMInstalled(const QString &);
 
protected slots:
  void OnResetPicksClicked();
 
  void OnActivateBtnClicked();
 
protected:
  QmitkSegmentAnythingToolGUI();
  ~QmitkSegmentAnythingToolGUI();
 
  void InitializeUI(QBoxLayout *mainLayout) override;
  
  void WriteStatusMessage(const QString&);
 
  void WriteErrorMessage(const QString&);
 
  void StatusMessageListener(const std::string&);
 
  void OnPreferenceChangedEvent(const mitk::IPreferences::ChangeEvent&);
 
  void ShowErrorMessage(const std::string&, QMessageBox::Icon = QMessageBox::Critical);
 
  void EnableAll(bool);
 
  void ShowProgressBar(bool);
 
  bool ActivateSAMDaemon();
 
  bool ValidatePrefences();
 
private:
  mitk::IPreferences* m_Preferences;
  Ui_QmitkSegmentAnythingGUIControls m_Controls;
  QmitkGPULoader m_GpuLoader;
  bool m_FirstPreviewComputation = true;
    const std::string WARNING_SAM_NOT_FOUND =
    "SAM is not detected in the selected python environment. Please reinstall SAM.";
};
#endif