mitkRandomImageSampler.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 mitkRandomImageSampler_h
#define mitkRandomImageSampler_h
 
#include "MitkCLUtilitiesExports.h"
 
//MITK
#include <mitkImage.h>
#include "mitkImageToImageFilter.h"
#include <itkImage.h>
 
namespace mitk
{
  enum RandomImageSamplerMode
  {
    SINGLE_ACCEPTANCE_RATE,
    CLASS_DEPENDEND_ACCEPTANCE_RATE,
    SINGLE_NUMBER_OF_ACCEPTANCE,
    CLASS_DEPENDEND_NUMBER_OF_ACCEPTANCE
  };
 
 
  class MITKCLUTILITIES_EXPORT RandomImageSampler : public ImageToImageFilter
  {
  public:
 
    mitkClassMacro( RandomImageSampler , ImageToImageFilter );
    itkFactorylessNewMacro(Self);
    itkCloneMacro(Self);
    itkSetMacro(SamplingMode, RandomImageSamplerMode);
    itkGetConstMacro(SamplingMode, RandomImageSamplerMode);
 
    itkSetMacro(AcceptanceRate, double);
    itkGetConstMacro(AcceptanceRate, double);
 
    //itkSetMacro(AcceptanceRateVector, std::vector<double>);
    void SetAcceptanceRateVector(std::vector<double> arg)
    {
      m_AcceptanceRateVector = arg;
    }
 
    itkGetConstMacro(AcceptanceRateVector, std::vector<double>);
 
    itkSetMacro(NumberOfSamples, unsigned int);
    itkGetConstMacro(NumberOfSamples, unsigned int);
 
    //itkSetMacro(NumberOfSamplesVector, std::vector<unsigned int>);
    void SetNumberOfSamplesVector(std::vector<unsigned int> arg)
    {
      m_NumberOfSamplesVector = arg;
    }
 
    itkGetConstMacro(NumberOfSamplesVector, std::vector<unsigned int>);
 
  private:
    RandomImageSampler();
    ~RandomImageSampler() override;
    void GenerateOutputInformation() override;
    void GenerateData() override;
 
    template<typename TPixel, unsigned int VImageDimension>
    void ItkImageProcessing(const itk::Image<TPixel, VImageDimension>* itkImage);
 
    template<typename TPixel, unsigned int VImageDimension>
    void ItkImageProcessingClassDependendSampling(const itk::Image<TPixel, VImageDimension>* itkImage);
 
    template<typename TPixel, unsigned int VImageDimension>
    void ItkImageProcessingFixedNumberSampling(const itk::Image<TPixel, VImageDimension>* itkImage);
 
    template<typename TPixel, unsigned int VImageDimension>
    void ItkImageProcessingClassDependendNumberSampling(const itk::Image<TPixel, VImageDimension>* itkImage);
 
    double m_AcceptanceRate;
    std::vector<double> m_AcceptanceRateVector;
    unsigned int m_NumberOfSamples;
    std::vector<unsigned int> m_NumberOfSamplesVector;
    RandomImageSamplerMode m_SamplingMode;
  };
} //END mitk namespace
#endif