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_Included
#define mitkFeedbackContourTool_h_Included

#include "mitkCommon.h"
#include "mitkContourModelUtils.h"
#include "mitkContourUtils.h" //TODO remove legacy support
#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();             // purposely hidden
    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 SetFeedbackContourColorDefault();

    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, int paintingPixelValue, bool setInvisibleAfterSuccess = true);

    void FillContourInSlice(ContourModel *projectedContour, Image *sliceImage, int paintingPixelValue = 1);

    void FillContourInSlice(ContourModel *projectedContour,
                            unsigned int timeStep,
                            Image *sliceImage,
                            int paintingPixelValue = 1);

  private:
    ContourModel::Pointer m_FeedbackContour;
    DataNode::Pointer m_FeedbackContourNode;
    bool m_FeedbackContourVisible;
  };

} // namespace

#endif