FMEwork/ITK/Sources/ITK/MLITK/ITKSupport/mlITKFlatStructElement.h

Go to the documentation of this file.

// **InsertLicense** code
//----------------------------------------------------------------------------------

//----------------------------------------------------------------------------------
#ifndef __mlITKFlatStructElement_H
#define __mlITKFlatStructElement_H

#include "mlInitSystemITKSupport.h"

#include "mlModuleIncludes.h"
#include "itkFlatStructuringElement.h"


// Be sure that all your modules are part of the ml. So collisions
// with names in system files or other libraries are minimized.
ML_START_NAMESPACE

//----------------------------------------------------------------------------------
//----------------------------------------------------------------------------------
class MLITK_SUPPORT_EXPORT ITKFlatStructElement : public Module{

public:

  ITKFlatStructElement();

  enum {
    Box     = 0,  
    Ball,         
    Cross,        
    Annulus,      
    Polygon,      

    NumFlatStructuringElements   

  } FlatStructureElementTypes;

  static const char * const FlatStructuringElementTypeNames[NumFlatStructuringElements];

  template <size_t DIM> const itk::FlatStructuringElement<DIM> &getFlatStructuringElement() const;

protected:

  virtual void activateAttachments();

  virtual void handleNotification(Field * field);

private:

  BaseField        *_outputBaseFld;

  EnumField        *_selectElementFld;

  ImageVectorField *_radiusFld;

  IntField         *_thicknessFld;

  BoolField        *_includeCenterFld;

  IntField         *_numLinesFld;


  itk::FlatStructuringElement<1> _dim1FlatStructuringElement;
  itk::FlatStructuringElement<2> _dim2FlatStructuringElement;
  itk::FlatStructuringElement<3> _dim3FlatStructuringElement;
  itk::FlatStructuringElement<4> _dim4FlatStructuringElement;
  itk::FlatStructuringElement<5> _dim5FlatStructuringElement;
  itk::FlatStructuringElement<6> _dim6FlatStructuringElement;

  ML_MODULE_CLASS_HEADER(ITKFlatStructElement)
};



template<> inline const itk::FlatStructuringElement<1> &ITKFlatStructElement::getFlatStructuringElement<1>() const { return _dim1FlatStructuringElement; };
template<> inline const itk::FlatStructuringElement<2> &ITKFlatStructElement::getFlatStructuringElement<2>() const { return _dim2FlatStructuringElement; };
template<> inline const itk::FlatStructuringElement<3> &ITKFlatStructElement::getFlatStructuringElement<3>() const { return _dim3FlatStructuringElement; };
template<> inline const itk::FlatStructuringElement<4> &ITKFlatStructElement::getFlatStructuringElement<4>() const { return _dim4FlatStructuringElement; };
template<> inline const itk::FlatStructuringElement<5> &ITKFlatStructElement::getFlatStructuringElement<5>() const { return _dim5FlatStructuringElement; };
template<> inline const itk::FlatStructuringElement<6> &ITKFlatStructElement::getFlatStructuringElement<6>() const { return _dim6FlatStructuringElement; };

ML_END_NAMESPACE

#endif