mitkCovarianceMatrixCalculator.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 __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() override;

  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