tools/clitkAnisotropicDiffusionGenericFilter.cxx

   1 /*=========================================================================
   2   Program:   vv                     http://www.creatis.insa-lyon.fr/rio/vv
   3
   4   Authors belong to:
   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
   8
   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.
  12
  13   It is distributed under dual licence
  14
  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
  20
  21 #include "clitkAnisotropicDiffusionGenericFilter.h"
  22
  23 // itk include
  24 #include <itkGradientAnisotropicDiffusionImageFilter.h>
  25 #include <itkCurvatureAnisotropicDiffusionImageFilter.h>
  26
  27 //#include <clitkCommon.h>
  28
  29 namespace clitk
  30 {
  31
  32 //--------------------------------------------------------------------
  33 AnisotropicDiffusionGenericFilter::AnisotropicDiffusionGenericFilter():
  34   ImageToImageGenericFilter<Self>("AnisotropicDiffusion")
  35 {
  36   InitializeImageType<2>();
  37   InitializeImageType<3>();
  38   InitializeImageType<4>();
  39 }
  40 //--------------------------------------------------------------------
  41
  42
  43 //--------------------------------------------------------------------
  44 template<unsigned int Dim>
  45 void AnisotropicDiffusionGenericFilter::InitializeImageType()
  46 {
  47   ADD_IMAGE_TYPE(Dim, float);
  48   ADD_IMAGE_TYPE(Dim, double);
  49 }
  50 //--------------------------------------------------------------------
  51
  52
  53 //--------------------------------------------------------------------
  54 void AnisotropicDiffusionGenericFilter::SetArgsInfo(const args_info_type & a)
  55 {
  56   mArgsInfo=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);
  63   }
  64   if (mArgsInfo.output_given) {
  65     SetOutputFilename(mArgsInfo.output_arg);
  66   }
  67 }
  68 //--------------------------------------------------------------------
  69
  70
  71 //--------------------------------------------------------------------
  72 // Update with the number of dimensions and the pixeltype
  73 //--------------------------------------------------------------------
  74 template<class InputImageType>
  75 void
  76 AnisotropicDiffusionGenericFilter::UpdateWithInputImageType()
  77 {
  78   // Reading input
  79   typename InputImageType::Pointer input = this->template GetInput<InputImageType>(0);
  80
  81   // Filter
  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);
  92     gadFilter->Update();
  93     this->template SetNextOutput<InputImageType>(gadFilter->GetOutput());
  94     break;
  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);
 100     cadFilter->Update();
 101     this->template SetNextOutput<InputImageType>(cadFilter->GetOutput());
 102     break;
 103   default:
 104     itkGenericExceptionMacro("Unhandled type");
 105     break;
 106   }
 107 }
 108 //--------------------------------------------------------------------
 109
 110
 111 }//end clitk
 112
 113 #endif  //#define clitkAnisotropicDiffusionGenericFilter_cxx