FloatImageType::Pointer ComputeSqrtMu(FloatImageType::Pointer mu)
{
typedef itk::SqrtImageFilter<FloatImageType,FloatImageType> SqrtType;
SqrtType::Pointer sqrtFilter = SqrtType::New();
sqrtFilter->SetInput(mu);
sqrtFilter->Update();
return sqrtFilter->GetOutput();
}
// -----------------------------------------------------------------------------
//
// -----------------------------------------------------------------------------
void ImageMath::execute()
{
//int err = 0;
dataCheck();
if(getErrorCondition() < 0) { return; }
DataContainer::Pointer m = getDataContainerArray()->getDataContainer(getSelectedCellArrayPath().getDataContainerName());
QString attrMatName = getSelectedCellArrayPath().getAttributeMatrixName();
//wrap m_RawImageData as itk::image
ImageProcessing::DefaultImageType::Pointer inputImage = ITKUtilitiesType::CreateItkWrapperForDataPointer(m, attrMatName, m_SelectedCellArray);
//define filter types
typedef itk::AddImageFilter<ImageProcessing::DefaultImageType, ImageProcessing::FloatImageType, ImageProcessing::FloatImageType> AddType;
typedef itk::SubtractImageFilter<ImageProcessing::DefaultImageType, ImageProcessing::FloatImageType, ImageProcessing::FloatImageType> SubtractType;
typedef itk::MultiplyImageFilter<ImageProcessing::DefaultImageType, ImageProcessing::FloatImageType, ImageProcessing::FloatImageType> MultiplyType;
typedef itk::DivideImageFilter<ImageProcessing::DefaultImageType, ImageProcessing::FloatImageType, ImageProcessing::FloatImageType> DivideType;
typedef itk::MinimumImageFilter<ImageProcessing::DefaultImageType, ImageProcessing::FloatImageType, ImageProcessing::FloatImageType> MinType;
typedef itk::MaximumImageFilter<ImageProcessing::DefaultImageType, ImageProcessing::FloatImageType, ImageProcessing::FloatImageType> MaxType;
typedef itk::BinaryFunctorImageFilter< ImageProcessing::DefaultImageType, ImageProcessing::FloatImageType, ImageProcessing::FloatImageType, ImageProcessing::Functor::Gamma<ImageProcessing::FloatPixelType> > GammaType;
typedef itk::LogImageFilter<ImageProcessing::DefaultImageType, ImageProcessing::FloatImageType> LogType;
typedef itk::ExpImageFilter<ImageProcessing::DefaultImageType, ImageProcessing::FloatImageType> ExpType;
typedef itk::SquareImageFilter<ImageProcessing::DefaultImageType, ImageProcessing::FloatImageType> SquareType;
typedef itk::SqrtImageFilter<ImageProcessing::DefaultImageType, ImageProcessing::FloatImageType> SqrtType;
typedef itk::InvertIntensityImageFilter<ImageProcessing::DefaultImageType, ImageProcessing::DefaultImageType> InvertType;
//set up filter to cap image range + round
typedef itk::UnaryFunctorImageFilter< ImageProcessing::FloatImageType, ImageProcessing::DefaultImageType, ImageProcessing::Functor::LimitsRound<ImageProcessing::FloatPixelType, ImageProcessing::DefaultPixelType> > LimitsRoundType;
LimitsRoundType::Pointer limitsRound = LimitsRoundType::New();
//apply selected operation
switch(m_Operator)
{
case 0://add
{
AddType::Pointer add = AddType::New();
add->SetInput1(inputImage);
add->SetConstant2(m_Value);
limitsRound->SetInput(add->GetOutput());
ITKUtilitiesType::SetITKFilterOutput(limitsRound->GetOutput(), m_NewCellArrayPtr.lock());
limitsRound->Update();
}
break;
case 1://subtract
{
SubtractType::Pointer subtract = SubtractType::New();
subtract->SetInput1(inputImage);
subtract->SetConstant2(m_Value);
limitsRound->SetInput(subtract->GetOutput());
ITKUtilitiesType::SetITKFilterOutput(limitsRound->GetOutput(), m_NewCellArrayPtr.lock());
limitsRound->Update();
}
break;
case 2://multiply
{
MultiplyType::Pointer multiply = MultiplyType::New();
multiply->SetInput1(inputImage);
multiply->SetConstant2(m_Value);
limitsRound->SetInput(multiply->GetOutput());
ITKUtilitiesType::SetITKFilterOutput(limitsRound->GetOutput(), m_NewCellArrayPtr.lock());
limitsRound->Update();
}
break;
case 3://divide
{
DivideType::Pointer divide = DivideType::New();
divide->SetInput1(inputImage);
divide->SetConstant2(m_Value);
limitsRound->SetInput(divide->GetOutput());
ITKUtilitiesType::SetITKFilterOutput(limitsRound->GetOutput(), m_NewCellArrayPtr.lock());
limitsRound->Update();
}
break;
case 4://min
{
MinType::Pointer minimum = MinType::New();
minimum->SetInput1(inputImage);
minimum->SetConstant2(m_Value);
limitsRound->SetInput(minimum->GetOutput());
ITKUtilitiesType::SetITKFilterOutput(limitsRound->GetOutput(), m_NewCellArrayPtr.lock());
limitsRound->Update();
}
break;
case 5://max
{
MaxType::Pointer maximum = MaxType::New();
maximum->SetInput1(inputImage);
maximum->SetConstant2(m_Value);
limitsRound->SetInput(maximum->GetOutput());
ITKUtilitiesType::SetITKFilterOutput(limitsRound->GetOutput(), m_NewCellArrayPtr.lock());
limitsRound->Update();
}
break;
case 6://gamma
{
GammaType::Pointer gamma = GammaType::New();
gamma->SetInput1(inputImage);
gamma->SetConstant2(m_Value);
limitsRound->SetInput(gamma->GetOutput());
ITKUtilitiesType::SetITKFilterOutput(limitsRound->GetOutput(), m_NewCellArrayPtr.lock());
limitsRound->Update();
}
break;
case 7://log
{
LogType::Pointer logfilter = LogType::New();
logfilter->SetInput(inputImage);
limitsRound->SetInput(logfilter->GetOutput());
ITKUtilitiesType::SetITKFilterOutput(limitsRound->GetOutput(), m_NewCellArrayPtr.lock());
limitsRound->Update();
}
break;
case 8://exp
{
ExpType::Pointer expfilter = ExpType::New();
expfilter->SetInput(inputImage);
limitsRound->SetInput(expfilter->GetOutput());
ITKUtilitiesType::SetITKFilterOutput(limitsRound->GetOutput(), m_NewCellArrayPtr.lock());
limitsRound->Update();
}
break;
case 9://square
{
SquareType::Pointer square = SquareType::New();
square->SetInput(inputImage);
limitsRound->SetInput(square->GetOutput());
ITKUtilitiesType::SetITKFilterOutput(limitsRound->GetOutput(), m_NewCellArrayPtr.lock());
limitsRound->Update();
}
break;
case 10://squareroot
{
SqrtType::Pointer sqrtfilter = SqrtType::New();
sqrtfilter->SetInput(inputImage);
limitsRound->SetInput(sqrtfilter->GetOutput());
ITKUtilitiesType::SetITKFilterOutput(limitsRound->GetOutput(), m_NewCellArrayPtr.lock());
limitsRound->Update();
}
break;
case 11://invert
{
InvertType::Pointer invert = InvertType::New();
invert->SetInput(inputImage);
ITKUtilitiesType::SetITKFilterOutput(invert->GetOutput(), m_NewCellArrayPtr.lock());
invert->Update();
}
break;
}
//array name changing/cleanup
if(m_SaveAsNewArray == false)
{
AttributeMatrix::Pointer attrMat = m->getAttributeMatrix(m_SelectedCellArrayPath.getAttributeMatrixName());
attrMat->removeAttributeArray(m_SelectedCellArrayPath.getDataArrayName());
attrMat->renameAttributeArray(m_NewCellArrayName, m_SelectedCellArrayPath.getDataArrayName());
}
/* Let the GUI know we are done with this filter */
notifyStatusMessage(getHumanLabel(), "Complete");
}