Method DoUpdatePreview

Modules/PythonSegmentation/src/mitknnInteractiveTool.cpp:530–648

}

void mitk::nnInteractiveTool::DoUpdatePreview(const Image* inputAtTimeStep, const Image* /*oldSegAtTimeStep*/, MultiLabelSegmentation* previewImage, TimeStepType timeStep)
{
  // This method assumes it is only called when an interaction has occurred or
  // when a session should be (re)initialized with a label mask. Otherwise,
  // calling this method will reset any existing preview content.

  if (previewImage == nullptr || m_Impl->GetPythonContext() == nullptr)
    return;

  // Color the preview with the selected target label's color (falling back to
  // the special preview color) so it matches the label being segmented.
  auto previewColor = this->GetSpecialPreviewColor();

  if (const auto* segmentation = this->GetTargetSegmentation(); segmentation != nullptr)
  {
    if (const auto* activeLabel = segmentation->GetActiveLabel(); activeLabel != nullptr)
      previewColor = activeLabel->GetColor();
  }

  this->SetPreviewLabel(1, previewColor);

  // An undo restored the target buffer in place; just repaint the preview from
  // it. This must run before the interactor branch (an interactor is typically
  // still enabled while undoing) and must not emit PreviewUpdatedEvent, so it
  // does not trigger the GUI's auto-confirm.
  if (m_Impl->UndoRefreshPending)
  {
    m_Impl->UndoRefreshPending = false;
    previewImage->UpdateGroupImage(previewImage->GetActiveLayer(), m_Impl->TargetBuffer, timeStep, 0);
    return;
  }

  const auto* interactor = m_Impl->GetEnabledInteractor();

  try
  {
    if (interactor != nullptr)
    {
      switch (interactor->GetType())
      {
        case InteractionType::Point:
        {
          auto point = static_cast<const PointInteractor*>(interactor)->GetLastPoint();
          m_Impl->AddPointInteraction(point.value(), inputAtTimeStep);
          break;
        }
        case InteractionType::Box:
        {
          auto box = static_cast<const BoxInteractor*>(interactor)->GetLastBox();
          m_Impl->AddBoxInteraction(box, inputAtTimeStep);
          break;
        }
        case InteractionType::Scribble:
        {
          auto scribbleInteractor = static_cast<const ScribbleInteractor*>(interactor);
          auto mask = scribbleInteractor->GetLastScribbleMask();
          if (mask.IsNull())
          {