mitkFeedbackContourTool.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 mitkFeedbackContourTool_h
#define mitkFeedbackContourTool_h
 
#include <mitkCommon.h>
#include <mitkContourModelUtils.h>
#include <mitkImage.h>
#include <mitkSegTool2D.h>
#include <MitkSegmentationExports.h>
 
#include <mitkDataNode.h>
 
#include <mitkImageCast.h>
 
namespace mitk
{
  class MITKSEGMENTATION_EXPORT FeedbackContourTool : public SegTool2D
  {
  public:
    mitkClassMacro(FeedbackContourTool, SegTool2D);
 
  protected:
    FeedbackContourTool(const char *); // purposely hidden
    ~FeedbackContourTool() override;
 
    const ContourModel *GetFeedbackContour() const;
 
    void InitializeFeedbackContour(bool isClosed);
 
    void ClearsCurrentFeedbackContour(bool isClosed);
 
    void UpdateCurrentFeedbackContour(const ContourModel* sourceModel, TimeStepType sourceTimeStep = 0);
    void UpdateFeedbackContour(const ContourModel* sourceModel, TimeStepType feedbackTimeStep, TimeStepType sourceTimeStep = 0);
 
    void AddVertexToCurrentFeedbackContour(const Point3D& point);
 
    void AddVertexToFeedbackContour(const Point3D& point, TimeStepType feedbackTimeStep);
 
    void SetFeedbackContourVisible(bool);
 
    void SetFeedbackContourColor(float r, float g, float b);
    void SetFeedbackContourColor(const Color& color);
    void SetFeedbackContourColorDefault();
    void SetFeedbackContourWidth(float width);
 
    void Deactivated() override;
 
    void Activated() override;
 
    ContourModel::Pointer ProjectContourTo2DSlice(const Image *slice,
                                                  const ContourModel *contourIn3D);
 
    ContourModel::Pointer BackProjectContourFrom2DSlice(const BaseGeometry *sliceGeometry,
                                                        const ContourModel *contourIn2D);
 
    void WriteBackFeedbackContourAsSegmentationResult(const InteractionPositionEvent* positionEvent, MultiLabelSegmentation::LabelValueType labelValue, bool addMode, bool setInvisibleAfterSuccess = true);
 
    mitk::Image::Pointer GenerateSliceWithContourUpdate(const MultiLabelSegmentation* seg, const PlaneGeometry* sliceGeometry,
      const ContourModel* contour, MultiLabelSegmentation::LabelValueType labelValue, TimePointType timePoint, bool addMode);
 
  private:
    ContourModel::Pointer m_FeedbackContour;
    DataNode::Pointer m_FeedbackContourNode;
    bool m_FeedbackContourVisible;
  };
 
} // namespace
 
#endif