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_DEFAULT_IMAGE_TYPES(Dim);
49 //--------------------------------------------------------------------
52 //--------------------------------------------------------------------
53 void AnisotropicDiffusionGenericFilter::SetArgsInfo(const args_info_type & a)
56 if (mArgsInfo.verbose_given)
57 SetIOVerbose(mArgsInfo.verbose_flag);
58 if (mArgsInfo.imagetypes_given && mArgsInfo.imagetypes_flag)
59 this->PrintAvailableImageTypes();
60 if (mArgsInfo.input_given) {
61 SetInputFilename(mArgsInfo.input_arg);
63 if (mArgsInfo.output_given) {
64 SetOutputFilename(mArgsInfo.output_arg);
67 //--------------------------------------------------------------------
70 //--------------------------------------------------------------------
71 // Update with the number of dimensions and the pixeltype
72 //--------------------------------------------------------------------
73 template<class InputImageType>
75 AnisotropicDiffusionGenericFilter::UpdateWithInputImageType()
78 typename InputImageType::Pointer input = this->template GetInput<InputImageType>(0);
81 typedef itk::CurvatureAnisotropicDiffusionImageFilter<InputImageType, InputImageType> CADIFType;
82 typename CADIFType::Pointer cadFilter = CADIFType::New();
83 typedef itk::GradientAnisotropicDiffusionImageFilter<InputImageType, InputImageType> GADIFType;
84 typename GADIFType::Pointer gadFilter = GADIFType::New();
85 switch(mArgsInfo.type_arg) {
86 case type_arg_Gradient:
87 gadFilter->SetInput(input);
88 gadFilter->SetNumberOfIterations(mArgsInfo.niterations_arg);
89 gadFilter->SetTimeStep(mArgsInfo.timestep_arg);
90 gadFilter->SetConductanceParameter(mArgsInfo.conductance_arg);
92 this->template SetNextOutput<InputImageType>(gadFilter->GetOutput());
93 case type_arg_Curvature:
94 cadFilter->SetInput(input);
95 cadFilter->SetNumberOfIterations(mArgsInfo.niterations_arg);
96 cadFilter->SetTimeStep(mArgsInfo.timestep_arg);
97 cadFilter->SetConductanceParameter(mArgsInfo.conductance_arg);
99 this->template SetNextOutput<InputImageType>(cadFilter->GetOutput());
102 //--------------------------------------------------------------------
107 #endif //#define clitkAnisotropicDiffusionGenericFilter_cxx