AnisotropicDiffusion.h

Go to the documentation of this file.

/*****************************************************************************
 * $CAMITK_LICENCE_BEGIN$
 *
 * CamiTK - Computer Assisted Medical Intervention ToolKit
 * (c) 2001-2021 Univ. Grenoble Alpes, CNRS, Grenoble INP, TIMC, 38000 Grenoble, France
 *
 * Visit http://camitk.imag.fr for more information
 *
 * This file is part of CamiTK.
 *
 * CamiTK is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License version 3
 * only, as published by the Free Software Foundation.
 *
 * CamiTK is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License version 3 for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * version 3 along with CamiTK.  If not, see <http://www.gnu.org/licenses/>.
 *
 * $CAMITK_LICENCE_END$
 ****************************************************************************/
#ifndef ANISOTROPICDIFFUSION_H
#define ANISOTROPICDIFFUSION_H
 
#include <QObject>
#include <Action.h>
#include <ImageComponent.h>
#include <ActionWidget.h>
 
#include "ITKFilterAPI.h"
 
class ITKFILTER_API AnisotropicDiffusion : public camitk::Action {
    Q_OBJECT
 
public:
    enum AnisoDiffType {GRADIENT, CURVATURE};
 
    // prefers Q_ENUM to Q_ENUMS if possible (Qt >= 5.5)
    Q_ENUM(AnisoDiffType)
 
    
    AnisotropicDiffusion(camitk::ActionExtension*);
 
    virtual ~AnisotropicDiffusion();
 
public slots:
    virtual camitk::Action::ApplyStatus apply();
 
    AnisotropicDiffusion::AnisoDiffType getDiffusionType();
 
private:
    virtual void process(camitk::ImageComponent*);
 
    vtkSmartPointer<vtkImageData> implementProcess(vtkSmartPointer<vtkImageData> img);
 
    template <class InputPixelType, class OutputPixelType, const int dim>
    vtkSmartPointer<vtkImageData> itkProcess(vtkSmartPointer<vtkImageData> img);
 
    template <class InputPixelType, class OutputPixelType, const int dim>
    vtkSmartPointer<vtkImageData> itkProcessGradientAnisotropicDiffusion(vtkSmartPointer<vtkImageData> img);
 
    template <class InputPixelType, class OutputPixelType, const int dim>
    vtkSmartPointer<vtkImageData> itkProcessCurvatureAnisotropicDiffusion(vtkSmartPointer<vtkImageData> img);
 
protected:
    bool keepOrgVoxelType;
    int numberOfIterations;
    double conductance;
};
 
#endif // ANISOTROPICDIFFUSION_H