itkLocalIntensityFilter.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 itkLocalIntensityFilter_h
#define itkLocalIntensityFilter_h

#include "itkImageToImageFilter.h"
#include "itkNumericTraits.h"
#include "itkArray.h"
#include "itkSimpleDataObjectDecorator.h"

namespace itk
{

  template< typename TInputImage >
  class ITK_TEMPLATE_EXPORT LocalIntensityFilter :
    public ImageToImageFilter< TInputImage, TInputImage >
  {
  public:
    typedef LocalIntensityFilter                           Self;
    typedef ImageToImageFilter< TInputImage, TInputImage > Superclass;
    typedef SmartPointer< Self >                           Pointer;
    typedef SmartPointer< const Self >                     ConstPointer;

    itkNewMacro(Self);

    itkTypeMacro(LocalIntensityFilter, ImageToImageFilter);

    typedef typename TInputImage::Pointer InputImagePointer;

    typedef typename TInputImage::RegionType RegionType;
    typedef typename TInputImage::SizeType   SizeType;
    typedef typename TInputImage::IndexType  IndexType;
    typedef typename TInputImage::PixelType  PixelType;

    typedef Image<unsigned short, TInputImage::ImageDimension> MaskImageType;

    itkStaticConstMacro(ImageDimension, unsigned int,
      TInputImage::ImageDimension);

    typedef typename NumericTraits< PixelType >::RealType RealType;

    typedef typename DataObject::Pointer DataObjectPointer;

    typedef SimpleDataObjectDecorator< RealType >  RealObjectType;
    typedef SimpleDataObjectDecorator< PixelType > PixelObjectType;

    RealType GetLocalPeak() const
    {
      return this->GetLocalPeakOutput()->Get();
    }
    RealObjectType * GetLocalPeakOutput();

    const RealObjectType * GetLocalPeakOutput() const;

    RealType GetGlobalPeak() const
    {
      return this->GetGlobalPeakOutput()->Get();
    }
    RealObjectType * GetGlobalPeakOutput();

    const RealObjectType * GetGlobalPeakOutput() const;

    RealType GetLocalMaximum() const
    {
      return this->GetLocalMaximumOutput()->Get();
    }
    RealObjectType * GetLocalMaximumOutput();

    const RealObjectType * GetLocalMaximumOutput() const;

    void SetMask(typename MaskImageType::Pointer mask)
    {
      m_Mask = mask;
    }

    itkSetMacro(Range, double);
    itkGetMacro(Range, double);
    itkGetConstMacro(Range, double);

    typedef ProcessObject::DataObjectPointerArraySizeType DataObjectPointerArraySizeType;
    using Superclass::MakeOutput;
    DataObjectPointer MakeOutput(DataObjectPointerArraySizeType idx) ITK_OVERRIDE;

#ifdef ITK_USE_CONCEPT_CHECKING
    // Begin concept checking
    itkConceptMacro(InputHasNumericTraitsCheck,
      (Concept::HasNumericTraits< PixelType >));
    // End concept checking
#endif

  protected:
    LocalIntensityFilter();
    ~LocalIntensityFilter() ITK_OVERRIDE {}
    void PrintSelf(std::ostream & os, Indent indent) const ITK_OVERRIDE;

    void AllocateOutputs() ITK_OVERRIDE;

    void BeforeThreadedGenerateData() ITK_OVERRIDE;

    void AfterThreadedGenerateData() ITK_OVERRIDE;

    void  ThreadedGenerateData(const RegionType &
      outputRegionForThread,
      ThreadIdType threadId) ITK_OVERRIDE;

    // Override since the filter needs all the data for the algorithm
    void GenerateInputRequestedRegion() ITK_OVERRIDE;

    // Override since the filter produces all of its output
    void EnlargeOutputRequestedRegion(DataObject *data) ITK_OVERRIDE;

  private:
    ITK_DISALLOW_COPY_AND_ASSIGN(LocalIntensityFilter);

    Array< RealType >       m_ThreadLocalMaximum;
    Array< RealType >       m_ThreadLocalPeakValue;
    Array< RealType >       m_ThreadGlobalPeakValue;
    typename MaskImageType::Pointer m_Mask;
    double m_Range;
  }; // end of class
} // end namespace itk

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkLocalIntensityFilter.hxx"
#endif

#endif