mitkWatershedTool.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 mitkWatershedTool_h_Included
#define mitkWatershedTool_h_Included

#include "mitkAutoMLSegmentationWithPreviewTool.h"
#include "mitkCommon.h"
#include <MitkSegmentationExports.h>

namespace us
{
  class ModuleResource;
}

namespace mitk
{
  class MITKSEGMENTATION_EXPORT WatershedTool : public AutoMLSegmentationWithPreviewTool
  {
  public:
    mitkClassMacro(WatershedTool, AutoMLSegmentationWithPreviewTool);
    itkFactorylessNewMacro(Self);
    itkCloneMacro(Self);

    const char** GetXPM() const override;
    const char* GetName() const override;
    us::ModuleResource GetIconResource() const override;

    void Activated() override;

    itkSetMacro(Threshold, double);
    itkGetConstMacro(Threshold, double);

    itkSetMacro(Level, double);
    itkGetConstMacro(Level, double);

  protected:
    WatershedTool() = default;
    ~WatershedTool() = default;

    LabelSetImage::Pointer ComputeMLPreview(const Image* inputAtTimeStep, TimeStepType timeStep) override;

    double m_Threshold = 0.0;
    double m_Level = 0.0;

private:
    template <typename TPixel, unsigned int VImageDimension>
    void ITKWatershed(const itk::Image<TPixel, VImageDimension>* originalImage, itk::SmartPointer<mitk::Image>& segmentation, bool inputChanged);

    itk::ProcessObject::Pointer m_MagFilter;
    itk::ProcessObject::Pointer m_WatershedFilter;
    mitk::Image::ConstPointer m_LastFilterInput;
  };

} // namespace

#endif