mitkRegionGrowingTool.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 mitkRegionGrowingTool_h_Included
#define mitkRegionGrowingTool_h_Included

#include "mitkFeedbackContourTool.h"
#include <MitkSegmentationExports.h>
#include <array>

namespace us
{
  class ModuleResource;
}

namespace mitk
{
  class MITKSEGMENTATION_EXPORT RegionGrowingTool : public FeedbackContourTool
  {
  public:
    mitkClassMacro(RegionGrowingTool, FeedbackContourTool);
    itkFactorylessNewMacro(Self);
    itkCloneMacro(Self);

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

    const char *GetName() const override;

  protected:
    RegionGrowingTool(); // purposely hidden
    ~RegionGrowingTool() override;

    void ConnectActionsAndFunctions() override;

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

    virtual void OnMousePressed(StateMachineAction *, InteractionEvent *interactionEvent);

    virtual void OnMousePressedInside();

    virtual void OnMousePressedOutside(StateMachineAction *, InteractionEvent *interactionEvent);

    virtual void OnMouseMoved(StateMachineAction *, InteractionEvent *interactionEvent);

    virtual void OnMouseReleased(StateMachineAction *, InteractionEvent *interactionEvent);

    template <typename TPixel, unsigned int imageDimension>
    void GetNeighborhoodAverage(const itk::Image<TPixel, imageDimension> *itkImage,
                                const itk::Index<imageDimension>& index,
                                ScalarType *result,
                                unsigned int neighborhood = 1);

    template <typename TPixel, unsigned int imageDimension>
    void IsInsideSegmentation(const itk::Image<TPixel, imageDimension> *itkImage,
                              const itk::Index<imageDimension>& index,
                              bool *result);

    template <typename TPixel, unsigned int imageDimension>
    void StartRegionGrowing(const itk::Image<TPixel, imageDimension> *itkImage,
                            const itk::Index<imageDimension>& seedPoint,
                            const std::array<ScalarType, 2>& thresholds,
                            mitk::Image::Pointer &outputImage);

    template <typename TPixel, unsigned int imageDimension>
    void CalculateInitialThresholds(const itk::Image<TPixel, imageDimension>* itkImage);

    Image::Pointer m_ReferenceSlice;
    Image::Pointer m_WorkingSlice;

    ScalarType m_SeedValue;
    itk::Index<3> m_SeedPoint;
    std::array<ScalarType, 2> m_ThresholdExtrema;
    std::array<ScalarType, 2> m_Thresholds;
    std::array<ScalarType, 2> m_InitialThresholds;
    Point2I m_LastScreenPosition;
    int m_ScreenYDifference;
    int m_ScreenXDifference;

  private:
    ScalarType m_MouseDistanceScaleFactor;
    int m_PaintingPixelValue;
    bool m_FillFeedbackContour;
    int m_ConnectedComponentValue;
  };

} // namespace

#endif