QmitkSegmentAnythingPreferencePage.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 QmitkSegmentAnythingPreferencePage_h
#define QmitkSegmentAnythingPreferencePage_h
 
#include <berryIQtPreferencePage.h>
#include <QmitknnUNetGPU.h>
#include <QString>
#include <QmitkSetupVirtualEnvUtil.h>
#include <QStandardPaths>
#include <ui_QmitkSegmentAnythingPreferencePage.h>
#include <QDir>
 
class QWidget;
 
namespace Ui
{
  class QmitkSegmentAnythingPreferencePage;
}
 
class QmitkSAMInstaller : public QmitkSetupVirtualEnvUtil
{
public:
  const QString VENV_NAME = ".sam";
  const QString SAM_VERSION = "1.0"; // currently, unused
  const std::vector<QString> PACKAGES = {QString("git+https://github.com/MIC-DKFZ/agent-sam.git@v0.2")};
  const QString STORAGE_DIR;
  inline QmitkSAMInstaller(
    const QString baseDir = QStandardPaths::writableLocation(QStandardPaths::GenericDataLocation) + QDir::separator() +
                            qApp->organizationName() + QDir::separator())
    : QmitkSetupVirtualEnvUtil(baseDir), STORAGE_DIR(baseDir){};
  bool SetupVirtualEnv(const QString &) override;
  QString GetVirtualEnvPath() override;
};
 
class QmitkSegmentAnythingPreferencePage : public QObject, public berry::IQtPreferencePage
{
  Q_OBJECT
  Q_INTERFACES(berry::IPreferencePage)
 
public:
  QmitkSegmentAnythingPreferencePage();
  ~QmitkSegmentAnythingPreferencePage() override;
 
  void Init(berry::IWorkbench::Pointer workbench) override;
  void CreateQtControl(QWidget* parent) override;
  QWidget* GetQtControl() const override;
  bool PerformOk() override;
  void PerformCancel() override;
  void Update() override;
 
private slots:
  void OnInstallBtnClicked();
  void OnClearInstall();
  std::pair<QString, QString> OnSystemPythonChanged(const QString &);
 
protected:
  void AutoParsePythonPaths();
 
  QString GetPythonPathFromUI(const QString &) const;
 
  void SetGPUInfo();
 
  int FetchSelectedGPUFromUI() const;
 
  void WriteStatusMessage(const QString &);
  void WriteErrorMessage(const QString &);
 
private:
  Ui::QmitkSegmentAnythingPreferencePage* m_Ui;
  QmitkSAMInstaller m_Installer;
  QWidget* m_Control;
  QmitkGPULoader m_GpuLoader;
  QString m_PythonPath;
  const QString CPU_ID = "cpu";
  const QStringList VALID_MODELS = {"vit_b", "vit_l", "vit_h"};
};
 
#endif