itkLabelSampler.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 itLabelSampler_h
#define itLabelSampler_h

#include "itkImageToImageFilter.h"

namespace itk
{
template< class TImage>
class LabelSampler:public ImageToImageFilter< TImage, TImage >
{
public:
  typedef LabelSampler             Self;
  typedef ImageToImageFilter< TImage, TImage > Superclass;
  typedef SmartPointer< Self >        Pointer;
  typedef std::map<int,int> outmap;

  itkNewMacro(Self); 
  itkTypeMacro(Self, ImageToImageFilter);
  itkSetMacro(AcceptRate, double);
  itkSetMacro(Label, int);
  itkGetMacro(LabelVoxelCountMap, outmap);
  itkGetMacro(NumberOfSampledVoxels,int);

  int GetLabelVoxelCount(int label)
  {
    if(m_LabelVoxelCountMap.empty())
      return 0;
    return m_LabelVoxelCountMap[label];
  }

protected:
  LabelSampler()
  :m_Label(-1)
  ,m_AcceptRate(1.0)
  ,m_UseLabelVoxelCount(false)
  {
    this->SetNumberOfRequiredInputs(1);
    this->SetNumberOfRequiredOutputs(1);
  }

  ~LabelSampler(){}

  virtual void GenerateData();

private:
  LabelSampler(const Self &); //purposely not implemented
  void operator=(const Self &);  //purposely not implemented

  int m_Label;
  int m_NumberOfSampledVoxels;
  double m_AcceptRate;
  bool m_UseLabelVoxelCount;
  std::map<int,int> m_LabelVoxelCountMap;
};
} //namespace ITK


#ifndef ITK_MANUAL_INSTANTIATION
#include <../src/Algorithms/itkLabelSampler.cpp>
#endif


#endif // itLabelSampler_h