mitkMultiLabelMaskGenerator.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 mitkMultiLabelMaskGenerator_h
#define mitkMultiLabelMaskGenerator_h
 
#include <MitkImageStatisticsExports.h>
#include <mitkMaskGenerator.h>
#include <mitkLabelSetImage.h>
 
 
namespace mitk
{
class MITKIMAGESTATISTICS_EXPORT MultiLabelMaskGenerator: public MaskGenerator
{
public:
  mitkClassMacro(MultiLabelMaskGenerator, MaskGenerator);
  itkNewMacro(Self);
 
  unsigned int GetNumberOfMasks() const override;
 
  itkSetConstObjectMacro(MultiLabelSegmentation, MultiLabelSegmentation);
 
protected:
  MultiLabelMaskGenerator() = default;
  ~MultiLabelMaskGenerator() = default;
 
  Image::ConstPointer DoGetMask(unsigned int maskID) override;
 
private:
    mitk::MultiLabelSegmentation::ConstPointer m_MultiLabelSegmentation;
};
 
}
 
#endif