mitkFastMarchingBaseTool.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 mitkFastMarchingBaseTool_h_Included
#define mitkFastMarchingBaseTool_h_Included

#include "mitkAutoSegmentationWithPreviewTool.h"
#include "mitkDataNode.h"
#include "mitkPointSet.h"
#include "mitkPointSetDataInteractor.h"
#include "mitkToolCommand.h"

#include "itkImage.h"

#include <MitkSegmentationExports.h>

namespace us
{
  class ModuleResource;
}

namespace mitk
{
  class MITKSEGMENTATION_EXPORT FastMarchingBaseTool : public AutoSegmentationWithPreviewTool
  {
  public:
    mitkClassMacro(FastMarchingBaseTool, AutoSegmentationWithPreviewTool);

    bool CanHandle(const BaseData* referenceData, const BaseData* workingData) const override;

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

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

    void SetUpperThreshold(double);

    void SetLowerThreshold(double);

    void SetStoppingValue(double);

    void SetSigma(double);

    void SetAlpha(double);

    void SetBeta(double);

    void ClearSeeds();


  protected:
    FastMarchingBaseTool(unsigned int toolDim);
    ~FastMarchingBaseTool() override;

    void ConnectActionsAndFunctions() override;

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

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

    void DoUpdatePreview(const Image* inputAtTimeStep, const Image* oldSegAtTimeStep, Image* previewImage, TimeStepType timeStep) override;

    template <typename TPixel, unsigned int VImageDimension>
    void DoITKFastMarching(const itk::Image<TPixel, VImageDimension>* inputImage,
      Image* segmentation, unsigned int timeStep, const BaseGeometry* inputGeometry);

    float m_LowerThreshold; // used in Threshold filter
    float m_UpperThreshold; // used in Threshold filter
    float m_StoppingValue;  // used in Fast Marching filter
    float m_Sigma;          // used in GradientMagnitude filter
    float m_Alpha;          // used in Sigmoid filter
    float m_Beta;           // used in Sigmoid filter

    DataNode::Pointer m_SeedsAsPointSetNode; // used to visualize the seed points
    PointSet::Pointer m_SeedsAsPointSet;

  private:
    unsigned int m_ToolDimension;

  };

} // namespace

#endif