mitkHistogramMatching.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 MITKHISTOGRAMMATCHING_H
#define MITKHISTOGRAMMATCHING_H

#include "MitkDeformableRegistrationExports.h"
#include "itkHistogramMatchingImageFilter.h"

#include "mitkImageAccessByItk.h"
#include "mitkRegistrationBase.h"

namespace mitk
{
  class MITKDEFORMABLEREGISTRATION_EXPORT HistogramMatching : public RegistrationBase
  {
  public:
    mitkClassMacro(HistogramMatching, RegistrationBase);

    itkFactorylessNewMacro(Self) itkCloneMacro(Self)

      
      void SetNumberOfMatchPoints(int matchPoints);

    void SetNumberOfHistogramLevels(int histogrammLevels);

    void SetThresholdAtMeanIntensity(bool on);

    virtual void GenerateData() override
    {
      if (this->GetInput())
      {
        AccessByItk(this->GetInput(), GenerateData2);
      }
    }

  protected:
    HistogramMatching();

    virtual ~HistogramMatching();

    template <typename TPixel, unsigned int VImageDimension>
    void GenerateData2(const itk::Image<TPixel, VImageDimension> *itkImage1);

    bool m_ThresholdAtMeanIntensity;
    int m_NumberOfHistogramLevels;
    int m_NumberOfMatchPoints;
  };

} // end namespace
#endif // MITKHISTOGRAMMATCHING_H