QmitkAdaptiveRegionGrowingToolGUI.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 QMITK_QmitkAdaptiveRegionGrowingToolGUI_H
#define QMITK_QmitkAdaptiveRegionGrowingToolGUI_H

#include "itkImage.h"
#include "mitkDataStorage.h"
#include "mitkGeometry3D.h"
#include "mitkPointSet.h"

#include "qwidget.h"
#include "ui_QmitkAdaptiveRegionGrowingToolGUIControls.h"

#include <MitkSegmentationUIExports.h>

#include "QmitkToolGUI.h"

#include "mitkAdaptiveRegionGrowingTool.h"

class DataNode;
class QmitkAdaptiveRegionGrowingToolGUIControls;

class MITKSEGMENTATIONUI_EXPORT QmitkAdaptiveRegionGrowingToolGUI : public QmitkToolGUI
{
  Q_OBJECT

public:
  mitkClassMacro(QmitkAdaptiveRegionGrowingToolGUI, QmitkToolGUI);

  itkFactorylessNewMacro(Self);

  itkCloneMacro(Self);

    QmitkAdaptiveRegionGrowingToolGUI(QWidget *parent = nullptr);

  virtual void CreateConnections();

  virtual void SetDataStorage(mitk::DataStorage *dataStorage);

  void SetDataNodeNames(std::string labledSegmentation,
                        std::string binaryImage,
                        /*std::string vesselTree,*/ std::string surface,
                        std::string maskedSegmentation);

  void EnableControls(bool enable);

  void SetInputImageNode(mitk::DataNode *node);

  void Deactivated();
  void Activated();

  Ui::QmitkAdaptiveRegionGrowingToolGUIControls m_Controls;

protected slots:

  void RunSegmentation();

  void ChangeLevelWindow(double newValue);

  void IncreaseSlider();

  void DecreaseSlider();

  void ConfirmSegmentation();

  void UseVolumeRendering(bool on);

  void SetLowerThresholdValue(double lowerThreshold);

  void SetUpperThresholdValue(double upperThreshold);

  void OnNewToolAssociated(mitk::Tool *);

protected:
  mitk::AdaptiveRegionGrowingTool::Pointer m_RegionGrow3DTool;

  ~QmitkAdaptiveRegionGrowingToolGUI() override;

  mitk::DataStorage *m_DataStorage;

  mitk::DataNode::Pointer m_InputImageNode;

  void OnPointAdded();

private:
  std::string m_NAMEFORORGIMAGE;
  std::string m_NAMEFORLABLEDSEGMENTATIONIMAGE;
  std::string m_NAMEFORBINARYIMAGE;
  std::string m_NAMEFORSURFACE;
  std::string m_NAMEFORMASKEDSEGMENTATION;

  mitk::ScalarType m_LOWERTHRESHOLD; // Hounsfield value
  mitk::ScalarType m_UPPERTHRESHOLD; // Hounsfield value
  mitk::ScalarType m_SeedPointValueMean;

  void RemoveHelperNodes();

  int m_DetectedLeakagePoint;

  bool m_CurrentRGDirectionIsUpwards; // defines fixed threshold (true = LOWERTHRESHOLD fixed, false = UPPERTHRESHOLD
                                      // fixed)

  int m_SeedpointValue;
  bool m_SliderInitialized;
  bool m_UseVolumeRendering;
  bool m_UpdateSuggestedThreshold;
  float m_SuggestedThValue;

  long m_PointSetAddObserverTag;
  long m_PointSetMoveObserverTag;

  template <typename TPixel, unsigned int VImageDimension>
  void StartRegionGrowing(itk::Image<TPixel, VImageDimension> *itkImage,
                          mitk::BaseGeometry *imageGeometry,
                          mitk::PointSet::PointType seedPoint);

  template <typename TPixel, unsigned int VImageDimension>
  void ITKThresholding(itk::Image<TPixel, VImageDimension> *inputImage);

  void InitializeLevelWindow();

  void EnableVolumeRendering(bool enable);

  void UpdateVolumeRenderingThreshold(int thValue);
};

#endif