QmitkTotalSegmentatorToolGUI.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 QmitkTotalSegmentatorToolGUI_h_Included
#define QmitkTotalSegmentatorToolGUI_h_Included
 
#include "QmitkMultiLabelSegWithPreviewToolGUIBase.h"
#include "QmitkSetupVirtualEnvUtil.h"
#include "QmitknnUNetGPU.h"
#include "ui_QmitkTotalSegmentatorGUIControls.h"
#include <MitkSegmentationUIExports.h>
#include <QMessageBox>
#include <QSettings>
#include <QStandardPaths>
#include <QDir>
 
class QmitkTotalSegmentatorToolInstaller : public QmitkSetupVirtualEnvUtil
{
public:
  const QString VENV_NAME = ".totalsegmentator_v2";
  const QString TOTALSEGMENTATOR_VERSION = "2.2.1";
  const std::vector<QString> PACKAGES = {QString("Totalsegmentator==") + TOTALSEGMENTATOR_VERSION,
                                         QString("setuptools")}; /* just in case */
  const QString STORAGE_DIR;
  inline QmitkTotalSegmentatorToolInstaller(
    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 MITKSEGMENTATIONUI_EXPORT QmitkTotalSegmentatorToolGUI : public QmitkMultiLabelSegWithPreviewToolGUIBase
{
  Q_OBJECT
 
public:
  mitkClassMacro(QmitkTotalSegmentatorToolGUI, QmitkMultiLabelSegWithPreviewToolGUIBase);
  itkFactorylessNewMacro(Self);
  itkCloneMacro(Self);
 
protected slots:
 
  void OnPreviewBtnClicked();
 
  void OnPythonPathChanged(const QString &);
 
  std::pair<QString, QString> OnSystemPythonChanged(const QString &);
 
  void OnInstallBtnClicked();
 
  void OnOverrideChecked(int);
 
  void OnClearInstall();
 
protected:
  QmitkTotalSegmentatorToolGUI();
  ~QmitkTotalSegmentatorToolGUI() = default;
 
  void ConnectNewTool(mitk::SegWithPreviewTool *newTool) override;
  void InitializeUI(QBoxLayout *mainLayout) override;
 
  void EnableAll(bool);
 
  void AutoParsePythonPaths();
 
  bool IsTotalSegmentatorInstalled(const QString &);
 
  void ShowErrorMessage(const std::string &, QMessageBox::Icon = QMessageBox::Critical);
 
  void WriteStatusMessage(const QString &);
 
  void WriteErrorMessage(const QString &);
 
  void SetGPUInfo();
 
  unsigned int FetchSelectedGPUFromUI() const;
 
  QString GetPythonPathFromUI(const QString &) const;
 
  QSettings m_Settings;
 
  QString m_PythonPath;
  QmitkGPULoader m_GpuLoader;
  Ui_QmitkTotalSegmentatorToolGUIControls m_Controls;
  bool m_FirstPreviewComputation = true;
  bool m_IsInstalled = false;
  EnableConfirmSegBtnFunctionType m_SuperclassEnableConfirmSegBtnFnc;
 
  const std::string WARNING_TOTALSEG_NOT_FOUND =
    "TotalSegmentator is not detected in the selected python environment.Please select a valid "
    "python environment or install TotalSegmentator.";
  const QStringList VALID_TASKS = {
    "total",
    "total_mr",
    "cerebral_bleed",
    "hip_implant",
    "coronary_arteries",
    "body",
    "lung_vessels",
    "pleural_pericard_effusion"
  };
  QmitkTotalSegmentatorToolInstaller m_Installer;
};
#endif