itkTotalVariationSingleIterationImageFilter.h

Go to the documentation of this file.

/*============================================================================
 
The Medical Imaging Interaction Toolkit (MITK)
 
Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.
 
Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.
 
============================================================================*/
 
#ifndef __itkTotalVariationSingleIterationImageFilter_h
#define __itkTotalVariationSingleIterationImageFilter_h
 
#include "itkImage.h"
#include "itkImageToImageFilter.h"
 
namespace itk
{
  template <class TInputImage, class TOutputImage>
  class TotalVariationSingleIterationImageFilter : public ImageToImageFilter<TInputImage, TOutputImage>
  {
  public:
    itkStaticConstMacro(InputImageDimension, unsigned int, TInputImage::ImageDimension);
    itkStaticConstMacro(OutputImageDimension, unsigned int, TOutputImage::ImageDimension);
 
    typedef TInputImage InputImageType;
    typedef TOutputImage OutputImageType;
 
    typedef itk::Image<float, InputImageDimension> LocalVariationImageType;
 
    typedef TotalVariationSingleIterationImageFilter Self;
    typedef ImageToImageFilter<InputImageType, OutputImageType> Superclass;
    typedef SmartPointer<Self> Pointer;
    typedef SmartPointer<const Self> ConstPointer;
 
    itkFactorylessNewMacro(Self);
    itkCloneMacro(Self);
 
      itkTypeMacro(TotalVariationSingleIterationImageFilter, ImageToImageFilter);
 
    typedef typename InputImageType::PixelType InputPixelType;
    typedef typename OutputImageType::PixelType OutputPixelType;
 
    typedef typename InputImageType::RegionType InputImageRegionType;
    typedef typename OutputImageType::RegionType OutputImageRegionType;
 
    typedef typename InputImageType::SizeType InputSizeType;
 
    void GenerateInputRequestedRegion() override;
 
    itkSetMacro(Lambda, double);
    itkGetMacro(Lambda, double);
 
    void SetOriginalImage(InputImageType *in) { this->m_OriginalImage = in; }
    typename InputImageType::Pointer GetOriginialImage() { return this->m_OriginalImage; }
  protected:
    TotalVariationSingleIterationImageFilter();
    ~TotalVariationSingleIterationImageFilter() override {}
    void PrintSelf(std::ostream &os, Indent indent) const override;
 
    void ThreadedGenerateData(const OutputImageRegionType &outputRegionForThread, ThreadIdType threadId) override;
 
    void BeforeThreadedGenerateData() override;
 
    typename LocalVariationImageType::Pointer m_LocalVariation;
 
    typename InputImageType::Pointer m_OriginalImage;
 
    double m_Lambda;
 
  private:
    TotalVariationSingleIterationImageFilter(const Self &);
 
    void operator=(const Self &);
  };
 
} // end namespace itk
 
#ifndef ITK_MANUAL_INSTANTIATION
#include "itkTotalVariationSingleIterationImageFilter.txx"
#endif
 
#endif //__itkTotalVariationSingleIterationImageFilter__