itkImageClassifierBase.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 itkImageClassifierBase_h
#define itkImageClassifierBase_h

#include "itkClassifierBase.h"
#include "itkMacro.h"
#include "itkImageRegionIterator.h"

namespace itk
{
template< typename TInputImage,
          typename TClassifiedImage >
class ITK_TEMPLATE_EXPORT ImageClassifierBase:
  public ClassifierBase< TInputImage >
{
public:
  typedef ImageClassifierBase           Self;
  typedef ClassifierBase< TInputImage > Superclass;
  typedef SmartPointer< Self >          Pointer;
  typedef SmartPointer< const Self >    ConstPointer;

  itkNewMacro(Self);

  itkTypeMacro(ImageClassifierBase, ClassifierBase);

  typedef TInputImage                        InputImageType;
  typedef typename TInputImage::Pointer      InputImagePointer;
  typedef typename TInputImage::ConstPointer InputImageConstPointer;

  typedef typename TClassifiedImage::Pointer ClassifiedImagePointer;

  typedef typename Superclass::MeasurementVectorType MeasurementVectorType;

  typedef typename Superclass::MembershipFunctionType MembershipFunctionType;

  typedef typename Superclass::MembershipFunctionPointer
  MembershipFunctionPointer;

  typedef typename Superclass::MembershipFunctionPointerVector
  MembershipFunctionPointerVector;

  typedef typename Superclass::DecisionRuleType DecisionRuleType;

  itkSetConstObjectMacro(InputImage, InputImageType);
  itkGetConstObjectMacro(InputImage, InputImageType);

  itkSetMacro(ClassifiedImage, ClassifiedImagePointer);

  itkGetConstMacro(ClassifiedImage, ClassifiedImagePointer);

  typedef typename TInputImage::PixelType InputImagePixelType;

  typedef typename TClassifiedImage::PixelType ClassifiedImagePixelType;

  typedef
  ImageRegionConstIterator< TInputImage >    InputImageConstIterator;
  typedef
  ImageRegionIterator< TClassifiedImage >    ClassifiedImageIterator;

  std::vector< double >
  GetPixelMembershipValue(const InputImagePixelType inputImagePixel);

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

  void Allocate();

  void GenerateData() ITK_OVERRIDE;

private:
  ITK_DISALLOW_COPY_AND_ASSIGN(ImageClassifierBase);

  typedef typename TInputImage::SizeType InputImageSizeType;

  InputImageConstPointer m_InputImage;
  ClassifiedImagePointer m_ClassifiedImage;

  virtual void Classify();
}; // class ImageClassifierBase
} // namespace itk

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

#endif