( maskingNode->GetData() );
//TODO Get 3D Surface of current time step
if(surface.IsNull())
{
MITK_ERROR << "Selection does not contain a surface";
QMessageBox::information( this, "Image and Surface Masking", "Selection does not contain a surface", QMessageBox::Ok );
this->EnableButtons();
return;
}
mitk::Image::Pointer maskImage = this->ConvertSurfaceToImage( referenceImage, surface );
//2. mask reference image with mask image
if(maskImage.IsNotNull() &&
referenceImage->GetLargestPossibleRegion().GetSize() == maskImage->GetLargestPossibleRegion().GetSize() )
{
resultImage = this->MaskImage( referenceImage, maskImage );
}
}
mitk::ProgressBar::GetInstance()->Progress();
if( resultImage.IsNull() )
{
MITK_ERROR << "Masking failed";
QMessageBox::information( this, "Image and Surface Masking", "Masking failed. For more information please see logging window.", QMessageBox::Ok );
this->EnableButtons();
mitk::ProgressBar::GetInstance()->Progress(4);
return;
}
//Add result to data storage
this->AddToDataStorage(
dataSelectionWidget->GetDataStorage(),
resultImage,
dataSelectionWidget->GetSelection(0)->GetName() + "_" + dataSelectionWidget->GetSelection(1)->GetName(),
dataSelectionWidget->GetSelection(0));
this->EnableButtons();
mitk::ProgressBar::GetInstance()->Progress();
}
mitk::Image::Pointer QmitkImageMaskingWidget::MaskImage(mitk::Image::Pointer referenceImage, mitk::Image::Pointer maskImage )
{
mitk::ScalarType backgroundValue = 0.0;
if (m_Controls.rbnMinimum->isChecked())
{
backgroundValue = referenceImage->GetStatistics()->GetScalarValueMin();
}
else if (m_Controls.rbnCustom->isChecked())
{
+ auto warningTitle = QStringLiteral("Invalid custom pixel value");
+
bool ok = false;
- backgroundValue = m_Controls.txtCustom->text().toDouble(&ok);
+ auto originalBackgroundValue = m_Controls.txtCustom->text().toDouble(&ok);
if (!ok)
{
- // Use zero as default if input is invalid
- backgroundValue = 0.0;
+ // Input is not even a number
+ QMessageBox::warning(nullptr, warningTitle, "Please enter a valid number as custom pixel value.");
+ return nullptr;
}
else
{
// Clamp to the numerical limits of the pixel/component type
double bottom, top;
AccessByItk_n(referenceImage, GetRange, (bottom, top));
- backgroundValue = std::max(bottom, std::min(backgroundValue, top));
+ backgroundValue = std::max(bottom, std::min(originalBackgroundValue, top));
// Get rid of decimals for integral numbers
auto type = referenceImage->GetPixelType().GetComponentType();
if (type != itk::ImageIOBase::FLOAT && type != itk::ImageIOBase::DOUBLE)
- backgroundValue = std::floor(backgroundValue);
+ backgroundValue = std::round(backgroundValue);
}
- // Sync potential changes of the custom value
- m_Controls.txtCustom->setText(QString("%1").arg(backgroundValue));
+ // Ask the user for permission before correcting their input
+ if (std::abs(originalBackgroundValue - backgroundValue) > 1e-4)
+ {
+ auto warningText = QString(
+ "The custom pixel value %1 lies not within the range of valid pixel values for the selected image.
"
+ "Apply the closest valid pixel value %2 instead?
").arg(originalBackgroundValue).arg(backgroundValue);
+
+ auto ret = QMessageBox::warning(
+ nullptr,
+ warningTitle,
+ warningText,
+ QMessageBox::StandardButton::Apply | QMessageBox::StandardButton::Cancel,
+ QMessageBox::StandardButton::Apply);
+
+ if (QMessageBox::StandardButton::Apply != ret)
+ return nullptr;
+
+ m_Controls.txtCustom->setText(QString("%1").arg(backgroundValue));
+ }
}
auto maskFilter = mitk::MaskImageFilter::New();
maskFilter->SetInput(referenceImage);
maskFilter->SetMask(maskImage);
maskFilter->OverrideOutsideValueOn();
maskFilter->SetOutsideValue(backgroundValue);
try
{
maskFilter->Update();
}
catch(const itk::ExceptionObject& e)
{
MITK_ERROR << e.GetDescription();
return nullptr;
}
return maskFilter->GetOutput();
}
mitk::Image::Pointer QmitkImageMaskingWidget::ConvertSurfaceToImage( mitk::Image::Pointer image, mitk::Surface::Pointer surface )
{
mitk::ProgressBar::GetInstance()->AddStepsToDo(2);
mitk::ProgressBar::GetInstance()->Progress();
mitk::SurfaceToImageFilter::Pointer surfaceToImageFilter = mitk::SurfaceToImageFilter::New();
surfaceToImageFilter->MakeOutputBinaryOn();
surfaceToImageFilter->SetInput(surface);
surfaceToImageFilter->SetImage(image);
try
{
surfaceToImageFilter->Update();
}
catch(itk::ExceptionObject& excpt)
{
MITK_ERROR << excpt.GetDescription();
return nullptr;
}
mitk::ProgressBar::GetInstance()->Progress();
mitk::Image::Pointer resultImage = mitk::Image::New();
resultImage = surfaceToImageFilter->GetOutput();
return resultImage;
}
void QmitkImageMaskingWidget::AddToDataStorage(mitk::DataStorage::Pointer dataStorage, mitk::Image::Pointer segmentation, const std::string& name, mitk::DataNode::Pointer parent )
{
auto dataNode = mitk::DataNode::New();
dataNode->SetName(name);
dataNode->SetData(segmentation);
if (parent.IsNotNull())
{
mitk::LevelWindow levelWindow;
parent->GetLevelWindow(levelWindow);
dataNode->SetLevelWindow(levelWindow);
}
dataStorage->Add(dataNode, parent);
}