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