mitkCovarianceMatrixCalculator.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_COVARIANCEMATRIXCALCULATOR_H__
#define __MITK_COVARIANCEMATRIXCALCULATOR_H__

// exports
#include "MitkAlgorithmsExtExports.h"

#include <mitkCommon.h>

#include <itkMatrix.h>
#include <itkObjectFactory.h>
#include <vector>

namespace mitk
{
  // forward declarations
  class Surface;
  struct CovarianceMatrixCalculatorData;

  class MITKALGORITHMSEXT_EXPORT CovarianceMatrixCalculator : public itk::Object
  {
  private:
    CovarianceMatrixCalculatorData *d;

  protected:
    // local typedefs

    typedef itk::Matrix<double, 3, 3> CovarianceMatrix;
    typedef std::vector<CovarianceMatrix> CovarianceMatrixList;
    typedef double Vertex[3];

    CovarianceMatrixList m_CovarianceMatrixList;

    void ComputeOrthonormalCoordinateSystem(
      const int index, Vertex normal, CovarianceMatrix &principalComponents, Vertex variances, Vertex curVertex);
    CovarianceMatrixCalculator();
    ~CovarianceMatrixCalculator();

  public:
    mitkClassMacroItkParent(CovarianceMatrixCalculator, itk::Object) itkFactorylessNewMacro(Self) itkCloneMacro(Self)

      void SetVoronoiScalingFator(const double factor);

    void EnableNormalization(bool state);

    double GetMeanVariance() const;

    const CovarianceMatrixList &GetCovarianceMatrices() const;

    void SetInputSurface(Surface *input);

    void ComputeCovarianceMatrices();
  };
}

#endif