QmitkTotalSegmentatorPreferencePage.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 QmitkTotalSegmentatorPreferencePage_h
#define QmitkTotalSegmentatorPreferencePage_h
 
#include <berryIQtPreferencePage.h>
#include "mitkIPreferences.h"
#include <ui_QmitkTotalSegmentatorPreferencePage.h>
#include "QmitkTotalSegmentatorToolInstaller.h"
#include <QMessageBox>
#include <QmitknnUNetGPU.h>
#include "QmitkToolInstallDialog.h"
 
namespace Ui
{
  class QmitkTotalSegmentatorPreferencePage;
}
 
class QmitkTotalSegmentatorPreferencePage : public QObject, public berry::IQtPreferencePage
{
  Q_OBJECT
  Q_INTERFACES(berry::IPreferencePage)
 
public:
  QmitkTotalSegmentatorPreferencePage();
  ~QmitkTotalSegmentatorPreferencePage() override = default;
 
  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:
  void ShowErrorMessage(const QString &, QMessageBox::Icon = QMessageBox::Critical);
 
  void WriteStatusMessage(const QString &);
 
  void WriteErrorMessage(const QString &message);
 
  void AutoParsePythonPaths();
 
  void SetDeviceInfo();
 
  void UpdateStatusLabel();
 
  QString GetPythonPathFromUI(const QString &pyUI) const;
 
  bool IsTotalSegmentatorInstalled(const QString &pythonPath);
 
  int FetchSelectedDeviceFromUI() const;
 
  void UpdateTotalSegPreferencePath();
 
  Ui::QmitkTotalSegmentatorPreferencePage *m_Ui;
  QWidget *m_Control;
  QString m_SysPythonPath;
  bool m_IsInstalled; // manual installed or not
  QmitkTotalSegmentatorToolInstaller m_Installer;
  QmitkGPULoader m_DeviceLoader;
  static const QString WARNING_TOTALSEG_NOT_FOUND;
  mitk::IPreferences *m_Preferences;
 
protected slots:
 
  void OnPythonPathChanged(const QString &pyEnv);
 
  std::pair<QString, QString> OnSystemPythonChanged(const QString &pyEnv);
 
  void OnOverrideBoxChecked(int state);
 
  void OnClearButtonClicked();
 
  void OnInstallButtonClicked();
};
 
#endif