1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93
|
/*=========================================================================
Program: Insight Segmentation & Registration Toolkit
Module: itkVectorAnisotropicDiffusionImageFilterTest.cxx
Language: C++
Date: $Date$
Version: $Revision$
Copyright (c) Insight Software Consortium. All rights reserved.
See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details.
This software is distributed WITHOUT ANY WARRANTY; without even
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE. See the above copyright notices for more information.
=========================================================================*/
#if defined(_MSC_VER)
#pragma warning ( disable : 4786 )
#endif
#include <iostream>
#include "itkImage.h"
#include "itkVectorGradientAnisotropicDiffusionImageFilter.h"
#include "itkVectorCurvatureAnisotropicDiffusionImageFilter.h"
#include "itkNullImageToImageFilterDriver.txx"
#include "itkVector.h"
inline std::ostream& operator<<(std::ostream &o, const itk::Vector<float, 3> &v)
{
o << "["<< v[0] << " " << v[1] << " " << v[2] << "]";
return o;
}
int itkVectorAnisotropicDiffusionImageFilterTest(int itkNotUsed(argc), char *itkNotUsed(argv) [] )
{
try
{
typedef itk::Image<itk::Vector<float, 3>, 2> ImageType;
// Set up Gradient diffusion filter
itk::VectorGradientAnisotropicDiffusionImageFilter<ImageType, ImageType>
::Pointer filter =
itk::VectorGradientAnisotropicDiffusionImageFilter<ImageType,
ImageType>
::New();
filter->SetNumberOfIterations(1);
filter->SetConductanceParameter(3.0f);
filter->SetTimeStep(0.125f);
filter->GetNumberOfIterations();
filter->GetConductanceParameter();
filter->GetTimeStep();
// Run Test
itk::Size<2> sz;
sz[0] = 100 ; //atoi(argv[1]);
sz[1] = 100 ; // atoi(argv[2]);
// sz[2] = 10;//atoi(argv[3]);
// sz[3] = 5;//atoi(argv[4]);
itk::NullImageToImageFilterDriver< ImageType, ImageType > test1;
test1.SetImageSize(sz);
test1.SetFilter(filter.GetPointer());
test1.Execute();
// Set up Curvature diffusion filter
itk::VectorCurvatureAnisotropicDiffusionImageFilter<ImageType, ImageType>
::Pointer filter2 =
itk::VectorCurvatureAnisotropicDiffusionImageFilter<ImageType, ImageType>
::New();
filter2->SetNumberOfIterations(1);
filter2->SetConductanceParameter(3.0f);
filter2->SetTimeStep(0.1f);
filter2->GetNumberOfIterations();
filter2->GetConductanceParameter();
filter2->GetTimeStep();
// Run Test
itk::NullImageToImageFilterDriver< ImageType, ImageType > test2;
test2.SetImageSize(sz);
test2.SetFilter(filter2.GetPointer());
test2.Execute();
}
catch(itk::ExceptionObject &err)
{
(&err)->Print(std::cerr);
return EXIT_FAILURE;
}
return EXIT_SUCCESS;
}
|