itkTotalVariationSingleIterationImageFilter.h

Go to the documentation of this file.

/*===================================================================

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center,
Division of Medical and Biological Informatics.
All rights reserved.

This software is distributed WITHOUT ANY WARRANTY; without
even the implied warranty of MERCHANTABILITY or FITNESS FOR
A PARTICULAR PURPOSE.

See LICENSE.txt or http://www.mitk.org for details.

===================================================================*/

#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;

    virtual void GenerateInputRequestedRegion() throw(InvalidRequestedRegionError);

    itkSetMacro(Lambda, double);
    itkGetMacro(Lambda, double);

    void SetOriginalImage(InputImageType *in) { this->m_OriginalImage = in; }
    typename InputImageType::Pointer GetOriginialImage() { return this->m_OriginalImage; }
  protected:
    TotalVariationSingleIterationImageFilter();
    virtual ~TotalVariationSingleIterationImageFilter() {}
    void PrintSelf(std::ostream &os, Indent indent) const;

    void ThreadedGenerateData(const OutputImageRegionType &outputRegionForThread, ThreadIdType threadId);

    void BeforeThreadedGenerateData();

    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__