mitkFeedbackContourTool.cpp

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.

============================================================================*/

#include "mitkFeedbackContourTool.h"
#include "mitkToolManager.h"

#include "mitkColorProperty.h"
#include "mitkProperties.h"
#include "mitkStringProperty.h"

#include "mitkBaseRenderer.h"
#include "mitkDataStorage.h"

#include "mitkRenderingManager.h"

mitk::FeedbackContourTool::FeedbackContourTool(const char *type) : SegTool2D(type), m_FeedbackContourVisible(false)
{
  m_FeedbackContour = ContourModel::New();
  m_FeedbackContour->SetClosed(true);
  m_FeedbackContourNode = DataNode::New();
  m_FeedbackContourNode->SetData(m_FeedbackContour);
  m_FeedbackContourNode->SetProperty("name", StringProperty::New("One of FeedbackContourTool's feedback nodes"));
  m_FeedbackContourNode->SetProperty("visible", BoolProperty::New(true));
  m_FeedbackContourNode->SetProperty("helper object", BoolProperty::New(true));
  m_FeedbackContourNode->SetProperty("layer", IntProperty::New(1000));
  m_FeedbackContourNode->SetProperty("contour.project-onto-plane", BoolProperty::New(false));
  m_FeedbackContourNode->SetProperty("contour.width", FloatProperty::New(1.0));
  // set to max short, max int doesn't work, max value somewhere hardcoded?
  m_FeedbackContourNode->SetProperty("layer", IntProperty::New(std::numeric_limits<short>::max()));
  m_FeedbackContourNode->SetProperty("fixedLayer", BoolProperty::New(true));

  SetFeedbackContourColorDefault();
}

mitk::FeedbackContourTool::~FeedbackContourTool()
{
}

void mitk::FeedbackContourTool::SetFeedbackContourColor(float r, float g, float b)
{
  m_FeedbackContourNode->SetProperty("contour.color", ColorProperty::New(r, g, b));
}

void mitk::FeedbackContourTool::SetFeedbackContourColorDefault()
{
  m_FeedbackContourNode->SetProperty("contour.color", ColorProperty::New(0.0 / 255.0, 255.0 / 255.0, 0.0 / 255.0));
}

void mitk::FeedbackContourTool::Deactivated()
{
  Superclass::Deactivated();
  DataStorage *storage = m_ToolManager->GetDataStorage();
  if (storage && m_FeedbackContourNode.IsNotNull())
  {
    storage->Remove(m_FeedbackContourNode);
    m_FeedbackContour->Clear();
    SetFeedbackContourVisible(false);
  }
}

void mitk::FeedbackContourTool::Activated()
{
  Superclass::Activated();

  SetFeedbackContourVisible(true);
}

mitk::ContourModel *mitk::FeedbackContourTool::GetFeedbackContour()
{
  return m_FeedbackContour;
}

void mitk::FeedbackContourTool::SetFeedbackContour(ContourModel::Pointer contour)
{
  m_FeedbackContour = contour;
  m_FeedbackContourNode->SetData(m_FeedbackContour);
}

void mitk::FeedbackContourTool::SetFeedbackContourVisible(bool visible)
{
  if (m_FeedbackContourVisible == visible)
    return; // nothing changed

  if (DataStorage *storage = m_ToolManager->GetDataStorage())
  {
    if (visible)
    {
      // Add the feedback contour node as a derived node of the first working data.
      // If there is no working data, the node is added at the top level.
      storage->Add(m_FeedbackContourNode, m_ToolManager->GetWorkingData(0));
    }
    else
    {
      storage->Remove(m_FeedbackContourNode);
    }
  }

  m_FeedbackContourVisible = visible;
}

mitk::ContourModel::Pointer mitk::FeedbackContourTool::ProjectContourTo2DSlice(Image *slice,
                                                                               ContourModel *contourIn3D,
                                                                               bool correctionForIpSegmentation,
                                                                               bool constrainToInside)
{
  return mitk::ContourModelUtils::ProjectContourTo2DSlice(
    slice, contourIn3D, correctionForIpSegmentation, constrainToInside);
}

mitk::ContourModel::Pointer mitk::FeedbackContourTool::BackProjectContourFrom2DSlice(const BaseGeometry *sliceGeometry,
                                                                                     ContourModel *contourIn2D,
                                                                                     bool correctionForIpSegmentation)
{
  return mitk::ContourModelUtils::BackProjectContourFrom2DSlice(
    sliceGeometry, contourIn2D, correctionForIpSegmentation);
}

void mitk::FeedbackContourTool::FillContourInSlice(ContourModel *projectedContour,
                                                   Image *sliceImage,
                                                   int paintingPixelValue)
{
  this->FillContourInSlice(projectedContour, 0, sliceImage, paintingPixelValue);
}

void mitk::FeedbackContourTool::FillContourInSlice(ContourModel *projectedContour,
                                                   unsigned int timeStep,
                                                   Image *sliceImage,
                                                   int paintingPixelValue)
{
  mitk::ContourModelUtils::FillContourInSlice(projectedContour, timeStep, sliceImage, sliceImage, paintingPixelValue);
}