1 /*=========================================================================
2 Program: vv http://www.creatis.insa-lyon.fr/rio/vv
5 - University of LYON http://www.universite-lyon.fr/
6 - Léon Bérard cancer center http://www.centreleonberard.fr
7 - CREATIS CNRS laboratory http://www.creatis.insa-lyon.fr
9 This software is distributed WITHOUT ANY WARRANTY; without even
10 the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
11 PURPOSE. See the copyright notices for more information.
13 It is distributed under dual licence
15 - BSD See included LICENSE.txt file
16 - CeCILL-B http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
17 ===========================================================================**/
18 #ifndef clitkAnisotropicDiffusionGenericFilter_cxx
19 #define clitkAnisotropicDiffusionGenericFilter_cxx
21 #include "clitkAnisotropicDiffusionGenericFilter.h"
24 #include <itkGradientAnisotropicDiffusionImageFilter.h>
25 #include <itkCurvatureAnisotropicDiffusionImageFilter.h>
27 //#include <clitkCommon.h>
32 //--------------------------------------------------------------------
33 AnisotropicDiffusionGenericFilter::AnisotropicDiffusionGenericFilter():
34 ImageToImageGenericFilter<Self>("AnisotropicDiffusion")
36 InitializeImageType<2>();
37 InitializeImageType<3>();
38 InitializeImageType<4>();
40 //--------------------------------------------------------------------
43 //--------------------------------------------------------------------
44 template<unsigned int Dim>
45 void AnisotropicDiffusionGenericFilter::InitializeImageType()
47 ADD_IMAGE_TYPE(Dim, float);
48 ADD_IMAGE_TYPE(Dim, double);
50 //--------------------------------------------------------------------
53 //--------------------------------------------------------------------
54 void AnisotropicDiffusionGenericFilter::SetArgsInfo(const args_info_type & a)
57 if (mArgsInfo.verbose_given)
58 SetIOVerbose(mArgsInfo.verbose_flag);
59 if (mArgsInfo.imagetypes_given && mArgsInfo.imagetypes_flag)
60 this->PrintAvailableImageTypes();
61 if (mArgsInfo.input_given) {
62 SetInputFilename(mArgsInfo.input_arg);
64 if (mArgsInfo.output_given) {
65 SetOutputFilename(mArgsInfo.output_arg);
68 //--------------------------------------------------------------------
71 //--------------------------------------------------------------------
72 // Update with the number of dimensions and the pixeltype
73 //--------------------------------------------------------------------
74 template<class InputImageType>
76 AnisotropicDiffusionGenericFilter::UpdateWithInputImageType()
79 typename InputImageType::Pointer input = this->template GetInput<InputImageType>(0);
82 typedef itk::CurvatureAnisotropicDiffusionImageFilter<InputImageType, InputImageType> CADIFType;
83 typename CADIFType::Pointer cadFilter = CADIFType::New();
84 typedef itk::GradientAnisotropicDiffusionImageFilter<InputImageType, InputImageType> GADIFType;
85 typename GADIFType::Pointer gadFilter = GADIFType::New();
86 switch(mArgsInfo.type_arg) {
87 case type_arg_Gradient:
88 gadFilter->SetInput(input);
89 gadFilter->SetNumberOfIterations(mArgsInfo.niterations_arg);
90 gadFilter->SetTimeStep(mArgsInfo.timestep_arg);
91 gadFilter->SetConductanceParameter(mArgsInfo.conductance_arg);
93 this->template SetNextOutput<InputImageType>(gadFilter->GetOutput());
95 case type_arg_Curvature:
96 cadFilter->SetInput(input);
97 cadFilter->SetNumberOfIterations(mArgsInfo.niterations_arg);
98 cadFilter->SetTimeStep(mArgsInfo.timestep_arg);
99 cadFilter->SetConductanceParameter(mArgsInfo.conductance_arg);
101 this->template SetNextOutput<InputImageType>(cadFilter->GetOutput());
104 itkGenericExceptionMacro("Unhandled type");
108 //--------------------------------------------------------------------
113 #endif //#define clitkAnisotropicDiffusionGenericFilter_cxx