mitkContourModelSetToImageFilter.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 mitkContourModelSetToImageFilter_h
#define mitkContourModelSetToImageFilter_h
 
#include <MitkSegmentationExports.h>
#include <mitkImageSource.h>
 
namespace mitk
{
  class ContourModelSet;
  class ContourModel;
 
  class MITKSEGMENTATION_EXPORT ContourModelSetToImageFilter : public ImageSource
  {
  public:
    mitkClassMacro(ContourModelSetToImageFilter, ImageSource);
    itkFactorylessNewMacro(Self);
    itkCloneMacro(Self);
 
    virtual void SetMakeOutputBinary(bool makeOutputBinary);
    virtual void SetMakeOutputLabelPixelType(bool makeOutputLabelPixelType);
    itkSetMacro(PaintingPixelValue, int);
    itkSetMacro(TimeStep, unsigned int);
 
    itkGetMacro(MakeOutputBinary, bool);
    itkGetMacro(MakeOutputLabelPixelType, bool);
    itkGetMacro(PaintingPixelValue, int);
 
    itkBooleanMacro(MakeOutputBinary);
    itkBooleanMacro(MakeOutputLabelPixelType);
 
    itk::DataObject::Pointer MakeOutput(DataObjectPointerArraySizeType idx) override;
 
    itk::DataObject::Pointer MakeOutput(const DataObjectIdentifierType &name) override;
 
    void GenerateInputRequestedRegion() override;
 
    void GenerateOutputInformation() override;
 
    void GenerateData() override;
 
    const mitk::ContourModelSet *GetInput(void);
 
    using itk::ProcessObject::SetInput;
    virtual void SetInput(const mitk::ContourModelSet *input);
 
    void SetImage(const mitk::Image *refImage);
 
    const mitk::Image *GetImage(void);
 
  protected:
    ContourModelSetToImageFilter();
 
    ~ContourModelSetToImageFilter() override;
 
    void InitializeOutputEmpty();
 
    bool m_MakeOutputBinary;
    bool m_MakeOutputLabelPixelType;
    int m_PaintingPixelValue;
 
    unsigned int m_TimeStep;
 
    const mitk::Image *m_ReferenceImage;
  };
 
  MITKSEGMENTATION_EXPORT Image::Pointer ConvertContourModelSetToLabelMask(const mitk::Image* refImage, mitk::ContourModelSet* contourSet);
 
  MITKSEGMENTATION_EXPORT Image::Pointer ConvertContourModelToLabelMask(const mitk::Image* refImage, mitk::ContourModel* contourModel);
}
#endif