itkMaskedStatisticsImageFilter.h

Go to the documentation of this file.

/*=========================================================================
*
*  Copyright Insight Software Consortium
*
*  Licensed under the Apache License, Version 2.0 (the "License");
*  you may not use this file except in compliance with the License.
*  You may obtain a copy of the License at
*
*         http://www.apache.org/licenses/LICENSE-2.0.txt
*
*  Unless required by applicable law or agreed to in writing, software
*  distributed under the License is distributed on an "AS IS" BASIS,
*  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*  See the License for the specific language governing permissions and
*  limitations under the License.
*
*=========================================================================*/
#ifndef __itkMaskedStatisticsImageFilter_h
#define __itkMaskedStatisticsImageFilter_h

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

namespace itk
{
  template< typename TInputImage , typename TMaskImage = ::itk::Image<unsigned char, TInputImage::ImageDimension> >
  class MaskedStatisticsImageFilter:
    public ImageToImageFilter< TInputImage, TInputImage >
  {
  public:
    typedef MaskedStatisticsImageFilter                          Self;
    typedef ImageToImageFilter< TInputImage, TInputImage > Superclass;
    typedef SmartPointer< Self >                           Pointer;
    typedef SmartPointer< const Self >                     ConstPointer;

    itkNewMacro(Self);

    itkTypeMacro(MaskedStatisticsImageFilter, ImageToImageFilter);

    typedef typename Superclass::InputImageType InputImageType;
    typedef typename Superclass::InputImagePointer InputImagePointer;

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

    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;

    typedef TMaskImage MaskImageType;
    typedef typename MaskImageType::ConstPointer     MaskImageConstPointer;
    typedef typename MaskImageType::RegionType  MaskImageRegionType;

    PixelType GetMinimum() const
    { return this->GetMinimumOutput()->Get(); }
    PixelObjectType * GetMinimumOutput();

    const PixelObjectType * GetMinimumOutput() const;

    PixelType GetMaximum() const
    { return this->GetMaximumOutput()->Get(); }
    PixelObjectType * GetMaximumOutput();

    const PixelObjectType * GetMaximumOutput() const;

    RealType GetMean() const
    { return this->GetMeanOutput()->Get(); }
    RealObjectType * GetMeanOutput();

    const RealObjectType * GetMeanOutput() const;

    RealType GetSigma() const
    { return this->GetSigmaOutput()->Get(); }
    RealObjectType * GetSigmaOutput();

    const RealObjectType * GetSigmaOutput() const;

    RealType GetVariance() const
    { return this->GetVarianceOutput()->Get(); }
    RealObjectType * GetVarianceOutput();

    const RealObjectType * GetVarianceOutput() const;

    RealType GetSum() const
    { return this->GetSumOutput()->Get(); }
    RealObjectType * GetSumOutput();

    const RealObjectType * GetSumOutput() const;

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

    itkSetConstObjectMacro(Mask, MaskImageType);
    itkGetConstObjectMacro(Mask, MaskImageType);

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

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

    void AllocateOutputs() override;

    void BeforeThreadedGenerateData() override;

    void AfterThreadedGenerateData() override;

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

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

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

  private:
    MaskedStatisticsImageFilter(const Self &); //purposely not implemented
    void operator=(const Self &);        //purposely not implemented

    Array< RealType >       m_ThreadSum;
    Array< RealType >       m_SumOfSquares;
    Array< SizeValueType >  m_Count;
    Array< PixelType >      m_ThreadMin;
    Array< PixelType >      m_ThreadMax;
    MaskImageConstPointer        m_Mask;
  }; // end of class
} // end namespace itk

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

#endif