itkLabelSampler.h

Go to the documentation of this file.

/*===================================================================

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center,
Division of Medical and Biological Informatics.
All rights reserved.

This software is distributed WITHOUT ANY WARRANTY; without
even the implied warranty of MERCHANTABILITY or FITNESS FOR
A PARTICULAR PURPOSE.

See LICENSE.txt or http://www.mitk.org for details.

===================================================================*/

#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