itkSegmentationModule.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    itkSegmentationModule.h
  Language:  C++
  Date:      $Date$
  Version:   $Revision$

  Copyright (c) Insight Software Consortium. All rights reserved.
  See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details.

     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notices for more information.

=========================================================================*/
#ifndef itkSegmentationModule_h
#define itkSegmentationModule_h

#include "itkProcessObject.h"
#include "itkImage.h"
#include "itkDataObjectDecorator.h"
#include "itkSpatialObject.h"

namespace itk
{

template <unsigned int NDimension>
class ITK_TEMPLATE_EXPORT SegmentationModule : public ProcessObject
{
public:
  ITK_DISALLOW_COPY_AND_ASSIGN(SegmentationModule);

  using Self = SegmentationModule;
  using Superclass = ProcessObject;
  using Pointer = SmartPointer<Self>;
  using ConstPointer = SmartPointer<const Self>;

  itkNewMacro(Self);

  itkTypeMacro(SegmentationModule, ProcessObject);

  static constexpr unsigned int Dimension = NDimension;

  using SpatialObjectType = SpatialObject< NDimension >;
  using SpatialObjectPointer = typename SpatialObjectType::Pointer;

  using ProcessObject::SetInput;
  void SetInput( const SpatialObjectType * input );
  const SpatialObjectType * GetInput() const;

  void SetFeature( const SpatialObjectType * feature );
  const SpatialObjectType * GetFeature() const;

  const SpatialObjectType * GetOutput() const;

  unsigned int GetExpectedNumberOfFeatures() const;

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

  /* Derived classes MUST implement the "void  GenerateData ()" method */

  SpatialObjectType * GetInternalOutput();
};

} // end namespace itk

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

#endif