mitkContourModelLiveWireInteractor.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 mitkContourModelLiveWireInteractor_h_Included
#define mitkContourModelLiveWireInteractor_h_Included

#include "mitkCommon.h"
#include "mitkContourModelInteractor.h"
#include <MitkSegmentationExports.h>

#include <mitkImageLiveWireContourModelFilter.h>

namespace mitk
{
  class MITKSEGMENTATION_EXPORT ContourModelLiveWireInteractor : public ContourModelInteractor
  {
  public:
    mitkClassMacro(ContourModelLiveWireInteractor, ContourModelInteractor);
    itkFactorylessNewMacro(Self);
    itkCloneMacro(Self);

        virtual void SetEditingContourModelNode(mitk::DataNode *_arg);

    virtual void SetWorkingImage(mitk::Image *_arg);

    void ConnectActionsAndFunctions() override;

  protected:
    ContourModelLiveWireInteractor();
    ~ContourModelLiveWireInteractor() override;

    bool OnCheckPointClick(const InteractionEvent *interactionEvent) override;
    bool IsHovering(const InteractionEvent *interactionEvent) override;

    void OnMovePoint(StateMachineAction *, InteractionEvent *interactionEvent) override;
    void OnDeletePoint(StateMachineAction *, InteractionEvent *interactionEvent) override;
    void OnFinishEditing(StateMachineAction *, InteractionEvent *interactionEvent) override;

    void SplitContourFromSelectedVertex(mitk::ContourModel *srcContour,
                                        const mitk::ContourModel::VertexType *nextPoint,
                                        const mitk::ContourModel::VertexType *previousPoint,
                                        int timestep);
    void SetRepulsivePoints(const mitk::ContourModel::VertexType *nextPoint,
                            mitk::ContourModel *contour,
                            int timestep);

    const float eps = 3.0;
    mitk::ImageLiveWireContourModelFilter::Pointer m_LiveWireFilter;
    mitk::Image::Pointer m_WorkingSlice;

    mitk::Point3D m_NextActiveVertexDown;
    mitk::Point3D m_NextActiveVertexUp;

    mitk::ContourModel::VertexIterator m_NextActiveVertexDownIter;
    mitk::ContourModel::VertexIterator m_NextActiveVertexUpIter;

    std::vector<itk::Index<2>> m_ContourBeingModified;

    mitk::DataNode::Pointer m_EditingContourNode;
    mitk::ContourModel::Pointer m_ContourLeft;
    mitk::ContourModel::Pointer m_ContourRight;
  };

} // namespace mitk

#endif // mitkContourModelLiveWireInteractor_h_Included