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

#include <mitkSegTool2D.h>
#include <mitkContourModelLiveWireInteractor.h>

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

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

    void ConfirmSegmentation();

    void ClearSegmentation();

  protected:
    LiveWireTool2D();
    ~LiveWireTool2D() override;

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

  private:
    void OnInitLiveWire(StateMachineAction *, InteractionEvent *interactionEvent);

    void OnAddPoint(StateMachineAction *, InteractionEvent *interactionEvent);

    void OnMouseMoved(StateMachineAction *, InteractionEvent *interactionEvent);

    bool OnCheckPoint(const InteractionEvent *interactionEvent);

    void OnFinish(StateMachineAction *, InteractionEvent *interactionEvent);

    void OnLastSegmentDelete(StateMachineAction *, InteractionEvent *interactionEvent);

    void OnMouseMoveNoDynamicCosts(StateMachineAction *, InteractionEvent *interactionEvent);

    void FinishTool();

    void EnableContourLiveWireInteraction(bool on);

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

    void ReleaseInteractors();

    void ReleaseHelperObjects();

    void RemoveHelperObjects();

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

    mitk::ContourModel::Pointer m_Contour;
    mitk::DataNode::Pointer m_ContourNode;

    mitk::ContourModel::Pointer m_LiveWireContour;
    mitk::DataNode::Pointer m_LiveWireContourNode;

    mitk::ContourModel::Pointer m_EditingContour;
    mitk::DataNode::Pointer m_EditingContourNode;
    mitk::ContourModelLiveWireInteractor::Pointer m_ContourInteractor;

    mitk::Image::Pointer m_WorkingSlice;

    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_LiveWireInteractors;

    PlaneGeometry::ConstPointer m_PlaneGeometry;
  };
}

#endif