Method DoUpdatePreview

Modules/PythonSegmentation/src/mitkTotalSegmentatorTool.cpp:152–223

}

void mitk::TotalSegmentatorTool::DoUpdatePreview(const Image *inputAtTimeStep,
                                                 const Image * /*oldSegAtTimeStep*/,
                                                 MultiLabelSegmentation *previewImage,
                                                 TimeStepType timeStep)
{
  m_LastErrorMessage.clear();

  // The base UpdatePreview catches itk::ExceptionObject (what mitkThrow produces)
  // and only forwards it to ErrorMessage, which no GUI here listens to. So catch
  // every failure locally and record it in m_LastErrorMessage; the GUI shows that
  // instead of the generic "cancelled or produced no result".
  try
  {
    // Programmer-guarantee guards: the GUI always injects both before running.
    if (m_ExecutablePath.empty())
      mitkThrow() << "TotalSegmentator executable path is not set.";

    if (!m_CommandRunner)
      mitkThrow() << "No command runner is set for the TotalSegmentator tool.";

    if (m_MitkTempDir.empty())
      m_MitkTempDir = IOUtil::CreateTemporaryDirectory("mitk-XXXXXX");

    const std::string inDir = IOUtil::CreateTemporaryDirectory("totalseg-in-XXXXXX", m_MitkTempDir);
    const std::string outDir = IOUtil::CreateTemporaryDirectory("totalseg-out-XXXXXX", m_MitkTempDir);

    // Remove this run's scratch directories on every exit path; only the parent
    // m_MitkTempDir persists (until the destructor), so repeated runs in one tool
    // activation do not pile up input copies and outputs.
    const ScopedDir inDirGuard{inDir};
    const ScopedDir outDirGuard{outDir};

    std::ofstream tmpStream;
    const std::string inputImagePath = IOUtil::CreateTemporaryFile(tmpStream, "XXXXXX.nii.gz", inDir);
    tmpStream.close();

    const std::string outputImagePath = outDir + IOUtil::GetDirectorySeparator() + "segmentation.nii.gz";

    IOUtil::Save(inputAtTimeStep, inputImagePath);

    const auto args = this->BuildArguments(inputImagePath, outputImagePath);
    const bool success = m_CommandRunner(m_ExecutablePath, args);

    if (!success)
      return; // Cancelled by the user or the process failed: leave the preview empty.

    if (!fs::exists(outputImagePath))
      mitkThrow() << "TotalSegmentator finished but did not produce an output file.";

    Image::Pointer outputImage = IOUtil::Load<Image>(outputImagePath);
    outputImage->SetGeometry(inputAtTimeStep->GetGeometry());

    auto outputBuffer = mitk::MultiLabelSegmentation::New();
    outputBuffer->InitializeByLabeledImage(outputImage);

    this->MapLabelsToSegmentation(outputBuffer, previewImage);
    previewImage->UpdateGroupImage(
      previewImage->GetActiveLayer(), outputBuffer->GetGroupImage(outputBuffer->GetActiveLayer()), timeStep);