mitkRicianNoiseModel.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 _MITK_RicianNoiseModel_H
#define _MITK_RicianNoiseModel_H

#include <mitkDiffusionNoiseModel.h>
#include <itkMersenneTwisterRandomVariateGenerator.h>

namespace mitk {

template< class ScalarType >
class RicianNoiseModel : public DiffusionNoiseModel< ScalarType >
{
public:

    RicianNoiseModel();
    ~RicianNoiseModel();

    typedef typename DiffusionNoiseModel< ScalarType >::PixelType      PixelType;

    void AddNoise(PixelType& pixel);

    void SetNoiseVariance(double var){ m_NoiseVariance = var; }
    double GetNoiseVariance(){ return m_NoiseVariance; }
    void SetSeed(int seed); 

protected:

    itk::Statistics::MersenneTwisterRandomVariateGenerator::Pointer m_RandGen;
    double      m_NoiseVariance;    

};

}

#include "mitkRicianNoiseModel.cpp"

#endif