Function loadMaskImage

Modules/MatchPointRegistration/cmdapps/MatchImage.cpp:208–249

};

mitk::Image::Pointer loadMaskImage(std::string filepath, std::string labelName = "")
{
  auto maskRaw = mitk::IOUtil::Load(filepath).front();

  if (auto maskImage = dynamic_cast<mitk::Image*>(maskRaw.GetPointer()))
    return maskImage;

  if (auto maskMultiLabelImage = dynamic_cast<mitk::MultiLabelSegmentation*>(maskRaw.GetPointer()))
  {
    if (maskMultiLabelImage->GetTotalNumberOfLabels() == 1)
      return mitk::CreateLabelMask(maskMultiLabelImage, maskMultiLabelImage->GetAllLabelValues().front());

    if (labelName.empty())
    {
      MITK_INFO << "Selected mask has multiple labels. Using first one. If you want a certain label, please specify via the --movingMaskLabel / --targetMaskLabel argument.";
      return mitk::CreateLabelMask(maskMultiLabelImage, maskMultiLabelImage->GetAllLabelValues().front());
    }

    mitk::MultiLabelSegmentation::LabelValueVectorType labelValues;
    for (mitk::MultiLabelSegmentation::GroupIndexType groupIndex = 0; groupIndex < maskMultiLabelImage->GetNumberOfGroups(); ++groupIndex)
    {
      auto groupLabels = maskMultiLabelImage->GetLabelValuesByName(groupIndex, labelName);
      labelValues.insert(labelValues.end(), groupLabels.begin(), groupLabels.end());
    }

    if (labelValues.size() == 0)
    {
      MITK_ERROR << "No label '" << labelName << "' found for the segmentation " << filepath;
      return nullptr;
    }
    if (labelValues.size() > 1)
    {
      MITK_ERROR << "Multiple labels called '" << labelName << "' found for the segmentation " << filepath;
      return nullptr;
    }

    return mitk::CreateLabelMask(maskMultiLabelImage, labelValues.front());
  }

  MITK_ERROR << "Selected image mask is neither an image nor a MultiLabelSegmentation: " << filepath;
  return nullptr;
}

template <typename TValueType>
map::core::MetaPropertyBase::Pointer