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 <MitkSegmentationUIExports.h>
#include <mitkIPreferences.h>
#include <QMutex>
#include <QMessageBox>
 
namespace Ui
{
  class QmitkTotalSegmentatorToolGUIControls;
}
 
/*
  \ingroup org_mitk_gui_qt_interactivesegmentation_internal
  \brief GUI for mitk::TotalSegmentatorTool.
*/
class MITKSEGMENTATIONUI_EXPORT QmitkTotalSegmentatorToolGUI : public QmitkMultiLabelSegWithPreviewToolGUIBase
{
  Q_OBJECT
 
public:
  mitkClassMacro(QmitkTotalSegmentatorToolGUI, QmitkMultiLabelSegWithPreviewToolGUIBase);
  itkFactorylessNewMacro(Self);
  itkCloneMacro(Self);
 
protected slots:
 
  void OnPreviewBtnClicked();
 
protected:
  QmitkTotalSegmentatorToolGUI();
  ~QmitkTotalSegmentatorToolGUI() override;
 
  void ConnectNewTool(mitk::SegWithPreviewTool* newTool) override;
  void InitializeUI(QBoxLayout* mainLayout) override;
 
  void EnableAll(bool);
  
  void ToggleLicensedTasks(bool activate);
 
  void ShowErrorMessage(const std::string&, QMessageBox::Icon = QMessageBox::Critical);
 
  void WriteStatusMessage(const QString&);
 
  void WriteErrorMessage(const QString&);
 
  void OnPreferenceChangedEvent(const mitk::IPreferences::ChangeEvent&);
 
  void DownloadStatusWorker(const bool isDownloading);
 
  QMutex m_Mutex;
  Ui::QmitkTotalSegmentatorToolGUIControls* m_Controls;
  bool m_FirstPreviewComputation = true;
  EnableConfirmSegBtnFunctionType m_SuperclassEnableConfirmSegBtnFnc;
  mitk::IPreferences* m_Preferences;
  const QStringList VALID_TASKS = {
    "total",
    "total_mr",
    "cerebral_bleed",
    "hip_implant",
    "body",
    "lung_vessels",
    "pleural_pericard_effusion",
    "head_glands_cavities",
    "head_muscles",
    "headneck_bones_vessels",
    "headneck_muscles",
    "liver_vessels",
    "oculomotor_muscles",
    "lung_nodules",
    "kidney_cysts",
    "breasts",
    "liver_segments",
    "liver_segments_mr",
    "craniofacial_structures"
  };
 
  const QStringList LICENSED_TASKS = {"heartchambers_highres",
                                      "appendicular_bones",
                                      "appendicular_bones_mr",
                                      "tissue_types",
                                      "tissue_types_mr",
                                      "tissue_4_types",
                                      "brain_structures",
                                      "vertebrae_body",
                                      "face",
                                      "face_mr",
                                      "thigh_shoulder_muscles",
                                      "thigh_shoulder_muscles_mr",
                                      "coronary_arteries"};
 
};
 
#endif