void StartSimulation(string testFileName)
{
mitk::DiffusionImage<short>::Pointer refImage = NULL;
if (!testFileName.empty())
CPPUNIT_ASSERT(refImage = dynamic_cast<mitk::DiffusionImage<short>*>(mitk::IOUtil::LoadDataNode(testFileName)->GetData()));
itk::AddArtifactsToDwiImageFilter< short >::Pointer artifactsToDwiFilter = itk::AddArtifactsToDwiImageFilter< short >::New();
artifactsToDwiFilter->SetUseConstantRandSeed(true);
artifactsToDwiFilter->SetInput(m_InputDwi->GetVectorImage());
artifactsToDwiFilter->SetParameters(m_Parameters);
CPPUNIT_ASSERT_NO_THROW(artifactsToDwiFilter->Update());
mitk::DiffusionImage<short>::Pointer testImage = mitk::DiffusionImage<short>::New();
testImage->SetVectorImage( artifactsToDwiFilter->GetOutput() );
testImage->SetReferenceBValue( m_Parameters.m_SignalGen.m_Bvalue);
testImage->SetDirections( m_Parameters.m_SignalGen.GetGradientDirections());
testImage->InitializeFromVectorImage();
if (refImage.IsNotNull())
{
if (!CompareDwi(testImage->GetVectorImage(), refImage->GetVectorImage()))
mitk::IOUtil::SaveBaseData(testImage, mitk::IOUtil::GetTempPath()+"testImage.dwi");
CPPUNIT_ASSERT_MESSAGE(testFileName, CompareDwi(testImage->GetVectorImage(), refImage->GetVectorImage()));
}
}
void StartSimulation(FiberfoxParameters<double> parameters, FiberBundleX::Pointer fiberBundle, mitk::DiffusionImage<short>::Pointer refImage, string message)
{
itk::TractsToDWIImageFilter< short >::Pointer tractsToDwiFilter = itk::TractsToDWIImageFilter< short >::New();
tractsToDwiFilter->SetUseConstantRandSeed(true);
tractsToDwiFilter->SetParameters(parameters);
tractsToDwiFilter->SetFiberBundle(fiberBundle);
tractsToDwiFilter->Update();
mitk::DiffusionImage<short>::Pointer testImage = mitk::DiffusionImage<short>::New();
testImage->SetVectorImage( tractsToDwiFilter->GetOutput() );
testImage->SetReferenceBValue(parameters.m_SignalGen.m_Bvalue);
testImage->SetDirections(parameters.m_SignalGen.GetGradientDirections());
testImage->InitializeFromVectorImage();
if (refImage.IsNotNull())
{
bool cond = CompareDwi(testImage->GetVectorImage(), refImage->GetVectorImage());
if (!cond)
{
MITK_INFO << "Saving test and rference image to " << mitk::IOUtil::GetTempPath();
mitk::IOUtil::SaveBaseData(testImage, mitk::IOUtil::GetTempPath()+"testImage.dwi");
mitk::IOUtil::SaveBaseData(refImage, mitk::IOUtil::GetTempPath()+"refImage.dwi");
}
MITK_TEST_CONDITION_REQUIRED(cond, message);
}
}
void StartSimulation(FiberfoxParameters<double> parameters, FiberBundle::Pointer fiberBundle, mitk::Image::Pointer refImage, string message)
{
itk::TractsToDWIImageFilter< short >::Pointer tractsToDwiFilter = itk::TractsToDWIImageFilter< short >::New();
tractsToDwiFilter->SetUseConstantRandSeed(true);
tractsToDwiFilter->SetParameters(parameters);
tractsToDwiFilter->SetFiberBundle(fiberBundle);
tractsToDwiFilter->Update();
mitk::Image::Pointer testImage = mitk::GrabItkImageMemory( tractsToDwiFilter->GetOutput() );
testImage->SetProperty( mitk::DiffusionPropertyHelper::GRADIENTCONTAINERPROPERTYNAME.c_str(), mitk::GradientDirectionsProperty::New( parameters.m_SignalGen.GetGradientDirections() ) );
testImage->SetProperty( mitk::DiffusionPropertyHelper::REFERENCEBVALUEPROPERTYNAME.c_str(), mitk::FloatProperty::New( parameters.m_SignalGen.m_Bvalue ) );
mitk::DiffusionPropertyHelper propertyHelper( testImage );
propertyHelper.InitializeImage();
if (refImage.IsNotNull())
{
if( static_cast<mitk::GradientDirectionsProperty*>( refImage->GetProperty(mitk::DiffusionPropertyHelper::GRADIENTCONTAINERPROPERTYNAME.c_str()).GetPointer() )->GetGradientDirectionsContainer().IsNotNull() )
{
ItkDwiType::Pointer itkTestImagePointer = ItkDwiType::New();
mitk::CastToItkImage(testImage, itkTestImagePointer);
ItkDwiType::Pointer itkRefImagePointer = ItkDwiType::New();
mitk::CastToItkImage(refImage, itkRefImagePointer);
bool cond = CompareDwi(itkTestImagePointer, itkRefImagePointer);
if (!cond)
{
MITK_INFO << "Saving test and rference image to " << mitk::IOUtil::GetTempPath();
mitk::IOUtil::SaveBaseData(testImage, mitk::IOUtil::GetTempPath()+"testImage.dwi");
mitk::IOUtil::SaveBaseData(refImage, mitk::IOUtil::GetTempPath()+"refImage.dwi");
}
MITK_TEST_CONDITION_REQUIRED(cond, message);
}
}
}
void StartSimulation(string testFileName)
{
mitk::DiffusionImage<short>::Pointer refImage = NULL;
if (!testFileName.empty())
CPPUNIT_ASSERT(refImage = dynamic_cast<mitk::DiffusionImage<short>*>(mitk::IOUtil::LoadDataNode(testFileName)->GetData()));
itk::AddArtifactsToDwiImageFilter< short >::Pointer artifactsToDwiFilter = itk::AddArtifactsToDwiImageFilter< short >::New();
artifactsToDwiFilter->SetUseConstantRandSeed(true);
artifactsToDwiFilter->SetInput(m_InputDwi->GetVectorImage());
artifactsToDwiFilter->SetParameters(m_Parameters);
CPPUNIT_ASSERT_NO_THROW(artifactsToDwiFilter->Update());
mitk::DiffusionImage<short>::Pointer testImage = mitk::DiffusionImage<short>::New();
testImage->SetVectorImage( artifactsToDwiFilter->GetOutput() );
testImage->SetB_Value(m_Parameters.m_Bvalue);
testImage->SetDirections(m_Parameters.GetGradientDirections());
testImage->InitializeFromVectorImage();
if (refImage.IsNotNull())
{
bool ok = CompareDwi(testImage->GetVectorImage(), refImage->GetVectorImage());
if (!ok)
{
mitk::IOUtil::SaveBaseData(testImage, "/tmp/test2.dwi");
mitk::IOUtil::SaveBaseData(refImage, "/tmp/ref2.dwi");
}
CPPUNIT_ASSERT_MESSAGE(testFileName, ok);
}
else
{
mitk::IOUtil::SaveBaseData(testImage, "/local/distortions2.dwi");
}
}