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 _MITK_CONTOURMODEL_SOURCE_H
#define _MITK_CONTOURMODEL_SOURCE_H

#include <MitkSegmentationExports.h>
#include <mitkImageSource.h>

namespace mitk
{
  class ContourModelSet;

  class MITKSEGMENTATION_EXPORT ContourModelSetToImageFilter : public ImageSource
  {
  public:
    mitkClassMacro(ContourModelSetToImageFilter, ImageSource);
    itkFactorylessNewMacro(Self);
    itkCloneMacro(Self);
    itkSetMacro(MakeOutputBinary, bool);
    itkGetMacro(MakeOutputBinary, bool);
    itkBooleanMacro(MakeOutputBinary);

    itkSetMacro(TimeStep, unsigned int);

    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;

    unsigned int m_TimeStep;

    const mitk::Image *m_ReferenceImage;
  };
}
#endif // #_MITK_CONTOURMODEL_SOURCE_H