mitkLiveWireTool2D.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 mitkCorrectorTool2D_h_Included
#define mitkCorrectorTool2D_h_Included

#include "mitkCommon.h"
#include "mitkSegTool2D.h"
#include <MitkSegmentationExports.h>
#include <mitkContourModel.h>
#include <mitkContourModelLiveWireInteractor.h>
#include <mitkImageLiveWireContourModelFilter.h>

namespace us
{
  class ModuleResource;
}

namespace mitk
{
  class MITKSEGMENTATION_EXPORT LiveWireTool2D : public SegTool2D
  {
  public:
    mitkClassMacro(LiveWireTool2D, SegTool2D);
    itkFactorylessNewMacro(Self) itkCloneMacro(Self)

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

    virtual const char *GetName() const override;

    void ConfirmSegmentation();

    void ClearSegmentation();

  protected:
    LiveWireTool2D();
    virtual ~LiveWireTool2D();

    void ConnectActionsAndFunctions() override;

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

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

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

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

    virtual bool OnCheckPoint(const InteractionEvent *interactionEvent);

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

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

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

    void FinishTool();

    virtual bool IsPositionEventInsideImageRegion(InteractionPositionEvent *positionEvent, BaseData *data);

    void EnableContourLiveWireInteraction(bool on);

    // the contour already set by the user
    mitk::ContourModel::Pointer m_Contour;

    // the corresponding datanode
    mitk::DataNode::Pointer m_ContourModelNode;

    // the current LiveWire computed contour
    mitk::ContourModel::Pointer m_LiveWireContour;

    // the corresponding datanode
    mitk::DataNode::Pointer m_LiveWireContourNode;

    // the contour for the editing portion
    mitk::ContourModel::Pointer m_EditingContour;

    // the corresponding datanode
    mitk::DataNode::Pointer m_EditingContourNode;

    // the corresponding contour interactor
    mitk::ContourModelLiveWireInteractor::Pointer m_ContourInteractor;

    // the current reference image
    mitk::Image::Pointer m_WorkingSlice;

    // the filter for live wire calculation
    mitk::ImageLiveWireContourModelFilter::Pointer m_LiveWireFilter;

    bool m_CreateAndUseDynamicCosts;

    std::vector<std::pair<mitk::DataNode::Pointer, mitk::PlaneGeometry::Pointer>> m_WorkingContours;
    std::vector<std::pair<mitk::DataNode::Pointer, mitk::PlaneGeometry::Pointer>> m_EditingContours;
    std::vector<mitk::ContourModelLiveWireInteractor::Pointer> m_LiveWireNodes;

    template <typename TPixel, unsigned int VImageDimension>
    void FindHighestGradientMagnitudeByITK(itk::Image<TPixel, VImageDimension> *inputImage,
                                           itk::Index<3> &index,
                                           itk::Index<3> &returnIndex);

  private:
    void RemoveHelperObjects();
    void ReleaseHelperObjects();
    void ReleaseInteractors();

    const PlaneGeometry *m_PlaneGeometry;
  };

} // namespace

#endif