mitkWatershedTool.h

Go to the documentation of this file.

/*===================================================================

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center,
Division of Medical and Biological Informatics.
All rights reserved.

This software is distributed WITHOUT ANY WARRANTY; without
even the implied warranty of MERCHANTABILITY or FITNESS FOR
A PARTICULAR PURPOSE.

See LICENSE.txt or http://www.mitk.org for details.

===================================================================*/

#ifndef mitkWatershedTool_h_Included
#define mitkWatershedTool_h_Included

#include "mitkAutoSegmentationTool.h"
#include "mitkCommon.h"
#include <MitkSegmentationExports.h>
#include <itkImage.h>

namespace us
{
  class ModuleResource;
}

namespace mitk
{
  class Image;

  class MITKSEGMENTATION_EXPORT WatershedTool : public AutoSegmentationTool
  {
  public:
    mitkClassMacro(WatershedTool, AutoSegmentationTool);
    itkFactorylessNewMacro(Self) itkCloneMacro(Self)

      void SetThreshold(double t)
    {
      m_Threshold = t;
    }

    void SetLevel(double l) { m_Level = l; }
    void DoIt();

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

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

  protected:
    WatershedTool(); // purposely hidden
    virtual ~WatershedTool();

    virtual void Activated() override;
    virtual void Deactivated() override;

    double m_Threshold;
    double m_Level;
  };

} // namespace

#endif