mitkSurfaceStampImageFilter.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 _mitkSurfaceStampImageFilter_h__
#define _mitkSurfaceStampImageFilter_h__

#include "MitkSegmentationExports.h"
#include "mitkCommon.h"
#include "mitkImageToImageFilter.h"
#include "mitkSurface.h"

#include <itkQuadEdgeMesh.h>

class vtkPolyData;

namespace mitk
{
  class MITKSEGMENTATION_EXPORT SurfaceStampImageFilter : public ImageToImageFilter
  {
  public:
    mitkClassMacro(SurfaceStampImageFilter, ImageToImageFilter);
    itkNewMacro(Self);

    itkSetMacro(MakeOutputBinary, bool);
    itkGetMacro(MakeOutputBinary, bool);
    itkBooleanMacro(MakeOutputBinary);

    itkSetMacro(OverwriteBackground, bool);
    itkGetMacro(OverwriteBackground, bool);
    itkBooleanMacro(OverwriteBackground);

    itkGetConstMacro(BackgroundValue, float);
    itkSetMacro(BackgroundValue, float);

    itkGetConstMacro(ForegroundValue, float);
    itkSetMacro(ForegroundValue, float);

    void GenerateInputRequestedRegion() override;

    void GenerateOutputInformation() override;

    void GenerateData() override;

    void SetSurface(mitk::Surface *surface);

    typedef itk::QuadEdgeMesh<double, 3> MeshType;

  protected:
    SurfaceStampImageFilter();

    ~SurfaceStampImageFilter() override;

    void SurfaceStamp(int time = 0);

    template <typename TPixel>
    void SurfaceStampProcessing(itk::Image<TPixel, 3> *input, MeshType *mesh);

    void SurfaceStampBinaryOutputProcessing(MeshType *mesh);

    bool m_MakeOutputBinary;
    bool m_OverwriteBackground;
    float m_BackgroundValue;
    float m_ForegroundValue;

    mitk::Surface::Pointer m_Surface;
  };
} // namespace mitk

#endif /* MITKCOONSPATCHFILTER_H_HEADER_INCLUDED_C10B22CD */